Upload agent.py

agent.py

@@ -78,12 +78,16 @@ class GAIAAgent(CodeAgent):
     # Convenience so the object itself can be *called* directly
     def __call__(self, question: str, **kwargs: Any) -> str:
         steps = self.run(question, **kwargs)
+        if isinstance(steps, str):
+            print("DEBUG: steps is a string:", repr(steps))
+            return steps.strip()
+        print("DEBUG: steps is a", type(steps))
         last_step = None
         for step in steps:
             last_step = step
-        # If last_step is a FinalAnswerStep with .answer, return it
+        print("DEBUG: last_step =", repr(last_step))
         answer = getattr(last_step, "answer", None)
-        # Always return as a string, with stripping for safety
+        print("DEBUG: answer =", repr(answer))
         if answer is not None:
             return str(answer).strip()
         return str(last_step).strip()