Kai Jennissen
commited on
added wiki-tool
Browse files- agent.py +2 -1
- 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 |
-
|
|
|
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
|