Update agent.py

agent.py

@@ -53,14 +53,15 @@ class GaiaAgent:
             return str(output)
 
     def answer_question(self, question: Dict) -> str:
-        q = question["question"]
-        print(f"📌 [QUESTION] ID: {question.get('id', 'N/A')} - {q}")
-
-        prompt = f"""You are a helpful agent answering a science question.
-Question: {q}
-Answer:"""
-
-        return self.generate(prompt).strip()
+        # Try different keys that might contain the question
+        q = question.get("question") or question.get("Question") or question.get("input")
+        if not q:
+            raise ValueError(f"No question text found in: {question}")
+    
+       prompt = f"""You are a helpful agent answering a science question.
+    Question: {q}
+    Answer:"""
+      return self.generate(prompt).strip()
 
     def run(self):
         print("🚀 [RUN] Starting submission...")