a3f61fcc0f
- DB: segments 用户输入文本（Alembic 0002） - Chat: 阶段检测/阶段提示/回复限制，编排与访谈/画像 prompts 调整 - Memoir: 忠实度检查 agent，叙事与分类等链路更新 - Core: agent 日志、Alembic 启动、LangChain/日志/配置等 - Story: time_hints；Memory 检索与相关测试 - Expo: 助手头像、会话页与消息拆分、实时会话与文案/i18n - Docs/scripts/tests: 迁移脚本、LLM JSON/记忆检索文档、新增单测
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
from sqlalchemy import (
|
||
JSON,
|
||
Boolean,
|
||
Column,
|
||
DateTime,
|
||
ForeignKey,
|
||
Integer,
|
||
String,
|
||
Text,
|
||
)
|
||
from sqlalchemy.orm import relationship
|
||
|
||
from app.core.db import Base, utc_now
|
||
|
||
|
||
class Conversation(Base):
|
||
__tablename__ = "conversations"
|
||
|
||
id = Column(String, primary_key=True)
|
||
user_id = Column(String, ForeignKey("users.id"), nullable=False)
|
||
started_at = Column(DateTime(timezone=True), default=utc_now)
|
||
last_message_at = Column(DateTime(timezone=True), nullable=True)
|
||
ended_at = Column(DateTime(timezone=True), nullable=True)
|
||
duration_seconds = Column(Integer, default=0)
|
||
summary = Column(Text, nullable=True)
|
||
status = Column(String, default="active")
|
||
current_topic = Column(String, nullable=True)
|
||
conversation_stage = Column(String, nullable=True)
|
||
deleted_at = Column(DateTime(timezone=True), nullable=True)
|
||
|
||
user = relationship("User", back_populates="conversations")
|
||
segments = relationship(
|
||
"Segment", back_populates="conversation", cascade="all, delete-orphan"
|
||
)
|
||
messages = relationship(
|
||
"ConversationMessage",
|
||
back_populates="conversation",
|
||
cascade="all, delete-orphan",
|
||
)
|
||
|
||
|
||
class Segment(Base):
|
||
__tablename__ = "segments"
|
||
|
||
id = Column(String, primary_key=True)
|
||
conversation_id = Column(String, ForeignKey("conversations.id"), nullable=False)
|
||
audio_url = Column(String, nullable=True)
|
||
# 用户输入正文：语音 ASR 结果或键盘输入（历史列名 transcript_text）
|
||
user_input_text = Column(Text, nullable=False)
|
||
audio_duration_seconds = Column(Integer, nullable=True)
|
||
created_at = Column(DateTime(timezone=True), default=utc_now)
|
||
processed = Column(Boolean, default=False)
|
||
topic_category = Column(String, nullable=True)
|
||
agent_response = Column(Text, nullable=True)
|
||
tts_audio_urls = Column(JSON, nullable=True)
|
||
|
||
conversation = relationship("Conversation", back_populates="segments")
|
||
|
||
|
||
class ConversationMessage(Base):
|
||
"""durable turn log aligned with Redis history shape (canonical chat source of truth)."""
|
||
|
||
__tablename__ = "conversation_messages"
|
||
|
||
id = Column(String, primary_key=True)
|
||
conversation_id = Column(String, ForeignKey("conversations.id"), nullable=False)
|
||
role = Column(String, nullable=False) # human / ai
|
||
content = Column(Text, nullable=False)
|
||
message_type = Column(String, nullable=False, default="text")
|
||
voice_session_id = Column(String, nullable=True)
|
||
duration_seconds = Column(Integer, nullable=True)
|
||
tts_audio_urls = Column(JSON, nullable=True)
|
||
segment_id = Column(String, ForeignKey("segments.id"), nullable=True)
|
||
created_at = Column(DateTime(timezone=True), default=utc_now)
|
||
|
||
conversation = relationship("Conversation", back_populates="messages")
|
||
segment = relationship("Segment", foreign_keys=[segment_id])
|