AR 15 M4
cac3d29273
Таблица thread_state (intent, step, slots) ведётся per-thread. В системный
промпт ветки дописывается текущее состояние, LLM возвращает служебный тег
[STATE: step=N; slots={...}] после основного ответа — парсер в chat_service
вырезает его и обновляет состояние. Если ветка решила, что тема ушла в другую,
она выдаёт [INTENT_CHANGE: code] — делаем один повторный вызов LLM с новой
веткой и сброшенным state (bouncing, MAX_BOUNCES=1). Если роутер сам выбрал
другую ветку, чем в thread_state, — state тоже сбрасывается. Промпт new_booking
переписан под 6-шаговый сценарий (имя → повод → специалист → время → подтверждение
→ запись), в «Песочнице» появился блок «Состояние треда» с intent/step/slots
и списком переходов.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
import logging
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from db.session import get_session
|
|
from models.requests import ThreadRenameRequest
|
|
from models.responses import (
|
|
MessageInfo,
|
|
SourceInfo,
|
|
ThreadDeleteResponse,
|
|
ThreadDetailResponse,
|
|
ThreadInfo,
|
|
ThreadListResponse,
|
|
ThreadStateInfo,
|
|
)
|
|
from services import chat_service
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter(prefix="/threads", tags=["threads"])
|
|
|
|
|
|
@router.get("", response_model=ThreadListResponse)
|
|
async def list_threads(session: AsyncSession = Depends(get_session)):
|
|
threads = await chat_service.list_threads(session)
|
|
return ThreadListResponse(
|
|
threads=[ThreadInfo(**t) for t in threads],
|
|
total=len(threads),
|
|
)
|
|
|
|
|
|
@router.get("/{thread_id}", response_model=ThreadDetailResponse)
|
|
async def get_thread(thread_id: int, session: AsyncSession = Depends(get_session)):
|
|
data = await chat_service.get_thread_detail(session, thread_id)
|
|
if data is None:
|
|
raise HTTPException(status_code=404, detail="Thread not found")
|
|
state = data.get("thread_state") or {}
|
|
return ThreadDetailResponse(
|
|
id=data["id"],
|
|
name=data["name"],
|
|
created_at=data["created_at"],
|
|
updated_at=data["updated_at"],
|
|
messages=[
|
|
MessageInfo(
|
|
id=m["id"],
|
|
role=m["role"],
|
|
text=m["text"],
|
|
created_at=m["created_at"],
|
|
sources=[SourceInfo(**s) for s in m["sources"]],
|
|
assembled_prompt=m["assembled_prompt"],
|
|
intent_code=m.get("intent_code", ""),
|
|
intent_name=m.get("intent_name", ""),
|
|
)
|
|
for m in data["messages"]
|
|
],
|
|
thread_state=ThreadStateInfo(
|
|
current_intent_code=state.get("current_intent_code"),
|
|
current_step=state.get("current_step", 0),
|
|
slots=state.get("slots", {}),
|
|
),
|
|
)
|
|
|
|
|
|
@router.patch("/{thread_id}", response_model=ThreadInfo)
|
|
async def rename_thread(
|
|
thread_id: int,
|
|
req: ThreadRenameRequest,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
data = await chat_service.rename_thread(session, thread_id, req.name)
|
|
if data is None:
|
|
raise HTTPException(status_code=404, detail="Thread not found")
|
|
return ThreadInfo(**data)
|
|
|
|
|
|
@router.delete("/{thread_id}", response_model=ThreadDeleteResponse)
|
|
async def delete_thread(thread_id: int, session: AsyncSession = Depends(get_session)):
|
|
deleted = await chat_service.delete_thread(session, thread_id)
|
|
if deleted is None:
|
|
raise HTTPException(status_code=404, detail="Thread not found")
|
|
return ThreadDeleteResponse(ok=True, deleted_messages=deleted)
|