solve messages

langraph_agent.py

@@ -8,7 +8,7 @@ from typing import Dict, Any, List, Optional
 from datetime import datetime
 from langgraph.graph import StateGraph, END
 from langchain_openai import ChatOpenAI
-from langchain_core.messages import HumanMessage, SystemMessage, AssistantMessage
+from langchain_core.messages import HumanMessage, SystemMessage, AIMessage
 
 from agent_state import AgentState, ConversationManager
 from prompts import SYSTEM_PROMPT, ERROR_HANDLING_PROMPT
@@ -141,7 +141,7 @@ class CyberLegalAgent:
                 elif msg["role"] == "user":
                     messages.append(HumanMessage(content=msg["content"]))
                 elif msg["role"] == "assistant":
-                    messages.append(AssistantMessage(content=msg["content"]))
+                    messages.append(AIMessage(content=msg["content"]))
             
             # Add system message only if not present in history
             if not has_system_message: