Changed to mini models of openai to make it faster
Browse files
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=[]
|
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 |
)
|