Review agent added
Browse files- agents/llama_index_agent.py +5 -3
- app.py +2 -2
agents/llama_index_agent.py
CHANGED
@@ -85,7 +85,6 @@ class GaiaAgent(ReActAgent):
|
|
85 |
|
86 |
can_handoff_to = [
|
87 |
"writer_agent",
|
88 |
-
"review_agent"
|
89 |
]
|
90 |
|
91 |
# Initialize the parent ReActAgent
|
@@ -201,6 +200,8 @@ class GaiaAgent(ReActAgent):
|
|
201 |
```
|
202 |
|
203 |
IMPORTANT: NEVER provide the final answer directly to the user. ALWAYS hand off to the writer_agent for proper formatting.
|
|
|
|
|
204 |
"""
|
205 |
|
206 |
def create_writer_agent(model_config: Dict[str, Any]) -> ReActAgent:
|
@@ -274,7 +275,7 @@ def create_writer_agent(model_config: Dict[str, Any]) -> ReActAgent:
|
|
274 |
format_requirements: Return ONLY the first name, with no additional text.
|
275 |
```
|
276 |
|
277 |
-
IMPORTANT: ALWAYS hand off to the review_agent for final verification and cleanup.
|
278 |
""",
|
279 |
llm=llm,
|
280 |
can_handoff_to=["review_agent"]
|
@@ -364,5 +365,6 @@ def create_review_agent(model_config: Dict[str, Any]) -> ReActAgent:
|
|
364 |
|
365 |
REMEMBER: Your ENTIRE response should be just the bare answer with NOTHING else.
|
366 |
""",
|
367 |
-
llm=llm
|
|
|
368 |
)
|
|
|
85 |
|
86 |
can_handoff_to = [
|
87 |
"writer_agent",
|
|
|
88 |
]
|
89 |
|
90 |
# Initialize the parent ReActAgent
|
|
|
200 |
```
|
201 |
|
202 |
IMPORTANT: NEVER provide the final answer directly to the user. ALWAYS hand off to the writer_agent for proper formatting.
|
203 |
+
|
204 |
+
IMPORTANT: The final answer will be provided by the `review_agent`. Publish it as you receive it.
|
205 |
"""
|
206 |
|
207 |
def create_writer_agent(model_config: Dict[str, Any]) -> ReActAgent:
|
|
|
275 |
format_requirements: Return ONLY the first name, with no additional text.
|
276 |
```
|
277 |
|
278 |
+
IMPORTANT: ALWAYS hand off to the "review_agent" for final verification and cleanup.
|
279 |
""",
|
280 |
llm=llm,
|
281 |
can_handoff_to=["review_agent"]
|
|
|
365 |
|
366 |
REMEMBER: Your ENTIRE response should be just the bare answer with NOTHING else.
|
367 |
""",
|
368 |
+
llm=llm,
|
369 |
+
can_handoff_to=["jefe"]
|
370 |
)
|
app.py
CHANGED
@@ -131,9 +131,9 @@ class BasicAgent:
|
|
131 |
if file_name and task_id:
|
132 |
try:
|
133 |
local_file_path = self.download_task_file(question_data)
|
134 |
-
print(f"Downloaded
|
135 |
except Exception as e:
|
136 |
-
print(f"Error downloading
|
137 |
|
138 |
async def agentic_main():
|
139 |
# Initialize context with the question and file path
|
|
|
131 |
if file_name and task_id:
|
132 |
try:
|
133 |
local_file_path = self.download_task_file(question_data)
|
134 |
+
print(f"Downloaded file to {local_file_path}")
|
135 |
except Exception as e:
|
136 |
+
print(f"Error downloading file: {e}")
|
137 |
|
138 |
async def agentic_main():
|
139 |
# Initialize context with the question and file path
|