Hugging Face's logo

Spaces:
ryefoxlime
/
AI_Agent
Sleeping

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

Update src/research_agent/tools/tool.py

Browse files
Files changed (1) hide show
  1. src/research_agent/tools/tool.py +2 -2
src/research_agent/tools/tool.py CHANGED
@@ -18,7 +18,7 @@ website_search_tool = WebsiteSearchTool(
18
  embedder=dict(
19
  provider="google",
20
  config=dict(
21
- model="gemini-embeddings-001",
22
  task_type="retrieval_document"
23
  ),
24
  ),
@@ -36,7 +36,7 @@ pdf_search_tool = PDFSearchTool(
36
  embedder=dict(
37
  provider="google",
38
  config=dict(
39
- model="gemini-embeddings-001",
40
  task_type="retrieval_document"
41
  ),
42
  ),
 
18
  embedder=dict(
19
  provider="google",
20
  config=dict(
21
+ model="gemini-embedding-001",
22
  task_type="retrieval_document"
23
  ),
24
  ),
 
36
  embedder=dict(
37
  provider="google",
38
  config=dict(
39
+ model="gemini-embedding-001",
40
  task_type="retrieval_document"
41
  ),
42
  ),