Edenbaba commited on
Commit
ae03d26
·
verified ·
1 Parent(s): 2cd9c6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -119,8 +119,8 @@ final_answer = FinalAnswerTool()
119
  visit_webpage = VisitWebpageTool()
120
  python_tool = PythonInterpreterTool()
121
 
122
- # web_search = DuckDuckGoSearchTool()
123
- google_search_tool = GoogleSearchTool()
124
 
125
  # If the agent does not answer, the model is overloaded,please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
126
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
@@ -139,7 +139,7 @@ with open("prompts.yaml", 'r') as stream:
139
 
140
  agent = CodeAgent(
141
  model=model,
142
- tools=[final_answer, my_custom_tool, get_current_time_in_timezone, visit_webpage, google_search_tool,
143
  python_tool, image_generation_tool, image_generation_tool2,
144
  suggest_menu, catering_service_tool, SuperheroPartyThemeTool()], ## add your tools here (don't remove final answer)
145
  max_steps=10,
 
119
  visit_webpage = VisitWebpageTool()
120
  python_tool = PythonInterpreterTool()
121
 
122
+ web_search_ddgs = DuckDuckGoSearchTool()
123
+ google_search_tool = GoogleSearchTool() # default tool name is "web_search"
124
 
125
  # If the agent does not answer, the model is overloaded,please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
126
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
 
139
 
140
  agent = CodeAgent(
141
  model=model,
142
+ tools=[final_answer, my_custom_tool, get_current_time_in_timezone, visit_webpage, google_search_tool, web_search_ddgs
143
  python_tool, image_generation_tool, image_generation_tool2,
144
  suggest_menu, catering_service_tool, SuperheroPartyThemeTool()], ## add your tools here (don't remove final answer)
145
  max_steps=10,