finalize medical consultation agent backend
This commit is contained in:
@@ -75,6 +75,7 @@ def run_api_contract_tests() -> None:
|
||||
assert auth_me.json()["data"]["user_id"] == "api_user_001"
|
||||
assert auth_me.json()["data"]["source"] == "django_user_center"
|
||||
assert auth_me.json()["data"]["display_name"] == "Swagger测试"
|
||||
assert "department_id" in auth_me.json()["data"]
|
||||
|
||||
openapi = client.get("/openapi.json")
|
||||
assert openapi.status_code == 200
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user