1

video-gen-api/app/services/private_portrait/reference_resolver.py

@@ -10,6 +10,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from app.enums.private_portrait import (
     PRIVATE_PORTRAIT_ASSET_URI_PREFIX,
     PrivatePortraitAssetStatus,
+    PRIVATE_PORTRAIT_ENABLED_ASSET_TYPES,
     PrivatePortraitAssetType,
     PrivatePortraitEventSource,
     PrivatePortraitEventStatus,
@@ -25,7 +26,6 @@ DOMAIN = "private_portrait"
 _ASSET_TYPE_TO_REFERENCE_TYPE = {
     PrivatePortraitAssetType.IMAGE.value: "image",
     PrivatePortraitAssetType.VIDEO.value: "video",
-    PrivatePortraitAssetType.AUDIO.value: "audio",
 }
 
 
@@ -260,24 +260,26 @@ async def resolve_private_portrait_references(
 
             asset_id = str(_ref_get(ref, "private_asset_id") or "")
             if not asset_id:
-                raise HTTPException(status_code=400, detail="真人素材引用缺少 private_asset_id")
+                raise HTTPException(status_code=400, detail="私域人像素材引用缺少 private_asset_id")
 
             asset = asset_map.get(asset_id)
             if not asset:
-                raise HTTPException(status_code=400, detail="真人素材不存在")
+                raise HTTPException(status_code=400, detail="私域人像素材不存在")
             if asset.user_id != user_id:
-                raise HTTPException(status_code=403, detail="真人素材不属于当前用户")
+                raise HTTPException(status_code=403, detail="私域人像素材不属于当前用户")
             if asset.deleted_at is not None:
-                raise HTTPException(status_code=400, detail="真人素材已删除")
+                raise HTTPException(status_code=400, detail="私域人像素材已删除")
+            if asset.asset_type not in PRIVATE_PORTRAIT_ENABLED_ASSET_TYPES:
+                raise HTTPException(status_code=400, detail="Audio 暂未开放，当前仅支持 Image/Video 私域素材")
             if asset.status != PrivatePortraitAssetStatus.ACTIVE.value:
-                raise HTTPException(status_code=400, detail=f"真人素材状态为 {asset.status}，Active 后才可用于生成")
+                raise HTTPException(status_code=400, detail=f"私域人像素材状态为 {asset.status}，Active 后才可用于生成")
             if not asset.remote_asset_id:
-                raise HTTPException(status_code=400, detail="真人素材缺少远程 AssetId")
+                raise HTTPException(status_code=400, detail="私域人像素材缺少远程 AssetId")
 
             expected_ref_type = _ASSET_TYPE_TO_REFERENCE_TYPE.get(asset.asset_type)
             ref_type = _normalize_ref_type(_ref_get(ref, "type"))
             if expected_ref_type and ref_type and ref_type != expected_ref_type:
-                raise HTTPException(status_code=400, detail=f"真人素材类型不匹配：引用为 {ref_type}，素材为 {expected_ref_type}")
+                raise HTTPException(status_code=400, detail=f"私域人像素材类型不匹配：引用为 {ref_type}，素材为 {expected_ref_type}")
 
             provider_url = f"{PRIVATE_PORTRAIT_ASSET_URI_PREFIX}{asset.remote_asset_id}"
             _ref_set(ref, "source", PrivatePortraitReferenceSource.PRIVATE_PORTRAIT_ASSET.value)