feat(api): vertical slice POST /v1/agent/chat and the Celery ping path

One request exercises every layer: JWT auth, quota reservation, a versioned
prompt through the LLM gateway, a cost row, quota commit or refund, and the
unified envelope. The trivial ping task proves the queue routing and the
tk_task transitions before any real long-running task is written.

Routes are mounted at the /api/v1 base path of 架构文档 §9.1; /v1/* is
rewritten to it at the ASGI layer so both spellings work from one route table.
This commit is contained in:
xsl
2026-08-14 00:26:39 +08:00
parent 71204affab
commit 746b66da01
16 changed files with 822 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
"""Celery 任务定义与四队列配置(gpu/cpu/io/beat)。"""
+54
View File
@@ -0,0 +1,54 @@
"""Celery application with the four queues from TD-11 / 架构文档 §3.1.
gpu 换装、发型、抠图 —— 并发上限由 GPU 容量决定,队列即容量 (§7.3)
cpu 报告视觉分析、渲染
io 外部 API 调用、推送、上传
beat 定时任务(单实例,防重)
Queue isolation is the point: 换装排队不能把报告生成拖慢。
"""
from __future__ import annotations
from celery import Celery
from kombu import Queue
from packages.shared.config import get_settings
from packages.shared.logging import configure_logging
settings = get_settings()
configure_logging(
level=settings.log_level,
json_output=settings.log_json,
extra_redact_keys=settings.extra_redact_keys,
)
celery_app = Celery(
"meida",
broker=settings.celery_broker_url,
backend=settings.celery_result_backend,
include=["apps.worker.tasks"],
)
celery_app.conf.update(
task_default_queue="cpu",
task_queues=(
Queue("gpu"),
Queue("cpu"),
Queue("io"),
Queue("beat"),
),
task_routes={
"meida.tryon.*": {"queue": "gpu"},
"meida.matting.*": {"queue": "gpu"},
"meida.report.*": {"queue": "cpu"},
"meida.notify.*": {"queue": "io"},
},
task_acks_late=True,
task_reject_on_worker_lost=True,
task_track_started=True,
worker_prefetch_multiplier=1,
broker_connection_retry_on_startup=True,
timezone="Asia/Shanghai",
result_expires=86400,
)
+68
View File
@@ -0,0 +1,68 @@
"""Celery task definitions.
Only the scaffold's ``ping`` task lives here. It proves the whole async path:
API creates ``tk_task`` → queue → worker transitions running → succeeded →
task centre reflects it. Real long-running tasks (report / tryon / matting /
render) follow exactly this shape and additionally call
``billing.commit()``/``refund()`` and ``compliance.publish_asset()``.
"""
from __future__ import annotations
import asyncio
from typing import Any
from apps.worker.celery_app import celery_app
from packages.domain.task.state import TaskStatus
from packages.shared.context import bind_context
from packages.shared.db import get_session_factory
from packages.shared.logging import get_logger
from packages.task.service import TaskService
logger = get_logger(__name__)
async def _run_ping(task_id: int, echo: str | None = None) -> dict[str, Any]:
factory = get_session_factory()
async with factory() as session:
service = TaskService(session)
task = await service.transition(
task_id=task_id, target=TaskStatus.RUNNING, stage_text="执行中"
)
payload = task.payload or {}
message = echo or str(payload.get("echo", "pong"))
await session.commit()
finished = await service.transition(
task_id=task_id,
target=TaskStatus.SUCCEEDED,
stage_text="已完成",
result={"echo": message},
)
await session.commit()
return {"task_id": str(finished.task_id), "echo": message}
async def _fail_ping(task_id: int, error: str) -> None:
factory = get_session_factory()
async with factory() as session:
service = TaskService(session)
await service.transition(
task_id=task_id,
target=TaskStatus.FAILED,
stage_text="执行失败",
error_message=error,
)
await session.commit()
@celery_app.task(name="meida.task.ping", bind=True, max_retries=2)
def ping_task(self: Any, task_id: int, echo: str | None = None) -> dict[str, Any]:
bind_context(task_id=task_id)
try:
return asyncio.run(_run_ping(task_id, echo))
except Exception as exc:
logger.exception("ping_task_failed", task_id=task_id)
# Real tasks additionally refund the quota reservation here (FR-BILL-05).
asyncio.run(_fail_ping(task_id, f"{type(exc).__name__}: {exc}"))
raise