update app.py
#194
by
vjtulsiyan
- opened
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
|
|
|
15 |
Args:
|
16 |
-
arg1: the
|
17 |
-
arg2: the
|
18 |
"""
|
19 |
-
|
|
|
|
|
|
|
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,
|