vjtulsiyan commited on
Commit
14daa10
·
verified ·
1 Parent(s): ae7a494

update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -11,12 +11,16 @@ from Gradio_UI import GradioUI
11
  @tool
12
  def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
  #Keep this format for the description / args / args description but feel free to modify the tool
14
- """A tool that does nothing yet
 
15
  Args:
16
- arg1: the first argument
17
- arg2: the second argument
18
  """
19
- return "What magic will you build ?"
 
 
 
20
 
21
  @tool
22
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +59,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,
 
11
  @tool
12
  def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
  #Keep this format for the description / args / args description but feel free to modify the tool
14
+ """A tool that takes the name of the person as an input and an interger number as another input. The tool checks if the number is greater than or eual to 18,
15
+ then it classifies the person as adult else as a minor.
16
  Args:
17
+ arg1: the name of the person
18
+ arg2: age of the person as integer
19
  """
20
+ if arg2 >= 18:
21
+ return f" {arg1} is classified as an adult as his age is {arg2} and its >= 18 years."
22
+ else:
23
+ return f" {arg1} is classified as a minor as his age is {arg2} and its < 18 years."
24
 
25
  @tool
26
  def get_current_time_in_timezone(timezone: str) -> str:
 
59
 
60
  agent = CodeAgent(
61
  model=model,
62
+ tools=[final_answer, my_custom_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,