at1300 commited on
Commit
cdfa443
·
verified ·
1 Parent(s): 565cdb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,6 +6,7 @@ import pandas as pd
6
  import yaml
7
 
8
  from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, DuckDuckGoSearchTool, Tool
 
9
 
10
  # (Keep Constants as is)
11
  # --- Constants ---
@@ -16,7 +17,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
16
  class BasicAgent:
17
  def __init__(self):
18
  websearchtool = WebSearchTool()
19
- search_tool = DuckDuckGoSearchTool()
20
  model = InferenceClientModel()
21
  # TODO: add additional tools and or subagents (e.g. OpenAI image recognition)
22
 
@@ -24,7 +25,7 @@ class BasicAgent:
24
  prompt_templates = yaml.safe_load(stream)
25
 
26
  master_agent = CodeAgent(
27
- tools=[websearchtool, search_tool],
28
  model=model,
29
  add_base_tools=True,
30
  planning_interval=3,
 
6
  import yaml
7
 
8
  from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, DuckDuckGoSearchTool, Tool
9
+ from tools import visit_webpage
10
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
 
17
  class BasicAgent:
18
  def __init__(self):
19
  websearchtool = WebSearchTool()
20
+ #search_tool = DuckDuckGoSearchTool()
21
  model = InferenceClientModel()
22
  # TODO: add additional tools and or subagents (e.g. OpenAI image recognition)
23
 
 
25
  prompt_templates = yaml.safe_load(stream)
26
 
27
  master_agent = CodeAgent(
28
+ tools=[websearchtool, visit_webpage],
29
  model=model,
30
  add_base_tools=True,
31
  planning_interval=3,