drAbreu commited on
Commit
1c0f5f2
·
1 Parent(s): e89cf5e

Review agent added

Browse files
Files changed (2) hide show
  1. agents/llama_index_agent.py +5 -3
  2. 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 audio file to {local_file_path}")
135
  except Exception as e:
136
- print(f"Error downloading audio file: {e}")
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