52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
|
# api/cost_router.py
|
||
|
|
"""成本估算接口（批次 3 自 advanced_router 拆分，D1）。"""
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException
|
||
|
|
from pydantic import BaseModel
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from shared.services.auth_service import get_current_active_user
|
||
|
|
from shared.database.database import get_db_session
|
||
|
|
from shared.models.identity import User
|
||
|
|
from moldinsight.services.task_query_service import TaskQueryService
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
class CostEstimateRequest(BaseModel):
|
||
|
|
task_id: str
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/cost-estimate")
|
||
|
|
async def estimate_cost(
|
||
|
|
body: CostEstimateRequest,
|
||
|
|
current_user: User = Depends(get_current_active_user),
|
||
|
|
db_session: AsyncSession = Depends(get_db_session),
|
||
|
|
):
|
||
|
|
"""模具成本估算：优先使用 LLM，未启用时降级为规则式估算"""
|
||
|
|
# 归属校验统一走 TaskQueryService（与 /api/status 共用，含 404/403 语义）
|
||
|
|
await TaskQueryService.ensure_task_access(db_session, body.task_id, current_user.id)
|
||
|
|
|
||
|
|
# 统一走任务视图：进行中读 Redis，完成态由 PG+RustFS 组装（Redis 大对象已瘦身）
|
||
|
|
task_data = await TaskQueryService.get_task_view(db_session, body.task_id)
|
||
|
|
if not task_data:
|
||
|
|
raise HTTPException(404, "任务不存在")
|
||
|
|
analysis_result = task_data.get("analysis_result")
|
||
|
|
if not analysis_result:
|
||
|
|
raise HTTPException(400, "该任务尚未完成分析")
|
||
|
|
detailed_context = {
|
||
|
|
"candidate_schemes": task_data.get("candidate_schemes", []),
|
||
|
|
"geometry_data": task_data.get("geometry_data", {}),
|
||
|
|
"metadata": {"selected_material": task_data.get("material")},
|
||
|
|
}
|
||
|
|
# 优先使用 LLM
|
||
|
|
from moldinsight.services.llm_service import llm_service
|
||
|
|
result = await llm_service.estimate_cost(analysis_result, detailed_context)
|
||
|
|
if result is not None:
|
||
|
|
result["source"] = "ai"
|
||
|
|
return {"status": "success", "data": result}
|
||
|
|
|
||
|
|
# LLM 未启用或失败，降级为规则估算
|
||
|
|
from moldinsight.services.cost_estimate_service import estimate_cost_by_rules
|
||
|
|
rules_result = estimate_cost_by_rules(analysis_result, detailed_context)
|
||
|
|
return {"status": "success", "data": rules_result}
|