finalize medical consultation agent backend

This commit is contained in:
刘金宝
2026-06-03 15:51:46 +08:00
parent 93d9e1c6a5
commit eb43573a44
33 changed files with 1063 additions and 281 deletions
+7 -1
View File
@@ -8,6 +8,8 @@ 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
@@ -15,7 +17,7 @@ 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
from app.models.training_record import TrainingRecord, TrainingScoreDetail
from app.schemas.evaluation import CreateEvaluationRequest
from app.schemas.session import (
ChatRequest,
@@ -130,9 +132,13 @@ async def run_demo_flow() -> None:
)
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()