58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
|
"""评测台：查询用户 segment 是否已完成回忆录 Phase1（topic_category 已写入）。"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from app.features.conversation.models import Conversation, Segment
|
|||
|
|
from app.features.evaluation.errors import (
|
|||
|
|
EvaluationBadRequestError,
|
|||
|
|
EvaluationNotFoundError,
|
|||
|
|
)
|
|||
|
|
from app.features.evaluation.schemas import MemoirPhase1ReadyOut
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MemoirReadinessService:
|
|||
|
|
def __init__(self, db: AsyncSession) -> None:
|
|||
|
|
self._db = db
|
|||
|
|
|
|||
|
|
async def memoir_phase1_ready_for_segments(
|
|||
|
|
self,
|
|||
|
|
*,
|
|||
|
|
conversation_id: str,
|
|||
|
|
segment_ids: list[str],
|
|||
|
|
) -> MemoirPhase1ReadyOut:
|
|||
|
|
cid = (conversation_id or "").strip()
|
|||
|
|
if not cid:
|
|||
|
|
raise EvaluationBadRequestError("conversation_id is required")
|
|||
|
|
ids = [s.strip() for s in segment_ids if (s or "").strip()]
|
|||
|
|
if not ids:
|
|||
|
|
raise EvaluationBadRequestError("segment_ids is required")
|
|||
|
|
|
|||
|
|
conv = await self._db.get(Conversation, cid)
|
|||
|
|
if not conv or conv.deleted_at is not None:
|
|||
|
|
raise EvaluationNotFoundError("conversation not found")
|
|||
|
|
|
|||
|
|
stmt = select(Segment).where(
|
|||
|
|
Segment.id.in_(ids),
|
|||
|
|
Segment.conversation_id == cid,
|
|||
|
|
)
|
|||
|
|
result = await self._db.execute(stmt)
|
|||
|
|
rows = list(result.scalars().all())
|
|||
|
|
found_ids = {s.id for s in rows}
|
|||
|
|
missing = [i for i in ids if i not in found_ids]
|
|||
|
|
if missing:
|
|||
|
|
raise EvaluationBadRequestError(
|
|||
|
|
"segment not in conversation: " + ", ".join(missing[:5])
|
|||
|
|
+ ("…" if len(missing) > 5 else "")
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
pending = [s.id for s in rows if s.topic_category is None]
|
|||
|
|
ready = len(pending) == 0
|
|||
|
|
return MemoirPhase1ReadyOut(
|
|||
|
|
ready=ready,
|
|||
|
|
checked_segment_ids=ids,
|
|||
|
|
pending_segment_ids=pending,
|
|||
|
|
)
|