cjw
0e6b3b1811
按 ROADMAP §3.1 治理批次推进的后端设计审查整改：
- 批次 0（安全）：/api/status/{task_id} 补 JWT 鉴权与任务归属校验；
pythonocc_available 真实探测；bcrypt 超 72 字节显式拒绝；
SECRET_KEY/RUSTFS_* 惰性校验，代码侧弱默认移除
- 批次 1（部署正确性）：主处理链路改走 RustFS（分派入参 stp_file_id 化，
worker 按 object_key 下载）；AUTO_MIGRATE 开关 + 迁移目录 alembic/→migrations/
修复包遮蔽（自动迁移此前从未真正生效）；OCC 镜像改 conda 原生执行 +
基础镜像 tag 锁定；compose 关键项改 ${VAR:?} 强制显式配置
- 批次 2（任务一致性）：删除 Redis 进程内存回退，PG 为任务状态单一事实源；
批量元数据入库（processing_tasks.batch_id，迁移 a3f8c2d91e47）；
型腔失败任务标 failed 不再静默 completed；事务边界收口
（数据本体写 flush-only、失败先回滚再置 failed、进度更新保留即时 commit）
- 批次 3（API 与代码结构）：592 行 advanced_router 拆为 design/cost/machining/
export 四子路由，请求体全量 Pydantic 化；ROUTE_MODULES + route_registry
（/api/health 呈现 degraded，DEBUG fail fast）；纯计算端点统一 to_thread；
StorageIntegrationService 按职责三拆；MAX_FILE_SIZE 接线生效、
celery 复用 Settings.redis_url；管理员重置密码改 JSON body（端到端断裂修复）；
openapi.json 重导出（76 paths）+ 前端 gen:api
- 批次 4（架构演进）：共享 ORM 按模块拆分（shared/models/base.py + identity.py、
moldinsight/models/、inventory/models/，删除三条无使用方的跨模块
relationship，跨模块桥接收敛为裸 FK 硬规则，无兼容 facade）；
OCC executor 重建补 cancel_futures=True（消除旧队列被慢恢复线程
并行消化的数据竞争）；OCC 吞吐方案设计先行
（docs/topics/performance/OCC_THROUGHPUT.md）；顺手清偿 D15
（vite.config.ts 未用参数致 npm run build 失败）
测试基线：125 passed, 2 skipped（pytest + sqlite+aiosqlite；归属边界、
路由契约、配置治理、鉴权回归等随批新增）
文档同步：STATUS / TECH_DEBT / ROADMAP / ARCHITECTURE / API_CONTRACT /
OPERATIONS / AGENTS
Co-Authored-By: Claude Code <noreply@anthropic.com>
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}
|