Spaces:
Sleeping
Sleeping
Updated to use openrouter and latest google embeddings
Browse files
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='
|
14 |
config=dict(
|
15 |
-
model="
|
16 |
),
|
17 |
),
|
18 |
embedder=dict(
|
19 |
provider="google",
|
20 |
config=dict(
|
21 |
-
model="models/embedding-
|
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='
|
32 |
config=dict(
|
33 |
-
model="
|
34 |
),
|
35 |
),
|
36 |
embedder=dict(
|
37 |
provider="google",
|
38 |
config=dict(
|
39 |
-
model="models/embedding-
|
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 |
),
|