prepare fastapi root layout for server deployment
This commit is contained in:
@@ -0,0 +1,162 @@
|
||||
from fastapi import APIRouter, Depends
|
||||
from sqlalchemy.orm import Session
|
||||
from starlette.responses import StreamingResponse
|
||||
|
||||
from app.core.response import ApiResponse, ok
|
||||
from app.core.user_context import UserContext, get_user_context
|
||||
from app.db.session import get_db
|
||||
from app.schemas.evaluation import CreateEvaluationRequest, EvaluationResponse
|
||||
from app.schemas.session import (
|
||||
ChatRequest,
|
||||
ChatResponse,
|
||||
CreateOrderRequest,
|
||||
CreateOrderResponse,
|
||||
CreateSessionRequest,
|
||||
CreateSessionResponse,
|
||||
OrderItemsResponse,
|
||||
SessionStatusResponse,
|
||||
SubmitDiagnosisRequest,
|
||||
SubmitDiagnosisResponse,
|
||||
SubmitTreatmentRequest,
|
||||
SubmitTreatmentResponse,
|
||||
HintRequest,
|
||||
HintResponse,
|
||||
)
|
||||
from app.services.evaluation_service import EvaluationService
|
||||
from app.services.order_service import OrderService
|
||||
from app.services.session_service import SessionService
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.post("", response_model=ApiResponse[CreateSessionResponse])
|
||||
def create_session(
|
||||
payload: CreateSessionRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""创建训练会话:初始化 user_id 隔离的训练会话和短期 memory。"""
|
||||
result = SessionService(db).create_session(ctx, payload)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/chat", response_model=ApiResponse[ChatResponse])
|
||||
async def chat(
|
||||
session_id: int,
|
||||
payload: ChatRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""非流式问诊:发送医生问题并返回 AI 病人回复。"""
|
||||
result = await SessionService(db).chat(ctx, session_id, payload.message)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/chat/stream", response_class=StreamingResponse)
|
||||
async def chat_stream(
|
||||
session_id: int,
|
||||
payload: ChatRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""流式问诊:返回 SSE 格式的 AI 病人增量回复。"""
|
||||
response = await SessionService(db).stream_chat(ctx, session_id, payload.message)
|
||||
db.commit()
|
||||
return StreamingResponse(
|
||||
response,
|
||||
media_type="text/event-stream",
|
||||
headers={
|
||||
"Cache-Control": "no-cache",
|
||||
"Connection": "keep-alive",
|
||||
"X-Accel-Buffering": "no",
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
@router.get("/{session_id}/order-items", response_model=ApiResponse[OrderItemsResponse])
|
||||
def list_order_items(
|
||||
session_id: int,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""检查项目列表:返回当前病例可申请项目,不返回检查结果。"""
|
||||
return ok(OrderService(db).list_order_items(session_id, ctx.user_id))
|
||||
|
||||
|
||||
@router.post("/{session_id}/orders", response_model=ApiResponse[CreateOrderResponse])
|
||||
def create_order(
|
||||
session_id: int,
|
||||
payload: CreateOrderRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""申请检查检验:从数据库读取并返回结构化结果。"""
|
||||
result = OrderService(db).create_order(session_id, ctx.user_id, payload.item_code)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/complete-inquiry", response_model=ApiResponse[SessionStatusResponse])
|
||||
def complete_inquiry(
|
||||
session_id: int,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""完成问诊:从问诊阶段进入诊断阶段。"""
|
||||
result = SessionService(db).complete_inquiry(ctx, session_id)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/hints", response_model=ApiResponse[HintResponse])
|
||||
async def generate_hints(
|
||||
session_id: int,
|
||||
payload: HintRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""新手模式提示:根据当前问诊上下文生成缺失维度和下一步问题。"""
|
||||
result = await SessionService(db).generate_hints(ctx, session_id, payload)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/diagnosis", response_model=ApiResponse[SubmitDiagnosisResponse])
|
||||
def submit_diagnosis(
|
||||
session_id: int,
|
||||
payload: SubmitDiagnosisRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""提交诊断:保存主要诊断、鉴别诊断和诊断依据。"""
|
||||
result = SessionService(db).submit_diagnosis(ctx, session_id, payload)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/treatment", response_model=ApiResponse[SubmitTreatmentResponse])
|
||||
def submit_treatment(
|
||||
session_id: int,
|
||||
payload: SubmitTreatmentRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""提交治疗方案:保存治疗、风险、沟通和随访内容。"""
|
||||
result = SessionService(db).submit_treatment(ctx, session_id, payload)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
|
||||
|
||||
@router.post("/{session_id}/evaluation", response_model=ApiResponse[EvaluationResponse])
|
||||
async def create_evaluation(
|
||||
session_id: int,
|
||||
payload: CreateEvaluationRequest,
|
||||
ctx: UserContext = Depends(get_user_context),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""生成评价报告:检索指南并调用 Scoring Agent 生成结构化评价。"""
|
||||
result = await EvaluationService(db).create_evaluation(ctx, session_id, payload)
|
||||
db.commit()
|
||||
return ok(result)
|
||||
Reference in New Issue
Block a user