feat(sprint3): редактор системного промпта и правил с версионированием
Операторы получают веб-редактор: правят системный промпт и правила,
сохраняют как новую версию, активируют, откатываются. Активная версия
используется в «Песочнице» на каждый /chat.
Принципы, согласованные заранее:
- Сохранённые версии не редактируются — только создание новой. Честный
откат: v1 всегда та же, что была при создании.
- Правила на этом этапе — свободный markdown (textarea). Переход на
структурированные правила (pattern → instruction) — в бэклог.
- Файл prompts/system_prompt.md становится сид-источником: при первом
старте, если таблица agent_configs пустая, из него создаётся v1 и
активируется. Дальше правда идёт из БД, файл не трогаем.
- rules_text конкатенируется с system_prompt в один system-message
через compose_full_system_prompt: "{prompt}\n\nДополнительные
правила:\n{rules}".
- Активную версию удалить нельзя — сначала активируют другую.
Модель и миграция:
- db/models/AgentConfig: id, version (unique/indexed), name (nullable),
system_prompt, rules_text, is_active (indexed), created_at.
Без updated_at — версии неизменяемы.
- Миграция b4450e33664d_add_agent_configs_table.
Сервисы и роутеры:
- services/config_service: ensure_seed (seed v1 из файла),
list/get/get_active/create (version=max+1, при activate атомарно
сбрасывает is_active у остальных и ставит новой),
activate_config (та же схема), delete_config (возвращает причину
отказа: not_found / active), compose_full_system_prompt.
- services/chat_service.send_message: берёт active_cfg, собирает
system_prompt через compose_full_system_prompt, пишет
thread.agent_config_id при создании треда (колонка была nullable
ещё со Спринта 2 — пригодилась именно здесь).
- routers/configs: GET /configs, GET /configs/active, GET /configs/{id},
POST /configs (activate-флаг), POST /configs/{id}/activate,
DELETE /configs/{id} (404 / 400 если активная).
- Pydantic: AgentConfigCreateRequest, AgentConfigInfo, ListResponse,
DeleteResponse.
- main.py: ensure_seed в lifespan после инициализации БД/Chroma/LLM.
UI:
- static/settings.html — трёхблочная страница: имя версии, textarea
промпта, textarea правил, «Сохранить как новую» + галка
«Сразу активировать», «Загрузить активную в редактор». Справа —
список версий с бейджем «активная», действиями «Активировать» /
«Удалить» (disabled у активной) / «Загрузить в редактор». При
первом заходе активная версия автоматом подгружается в редактор.
- В nav на index.html и sandbox.html добавлена ссылка «Настройки».
- В шапке «Песочницы» — зелёный кликабельный бейдж «активная: vN · имя»
(ведёт на /settings.html), обновляется раз в 15 с.
E2E проверено: создана v2 с правилом «ВСЕГДА начинай со слов СПАСИБО
ЗА ВОПРОС», активирована; следующий /chat вернул ответ, начинающийся
ровно с этой фразы; assembled_prompt содержит блок «Дополнительные
правила». После отката на v1 тест-v2 удалена.
SPRINTS.md: Спринт 3 помечен закрытым.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -7,6 +7,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from sqlalchemy.orm import selectinload
|
||||
|
||||
from db.models import Message, Thread
|
||||
from services import config_service
|
||||
from services.llm_client import LLMClient
|
||||
from services.vectorstore import VectorStoreService
|
||||
|
||||
@@ -50,8 +51,14 @@ async def send_message(
|
||||
max_tokens: int | None = None,
|
||||
) -> dict:
|
||||
"""Добавить реплику пациента в тред, получить ответ ассистента, сохранить оба сообщения."""
|
||||
active_cfg = await config_service.get_active_config(session)
|
||||
system_prompt = config_service.compose_full_system_prompt(active_cfg) if active_cfg else None
|
||||
|
||||
if thread_id is None:
|
||||
thread = Thread(name=_auto_thread_name(text))
|
||||
thread = Thread(
|
||||
name=_auto_thread_name(text),
|
||||
agent_config_id=active_cfg.id if active_cfg else None,
|
||||
)
|
||||
session.add(thread)
|
||||
await session.flush()
|
||||
else:
|
||||
@@ -81,6 +88,7 @@ async def send_message(
|
||||
question=text,
|
||||
sources=retrieved,
|
||||
history=history,
|
||||
system_prompt=system_prompt,
|
||||
temperature=temperature,
|
||||
max_tokens=max_tokens,
|
||||
)
|
||||
|
||||
@@ -0,0 +1,115 @@
|
||||
"""Версионируемые конфигурации агента: создание, активация, чтение."""
|
||||
import logging
|
||||
from pathlib import Path
|
||||
|
||||
from sqlalchemy import func, select, update
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from db.models import AgentConfig
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
SEED_PROMPT_PATH = Path(__file__).resolve().parent.parent / "prompts" / "system_prompt.md"
|
||||
|
||||
|
||||
def _load_seed_prompt() -> str:
|
||||
try:
|
||||
return SEED_PROMPT_PATH.read_text(encoding="utf-8").strip()
|
||||
except FileNotFoundError:
|
||||
logger.warning("Seed prompt file not found at %s — creating empty v1", SEED_PROMPT_PATH)
|
||||
return ""
|
||||
|
||||
|
||||
async def ensure_seed(session: AsyncSession) -> None:
|
||||
"""Если таблица пустая, создать v1 из prompts/system_prompt.md и активировать."""
|
||||
count = (await session.execute(select(func.count(AgentConfig.id)))).scalar_one()
|
||||
if count > 0:
|
||||
return
|
||||
|
||||
seed_text = _load_seed_prompt()
|
||||
seed = AgentConfig(
|
||||
version=1,
|
||||
name="Исходная версия (из prompts/system_prompt.md)",
|
||||
system_prompt=seed_text,
|
||||
rules_text="",
|
||||
is_active=True,
|
||||
)
|
||||
session.add(seed)
|
||||
await session.commit()
|
||||
logger.info("Seeded agent_configs with v1 from %s", SEED_PROMPT_PATH.name)
|
||||
|
||||
|
||||
async def list_configs(session: AsyncSession) -> list[AgentConfig]:
|
||||
stmt = select(AgentConfig).order_by(AgentConfig.version.desc())
|
||||
return list((await session.execute(stmt)).scalars().all())
|
||||
|
||||
|
||||
async def get_config(session: AsyncSession, config_id: int) -> AgentConfig | None:
|
||||
return await session.get(AgentConfig, config_id)
|
||||
|
||||
|
||||
async def get_active_config(session: AsyncSession) -> AgentConfig | None:
|
||||
stmt = select(AgentConfig).where(AgentConfig.is_active.is_(True)).limit(1)
|
||||
return (await session.execute(stmt)).scalar_one_or_none()
|
||||
|
||||
|
||||
async def create_config(
|
||||
session: AsyncSession,
|
||||
system_prompt: str,
|
||||
rules_text: str,
|
||||
name: str | None = None,
|
||||
activate: bool = False,
|
||||
) -> AgentConfig:
|
||||
"""Создать новую версию. При activate=True — сразу сделать активной."""
|
||||
next_version = (await session.execute(select(func.coalesce(func.max(AgentConfig.version), 0)))).scalar_one() + 1
|
||||
|
||||
if activate:
|
||||
await session.execute(
|
||||
update(AgentConfig).where(AgentConfig.is_active.is_(True)).values(is_active=False)
|
||||
)
|
||||
|
||||
cfg = AgentConfig(
|
||||
version=next_version,
|
||||
name=(name or "").strip() or None,
|
||||
system_prompt=system_prompt,
|
||||
rules_text=rules_text or "",
|
||||
is_active=activate,
|
||||
)
|
||||
session.add(cfg)
|
||||
await session.commit()
|
||||
await session.refresh(cfg)
|
||||
return cfg
|
||||
|
||||
|
||||
async def activate_config(session: AsyncSession, config_id: int) -> AgentConfig | None:
|
||||
cfg = await session.get(AgentConfig, config_id)
|
||||
if cfg is None:
|
||||
return None
|
||||
await session.execute(
|
||||
update(AgentConfig).where(AgentConfig.is_active.is_(True)).values(is_active=False)
|
||||
)
|
||||
cfg.is_active = True
|
||||
await session.commit()
|
||||
await session.refresh(cfg)
|
||||
return cfg
|
||||
|
||||
|
||||
async def delete_config(session: AsyncSession, config_id: int) -> tuple[bool, str]:
|
||||
"""Удалить версию. Нельзя удалить активную. Возвращает (ok, reason_if_fail)."""
|
||||
cfg = await session.get(AgentConfig, config_id)
|
||||
if cfg is None:
|
||||
return False, "not_found"
|
||||
if cfg.is_active:
|
||||
return False, "active"
|
||||
await session.delete(cfg)
|
||||
await session.commit()
|
||||
return True, ""
|
||||
|
||||
|
||||
def compose_full_system_prompt(cfg: AgentConfig) -> str:
|
||||
"""Собрать из system_prompt + rules_text единую строку для LLM."""
|
||||
base = (cfg.system_prompt or "").strip()
|
||||
rules = (cfg.rules_text or "").strip()
|
||||
if not rules:
|
||||
return base
|
||||
return f"{base}\n\nДополнительные правила:\n{rules}"
|
||||
Reference in New Issue
Block a user