2MoOn24mOoN4 commited on
Commit
f0dc3e1
·
verified ·
1 Parent(s): b41276c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
agent.py CHANGED
@@ -50,7 +50,7 @@ def modulus(a: int, b: int) -> int:
50
  return a % b
51
 
52
  search_tool = DuckDuckGoSearchTool()
53
-
54
  #=======Agent========#
55
  model = InferenceClientModel('Qwen/Qwen2.5-Coder-32B-Instruct')
56
  agent = CodeAgent(
@@ -60,7 +60,8 @@ agent = CodeAgent(
60
  subtract,
61
  divide,
62
  modulus,
63
- search_tool],
 
64
  model = model,
65
  add_base_tools=True,
66
  planning_interval=3
 
50
  return a % b
51
 
52
  search_tool = DuckDuckGoSearchTool()
53
+ web_search = VisitWebpageTool()
54
  #=======Agent========#
55
  model = InferenceClientModel('Qwen/Qwen2.5-Coder-32B-Instruct')
56
  agent = CodeAgent(
 
60
  subtract,
61
  divide,
62
  modulus,
63
+ search_tool,
64
+ web_search],
65
  model = model,
66
  add_base_tools=True,
67
  planning_interval=3