"""Chat session model for grouping chat messages per participant-agent pair.""" import uuid from datetime import datetime from sqlalchemy import Boolean, DateTime, String, ForeignKey, func, UniqueConstraint from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.orm import Mapped, mapped_column from app.database import Base class ChatSession(Base): """A named session grouping chat messages between a user and an agent. source_channel: 'web' | 'feishu' | 'discord' | 'slack' external_conv_id: original channel conversation ID (e.g. 'feishu_p2p_ou_xxx'). Unique per agent — used for reliable find-or-create without in-process caching. is_group: True for group chat sessions (Feishu group, WeCom group, Slack channel, etc.). Group sessions have user_id=NULL and only appear in the 'all sessions' view. group_name: Display name for group chat sessions (e.g. the group/channel name from IM platform). """ __tablename__ = "chat_sessions" __table_args__ = ( UniqueConstraint("agent_id", "external_conv_id", name="uq_chat_sessions_agent_ext_conv"), ) id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) agent_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("agents.id"), nullable=False, index=True) # user_id: for P2P sessions this is the user; for group sessions this is the agent creator (placeholder) user_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=False, index=True) title: Mapped[str] = mapped_column(String(200), nullable=False, default="New Session") source_channel: Mapped[str] = mapped_column(String(20), nullable=False, default="web") external_conv_id: Mapped[str | None] = mapped_column(String(200), nullable=True) # Group chat support: group sessions have user_id=NULL and show group_name instead is_group: Mapped[bool] = mapped_column(Boolean, default=False, server_default="false") group_name: Mapped[str | None] = mapped_column(String(200), nullable=True) # Participant identity (unified User/Agent identity) participant_id: Mapped[uuid.UUID | None] = mapped_column(UUID(as_uuid=True), ForeignKey("participants.id"), nullable=True) # For agent-to-agent sessions: the other agent in the conversation peer_agent_id: Mapped[uuid.UUID | None] = mapped_column(UUID(as_uuid=True), ForeignKey("agents.id"), nullable=True) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), index=True) last_message_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))