feat: 手术视频消耗、待确认与持久化改造

- 新增 Alembic 初始迁移、领域明细模型及归档持久化与重试链路\n- 拆分视频会话注册表、分类处理、推理时间窗聚合与流处理\n- 消耗日志:TSV/Markdown 含 top2/top3;item_id 优先产品编码;待确认记「待确认」行,语音确认后落正式行并更新汇总\n- 待确认时内存/DB 明细为占位行,确认后替换;拒绝时移除占位\n- 分类 probs 先 detach/cpu 再转 NumPy,修复 MPS/CUDA 上推理被静默跳过\n- 补充集成测试、归档与设备张量等单测

Made-with: Cursor
This commit is contained in:
Kevin
2026-04-23 20:42:21 +08:00
parent 69980d8073
commit 3d7bd70355
55 changed files with 4544 additions and 2050 deletions

View File

@@ -0,0 +1,101 @@
"""ArchivePersister指数退避、重试上限与 durable fallback 恢复。"""
from __future__ import annotations
import json
from datetime import datetime, timezone
import pytest
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
from app.config import Settings
from app.domain.consumption import SurgeryConsumptionStored
from app.repositories.surgery_results import SurgeryResultRepository
from app.services.video.archive_persister import ArchivePersister
class _AlwaysFailRepo(SurgeryResultRepository):
def __init__(self) -> None:
super().__init__()
self.calls = 0
async def save_final_result(self, session: AsyncSession, **kwargs: object) -> None:
self.calls += 1
raise RuntimeError("db down")
def _detail(item_id: str = "纱布") -> SurgeryConsumptionStored:
return SurgeryConsumptionStored(
item_id=item_id,
item_name=item_id,
qty=1,
doctor_id="vision",
timestamp=datetime(2026, 4, 23, 12, 0, tzinfo=timezone.utc),
source="vision",
)
@pytest.mark.asyncio
async def test_persist_or_archive_writes_durable_fallback(
tmp_path,
sqlite_session_factory: async_sessionmaker[AsyncSession],
) -> None:
fallback_dir = tmp_path / "pending_archive"
settings = Settings(archive_persist_durable_fallback_dir=str(fallback_dir))
repo = _AlwaysFailRepo()
persister = ArchivePersister(
settings=settings,
repository=repo,
session_factory=sqlite_session_factory,
)
ok = await persister.persist_or_archive("abc123", [_detail("纱布")])
assert ok is False
path = fallback_dir / "abc123.json"
assert path.exists()
payload = json.loads(path.read_text(encoding="utf-8"))
assert payload["surgery_id"] == "abc123"
assert payload["details"][0]["item_id"] == "纱布"
assert persister.archived_details("abc123") is not None
@pytest.mark.asyncio
async def test_recover_from_durable_fallback_reloads_pending_archive(
tmp_path,
sqlite_session_factory: async_sessionmaker[AsyncSession],
) -> None:
fallback_dir = tmp_path / "pending_archive"
fallback_dir.mkdir()
payload = {
"surgery_id": "recov01",
"saved_at": "2026-04-23T08:00:00+00:00",
"details": [
{
"item_id": "缝线",
"item_name": "缝线",
"qty": 1,
"doctor_id": "vision",
"timestamp": "2026-04-23T08:00:00+00:00",
"source": "vision",
}
],
}
(fallback_dir / "recov01.json").write_text(
json.dumps(payload, ensure_ascii=False), encoding="utf-8"
)
settings = Settings(archive_persist_durable_fallback_dir=str(fallback_dir))
persister = ArchivePersister(
settings=settings,
repository=SurgeryResultRepository(),
session_factory=sqlite_session_factory,
)
loaded = await persister.recover_from_durable_fallback()
assert loaded == 1
details = persister.archived_details("recov01")
assert details is not None
assert details[0].item_id == "缝线"
# 下一次 retry 应成功落库并清理内存 + durable 文件。
ok = await persister.try_persist_archive("recov01")
assert ok is True
assert persister.archived_details("recov01") is None
assert not (fallback_dir / "recov01.json").exists()