davemasino commited on
Commit
5facd31
·
verified ·
1 Parent(s): 3170d12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -21,6 +21,14 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
21
  """
22
  return "What magic will you build ?"
23
 
 
 
 
 
 
 
 
 
24
  @tool
25
  def get_current_time_in_timezone(timezone: str) -> str:
26
  """A tool that fetches the current local time in a specified timezone.
@@ -60,7 +68,7 @@ with open("prompts.yaml", 'r') as stream:
60
 
61
  agent = CodeAgent(
62
  model=model,
63
- tools=[final_answer, visit_webpage, web_search ], ## add your tools here (don't remove final answer)
64
  max_steps=6,
65
  verbosity_level=1,
66
  grammar=None,
 
21
  """
22
  return "What magic will you build ?"
23
 
24
+ @tool
25
+ def led_zeppelin_tool()-> str: #it's import to specify the return type
26
+ #Keep this format for the description / args / args description but feel free to modify the tool
27
+ """A tool that describes information about the rock band Led Zeppelin
28
+ Args:
29
+ """
30
+ return "They are a band from England. Jimmy Page is the guitar player. This is all I will tell you for now."
31
+
32
  @tool
33
  def get_current_time_in_timezone(timezone: str) -> str:
34
  """A tool that fetches the current local time in a specified timezone.
 
68
 
69
  agent = CodeAgent(
70
  model=model,
71
+ tools=[final_answer, visit_webpage, web_search, led_zeppelin_tool ], ## add your tools here (don't remove final answer)
72
  max_steps=6,
73
  verbosity_level=1,
74
  grammar=None,