张胜宇
e239eb778b
1) 客服 Agent 四份交付文档 + 构建脚手架：品牌由包装占位 XX科技 / 旧名 南方财富 统一为南方基金（热线 400-889-8899 / 官网 nffund.com），系统名改为「智能服务系统」； 同步追加 §0.4 修订记录行，工程记录行保留原占位字面以支撑硬编码扫描验收。 2) 开发文档：清理 28 份已作废/残留文档（14 份移出归档 + 14 份仓库副本）， 新增《文档规整方案与开发前待决事项-2026-09-17》。 3) 客服agent 四份交付文档首次纳入本分支。
55 lines
2.4 KiB
Python
55 lines
2.4 KiB
Python
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.model.conversation import ConversationFeedback, ConversationMessage
|
|
from app.model.platform import AgentRun
|
|
|
|
|
|
class ConversationRepository:
|
|
def __init__(self, session: AsyncSession) -> None:
|
|
self.session = session
|
|
|
|
async def messages(
|
|
self, session_id: str, user_id: int, limit: int, before: int | None = None
|
|
) -> list[ConversationMessage]:
|
|
"""按会话取一页消息（默认 `id DESC`，即最新一页）。
|
|
|
|
`before` 与 `PlatformRepository.rows` 的游标语义一致（`id < before`，取更旧的
|
|
一页）；`None` 表示不翻页，行为与加游标前逐字相同。游标的合法性由入口的
|
|
`parse_cursor` 负责，这里只接受已验证的整数边界。
|
|
"""
|
|
query = select(ConversationMessage).where(
|
|
ConversationMessage.session_id == session_id,
|
|
ConversationMessage.customer_id == user_id,
|
|
)
|
|
if before is not None:
|
|
query = query.where(ConversationMessage.id < before)
|
|
return list(await self.session.scalars(
|
|
query.order_by(ConversationMessage.id.desc()).limit(limit)))
|
|
|
|
async def message(self, message_id: int, user_id: int) -> ConversationMessage | None:
|
|
statement = select(ConversationMessage).where(
|
|
ConversationMessage.id == message_id, ConversationMessage.customer_id == user_id,
|
|
).with_for_update()
|
|
row: ConversationMessage | None = await self.session.scalar(statement)
|
|
return row
|
|
|
|
async def feedback(self, message_id: int, user_id: int) -> ConversationFeedback | None:
|
|
statement = select(ConversationFeedback).where(
|
|
ConversationFeedback.message_id == message_id,
|
|
ConversationFeedback.customer_id == user_id,
|
|
)
|
|
row: ConversationFeedback | None = await self.session.scalar(statement)
|
|
return row
|
|
|
|
async def run_result(
|
|
self, run_id: str, user_id: int
|
|
) -> tuple[AgentRun, ConversationMessage | None] | None:
|
|
run = await self.session.scalar(select(AgentRun).where(
|
|
AgentRun.run_id == run_id, AgentRun.user_id == user_id))
|
|
if run is None:
|
|
return None
|
|
message = (await self.session.get(ConversationMessage, run.result_message_id)
|
|
if run.result_message_id else None)
|
|
return run, message
|