Hugging Face's logo

Spaces:
ryefoxlime
/
AI_Agent
Running

App Files Community
ryefoxlime commited on
Commit
824c45b
·
verified ·
1 Parent(s): 618bf37

Made some changes to solve errors

Browse files
Files changed (1) hide show
  1. src/research_agent/tools/tool.py +4 -4
src/research_agent/tools/tool.py CHANGED
@@ -10,9 +10,9 @@ search_tool = SerperDevTool(
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(
@@ -28,9 +28,9 @@ website_search_tool = WebsiteSearchTool(
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(
 
10
  website_search_tool = WebsiteSearchTool(
11
  config=dict(
12
  llm=dict(
13
+ provider='google',
14
  config=dict(
15
+ model="gemini/gemini-2.0-flash-001",
16
  ),
17
  ),
18
  embedder=dict(
 
28
  pdf_search_tool = PDFSearchTool(
29
  config=dict(
30
  llm=dict(
31
+ provider='google',
32
  config=dict(
33
+ model="gemini/gemini-2.0-flash-001",
34
  ),
35
  ),
36
  embedder=dict(