Update app.py

app.py

@@ -26,11 +26,38 @@ class GaiaAgent:
             tools=tools
         )
 
-    def __call__(self, query: str) -> str:
-        return self.agent.run(query)
-
-
-
+    #def __call__(self, query: str) -> str:
+        #return self.agent.run(query)
+
+    #def __call__(self, query: str) -> str:
+        #result = self.agent.run(query)
+        #return result.get("output", "No output returned")
+
+    def __call__(self, question: str) -> str:
+        prompt = f"""You are an agent solving the GAIA benchmark and you are required to provide exact answers.
+        Rules to follow:
+        1. Return only the exact requested answer: no explanation and no reasoning.
+        2. For yes/no questions, return exactly "Yes" or "No".
+        3. For dates, use the exact format requested.
+        4. For numbers, use the exact number, no other format.
+        5. For names, use the exact name as found in sources.
+        6. If the question has an associated file, download the file first using the task ID.
+
+        Examples of good responses:
+        - "42"
+        - "Pinco Palla"
+        - "Yes"
+        - "October 5, 2001"
+        - "Buenos Aires"
+
+        Never include phrases like "the answer is..." or "Based on my research".
+        Only return the exact answer.
+
+        QUESTION:
+        {question}
+        """
+        result = self.agent.run(prompt)
+        return result.get("output", "No output returned")