Hugging Face's logo

Spaces:
ryefoxlime
/
AI_Agent
Sleeping

App Files Community
ryefoxlime commited on
Commit
8e00063
·
verified ·
1 Parent(s): da3e4d2

Updated the models

Browse files
Files changed (1) hide show
  1. src/research_agent/tools/tool.py +4 -4
src/research_agent/tools/tool.py CHANGED
@@ -12,13 +12,13 @@ website_search_tool = WebsiteSearchTool(
12
  llm=dict(
13
  provider='google',
14
  config=dict(
15
- model="gemini/gemini-2.0-flash-001",
16
  ),
17
  ),
18
  embedder=dict(
19
  provider="google",
20
  config=dict(
21
- model="models/text-embedding-004/",
22
  task_type="retrieval_document"
23
  ),
24
  ),
@@ -30,13 +30,13 @@ pdf_search_tool = PDFSearchTool(
30
  llm=dict(
31
  provider='google',
32
  config=dict(
33
- model="gemini/gemini-2.0-flash-001",
34
  ),
35
  ),
36
  embedder=dict(
37
  provider="google",
38
  config=dict(
39
- model="models/text-embedding-004/",
40
  task_type="retrieval_document"
41
  ),
42
  ),
 
12
  llm=dict(
13
  provider='google',
14
  config=dict(
15
+ model="gemini/gemini-2.5-flash",
16
  ),
17
  ),
18
  embedder=dict(
19
  provider="google",
20
  config=dict(
21
+ model="gemini-embeddings-001",
22
  task_type="retrieval_document"
23
  ),
24
  ),
 
30
  llm=dict(
31
  provider='google',
32
  config=dict(
33
+ model="gemini/gemini-2.5-flash",
34
  ),
35
  ),
36
  embedder=dict(
37
  provider="google",
38
  config=dict(
39
+ model="gemini-embeddings-001",
40
  task_type="retrieval_document"
41
  ),
42
  ),