SergeyO7 commited on
Commit
b518461
·
verified ·
1 Parent(s): 6557ac2

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
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
  )