Files
RAG_helper/db/models/thread_state.py
T
AR 15 M4 cac3d29273 feat(sprint5): state machine + bouncing — thread_state и служебные теги
Таблица 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>
2026-04-24 12:12:36 +05:00

31 lines
1.2 KiB
Python

from datetime import datetime, timezone
from sqlalchemy import DateTime, ForeignKey, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column
from db.base import Base
def _utcnow() -> datetime:
return datetime.now(timezone.utc)
class ThreadState(Base):
"""Состояние треда для state machine (Спринт 5).
Одна строка на тред: какая ветка сейчас ведёт разговор, на каком шаге она
внутри своего сценария и какие слоты собраны. `slots_json` — произвольный
JSON, формат определяется конкретной веткой.
"""
__tablename__ = "thread_state"
thread_id: Mapped[int] = mapped_column(
ForeignKey("threads.id", ondelete="CASCADE"), primary_key=True
)
current_intent_code: Mapped[str | None] = mapped_column(String(50), nullable=True)
current_step: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
slots_json: Mapped[str] = mapped_column(Text, nullable=False, default="{}")
updated_at: Mapped[datetime] = mapped_column(
DateTime(timezone=True), default=_utcnow, onupdate=_utcnow, nullable=False
)