真人素材库修复相关BUG

video-gen-api/app/api/v1/generation_ai.py

@@ -36,6 +36,7 @@ from app.services.generation_billing_service import (
 from app.services.generation_history_delete_service import batch_delete_generation_history_items
 from app.services.generation_log_service import log_task_event
 from app.services.generation_refund_service import mark_chat_generation_task_failed_and_refund_once
+from app.services.private_portrait.reference_resolver import batch_resolve_private_portrait_reference_display_urls, resolve_private_portrait_reference_display_urls
 from app.services.resource_capacity_service import assert_user_resource_capacity_available
 from app.tasks.celery_app import celery_app
 
@@ -173,7 +174,8 @@ async def create_task(
         await db.commit()
         raise HTTPException(status_code=503, detail="任务队列投递失败，请稍后重试")
 
-    return record_to_out(task)
+    refs = await resolve_private_portrait_reference_display_urls(db, record_to_out(task).media_references, user_id=current_user.id)
+    return record_to_out(task, media_references=refs)
 
 
 @router.get(
@@ -280,7 +282,12 @@ async def list_tasks(
         )
     else:
         items_sorted = items
-    return GenerationAITaskListOut(total=total, items=[record_to_out(task=i, is_admin=is_admin) for i in items_sorted])
+    refs_map = await batch_resolve_private_portrait_reference_display_urls(
+        db,
+        {item.id: record_to_out(task=item, is_admin=is_admin).media_references for item in items_sorted},
+        user_id=None if is_admin else current_user.id,
+    )
+    return GenerationAITaskListOut(total=total, items=[record_to_out(task=i, is_admin=is_admin, media_references=refs_map.get(i.id)) for i in items_sorted])
 
 
 @router.get(
@@ -521,7 +528,8 @@ async def get_task(
     task = result.scalar_one_or_none()
     if not task:
         raise HTTPException(status_code=404, detail="任务不存在")
-    return record_to_out(task)
+    refs = await resolve_private_portrait_reference_display_urls(db, record_to_out(task).media_references, user_id=current_user.id)
+    return record_to_out(task, media_references=refs)
 
 
 @router.delete(