xsl
e8208d2972
POST /reports 骨架：资格校验→终身配额预扣→tk_task→cpu 队列；动态装配 流水线（dotted-path 注册、缺阶段即 CONFIG_ERROR）；中间态逐段落库、失败 退款、derived_features 跨重试/重生成复用（T-P3-07/09）。 实现：packages/report 的 pipeline.py（Stage 契约保持不变，新增 assemble_stages/ run_pipeline/StageFailureError，ReportContext 增可选 task_id 参数、向后兼容）、 repo.py、schemas.py、service.py（重放优先于资格校验、reserve+建行+建任务同事务、 提交后才投递、并发同键落败回滚）；apps/api/routers/reports.py 填桩（send_task 按 任务名投递，不 import apps.worker）；apps/worker/report_tasks.py（worker 侧自建 gateway 装配、状态归一 QUEUED→RUNNING、成功 commit 配额+成本聚合写 model_info、 基础设施异常重试一次、重试耗尽兜底失败退款）。测试：单测 6 个（装配缺失/空配置/ 协议违约/顺序执行/失败保留产物），集成 6 场景（预扣 lifetime/PENDING、成功 committed+两阶段键、同 request_key 幂等不重复扣、二次首生成 40920、阶段失败 refund+derived_features 保留、regenerate 走 m: 月配额 version+1 复用视觉中间态）。 偏差说明： 1) apps/worker/celery_app.py 的 include 只含 apps.worker.tasks（禁改文件），真 worker 进程注册 meida.report.generate 需协调者合并时把 include 补上 apps.worker.report_tasks；API 侧已按任务名 send_task，两侧只靠名字衔接，测试 不受影响。2) ReportContext 增加可选 task_id 关键字参数（失败结案要推动 tk_task），Stage 协议/DEFAULT_STAGE_PATHS/stage_names 未动，T16/T17 兼容。 3) JSONB 深变更用 flag_modified 显式标脏（重新赋相等新 dict 会被 flush 的按值 比较跳过，集成测试抓到过该坑）。4) POST 后 tk_task 保持 PENDING（任务文档 §4 场景 1 口径），QUEUED→RUNNING 由 worker 入口归一。
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
"""报告域的请求/响应模型。
|
|
|
|
Wire 响应统一走 ok() 信封（架构文档 §9.2），这里的模型只描述 ``data`` 部分；
|
|
雪花 ID 一律以字符串返回，避免 JS 精度丢失（NFR-MAINT-01 契约先行）。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ReportCreateRequest(BaseModel):
|
|
questionnaire: dict[str, object] = Field(
|
|
default_factory=dict, description="问卷答案；字段结构由产品定稿后在路由层收紧"
|
|
)
|
|
photo_object_key: str = Field(
|
|
min_length=1, max_length=255, description="临时桶里已上传照片的对象键（report_photo 场景）"
|
|
)
|
|
request_key: str | None = Field(
|
|
default=None, max_length=128, description="幂等键；相同键返回同一任务且不重复扣配额"
|
|
)
|
|
|
|
|
|
class ReportRegenerateRequest(BaseModel):
|
|
request_key: str | None = Field(
|
|
default=None, max_length=128, description="幂等键；相同键返回同一任务且不重复扣配额"
|
|
)
|
|
|
|
|
|
class ReportSubmitResponse(BaseModel):
|
|
task_id: str
|
|
report_id: str
|
|
status: str = Field(description="tk_task 状态；报告本体状态与其同步推进")
|
|
version: int
|
|
|
|
|
|
class ReportItem(BaseModel):
|
|
report_id: str
|
|
tier: str
|
|
status: str
|
|
version: int
|
|
result: dict[str, object] = Field(
|
|
description="按阶段键组织的产物；failed 报告只透错误摘要（中间态留给流水线复用）"
|
|
)
|
|
created_at: datetime
|
|
completed_at: datetime | None
|