DTStudios commited on
Commit
98d293e
·
verified ·
1 Parent(s): 3cba5f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -20,7 +20,7 @@ def calculator(a: int, b: int) -> int:
20
  """
21
  return a * b
22
 
23
- print(calculator.to_string())
24
 
25
  @tool
26
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -79,10 +79,13 @@ tool_registry = {
79
  "enabled": True
80
  }
81
  }
82
-
 
 
 
83
  agent = CodeAgent(
84
  model=model,
85
- tools=[final_answer], ## add your tools here (don't remove final answer)
86
  max_steps=6,
87
  verbosity_level=1,
88
  grammar=None,
 
20
  """
21
  return a * b
22
 
23
+ print(calculator.tool_schema())
24
 
25
  @tool
26
  def get_current_time_in_timezone(timezone: str) -> str:
 
79
  "enabled": True
80
  }
81
  }
82
+ from registry import tool_registry
83
+
84
+ agent_tools = [entry["tool"] for entry in tool_registry.values() if entry["enabled"]
85
+
86
  agent = CodeAgent(
87
  model=model,
88
+ tools=[final_answer], [calculator], [get_current_time_in_timezone]
89
  max_steps=6,
90
  verbosity_level=1,
91
  grammar=None,