Update app.py

app.py

@@ -11,6 +11,43 @@ import pandas as pd
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
+class BasicAgent:
+    """
+    This is the agent class that the GAIA test harness will use.
+    """
+    def __init__(self):
+        # The compiled LangGraph app is our agent executor
+        self.agent_executor = app
+
+    def run(self, question: str) -> str:
+        """
+        This method is called by the test script with each question.
+        It runs the LangGraph agent and returns the final answer.
+        """
+        print(f"Agent received question (first 80 chars): {question[:80]}...")
+        try:
+            # Format the input for our graph
+            inputs = {"messages": [HumanMessage(content=question)]}
+
+            # Stream the response to get the final answer
+            final_response = ""
+            for s in self.agent_executor.stream(inputs, {"recursion_limit": 15}):
+                if "agent" in s:
+                    # The final answer is the content of the last message from the agent node
+                    if s["agent"]["messages"][-1].content:
+                         final_response = s["agent"]["messages"][-1].content
+
+            # A fallback in case the agent finishes without a clear message
+            if not final_response:
+                final_response = "Agent finished but did not produce a final answer."
+
+            print(f"Agent returning final answer (first 80 chars): {final_response[:80]}...")
+            return final_response
+
+        except Exception as e:
+            print(f"An error occurred in agent execution: {e}")
+            return f"Error: {e}"
+
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):