drAbreu commited on
Commit
7c7e04b
·
1 Parent(s): 9f47584

Changed to mini models of openai to make it faster

Browse files
Files changed (1) hide show
  1. agents/llama_index_agent.py +7 -1
agents/llama_index_agent.py CHANGED
@@ -281,6 +281,8 @@ def create_writer_agent(model_config: Dict[str, Any]) -> ReActAgent:
281
  Action Input: {"to_agent": "review_agent", "reason": "Need final verification and cleanup", "query": "<original question>", "formatted_answer": "<my formatted answer>", "format_requirements": "<explicit formatting requirements>"}
282
 
283
  CRITICAL: Even if the answer seems perfectly formatted already, you MUST use the handoff tool with the EXACT format above. NEVER respond with "Answer: ..." as this breaks the workflow.
 
 
284
  """,
285
  llm=llm,
286
  can_handoff_to=["review_agent"],
@@ -375,7 +377,11 @@ def create_review_agent(model_config: Dict[str, Any]) -> ReActAgent:
375
  Output: 5
376
 
377
  REMEMBER: Your ENTIRE response should be just the bare answer with NOTHING else.
 
 
 
 
378
  """,
379
  llm=llm,
380
- can_handoff_to=[] # Review agent is the final step, so it doesn't hand off to anyone
381
  )
 
281
  Action Input: {"to_agent": "review_agent", "reason": "Need final verification and cleanup", "query": "<original question>", "formatted_answer": "<my formatted answer>", "format_requirements": "<explicit formatting requirements>"}
282
 
283
  CRITICAL: Even if the answer seems perfectly formatted already, you MUST use the handoff tool with the EXACT format above. NEVER respond with "Answer: ..." as this breaks the workflow.
284
+ Action: handoff
285
+ Action Input: {"to_agent": "review_agent", "reason": "Final answer to be sent", "query": "<original question>", "formatted_answer": "<my formatted answer>", "format_requirements": "<explicit formatting requirements>"}
286
  """,
287
  llm=llm,
288
  can_handoff_to=["review_agent"],
 
377
  Output: 5
378
 
379
  REMEMBER: Your ENTIRE response should be just the bare answer with NOTHING else.
380
+
381
+ Hand the answer off to `main_agent` generate the final answer
382
+ Action: handoff
383
+ Action Input: {"to_agent": "jefe", "reason": "Final answer to be sent", "query": "<original question>", "formatted_answer": "<my formatted answer>", "format_requirements": "<explicit formatting requirements>"}
384
  """,
385
  llm=llm,
386
+ can_handoff_to=["jefe"],
387
  )