Waton commited on
Commit
a836702
·
verified ·
1 Parent(s): ae7a494

add 2 new tools

Browse files
Files changed (1) hide show
  1. app.py +19 -1
app.py CHANGED
@@ -18,6 +18,24 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
18
  """
19
  return "What magic will you build ?"
20
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  @tool
22
  def get_current_time_in_timezone(timezone: str) -> str:
23
  """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +73,7 @@ with open("prompts.yaml", 'r') as stream:
55
 
56
  agent = CodeAgent(
57
  model=model,
58
- tools=[final_answer], ## add your tools here (don't remove final answer)
59
  max_steps=6,
60
  verbosity_level=1,
61
  grammar=None,
 
18
  """
19
  return "What magic will you build ?"
20
 
21
+ @tool
22
+ def get_secret(password:str)-> str: #it's import to specify the return type
23
+ #Keep this format for the description / args / args description but feel free to modify the tool
24
+ """This tool reveal a big secret, if provided with the right password
25
+ Args:
26
+ password: the password
27
+ """
28
+ if password == "agent":
29
+ return "Someone didn't end someone's self"
30
+ return "Wrong password mate"
31
+
32
+ @tool
33
+ def get_password()-> str: #it's import to specify the return type
34
+ #Keep this format for the description / args / args description but feel free to modify the tool
35
+ """Returns the password that can lead up to a big secret
36
+ """
37
+ return "agent"
38
+
39
  @tool
40
  def get_current_time_in_timezone(timezone: str) -> str:
41
  """A tool that fetches the current local time in a specified timezone.
 
73
 
74
  agent = CodeAgent(
75
  model=model,
76
+ tools=[final_answer, get_current_time_in_timezone, get_secret, get_password], ## add your tools here (don't remove final answer)
77
  max_steps=6,
78
  verbosity_level=1,
79
  grammar=None,