Ferocious0xide commited on
Commit
66459aa
·
verified ·
1 Parent(s): addb0c2

Update app.py to reflect the appropriate tool call for web search

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -5,7 +5,7 @@ import yaml
5
  from tools.final_answer import FinalAnswerTool
6
  from tools.book_tool import BookSearchTool, BookDetailsTool
7
  from tools.arxiv_tool import ArxivSearchTool, LatestPapersTool
8
- from tools.search_tool import DuckDuckGoSearchTool, VisitWebpageTool
9
  from scheduler import initialize_scheduler
10
  from Gradio_UI import GradioUI
11
 
@@ -57,7 +57,6 @@ def initialize_agent():
57
  arxiv_search = ArxivSearchTool()
58
  latest_papers = LatestPapersTool()
59
  web_search = DuckDuckGoSearchTool(max_results=5) # Initialize DDG search with 5 results
60
- visit_webpage = VisitWebpageTool()
61
 
62
  # Initialize the model
63
  model = HfApiModel(
@@ -82,8 +81,7 @@ def initialize_agent():
82
  book_details,
83
  arxiv_search,
84
  latest_papers,
85
- web_search,
86
- visit_webpage
87
  ],
88
  max_steps=8,
89
  verbosity_level=1,
 
5
  from tools.final_answer import FinalAnswerTool
6
  from tools.book_tool import BookSearchTool, BookDetailsTool
7
  from tools.arxiv_tool import ArxivSearchTool, LatestPapersTool
8
+ from tools.web_search import DuckDuckGoSearchTool
9
  from scheduler import initialize_scheduler
10
  from Gradio_UI import GradioUI
11
 
 
57
  arxiv_search = ArxivSearchTool()
58
  latest_papers = LatestPapersTool()
59
  web_search = DuckDuckGoSearchTool(max_results=5) # Initialize DDG search with 5 results
 
60
 
61
  # Initialize the model
62
  model = HfApiModel(
 
81
  book_details,
82
  arxiv_search,
83
  latest_papers,
84
+ web_search
 
85
  ],
86
  max_steps=8,
87
  verbosity_level=1,