amritzeon commited on
Commit
c360166
·
verified ·
1 Parent(s): 6f011be

Update app.py

Browse files

Adsded web_search tool in CodeAgent class call

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -17,8 +17,8 @@ def web_search(arg1:str)-> str: #it's import to specify the return type
17
  arg1: the first argument
18
  arg2: the second argument
19
  """
20
- client = TavilyClient("tvly-dev-2wT5SA4C7gtWyKi7WzcP8C00P")
21
- response = client.search(query)
22
  return(response)
23
 
24
  @tool
@@ -59,7 +59,7 @@ with open("prompts.yaml", 'r') as stream:
59
 
60
  agent = CodeAgent(
61
  model=model,
62
- tools=[final_answer,image_generation_tool,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
63
  max_steps=6,
64
  verbosity_level=1,
65
  grammar=None,
 
17
  arg1: the first argument
18
  arg2: the second argument
19
  """
20
+ client = TavilyClient("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")
21
+ response = client.search(arg1)
22
  return(response)
23
 
24
  @tool
 
59
 
60
  agent = CodeAgent(
61
  model=model,
62
+ tools=[final_answer,image_generation_tool,get_current_time_in_timezone,web_search], ## add your tools here (don't remove final answer)
63
  max_steps=6,
64
  verbosity_level=1,
65
  grammar=None,