Update pages/2_Consult.py

pages/2_Consult.py

@@ -1,17 +1,18 @@
 # /home/user/app/pages/2_Consult.py
 import streamlit as st
-from langchain_core.messages import HumanMessage, AIMessage
+from langchain_core.messages import HumanMessage, AIMessage, SystemMessage # <--- CORRECTED IMPORT
 from datetime import datetime
 from typing import List, Optional, Dict, Any
 from sqlmodel import select
 
 from config.settings import settings
-from agent import get_agent_executor # Assumes this is your LangChain agent
+from agent import get_agent_executor 
 from models import ChatMessage, ChatSession
 from models.db import get_session_context
 from services.logger import app_logger
 from services.metrics import log_consultation_start
 
+
 # --- Authentication Check ---
 if not st.session_state.get("authenticated_user_id"):
     st.warning("Please log in to access the consultation page.")