init

agent.py

@@ -0,0 +1,87 @@
+from llama_index.llms.openai import OpenAI
+from llama_index.tools.wikipedia.base import WikipediaToolSpec
+from llama_index.core.llms import ChatMessage
+from llama_index.core.agent import ReActAgent
+import logging
+from llama_index.llms.deepinfra import DeepInfraLLM
+import os
+from llama_index.tools.tavily_research import TavilyToolSpec
+from llama_index.core.prompts import PromptTemplate
+import requests
+import json
+
+class CuongBasicAgent:
+    """
+    Agent using LlamaIndex to fetch data from the web and answer GAIA benchmark questions.
+    """
+    def __init__(self):
+        system_prompt = """
+        Value: You are an advanced assistant designed to help with a variety of tasks, including answering questions, providing summaries, and performing other types of analyses.
+
+        ## Tools
+
+        You have access to a wide variety of tools. You are responsible for using the tools in any sequence you deem appropriate to complete the task at hand.
+        This may require breaking the task into subtasks and using different tools to complete each subtask.
+
+        You have access to the following tools:
+        {tool_desc}
+
+
+        ## Output Format
+
+        Please answer in the same language as the question and use the following format:
+
+        ```
+        Thought: The current language of the user is: (user's language). I need to use a tool to help me answer the question.
+        Action: tool name (one of {tool_names}) if using a tool.
+        Action Input: the input to the tool, in a JSON format representing the kwargs (e.g. {{"input": "hello world", "num_beams": 5}})
+        ```
+
+        Please ALWAYS start with a Thought.
+
+        NEVER surround your response with markdown code markers. You may use code markers within your response if you need to.
+
+        Please use a valid JSON format for the Action Input. Do NOT do this {{'input': 'hello world', 'num_beams': 5}}.
+
+        If this format is used, the tool will respond in the following format:
+
+        ```
+        Observation: tool response
+        ```
+
+        You should keep repeating the above format till you have enough information to answer the question without using any more tools. At that point, you MUST respond in one of the following two formats:
+
+        ```
+        Thought: I can answer without using any more tools. I'll use the user's language to answer
+        Answer: [your answer here (In the same language as the user's question)]
+        ```
+
+        ```
+        Thought: I cannot answer the question with the provided tools.
+        Answer: [your answer here (In the same language as the user's question)]
+        ```
+
+        The answer should be concise and to the point. For example, if the answer is a number, just return the number without any additional text. If the question is "What is the capital of France?", the answer should be "Paris".
+
+        If the question includes guidelines regarding the format of the answer, please follow those guidelines faithfully
+        ## Current Conversation
+
+        Below is the current conversation consisting of interleaving human and assistant messages.
+        """
+        react_system_prompt = PromptTemplate(system_prompt)
+        #llm = DeepInfraLLM(
+        #    model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",api_key=os.getenv("DEEPINFRA_API_KEY"))
+
+        llm = OpenAI(model='gpt-4.1')
+        agent = ReActAgent.from_tools(
+            llm=llm,
+            tools=WikipediaToolSpec().to_tool_list() + TavilyToolSpec(api_key=os.getenv('TAVILY_API_KEY')).to_tool_list(),
+            verbose=True,
+        )
+
+        agent.update_prompts({"agent_worker:system_prompt": react_system_prompt})
+        self.agent = agent
+
+    def __call__(self, question: str) -> str:
+        answer = self.agent.query(question)
+        return str(answer)

app.py

@@ -3,22 +3,12 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from agent import CuongBasicAgent
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        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."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
-
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -30,6 +20,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     if profile:
         username= f"{profile.username}"
         print(f"User logged in: {username}")
+        if username != os.getenv("whoami"):
+            return f"You are not {os.getenv('whoami')}", None
     else:
         print("User not logged in.")
         return "Please Login to Hugging Face with the button.", None
@@ -37,10 +29,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
+    files_url = f"{api_url}/files/"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = CuongBasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -79,8 +72,16 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+
+        # Manage questions with files
+        if item.get("file_name") and os.path.splitext(item.get("file_name"))[1] in ['.py', '.txt', '.json']:
+            file = requests.get(files_url+task_id, timeout=15).text
+            complete_question_text = f'{question_text}\nThis is the accompanying file:\n{file}'
+        else:
+            complete_question_text = question_text
+            
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(complete_question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:
@@ -99,7 +100,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # 5. Submit
     print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
     try:
-        response = requests.post(submit_url, json=submission_data, timeout=60)
+        response = requests.post(submit_url, json=submission_data, timeout=180)
         response.raise_for_status()
         result_data = response.json()
         final_status = (

requirements.txt

@@ -1,2 +1,7 @@
 gradio
-requests
+requests
+pandas
+llama-index-llms-openai
+llama-index-tools-wikipedia
+llama-index-llms-deepinfra
+llama-index-tools-tavily_research