Arslan1997 commited on
Commit
f85c79e
·
1 Parent(s): 11794cc
Files changed (2) hide show
  1. src/agents/agents.py +6 -1
  2. test_datasets.ipynb +0 -0
src/agents/agents.py CHANGED
@@ -2004,8 +2004,13 @@ class auto_analyst(dspy.Module):
2004
  response = await self.agents['basic_qa_agent'](**inputs)
2005
  yield 'basic_qa_agent', inputs, response
2006
  return
 
2007
 
2008
- plan_list = [agent.strip() for agent in plan_text.split("->") if agent.strip()]
 
 
 
 
2009
 
2010
 
2011
 
 
2004
  response = await self.agents['basic_qa_agent'](**inputs)
2005
  yield 'basic_qa_agent', inputs, response
2006
  return
2007
+
2008
 
2009
+ plan_list = []
2010
+ for agent in [a.strip() for a in plan_text.split("->") if a.strip()]:
2011
+ if not agent.startswith("planner_"):
2012
+ agent = "planner_" + agent
2013
+ plan_list.append(agent)
2014
 
2015
 
2016
 
test_datasets.ipynb ADDED
File without changes