修复LLM base64资源定位BUG

2026-07-17 14:58:05 +08:00
parent 8ca6f0fedb
commit 2d3d096fd7
1 changed files with 8 additions and 2 deletions
@@ -46,6 +46,12 @@ async def media_to_base64(url: str, fallback_mime: str = "image/png", max_mb: in
        return f"data:{mime};base64,{b64}"

    # 本地路径
+    if url.startswith("/generate/"):
+        type_pre = '/generate/videos/' if url.startswith("/generate/videos/") else '/generate/images/'
+        type_path = settings.STORAGE_LOCAL_PATH if url.startswith("/generate/videos/") else settings.STORAGE_IMAGE_LOCAL_PATH
+        relative_path = url.replace(type_pre, "", 1).lstrip("/")
+        file_path = os.path.join(type_path, relative_path)
+    else:
        relative_path = url.replace("/uploads/", "", 1).lstrip("/")
        file_path = os.path.join(settings.UPLOAD_LOCAL_PATH, relative_path)
    mime = mimetypes.guess_type(file_path)[0] or fallback_mime