Kai Jennissen commited on
Commit
265eb35
·
unverified ·
1 Parent(s): 42b982f

added wiki-tool

Browse files
Files changed (2) hide show
  1. agent.py +2 -1
  2. requirements.in +2 -1
agent.py CHANGED
@@ -4,6 +4,7 @@ from smolagents import (
4
  DuckDuckGoSearchTool,
5
  VisitWebpageTool,
6
  InferenceClientModel,
 
7
  )
8
  from dotenv import load_dotenv
9
  from tracing import setup_tracing
@@ -45,7 +46,7 @@ def get_agent():
45
 
46
  # Create web agent
47
  web_agent = ToolCallingAgent(
48
- tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
49
  model=llm_qwen,
50
  max_steps=10,
51
  name="Web_Agent",
 
4
  DuckDuckGoSearchTool,
5
  VisitWebpageTool,
6
  InferenceClientModel,
7
+ WikipediaSearchTool,
8
  )
9
  from dotenv import load_dotenv
10
  from tracing import setup_tracing
 
46
 
47
  # Create web agent
48
  web_agent = ToolCallingAgent(
49
+ tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), WikipediaSearchTool()],
50
  model=llm_qwen,
51
  max_steps=10,
52
  name="Web_Agent",
requirements.in CHANGED
@@ -1,4 +1,5 @@
 
1
  gradio[oauth]
2
  requests
3
  smolagents[gradio,litellm,openai,telemetry,toolkit,torch,transformers,vision]
4
- duckduckgo_search>=7.0.0,<8.0.0
 
1
+ duckduckgo_search>=7.0.0,<8.0.0
2
  gradio[oauth]
3
  requests
4
  smolagents[gradio,litellm,openai,telemetry,toolkit,torch,transformers,vision]
5
+ wikipedia-api