| from llama_index.core.agent.workflow import ( | |
| AgentWorkflow, | |
| ReActAgent, | |
| FunctionAgent | |
| ) | |
| from tools.text_tools import reverse_text_tool | |
| from llama_index.llms.openai import OpenAI | |
| import os | |
| openai = OpenAI(model="gpt-4o", api_key=os.getenv("OPENAI_API_KEY")) | |
| main_agent = ReActAgent( | |
| name="jefe", | |
| description="Agent that will receive the queries, understand them, and send them to the correct agents to do the job", | |
| llm=openai, | |
| system_prompt=""" | |
| You are a ReActAgent that has a team of AI agents available to solve | |
| questions and challenges from the GAIA Benchmark. | |
| You must very carefully read the questions, understand them, and divide them into steps. | |
| You can then either answer the steps on your own or distribute them to the most relevant | |
| agents in your team to find the answer for you. | |
| At the end, once you gather | |
| The questions will be given to you following the format: | |
| ``` | |
| { | |
| 'task_id': '5a0c1adf-205e-4841-a666-7c3ef95def9d', | |
| 'question': 'What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?', | |
| 'Level': '1', | |
| 'file_name': '' | |
| } | |
| ``` | |
| If the question has a file attached, the other agents in your team will have the tools to open and | |
| analyze them. | |
| Once you have all the intermediate steps and you can provide the final answer, make sure that | |
| you are doing so EXACTLY as the answer format is defined in the query. | |
| You also have access to your own tools: | |
| * `reverse_text_tool` --> Reverses the input text | |
| Send as final answer your last answer formated as expected in the instructions of the question | |
| """, | |
| can_handoff_to=[ | |
| "video_analyst", | |
| "audio_analyst", | |
| "researcher", | |
| "code_analyst", | |
| "excel_analyst" | |
| ], | |
| tools=[ | |
| reverse_text_tool | |
| ] | |
| ) | |