Spaces:
Running
Running
Update agent.py
Browse files
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 =
|
|
|
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 |
)
|