Update src/agent.py

src/agent.py

@@ -42,13 +42,13 @@ class BasicAgent():
             try:
                 for step in self.agent.run(input_text):
                     if isinstance(step, str):
+                        final_answer = step
+                        print(f"Final Answer (string step): {final_answer}")
                         continue
                     step_name = step.__class__.__name__
                     output = getattr(step, "output", None)
                     if output:
                         print(f"Step: {step_name} Output: {step.output}")
-                    print(f"Step: {step_name} Output: {step.output}")
-
                     self.token_rate_limiter.maybe_wait(self.expected_tokens_per_step)
                     tokens_used = getattr(step, "token_usage", None)
                     if tokens_used:
@@ -56,7 +56,7 @@ class BasicAgent():
 
                     if step_name == "FinalAnswerStep":
                         final_answer = step.output
-                        print(f"Final Answer: {final_answer}")
+                        print(f"Final Answer (FinalAnswerStep): {final_answer}")
                 break
             except Exception as e:
                 if "overload" in str(e).lower() or "rate limit" in str(e).lower():