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
|