ryefoxlime commited on
Commit
618bf37
·
verified ·
1 Parent(s): 19a426a

Updated to use openrouter and latest google embeddings

Browse files
Files changed (1) hide show
  1. src/research_agent/tools/tool.py +6 -6
src/research_agent/tools/tool.py CHANGED
@@ -10,15 +10,15 @@ search_tool = SerperDevTool(
10
  website_search_tool = WebsiteSearchTool(
11
  config=dict(
12
  llm=dict(
13
- provider='google',
14
  config=dict(
15
- model="gemini/gemini-1.5-flash-002",
16
  ),
17
  ),
18
  embedder=dict(
19
  provider="google",
20
  config=dict(
21
- model="models/embedding-001/",
22
  task_type="retrieval_document"
23
  ),
24
  ),
@@ -28,15 +28,15 @@ website_search_tool = WebsiteSearchTool(
28
  pdf_search_tool = PDFSearchTool(
29
  config=dict(
30
  llm=dict(
31
- provider='google',
32
  config=dict(
33
- model="gemini/gemini-1.5-flash-002",
34
  ),
35
  ),
36
  embedder=dict(
37
  provider="google",
38
  config=dict(
39
- model="models/embedding-001/",
40
  task_type="retrieval_document"
41
  ),
42
  ),
 
10
  website_search_tool = WebsiteSearchTool(
11
  config=dict(
12
  llm=dict(
13
+ provider='openrouter',
14
  config=dict(
15
+ model="google/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
  ),
 
28
  pdf_search_tool = PDFSearchTool(
29
  config=dict(
30
  llm=dict(
31
+ provider='openrouter',
32
  config=dict(
33
+ model="google/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
  ),