Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,7 @@ import traceback
|
|
| 6 |
|
| 7 |
# Import smol-agent and tool components
|
| 8 |
from smolagents import CodeAgent, LiteLLMModel, tool
|
| 9 |
-
|
| 10 |
-
from smolagents.tools import DuckDuckGoSearch
|
| 11 |
from unstructured.partition.auto import partition
|
| 12 |
|
| 13 |
# --- Constants ---
|
|
@@ -56,7 +55,7 @@ class GaiaSmolAgent:
|
|
| 56 |
# The agent will make these available to the script it runs.
|
| 57 |
self.executor_agent = CodeAgent(
|
| 58 |
model=self.planner_model,
|
| 59 |
-
tools=[file_reader,
|
| 60 |
add_base_tools=True, # Provides a python interpreter
|
| 61 |
)
|
| 62 |
print("GaiaSmolAgent initialized successfully.")
|
|
|
|
| 6 |
|
| 7 |
# Import smol-agent and tool components
|
| 8 |
from smolagents import CodeAgent, LiteLLMModel, tool
|
| 9 |
+
from smolagents import DuckDuckGoSearchTool
|
|
|
|
| 10 |
from unstructured.partition.auto import partition
|
| 11 |
|
| 12 |
# --- Constants ---
|
|
|
|
| 55 |
# The agent will make these available to the script it runs.
|
| 56 |
self.executor_agent = CodeAgent(
|
| 57 |
model=self.planner_model,
|
| 58 |
+
tools=[file_reader, DuckDuckGoSearchTool()],
|
| 59 |
add_base_tools=True, # Provides a python interpreter
|
| 60 |
)
|
| 61 |
print("GaiaSmolAgent initialized successfully.")
|