Simon
101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 会话标注模型
|
||
# =============================================================================
|
||
# 说明：对应数据库 conversation_annotations 表
|
||
# 存储坐席对AI回复的标注数据，用于模型优化
|
||
# =============================================================================
|
||
|
||
import uuid
|
||
from datetime import datetime
|
||
|
||
from sqlalchemy import DateTime, Index, Integer, String, Text
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
from app.database import Base
|
||
|
||
|
||
class ConversationAnnotation(Base):
|
||
"""会话标注模型 — 对应 conversation_annotations 表。
|
||
|
||
存储坐席对AI回复的标注数据，用于持续优化AI能力。
|
||
|
||
Attributes:
|
||
id: 标注ID（UUID）
|
||
conversation_id: 会话ID
|
||
agent_id: 坐席ID
|
||
message_id: 被标注的消息ID（AI回复）
|
||
feedback: 反馈类型（useful=有用/useless=无用）
|
||
comment: 备注（可选）
|
||
created_at: 创建时间
|
||
"""
|
||
|
||
# 表名
|
||
__tablename__ = "conversation_annotations"
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 字段定义
|
||
# --------------------------------------------------------------------------
|
||
|
||
# 主键
|
||
id: Mapped[str] = mapped_column(
|
||
String(36),
|
||
primary_key=True,
|
||
default=lambda: str(uuid.uuid4()),
|
||
)
|
||
|
||
# 会话ID
|
||
conversation_id: Mapped[str] = mapped_column(
|
||
String(36),
|
||
nullable=False,
|
||
index=True,
|
||
comment="会话ID",
|
||
)
|
||
|
||
# 坐席ID
|
||
agent_id: Mapped[str] = mapped_column(
|
||
String(36),
|
||
nullable=False,
|
||
comment="坐席ID",
|
||
)
|
||
|
||
# 被标注的消息ID
|
||
message_id: Mapped[str] = mapped_column(
|
||
String(36),
|
||
nullable=False,
|
||
comment="被标注的消息ID",
|
||
)
|
||
|
||
# 反馈类型
|
||
feedback: Mapped[str] = mapped_column(
|
||
String(20),
|
||
nullable=False,
|
||
comment="useful=有用/useless=无用",
|
||
)
|
||
|
||
# 备注
|
||
comment: Mapped[str] = mapped_column(
|
||
Text,
|
||
nullable=True,
|
||
comment="备注",
|
||
)
|
||
|
||
# 创建时间
|
||
created_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True),
|
||
nullable=False,
|
||
default=datetime.now,
|
||
comment="创建时间",
|
||
)
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 索引定义
|
||
# --------------------------------------------------------------------------
|
||
__table_args__ = (
|
||
Index("idx_annotation_conversation", "conversation_id"),
|
||
Index("idx_annotation_message", "message_id"),
|
||
)
|
||
|
||
def __repr__(self) -> str:
|
||
"""标注对象的字符串表示。"""
|
||
return f"<ConversationAnnotation(id={self.id}, feedback={self.feedback})>"
|