Files
RAG_helper/migrations/versions/3f1d9a5b7c42_add_thread_state.py
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

39 lines
1.3 KiB
Python

"""add thread_state for per-thread state machine
Revision ID: 3f1d9a5b7c42
Revises: cd0a88ef9080
Create Date: 2026-04-24 10:00:00.000000
Таблица состояния треда (одна строка на тред): текущая ветка, шаг внутри ветки
и собранные слоты (JSON). Нужна для state machine в Спринте 5.
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = '3f1d9a5b7c42'
down_revision: Union[str, None] = 'cd0a88ef9080'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.create_table(
'thread_state',
sa.Column('thread_id', sa.Integer(), nullable=False),
sa.Column('current_intent_code', sa.String(length=50), nullable=True),
sa.Column('current_step', sa.Integer(), nullable=False, server_default='0'),
sa.Column('slots_json', sa.Text(), nullable=False, server_default='{}'),
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
sa.ForeignKeyConstraint(
['thread_id'], ['threads.id'], name='fk_thread_state_thread_id', ondelete='CASCADE',
),
sa.PrimaryKeyConstraint('thread_id'),
)
def downgrade() -> None:
op.drop_table('thread_state')