ajac-zero commited on
Commit
f1f3f30
·
1 Parent(s): 26aaa97
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import gradio as gr
3
  import requests
4
  import pandas as pd
5
- from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
6
 
7
  # (Keep Constants as is)
8
  # --- Constants ---
@@ -27,10 +27,13 @@ class BasicAgent:
27
  def __init__(self):
28
  model = AzureOpenAIServerModel("gpt-4o")
29
  search_tool = DuckDuckGoSearchTool()
 
 
 
30
 
31
  self.agent = CodeAgent(
32
  model=model,
33
- tools=[search_tool]
34
  )
35
 
36
  def __call__(self, question: str) -> str:
 
2
  import gradio as gr
3
  import requests
4
  import pandas as pd
5
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, PythonInterpreterTool, FinalAnswerTool
6
 
7
  # (Keep Constants as is)
8
  # --- Constants ---
 
27
  def __init__(self):
28
  model = AzureOpenAIServerModel("gpt-4o")
29
  search_tool = DuckDuckGoSearchTool()
30
+ visit_page_tool = VisitWebpageTool()
31
+ interpreter_tool = PythonInterpreterTool()
32
+ final_answer_tool = FinalAnswerTool()
33
 
34
  self.agent = CodeAgent(
35
  model=model,
36
+ tools=[search_tool, visit_page_tool, interpreter_tool, final_answer_tool]
37
  )
38
 
39
  def __call__(self, question: str) -> str: