fixing messages

app.py

@@ -52,14 +52,22 @@ class AgentState(TypedDict):
 
 def assistant(state: AgentState):
     # Prepare messages for the agent
+
     messages = [{"role": "user", "content": msg.content} for msg in state["messages"]]
     
     # Invoke the agent with the prepared messages
     response = agent.invoke({"messages": messages})
     print(response)
-    # Extract the response content from the AIMessage object
-    response_content = response
-    # response_content = response.messages[-1].content  # Access the content attribute directly
+    
+    # Ensure the response is a list of message dictionaries
+    response_messages = [
+        {"role": "assistant", "content": msg.content} for msg in response["messages"]
+    ]
+    print(response_messages)
+
+    # Extract the response content from the last message
+    response_content = response_messages[-1]["content"]
+    print(response_content)
     
     return {
         "messages": [response_content],