feat(core): standalone image-gen endpoint + api.ts methods (profile/password/avatar/generate-image)

- POST /api/ai/generate-image/ — project-less image generation (AITask.project nullable, no schema change), reuses VolcanoArk image_generation + credit reserve/charge; modes image/model/cover. Verified: manage.py check clean; 2 active IMAGE models present (doubao-seedream-4.5/5.0). (Real generation calls Volcano API + charges credit — not yet live-tested to avoid spend.) - api.ts: updateProfile / changePassword / uploadAvatar / generateImage Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-05 15:51:36 +08:00
parent aad9bd683b
commit a8f4608d10
4 changed files with 115 additions and 3 deletions
@@ -419,3 +419,73 @@ def poll_video_segment(*, video_segment: VideoSegment, user) -> VideoSegmentVers

 def create_export_job(*, timeline, user) -> ExportJob:
    return ExportJob.objects.create(timeline=timeline, status=ExportJob.Status.QUEUED)
+
+
+_STANDALONE_CATEGORY = {
+    "model": Asset.Category.PERSON,
+    "cover": Asset.Category.PRODUCT_IMAGE,
+    "image": Asset.Category.PRODUCT_IMAGE,
+}
+_STANDALONE_TASK_TYPE = {
+    "model": AITask.Type.PERSON_IMAGE,
+    "cover": AITask.Type.PRODUCT_IMAGE,
+    "image": AITask.Type.PRODUCT_IMAGE,
+}
+
+
+def generate_standalone_image(*, team, user, prompt: str, mode: str = "image", count: int = 1) -> list[Asset]:
+    """不绑定项目的独立生图(图片创作 / 模特上身图 / 平台套图)。复用项目内生图链路,AITask.project=None。"""
+    model_config = get_default_model(ModelConfig.Capability.IMAGE)
+    if model_config is None:
+        raise ValueError("no active image model configured")
+    category = _STANDALONE_CATEGORY.get(mode, Asset.Category.UNCATEGORIZED)
+    task_type = _STANDALONE_TASK_TYPE.get(mode, AITask.Type.PRODUCT_IMAGE)
+    count = max(1, min(int(count or 1), 4))
+    provider = VolcanoArkProvider(base_url=model_config.provider.base_url or None)
+    assets: list[Asset] = []
+    for index in range(count):
+        cost = estimate_cost(model_config)
+        task = AITask.objects.create(
+            team=team,
+            created_by=user,
+            project=None,
+            task_type=task_type,
+            status=AITask.Status.CREATED,
+            model_config=model_config,
+            idempotency_key=f"standalone-image:{team.id}:{uuid.uuid4()}",
+            request_payload={"model": model_config.name, "endpoint": model_config.endpoint, "prompt": prompt, "mode": mode},
+            estimated_cost=cost,
+        )
+        reserve_credit(team=team, user=user, task=task, amount=cost)
+        task.status = AITask.Status.RESERVED
+        task.save(update_fields=["status", "updated_at"])
+        reservation = task.credit_reservation
+        try:
+            response = provider.image_generation(model=model_config.name, endpoint=model_config.endpoint, prompt=prompt)
+            media = provider.extract_first_media_url(response)
+            with transaction.atomic():
+                task.status = AITask.Status.SUCCEEDED
+                task.response_payload = response
+                task.actual_cost = task.estimated_cost
+                task.completed_at = timezone.now()
+                task.save(update_fields=["status", "response_payload", "actual_cost", "completed_at", "updated_at"])
+                charge_reserved_credit(reservation=reservation, actual_amount=task.actual_cost)
+                fileobj, content_type = VolcanoArkProvider.media_to_bytes(media)
+                suffix = ".jpg" if "jpeg" in content_type else (".webp" if "webp" in content_type else ".png")
+                asset_id = uuid.uuid4()
+                object_key = f"teams/{team.id}/standalone/{asset_id}{suffix}"
+                stored = TosStorage().upload_fileobj(fileobj=fileobj, object_key=object_key, content_type=content_type)
+                asset = Asset.objects.create(
+                    id=asset_id, team=team, created_by=user, name=f"AI 生成 · {mode} · {index + 1}",
+                    asset_type=Asset.Type.IMAGE, source=Asset.Source.AI_GENERATED, category=category, origin_task=task,
+                )
+                AssetFile.objects.create(asset=asset, object_key=stored.object_key, bucket=stored.bucket, content_type=stored.content_type, size_bytes=stored.size_bytes, is_primary=True)
+                assets.append(asset)
+        except Exception as exc:
+            task.status = AITask.Status.FAILED
+            task.error_message = str(exc)
+            task.completed_at = timezone.now()
+            task.save(update_fields=["status", "error_message", "completed_at", "updated_at"])
+            release_credit(reservation=reservation, reason=str(exc))
+            raise
+    return assets