seaislee1209andClaude Opus 4.8
c8f3f91d38
后端: - 独立生图三类归类:model+product→model_tryon(模特上身图,引用模特库,不送审) / cover→platform_kit(平台套图) / image→free_create(自由创作);生成演员(model 无 product)仍→person(视频角色,留期3) - 成组:每次提交一个 batch_id 串起整批,落 asset.metadata;另记 mode + model_entity_id(上身图溯源模特库) - generate-image 端点透传 model_entity_id;资产库 _tab_q+summary 加 tryon/kits/creations 三类桶 - 单测 StandaloneCategoryTests 直跑 worker 验四态归类全绿(绕开异步 .delay) 前端: - 资产库 library.tsx 加三类 tab(模特上身图/平台套图/自由创作)+ 计数 + 筛选维度 - ai-tools 模特选择器数据源 person→模特库(listModels 映射,选中传形象图当参考 = 引用模特库),ActorLibrary 同源 - api.ts submitGenerateImage 加 model_entity_id 验收:tsc+build 全绿;无头 0 console error(资产库三类 tab 各归类正确、模特选择器 5 张取自模特库) 基线既有 3 失败(StandaloneImageReferenceTests 异步漂移)零新增 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
86 lines
4.3 KiB
Python
86 lines
4.3 KiB
Python
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
from rest_framework.viewsets import ReadOnlyModelViewSet
|
|
|
|
from apps.assets.serializers import AssetSerializer
|
|
from apps.common.api import TeamScopedViewSetMixin, get_current_team
|
|
from apps.common.celery_health import require_worker
|
|
|
|
from .models import AITask, ModelConfig
|
|
from .serializers import AITaskSerializer, ModelConfigSerializer
|
|
from .services import enqueue_standalone_images
|
|
|
|
|
|
class GenerateImageView(APIView):
|
|
"""独立生图(不绑项目)· 图片创作/模特图/平台套图共用 —— **异步**。
|
|
|
|
POST /api/ai/generate-image/ 提交生成,秒级返回 RESERVED 任务列表(慢出图交给 worker)。
|
|
GET /api/ai/generate-image/?ids=… 轮询这些任务的状态;成功的任务带回成图 asset。
|
|
"""
|
|
|
|
def post(self, request):
|
|
require_worker() # 异步出图依赖 worker 兜底执行,没 worker 直接拒绝(否则任务永远 RESERVED)
|
|
prompt = str(request.data.get("prompt") or "").strip()
|
|
if not prompt:
|
|
return Response({"detail": "prompt 不能为空"}, status=status.HTTP_400_BAD_REQUEST)
|
|
mode = str(request.data.get("mode") or "image")
|
|
try:
|
|
count = int(request.data.get("count") or 1)
|
|
except (TypeError, ValueError):
|
|
count = 1
|
|
product_id = str(request.data.get("product_id") or "").strip() or None
|
|
reference_product = bool(request.data.get("reference_product"))
|
|
model_id = str(request.data.get("model_id") or "").strip() or None
|
|
model_entity_id = str(request.data.get("model_entity_id") or "").strip() or None
|
|
ratio = str(request.data.get("ratio") or "").strip() or None
|
|
team = get_current_team(request.user)
|
|
try:
|
|
tasks = enqueue_standalone_images(team=team, user=request.user, prompt=prompt, mode=mode, count=count, product_id=product_id, reference_product=reference_product, model_id=model_id, model_entity_id=model_entity_id, ratio=ratio)
|
|
except ValueError as exc: # 无可用模型 / 余额不足等,立即反馈
|
|
return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
|
|
return Response(
|
|
{"tasks": [{"id": str(t.id), "status": t.status} for t in tasks]},
|
|
status=status.HTTP_202_ACCEPTED,
|
|
)
|
|
|
|
def get(self, request):
|
|
team = get_current_team(request.user)
|
|
ids = [s for s in str(request.query_params.get("ids") or "").split(",") if s]
|
|
if not ids:
|
|
return Response({"tasks": []})
|
|
tasks = AITask.objects.filter(team=team, id__in=ids).prefetch_related(
|
|
"generated_assets", "generated_assets__files"
|
|
)
|
|
data = [
|
|
{
|
|
"id": str(t.id),
|
|
"status": t.status,
|
|
"error_message": t.error_message,
|
|
"assets": AssetSerializer(
|
|
[a for a in t.generated_assets.all() if not a.is_deleted], many=True
|
|
).data,
|
|
}
|
|
for t in tasks
|
|
]
|
|
return Response({"tasks": data})
|
|
|
|
|
|
class AITaskViewSet(TeamScopedViewSetMixin, ReadOnlyModelViewSet):
|
|
# 序列化器不含 request_payload/response_payload(单条可达 3MB+ base64 图),defer 掉:
|
|
# 否则只为序列化 14 个小字段也会把几十 MB blob 从库里拉回(远程库实测 40 条要 30s+)。
|
|
queryset = AITask.objects.select_related("team", "project", "model_config", "model_config__provider").defer("request_payload", "response_payload").all()
|
|
serializer_class = AITaskSerializer
|
|
search_fields = ["idempotency_key", "provider_task_id", "project__name"]
|
|
ordering_fields = ["created_at", "updated_at", "completed_at"]
|
|
|
|
|
|
class ModelConfigViewSet(ReadOnlyModelViewSet):
|
|
# 按创建序固定排序:最早创建的 active 模型排第一 = 前端选择器默认项,与 get_default_model 口径一致
|
|
# (否则 DB 默认序不稳定,可能默认选到 Gemini 等;用户要默认 = 豆包 2.0 Pro,它最早创建)
|
|
queryset = ModelConfig.objects.select_related("provider").filter(status=ModelConfig.Status.ACTIVE).order_by("created_at")
|
|
serializer_class = ModelConfigSerializer
|
|
search_fields = ["name", "display_name", "capability"]
|
|
ordering_fields = ["created_at", "display_name"]
|
|
|