Spaces:
Sleeping
Sleeping
added wikipedia retreiver tool
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import yaml
|
|
7 |
from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, GoogleSearchTool, FinalAnswerTool, InferenceClientModel
|
8 |
|
9 |
from tools import translator, image_qa
|
10 |
-
|
11 |
|
12 |
# (Keep Constants as is)
|
13 |
# --- Constants ---
|
@@ -27,7 +27,7 @@ class BasicAgent:
|
|
27 |
)
|
28 |
self.agent = CodeAgent(
|
29 |
model=model,
|
30 |
-
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(),FinalAnswerTool(),translator, image_qa],
|
31 |
prompt_templates=prompt_templates,
|
32 |
additional_authorized_imports=['datetime','numpy','pandas'],
|
33 |
max_steps=3
|
|
|
7 |
from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, GoogleSearchTool, FinalAnswerTool, InferenceClientModel
|
8 |
|
9 |
from tools import translator, image_qa
|
10 |
+
from retreiver import retriever_tool
|
11 |
|
12 |
# (Keep Constants as is)
|
13 |
# --- Constants ---
|
|
|
27 |
)
|
28 |
self.agent = CodeAgent(
|
29 |
model=model,
|
30 |
+
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(),FinalAnswerTool(),translator, image_qa, retriever_tool],
|
31 |
prompt_templates=prompt_templates,
|
32 |
additional_authorized_imports=['datetime','numpy','pandas'],
|
33 |
max_steps=3
|