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
+27 -1
View File
@@ -3,7 +3,7 @@ from sqlalchemy.orm import Session, selectinload
from app.models.source_case import CaseBase, CaseExamItem, ScoringRule, TeachingCase, TraditionalCase
from app.models.training import SessionOrder, SessionSubmission, TrainingSession
from app.models.training_record import TrainingRecord
from app.models.training_record import TrainingRecord, TrainingScoreDetail
class CaseRepository:
@@ -64,6 +64,7 @@ class CaseRepository:
"training_session": len(session_ids),
"training_order": self._count_training_orders(case_id, session_ids),
"training_submission": self._count_by_sessions(SessionSubmission, SessionSubmission.session_id, session_ids),
"training_score_detail": self._count_score_details(case_id, session_ids),
"training_record": self._count_training_records(case_id, session_ids),
}
@@ -75,6 +76,7 @@ class CaseRepository:
deleted["training_submission"] = self._delete_by_sessions(
SessionSubmission, SessionSubmission.session_id, session_ids
)
deleted["training_score_detail"] = self._delete_score_details(case_id, session_ids)
deleted["training_record"] = self._delete_training_records(case_id, session_ids)
deleted["training_session"] = self._delete_where(TrainingSession, TrainingSession.case_id == case_id)
deleted["case_exam_item"] = self._delete_where(CaseExamItem, CaseExamItem.case_id == case_id)
@@ -115,6 +117,13 @@ class CaseRepository:
)
return self._count(TrainingRecord, TrainingRecord.case_id == case_id)
def _count_score_details(self, case_id: int, session_ids: list[int]) -> int:
"""病例删除预览:统计该病例评价记录下的评分明细。"""
record_ids = self._record_ids(case_id, session_ids)
if not record_ids:
return 0
return self._count(TrainingScoreDetail, TrainingScoreDetail.record_id.in_(record_ids))
def _delete_where(self, model: type, *criteria) -> int:
"""病例删除执行:按条件删除单表记录并返回影响行数。"""
result = self.db.execute(delete(model).where(*criteria))
@@ -144,6 +153,23 @@ class CaseRepository:
)
return self._delete_where(TrainingRecord, TrainingRecord.case_id == case_id)
def _delete_score_details(self, case_id: int, session_ids: list[int]) -> int:
"""病例删除执行:先删除评价明细,避免阻塞训练记录删除。"""
record_ids = self._record_ids(case_id, session_ids)
if not record_ids:
return 0
return self._delete_where(TrainingScoreDetail, TrainingScoreDetail.record_id.in_(record_ids))
def _record_ids(self, case_id: int, session_ids: list[int]) -> list[int]:
"""病例删除:读取该病例关联的训练记录 ID 集合。"""
if session_ids:
stmt = select(TrainingRecord.id).where(
or_(TrainingRecord.case_id == case_id, TrainingRecord.session_id.in_(session_ids))
)
else:
stmt = select(TrainingRecord.id).where(TrainingRecord.case_id == case_id)
return [int(item) for item in self.db.scalars(stmt).all()]
def get_exam_items(self, case_id: int) -> list[CaseExamItem]:
"""检查项目:读取当前病例下全部可申请检查检验项目。"""
stmt = select(CaseExamItem).where(CaseExamItem.case_id == case_id).order_by(CaseExamItem.display_order)
@@ -1,7 +1,7 @@
from sqlalchemy import select
from sqlalchemy import delete, select
from sqlalchemy.orm import Session
from app.models.training_record import TrainingRecord
from app.models.training_record import TrainingRecord, TrainingScoreDetail
class EvaluationRepository:
@@ -16,6 +16,20 @@ class EvaluationRepository:
self.db.flush()
return record
def replace_score_details(self, record_id: int, details: list[TrainingScoreDetail]) -> list[TrainingScoreDetail]:
"""评分明细保存:按训练记录覆盖写入维度评分明细。"""
self.db.execute(delete(TrainingScoreDetail).where(TrainingScoreDetail.record_id == record_id))
for detail in details:
detail.record_id = record_id
self.db.add(detail)
self.db.flush()
return details
def list_score_details(self, record_id: int) -> list[TrainingScoreDetail]:
"""评分明细读取:按训练记录查询全部维度明细。"""
stmt = select(TrainingScoreDetail).where(TrainingScoreDetail.record_id == record_id).order_by(TrainingScoreDetail.id)
return list(self.db.scalars(stmt).all())
def get_by_session(self, session_id: int, user_id: str) -> TrainingRecord | None:
"""评价读取:按会话 ID 和外部 user_id 查询训练记录。"""
stmt = select(TrainingRecord).where(
@@ -1,25 +0,0 @@
from sqlalchemy import select
from sqlalchemy.orm import Session
from app.models.user import UserLearningProfile
class UserLearningProfileRepository:
"""学习档案仓储:维护用户训练评价聚合数据。"""
def __init__(self, db: Session) -> None:
self.db = db
def get_profile(self, user_id: str, tenant_id: str | None) -> UserLearningProfile | None:
"""档案读取:按 user_id 和 tenant_id 获取学习档案。"""
stmt = select(UserLearningProfile).where(
UserLearningProfile.user_id == user_id,
UserLearningProfile.tenant_id == tenant_id,
)
return self.db.scalar(stmt)
def save(self, profile: UserLearningProfile) -> UserLearningProfile:
"""档案保存:创建或更新用户学习档案。"""
self.db.add(profile)
self.db.flush()
return profile
@@ -50,7 +50,7 @@ class SourceCaseRepository:
return self.db.scalar(stmt)
def get_department_name(self, department_id: int | None) -> str:
"""科室名称:兼容当前 demo 的 departments,源库无科室表时返回空字符串"""
"""科室名称:按用户端 department 表读取科室名称"""
if not department_id:
return ""
department = self.db.scalar(select(Department).where(Department.id == department_id))