pedrogaprieto commited on
Commit
42e1cb9
·
verified ·
1 Parent(s): ec38724

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import ToolCallingAgent, LiteLLMModel, VisitWebpageTool, FinalAnswerTool
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
@@ -25,12 +25,12 @@ llm = LiteLLMModel(
25
 
26
 
27
  # Create Alfred with all the tools
28
- agent = ToolCallingAgent(
29
  tools=[web_search_tool, VisitWebpageTool(), FinalAnswerTool()],
30
  model=llm,
31
  max_steps=10,
32
  #add_base_tools=False, # Add any additional base tools
33
- #planning_interval=3 # Enable planning every 3 steps
34
  )
35
 
36
  #agent = ToolCallingAgent(tools=[search_tool], model=llm)
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, LiteLLMModel, VisitWebpageTool, FinalAnswerTool
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
 
25
 
26
 
27
  # Create Alfred with all the tools
28
+ agent = CodeAgent(
29
  tools=[web_search_tool, VisitWebpageTool(), FinalAnswerTool()],
30
  model=llm,
31
  max_steps=10,
32
  #add_base_tools=False, # Add any additional base tools
33
+ planning_interval=3 # Enable planning every 3 steps
34
  )
35
 
36
  #agent = ToolCallingAgent(tools=[search_tool], model=llm)