vietvo commited on
Commit
2050dd3
·
verified ·
1 Parent(s): 6d6ba1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -17,7 +17,6 @@ from Gradio_UI import GradioUI
17
  # arg2: the second argument
18
  # """
19
  # return "What magic will you build ?"
20
-
21
  def calculator(a: int, b: int)-> int: #it's import to specify the return type
22
  #Keep this format for the description / args / args description but feel free to modify the tool
23
  """Multiply two integers
@@ -64,7 +63,7 @@ with open("prompts.yaml", 'r') as stream:
64
 
65
  agent = CodeAgent(
66
  model=model,
67
- tools=[final_answer], ## add your tools here (don't remove final answer)
68
  max_steps=6,
69
  verbosity_level=1,
70
  grammar=None,
 
17
  # arg2: the second argument
18
  # """
19
  # return "What magic will you build ?"
 
20
  def calculator(a: int, b: int)-> int: #it's import to specify the return type
21
  #Keep this format for the description / args / args description but feel free to modify the tool
22
  """Multiply two integers
 
63
 
64
  agent = CodeAgent(
65
  model=model,
66
+ tools=[calculator, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
67
  max_steps=6,
68
  verbosity_level=1,
69
  grammar=None,