fix(chat): store trajectory history, remove final_answer event (#1295)

Signed-off-by: Radek Ježek <radek.jezek@ibm.com>

Author: jezekra1

agents/chat/src/chat/agent.py

@@ -292,16 +292,18 @@ async def chat(
             continue
 
         last_step = event.state.steps[-1] if event.state.steps else None
-        if last_step and last_step.tool is not None:
+        if last_step and last_step.tool is not None and last_step.tool.name != FinalAnswerTool.name:
             trajectory_content = TrajectoryContent(
                 input=last_step.input,
                 output=last_step.output,
                 error=last_step.error,
             )
-            yield trajectory.trajectory_metadata(
+            metadata = trajectory.trajectory_metadata(
                 title=last_step.tool.name,
                 content=trajectory_content.model_dump_json(),
             )
+            yield metadata
+            await context.store(AgentMessage(metadata=metadata))
 
             if isinstance(last_step.output, FileCreatorToolOutput):
                 result = last_step.output.result
@@ -338,5 +340,6 @@ def serve():
     except KeyboardInterrupt:
         pass
 
+
 if __name__ == "__main__":
     serve()