MickyWin22 commited on
Commit
f368ef4
·
verified ·
1 Parent(s): c2f952d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- # Corrected import for the search tool
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, DuckDuckGoSearch()],
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.")