ronnybehrens commited on
Commit
3d93523
·
verified ·
1 Parent(s): 2a53f00

Added visit web page and python interpreter tools

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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 CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, WikipediaSearchTool
7
  from huggingface_hub import login
8
 
9
  # Login to HF
@@ -16,12 +16,15 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
16
  # ---- Tools ----
17
  web_search_tool = DuckDuckGoSearchTool(max_results=10)
18
  wiki_search_tool = WikipediaSearchTool(user_agent="HuggingFaceSmolagentsExercise", language="en", content_type="text", extract_format="WIKI")
 
 
 
19
 
20
  # --- Basic Agent Definition ---
21
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
22
  class BasicAgent:
23
  def __init__(self):
24
- self.agent = CodeAgent(tools=[web_search_tool, wiki_search_tool],
25
  model=InferenceClientModel(),
26
  )
27
  print("BasicAgent initialized.")
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, WikipediaSearchTool, VisitWebpageTool, PythonInterpreterTool
7
  from huggingface_hub import login
8
 
9
  # Login to HF
 
16
  # ---- Tools ----
17
  web_search_tool = DuckDuckGoSearchTool(max_results=10)
18
  wiki_search_tool = WikipediaSearchTool(user_agent="HuggingFaceSmolagentsExercise", language="en", content_type="text", extract_format="WIKI")
19
+ web_page_tool = VisitWebpageTool()
20
+ python_interpreter = PythonInterpreterTool()
21
+
22
 
23
  # --- Basic Agent Definition ---
24
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
25
  class BasicAgent:
26
  def __init__(self):
27
+ self.agent = CodeAgent(tools=[web_search_tool, wiki_search_tool, web_page_tool, python_interpreter],
28
  model=InferenceClientModel(),
29
  )
30
  print("BasicAgent initialized.")