ChIrish06 commited on
Commit
33da7c6
·
verified ·
1 Parent(s): 559e26f

Update app.py

Browse files

Adding a duckduckgo search tool

Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -156,6 +156,15 @@ def tool_visit_webpage(url:str) -> str:
156
  vst = VisitWebpageTool()
157
  ret = vst(url)
158
  return ret
 
 
 
 
 
 
 
 
 
159
 
160
  final_answer = FinalAnswerTool()
161
 
@@ -185,7 +194,8 @@ agent = CodeAgent(
185
  model=model,
186
  tools=[final_answer, image_generation_tool,webpage_contents_get,
187
  webpage_header_get, webpage_contents_soup_links, webpage_contents_soup_paragraphs,
188
- summarize_and_answer_from_web, webpage_json_get, tool_visit_webpage], ## add your tools here (don't remove final answer)
 
189
  max_steps=6,
190
  verbosity_level=1,
191
  grammar=None,
 
156
  vst = VisitWebpageTool()
157
  ret = vst(url)
158
  return ret
159
+ def duckduckgo_search_tool(query:str) -> str:
160
+ '''
161
+ A tool used to initialize the ddg search tool, and to return the top 10 finings
162
+ Args:
163
+ query: A string to pass to duckduckgo
164
+ '''
165
+ ddg = DuckDuckGoSearchTool()
166
+ ret = ddg(query)
167
+ return ret
168
 
169
  final_answer = FinalAnswerTool()
170
 
 
194
  model=model,
195
  tools=[final_answer, image_generation_tool,webpage_contents_get,
196
  webpage_header_get, webpage_contents_soup_links, webpage_contents_soup_paragraphs,
197
+ summarize_and_answer_from_web, webpage_json_get, tool_visit_webpage,
198
+ duckduckgo_search_tool], ## add your tools here (don't remove final answer)
199
  max_steps=6,
200
  verbosity_level=1,
201
  grammar=None,