prepare fastapi root layout for server deployment

2026-06-04 10:55:23 +08:00
parent eb43573a44
commit b46e43aadc
103 changed files with 347 additions and 197 deletions
@@ -0,0 +1,154 @@
+import asyncio
+import os
+import sys
+from pathlib import Path
+
+os.environ.setdefault("DATABASE_URL", "sqlite:///./storage/test_demo_flow.db")
+os.environ.setdefault("RUNTIME_MEMORY_BACKEND", "memory")
+os.environ.setdefault("LLM_MOCK_ENABLED", "true")
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+if os.getenv("DATABASE_URL") == "sqlite:///./storage/test_demo_flow.db":
+    Path("storage/test_demo_flow.db").unlink(missing_ok=True)
+
+from sqlalchemy import select
+
+from app.core.context import UserContext
+from app.core.exceptions import AppError
+from app.db.session import SessionLocal
+from app.models.source_case import CaseBase
+from app.models.training_record import TrainingRecord, TrainingScoreDetail
+from app.schemas.evaluation import CreateEvaluationRequest
+from app.schemas.session import (
+    ChatRequest,
+    CreateOrderRequest,
+    CreateSessionRequest,
+    SubmitDiagnosisRequest,
+    SubmitTreatmentRequest,
+)
+from app.services.evaluation_service import EvaluationService
+from app.services.order_service import OrderService
+from app.services.pdf_export_service import PdfExportService
+from app.services.runtime_memory import runtime_memory
+from app.services.session_service import SessionService
+from scripts.init_demo_db import init_database
+
+
+async def run_demo_flow() -> None:
+    """完整闭环：验证第一版 Demo 的核心训练链路可跑通。"""
+    init_database()
+    ctx = UserContext(user_id="demo_flow_user", tenant_id="demo_tenant", entry_scene="service_test")
+    with SessionLocal() as db:
+        case = db.scalar(select(CaseBase).where(CaseBase.title == "支气管肺炎 - 6岁男性患儿"))
+        assert case is not None
+
+        session_service = SessionService(db)
+        order_service = OrderService(db)
+        evaluation_service = EvaluationService(db)
+        pdf_service = PdfExportService(db)
+
+        created = session_service.create_session(
+            ctx,
+            CreateSessionRequest(
+                case_id=case.id,
+                training_type="diagnosis_treatment",
+                mode="practice",
+                score_type="percentage",
+            ),
+        )
+        db.commit()
+        assert created.status == "inquiry"
+
+        chat = await session_service.chat(ctx, created.session_id, ChatRequest(message="孩子最高体温多少？").message)
+        db.commit()
+        assert chat.reply
+
+        order = order_service.create_order(created.session_id, ctx.user_id, CreateOrderRequest(item_code="chest_xray").item_code)
+        db.commit()
+        assert order.is_key is True
+        tool_count_before = len([item for item in runtime_memory.get_messages(f"mem:{created.session_code}") if item.get("role") == "tool"])
+
+        duplicate_order = order_service.create_order(created.session_id, ctx.user_id, "chest_xray")
+        db.commit()
+        tool_count_after = len([item for item in runtime_memory.get_messages(f"mem:{created.session_code}") if item.get("role") == "tool"])
+        assert duplicate_order.already_ordered is True
+        assert tool_count_after == tool_count_before
+
+        try:
+            order_service.create_order(created.session_id, ctx.user_id, "not_exists")
+        except AppError as exc:
+            assert exc.code == "ORDER_ITEM_NOT_FOUND"
+        else:
+            raise AssertionError("invalid order item should raise AppError")
+
+        try:
+            order_service.list_order_items(created.session_id, "another_user")
+        except AppError as exc:
+            assert exc.code == "SESSION_NOT_FOUND"
+        else:
+            raise AssertionError("cross user access should raise AppError")
+
+        status = session_service.complete_inquiry(ctx, created.session_id)
+        db.commit()
+        assert status.status == "diagnosis"
+
+        diagnosis = session_service.submit_diagnosis(
+            ctx,
+            created.session_id,
+            SubmitDiagnosisRequest(
+                primary_diagnosis="支气管肺炎",
+                differential_diagnoses=["毛细支气管炎", "支气管哮喘急性发作"],
+                diagnosis_basis="发热咳嗽伴喘息，肺部湿啰音，胸片异常，炎症指标升高。",
+            ),
+        )
+        db.commit()
+        assert diagnosis.status == "treatment"
+
+        treatment = session_service.submit_treatment(
+            ctx,
+            created.session_id,
+            SubmitTreatmentRequest(
+                treatment_principle="抗感染、止咳平喘、改善氧合、严密观察病情变化。",
+                treatment_measures="根据病情选择抗感染治疗，必要时雾化吸入，监测体温、呼吸和血氧。",
+                risk_plan="关注低氧、呼吸困难加重、持续高热和精神反应差。",
+                communication="向家属说明病情、用药注意事项和复诊指征。",
+                follow_up="治疗后复查体温、呼吸情况和必要炎症指标。",
+            ),
+        )
+        db.commit()
+        assert treatment.status == "evaluating"
+
+        try:
+            await session_service.chat(ctx, created.session_id, "治疗后还能问诊吗？")
+        except AppError as exc:
+            assert exc.code == "SESSION_STATUS_INVALID"
+        else:
+            raise AssertionError("chat after treatment submission should raise AppError")
+
+        evaluation = await evaluation_service.create_evaluation(
+            ctx,
+            created.session_id,
+            CreateEvaluationRequest(score_type="percentage"),
+        )
+        db.commit()
+        assert evaluation.total_score > 0
+        assert evaluation.score_details
+        training_record = db.scalar(select(TrainingRecord).where(TrainingRecord.session_id == created.session_id))
+        assert training_record is not None
+        assert training_record.external_user_id == ctx.user_id
+        score_details = list(db.scalars(select(TrainingScoreDetail).where(TrainingScoreDetail.record_id == training_record.id)).all())
+        assert score_details
+        assert score_details[0].dimension
+
+        export = pdf_service.export(evaluation.evaluation_id, ctx.user_id)
+        db.commit()
+        assert Path(export.file_path).exists()
+        assert Path(export.file_path).stat().st_size > 1000
+
+        history = evaluation_service.list_history(ctx.user_id)
+        assert history.items
+
+
+if __name__ == "__main__":
+    asyncio.run(run_demo_flow())
+    print("demo flow test passed")