Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -278,8 +278,8 @@ class MagAgent:
|
|
278 |
# )
|
279 |
|
280 |
# Load prompt templates
|
281 |
-
with open("prompts.yaml", 'r') as stream:
|
282 |
-
prompt_templates = yaml.safe_load(stream)
|
283 |
|
284 |
# Initialize rate limiter for DuckDuckGoSearchTool
|
285 |
search_rate_limiter = Limiter(rate=10/60, capacity=10, storage=MemoryStorage()) if not rate_limiter else rate_limiter
|
@@ -301,7 +301,7 @@ class MagAgent:
|
|
301 |
# ImageAnalysisTool,
|
302 |
],
|
303 |
verbosity_level=2,
|
304 |
-
prompt_templates=prompt_templates,
|
305 |
add_base_tools=False,
|
306 |
max_steps=15
|
307 |
)
|
|
|
278 |
# )
|
279 |
|
280 |
# Load prompt templates
|
281 |
+
# with open("prompts.yaml", 'r') as stream:
|
282 |
+
# prompt_templates = yaml.safe_load(stream)
|
283 |
|
284 |
# Initialize rate limiter for DuckDuckGoSearchTool
|
285 |
search_rate_limiter = Limiter(rate=10/60, capacity=10, storage=MemoryStorage()) if not rate_limiter else rate_limiter
|
|
|
301 |
# ImageAnalysisTool,
|
302 |
],
|
303 |
verbosity_level=2,
|
304 |
+
# prompt_templates=prompt_templates,
|
305 |
add_base_tools=False,
|
306 |
max_steps=15
|
307 |
)
|