Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,7 @@ import requests
|
|
| 5 |
import inspect
|
| 6 |
import pandas as pd
|
| 7 |
from smolagents import LiteLLMModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool,VisitWebpageTool
|
|
|
|
| 8 |
|
| 9 |
# (Keep Constants as is)
|
| 10 |
# --- Constants ---
|
|
@@ -20,7 +21,7 @@ class BasicAgent:
|
|
| 20 |
)
|
| 21 |
self.agent = CodeAgent(
|
| 22 |
model=model,
|
| 23 |
-
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(),VisitWebpageTool()],
|
| 24 |
add_base_tools=True,
|
| 25 |
)
|
| 26 |
print("BasicAgent initialized.")
|
|
|
|
| 5 |
import inspect
|
| 6 |
import pandas as pd
|
| 7 |
from smolagents import LiteLLMModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool,VisitWebpageTool
|
| 8 |
+
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 9 |
|
| 10 |
# (Keep Constants as is)
|
| 11 |
# --- Constants ---
|
|
|
|
| 21 |
)
|
| 22 |
self.agent = CodeAgent(
|
| 23 |
model=model,
|
| 24 |
+
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(),VisitWebpageTool(),TavilySearchResults()],
|
| 25 |
add_base_tools=True,
|
| 26 |
)
|
| 27 |
print("BasicAgent initialized.")
|