SergeyO7 commited on
Commit
a518e5e
·
verified ·
1 Parent(s): ce96e25

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
agent.py CHANGED
@@ -266,7 +266,8 @@ class PythonCodeReaderTool(Tool):
266
  class MagAgent:
267
  def __init__(self, rate_limiter: Optional[Limiter] = None):
268
  """Initialize the MagAgent with search tools."""
269
- self.rate_limiter = rate_limiter
 
270
 
271
  print("Initializing MagAgent with search tools...")
272
  model = LiteLLMModel(
@@ -296,7 +297,7 @@ class MagAgent:
296
  # ImageAnalysisTool,
297
  ],
298
  verbosity_level=2,
299
- prompt_templates=prompt_templates
300
  add_base_tools=True,
301
  max_steps=15
302
  )
 
266
  class MagAgent:
267
  def __init__(self, rate_limiter: Optional[Limiter] = None):
268
  """Initialize the MagAgent with search tools."""
269
+ self.rate_limiter =
270
+ rate_limiter
271
 
272
  print("Initializing MagAgent with search tools...")
273
  model = LiteLLMModel(
 
297
  # ImageAnalysisTool,
298
  ],
299
  verbosity_level=2,
300
+ prompt_templates=prompt_templates,
301
  add_base_tools=True,
302
  max_steps=15
303
  )