GoldsWolf commited on
Commit
21209b1
·
verified ·
1 Parent(s): dcee3c9

Update app.py fix errors

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,6 +4,7 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
7
 
8
  from Gradio_UI import GradioUI
9
 
@@ -26,7 +27,6 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
26
  "{}: Your spaceship’s toaster gains sentience tomorrow. Be nice to it.".format(arg1)
27
  ]
28
 
29
- import random
30
  return random.choice(fortunes)
31
 
32
  @tool
@@ -66,7 +66,7 @@ with open("prompts.yaml", 'r') as stream:
66
 
67
  agent = CodeAgent(
68
  model=model,
69
- tools=[final_answer], ## add your tools here (don't remove final answer)
70
  max_steps=6,
71
  verbosity_level=1,
72
  grammar=None,
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+ import random
8
 
9
  from Gradio_UI import GradioUI
10
 
 
27
  "{}: Your spaceship’s toaster gains sentience tomorrow. Be nice to it.".format(arg1)
28
  ]
29
 
 
30
  return random.choice(fortunes)
31
 
32
  @tool
 
66
 
67
  agent = CodeAgent(
68
  model=model,
69
+ tools=[my_custom_tool,final_answer], ## add your tools here (don't remove final answer)
70
  max_steps=6,
71
  verbosity_level=1,
72
  grammar=None,