yetessam commited on
Commit
e55e264
·
verified ·
1 Parent(s): edeb4b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -9,7 +9,7 @@ from checks.endpoint_check import is_huggingface_endpoint, is_public_endpoint
9
  import os
10
  import pytz # Had to give it permission in Code agent
11
 
12
- from tools.tools import tools
13
 
14
  from agents.model import load_huggingface_model
15
  from agents.prompts import load_prompts
@@ -56,10 +56,11 @@ model = LiteLLMModel(
56
 
57
  # Load prompts
58
  combined_prompts = load_prompts()
 
59
 
60
  agent = CodeAgent(
61
  model=model,
62
- tools=[final_answer, polite_guard, web_search, get_the_current_time_in_timezone ], ## add your tools here (don't remove final answer)
63
  max_steps=6,
64
  verbosity_level=3,
65
  grammar=None,
 
9
  import os
10
  import pytz # Had to give it permission in Code agent
11
 
12
+ from tools.tools import load_tools
13
 
14
  from agents.model import load_huggingface_model
15
  from agents.prompts import load_prompts
 
56
 
57
  # Load prompts
58
  combined_prompts = load_prompts()
59
+ combined_tools = load_tools() ## from the tools folder (don't remove final answer)
60
 
61
  agent = CodeAgent(
62
  model=model,
63
+ tools=combined_tools,
64
  max_steps=6,
65
  verbosity_level=3,
66
  grammar=None,