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

Review agent added

Browse files
Files changed (1) hide show
  1. agents/llama_index_agent.py +4 -1
agents/llama_index_agent.py CHANGED
@@ -84,7 +84,8 @@ class GaiaAgent(ReActAgent):
84
  system_prompt = self._get_default_system_prompt()
85
 
86
  can_handoff_to = [
87
- "writer_agent"
 
88
  ]
89
 
90
  # Initialize the parent ReActAgent
@@ -358,6 +359,8 @@ def create_review_agent(model_config: Dict[str, Any]) -> ReActAgent:
358
  Output: Tom Hanks, Meg Ryan, Bill Pullman
359
  - Input: "She published studio albums "Album 1", "Album 2", "Album 3", so in total 3."
360
  Output: 3
 
 
361
 
362
  REMEMBER: Your ENTIRE response should be just the bare answer with NOTHING else.
363
  """,
 
84
  system_prompt = self._get_default_system_prompt()
85
 
86
  can_handoff_to = [
87
+ "writer_agent",
88
+ "review_agent"
89
  ]
90
 
91
  # Initialize the parent ReActAgent
 
359
  Output: Tom Hanks, Meg Ryan, Bill Pullman
360
  - Input: "She published studio albums "Album 1", "Album 2", "Album 3", so in total 3."
361
  Output: 3
362
+ - Input: Misa Criolla (2000), Acústico (2002), Corazón Libre (2005), Cantora 1 (2009), and Cantora 2 (2009)
363
+ Output 5
364
 
365
  REMEMBER: Your ENTIRE response should be just the bare answer with NOTHING else.
366
  """,