45 lines
2.6 KiB
Python
45 lines
2.6 KiB
Python
"""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))
|