100 lines
5 KiB
Python
100 lines
5 KiB
Python
|
|
"""Локальное, детерминированное редактирование учебных сценариев.
|
|||
|
|
|
|||
|
|
Это копирование утверждённого шаблона, а не генерация нового сюжета ИИ.
|
|||
|
|
Никаких сетевых вызовов в этом модуле нет.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from copy import deepcopy
|
|||
|
|
|
|||
|
|
from pydantic import ValidationError
|
|||
|
|
|
|||
|
|
from app.scenarios.loader import ScenarioError, _derive_ground_truth
|
|||
|
|
from app.scenarios.schema import Scenario
|
|||
|
|
|
|||
|
|
DERIVED_TRUTH = {"incident_type", "dds", "incident_code", "notify", "required_facts"}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def template_copy(source: Scenario, draft_id: str, title: str | None = None) -> dict:
|
|||
|
|
"""Сделать полностью самостоятельный шаблонный черновик из сценария."""
|
|||
|
|
body = source.model_dump(mode="json")
|
|||
|
|
body["id"] = draft_id
|
|||
|
|
body["title"] = title or f"{source.title} — копия"
|
|||
|
|
body["extends"] = None # чек-лист уже развёрнут загрузчиком
|
|||
|
|
body["ticket"] = None
|
|||
|
|
body["position"] = None
|
|||
|
|
for key in DERIVED_TRUTH:
|
|||
|
|
body["ground_truth"].pop(key, None)
|
|||
|
|
return body
|
|||
|
|
|
|||
|
|
|
|||
|
|
def merge_patch(body: dict, patch: dict) -> dict:
|
|||
|
|
"""JSON Merge Patch: словари сливаются, списки заменяются целиком.
|
|||
|
|
|
|||
|
|
`id` — идентичность записи, а вычисляемые поля эталона не принимаются от
|
|||
|
|
клиента. Некорректные обычные поля можно временно сохранить в черновике;
|
|||
|
|
утверждение всё равно потребует полной проверки.
|
|||
|
|
"""
|
|||
|
|
if "id" in patch and patch["id"] != body["id"]:
|
|||
|
|
raise ScenarioError("id сценария менять нельзя")
|
|||
|
|
truth = patch.get("ground_truth")
|
|||
|
|
if isinstance(truth, dict) and DERIVED_TRUTH & truth.keys():
|
|||
|
|
raise ScenarioError("вычисляемые поля ground_truth нельзя задавать вручную")
|
|||
|
|
|
|||
|
|
result = deepcopy(body)
|
|||
|
|
|
|||
|
|
def apply(target: dict, changes: dict) -> None:
|
|||
|
|
for key, value in changes.items():
|
|||
|
|
if value is None:
|
|||
|
|
target.pop(key, None)
|
|||
|
|
elif isinstance(value, dict) and isinstance(target.get(key), dict):
|
|||
|
|
apply(target[key], value)
|
|||
|
|
else:
|
|||
|
|
target[key] = deepcopy(value)
|
|||
|
|
|
|||
|
|
apply(result, patch)
|
|||
|
|
result["id"] = body["id"]
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
def validate(body: dict) -> Scenario:
|
|||
|
|
"""Проверить схему и связи фактов, затем пересчитать эталон по ЕКП."""
|
|||
|
|
declared = body.get("ground_truth") or {}
|
|||
|
|
if not isinstance(declared, dict):
|
|||
|
|
raise ScenarioError("ground_truth: ожидается объект")
|
|||
|
|
forbidden = DERIVED_TRUTH & declared.keys()
|
|||
|
|
if forbidden:
|
|||
|
|
raise ScenarioError(
|
|||
|
|
f"{', '.join(sorted(forbidden))} в ground_truth выводится кодом и руками не пишется"
|
|||
|
|
)
|
|||
|
|
try:
|
|||
|
|
scenario = Scenario.model_validate(body)
|
|||
|
|
except ValidationError as exc:
|
|||
|
|
first = exc.errors()[0]
|
|||
|
|
where = ".".join(str(part) for part in first["loc"])
|
|||
|
|
raise ScenarioError(f"{where}: {first['msg']}") from exc
|
|||
|
|
|
|||
|
|
if not scenario.title.strip() or len(scenario.title) > 200:
|
|||
|
|
raise ScenarioError("title: требуется название длиной от 1 до 200 символов")
|
|||
|
|
if not scenario.first_line.strip():
|
|||
|
|
raise ScenarioError("first_line: требуется первая реплика")
|
|||
|
|
if not scenario.facts or not scenario.checklist:
|
|||
|
|
raise ScenarioError("сценарию нужны факты и чек-лист для оценки занятия")
|
|||
|
|
|
|||
|
|
known = scenario.fact_ids()
|
|||
|
|
question_ids = {item.id for item in scenario.checklist}
|
|||
|
|
if len(known) != len(scenario.facts):
|
|||
|
|
raise ScenarioError("id фактов должны быть уникальны")
|
|||
|
|
if len(question_ids) != len(scenario.checklist):
|
|||
|
|
raise ScenarioError("id пунктов чек-листа должны быть уникальны")
|
|||
|
|
for item in scenario.checklist:
|
|||
|
|
if not item.question:
|
|||
|
|
raise ScenarioError(f"у пункта {item.id} нет текста вопроса")
|
|||
|
|
if item.fact and item.fact not in known:
|
|||
|
|
raise ScenarioError(f"пункт {item.id} ссылается на факт {item.fact}, которого нет")
|
|||
|
|
for fact in scenario.facts:
|
|||
|
|
question_id = fact.reveal_on.question if fact.reveal_on else None
|
|||
|
|
if question_id and question_id not in question_ids:
|
|||
|
|
raise ScenarioError(f"факт {fact.id} раскрывается несуществующим вопросом {question_id}")
|
|||
|
|
if fact.refine_on and fact.refine_on not in question_ids:
|
|||
|
|
raise ScenarioError(f"факт {fact.id} уточняется несуществующим вопросом {fact.refine_on}")
|
|||
|
|
return _derive_ground_truth(scenario)
|