Initial commit

.gitignore

@@ -0,0 +1,3 @@
+.env
+notebooks/
+.venv/

agents/llama_index_agent.py

@@ -0,0 +1,57 @@
+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
+    ]
+)

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-
+from agents.llama_index_agent import main_agent
+import asyncio
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -15,7 +16,12 @@ class BasicAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
+        base_agent = main_agent()
+        async def agentic_main():
+            response = await base_agent.run(question)
+        response = asyncio.run(agentic_main())
+        print(response)
+        exit()
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 

requirements.txt

@@ -1,2 +1,11 @@
 gradio
-requests
+requests
+llama-index
+llama-index-tools-wikipedia
+llama-index-tools-tavily-research
+nest_asyncio
+certifi
+board_to_fen
+keras==2.11
+tensorflow==2.13.0rc0
+numpy==1.23.5

tools/text_tools.py

@@ -0,0 +1,13 @@
+from llama_index.core.tools import FunctionTool
+
+
+def reverse_text(text: str) -> str:
+    """It returns the reversed string of text in the input."""
+    return text[::-1]
+
+
+reverse_text_tool = FunctionTool.from_defaults(
+    reverse_text,
+    name="reverse_text_tool",
+    description="It returns the reversed string of text in the input.",
+)

youtube_analysis.py

@@ -0,0 +1 @@
+