Coool2 commited on
Commit
671c833
·
1 Parent(s): 08de0c8

Update agent2.py

Browse files
Files changed (1) hide show
  1. agent2.py +3 -4
agent2.py CHANGED
@@ -15,7 +15,6 @@ import helium
15
  from PIL import Image
16
  from io import BytesIO
17
  from time import sleep
18
- from smolagents import PythonInterpreterTool, SpeechToTextTool
19
 
20
  # Langfuse observability imports
21
  from opentelemetry.sdk.trace import TracerProvider
@@ -280,10 +279,10 @@ Your final answer should be as few words as possible, a number, or a comma-separ
280
  ]
281
 
282
  self.agent = CodeAgent(
283
- tools=base_tools + [PythonInterpreterTool(), SpeechToTextTool()],
284
  model=self.model,
285
- add_base_tools=False,
286
- planning_interval=2,
287
  additional_authorized_imports=["helium", "requests", "BeautifulSoup", "json"],
288
  step_callbacks=[save_screenshot_callback] if self.driver else [],
289
  max_steps=10,
 
15
  from PIL import Image
16
  from io import BytesIO
17
  from time import sleep
 
18
 
19
  # Langfuse observability imports
20
  from opentelemetry.sdk.trace import TracerProvider
 
279
  ]
280
 
281
  self.agent = CodeAgent(
282
+ tools=base_tools,
283
  model=self.model,
284
+ add_base_tools=True,
285
+ planning_interval=3,
286
  additional_authorized_imports=["helium", "requests", "BeautifulSoup", "json"],
287
  step_callbacks=[save_screenshot_callback] if self.driver else [],
288
  max_steps=10,