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 |
)
|