KingAj23 commited on
Commit
130356b
·
verified ·
1 Parent(s): e3c6999

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -9,15 +9,6 @@ from Gradio_UI import GradioUI
9
 
10
 
11
  # Below is an example of a tool that does nothing. Amaze us with your creativity!
12
- @tool
13
- def my_custom_tool(arg1:str, arg2:int)-> str: # it's important to specify the return type
14
- # Keep this format for the tool description / args description but feel free to modify the tool
15
- """A tool that does nothing yet
16
- Args:
17
- arg1: the first argument
18
- arg2: the second argument
19
- """
20
- return "What magic will you build ?"
21
 
22
  @tool
23
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -81,6 +72,11 @@ agent = CodeAgent(
81
  description=None,
82
  prompt_templates=prompt_templates # Pass system prompt to CodeAgent
83
  )
 
 
 
 
84
 
 
85
 
86
  GradioUI(agent).launch()
 
9
 
10
 
11
  # Below is an example of a tool that does nothing. Amaze us with your creativity!
 
 
 
 
 
 
 
 
 
12
 
13
  @tool
14
  def get_current_time_in_timezone(timezone: str) -> str:
 
72
  description=None,
73
  prompt_templates=prompt_templates # Pass system prompt to CodeAgent
74
  )
75
+ @tool
76
+ def calculator(a: int, b: int) -> int:
77
+ """Multiply two integers."""
78
+ return a * b
79
 
80
+ print(calculator.to_string())
81
 
82
  GradioUI(agent).launch()