Display messages in logs

app.py

@@ -62,7 +62,7 @@ class BasicAgent:
         fixed_answer = self.assistant(state)["messages"][0].content
         print(f"Last message: {self.assistant(state)['messages'][0]}")
         print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer, state
+        return fixed_answer, state["messages"]
 
     def assistant(self, state: State):
         # System message