Update models/chat.py

models/chat.py

@@ -1,42 +1,38 @@
+# /home/user/app/models/chat.py
 from sqlmodel import SQLModel, Field, Relationship
-from typing import Optional, List, TYPE_CHECKING # Ensure TYPE_CHECKING is imported
+from typing import Optional, List, TYPE_CHECKING, Dict, Any # Ensure TYPE_CHECKING is imported
 from datetime import datetime
 
-# Forward reference for type hinting to avoid circular imports
 if TYPE_CHECKING:
-    from .user import User # For the 'user' attribute in ChatSession
+    from .user import User
 
 class ChatSessionBase(SQLModel):
-    user_id: int = Field(foreign_key="user.id") # Foreign key to the 'user' table's 'id' column
+    user_id: int = Field(foreign_key="user.id")
     start_time: datetime = Field(default_factory=datetime.utcnow)
     title: Optional[str] = None
+    # NEW: For storing initial context provided by the clinician
+    patient_context_summary: Optional[str] = Field(default=None, description="Brief summary of patient context for this session")
+    # NEW: Could store structured context as JSON if more detail is needed
+    # patient_context_structured: Optional[Dict[str, Any]] = Field(default=None, sa_column=Column(JSON)) # If using SQLAlchemy Column for JSON
 
 class ChatSession(ChatSessionBase, table=True):
     id: Optional[int] = Field(default=None, primary_key=True)
-    
-    # ------------ RELATIONSHIP DEFINITION ------------
-    # A ChatSession belongs to one User.
-    # `back_populates="chat_sessions"` links this to the `chat_sessions` attribute in the User model.
     user: Optional["User"] = Relationship(back_populates="chat_sessions")
-    
-    # A ChatSession can have many ChatMessage objects.
     messages: List["ChatMessage"] = Relationship(back_populates="session")
-    # -------------------------------------------------
 
 class ChatMessageBase(SQLModel):
-    # Foreign key to the 'chatsession' table's 'id' column.
-    # SQLModel infers table names from class names (lowercase).
     session_id: int = Field(foreign_key="chatsession.id")
-    role: str
+    role: str # "user", "assistant", "system", "tool"
     content: str
     timestamp: datetime = Field(default_factory=datetime.utcnow)
     tool_call_id: Optional[str] = None
     tool_name: Optional[str] = None
+    # NEW: For AI to optionally provide source/confidence
+    # source_references: Optional[List[str]] = Field(default=None, sa_column=Column(JSON)) # If using SQLAlchemy Column for JSON
+    # confidence_score: Optional[float] = None
 
 class ChatMessage(ChatMessageBase, table=True):
     id: Optional[int] = Field(default=None, primary_key=True)
-    
-    # A ChatMessage belongs to one ChatSession.
     session: Optional["ChatSession"] = Relationship(back_populates="messages")
 
 class ChatMessageCreate(ChatMessageBase):