Spaces:
Sleeping
Sleeping
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 |
)
|