✨ The model ultimately renders an image from the S3 bucket.

Author: Zhi-a

backend/apps/file_management_app.py

@@ -1,5 +1,6 @@
 import logging
 import re
+import base64
 from http import HTTPStatus
 from typing import List, Optional
 from urllib.parse import urlparse, urlunparse, unquote, quote
@@ -149,7 +150,16 @@ async def process_files(
 @file_management_config_router.get("/download/{object_name:path}")
 async def get_storage_file(
     object_name: str = PathParam(..., description="File object name"),
-    download: str = Query("ignore", description="How to get the file"),
+    download: str = Query(
+        "ignore",
+        description=(
+            "How to get the file: "
+            "'ignore' (default, return file info), "
+            "'stream' (return file stream), "
+            "'redirect' (redirect to download URL), "
+            "'base64' (return base64-encoded content for images)."
+        ),
+    ),
     expires: int = Query(3600, description="URL validity period (seconds)"),
     filename: Optional[str] = Query(None, description="Original filename for download (optional)")
 ):
@@ -192,6 +202,28 @@ async def get_storage_file(
                     "ETag": f'"{object_name}"',
                 }
             )
+        elif download == "base64":
+            # Return base64 encoded file content (primarily for images)
+            file_stream, content_type = await get_file_stream_impl(object_name=object_name)
+            try:
+                data = file_stream.read()
+            except Exception as exc:
+                logger.error("Failed to read file stream for base64: %s", str(exc))
+                raise HTTPException(
+                    status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
+                    detail="Failed to read file content for base64 encoding",
+                )
+
+            base64_content = base64.b64encode(data).decode("utf-8")
+            return JSONResponse(
+                status_code=HTTPStatus.OK,
+                content={
+                    "success": True,
+                    "base64": base64_content,
+                    "content_type": content_type,
+                    "object_name": object_name,
+                },
+            )
         else:
             # return file metadata
             return await get_file_url_impl(object_name=object_name, expires=expires)

backend/services/image_service.py

@@ -1,5 +1,6 @@
 import logging
 from http import HTTPStatus
+from urllib.parse import quote
 
 import aiohttp
 
@@ -16,8 +17,9 @@
 async def proxy_image_impl(decoded_url: str):
     # Create session to call the data processing service
     async with aiohttp.ClientSession() as session:
-        # Call the data processing service to load the image
-        data_process_url = f"{DATA_PROCESS_SERVICE}/tasks/load_image?url={decoded_url}"
+        # Encode URL for query string safety
+        encoded_url = quote(decoded_url, safe="")
+        data_process_url = f"{DATA_PROCESS_SERVICE}/tasks/load_image?url={encoded_url}"
 
         async with session.get(data_process_url) as response:
             if response.status != HTTPStatus.OK:
@@ -29,6 +31,7 @@ async def proxy_image_impl(decoded_url: str):
             result = await response.json()
             return result
 
+
 def get_vlm_model(tenant_id: str):
     # Get the tenant config
     vlm_model_config = tenant_config_manager.get_model_config(

frontend/app/[locale]/chat/streaming/chatStreamFinalMessage.tsx

@@ -272,6 +272,9 @@ export function ChatStreamFinalMessage({
                 content={message.finalAnswer || message.content || ""}
                 searchResults={message?.searchResults}
                 onCitationHover={onCitationHover}
+                // For historical messages, content already represents the final answer
+                // when finalAnswer is not present, so enable S3 resolution in both cases.
+                resolveS3Media={Boolean(message.finalAnswer || message.content)}
               />
 
               {/* Button group - only show when hideButtons is false and message is complete */}

frontend/app/[locale]/chat/streaming/chatStreamMessage.tsx

@@ -231,6 +231,7 @@ export function ChatStreamMessage({
                   <MarkdownRenderer
                     content={message.finalAnswer}
                     searchResults={message?.searchResults}
+                    resolveS3Media
                   />
 
                   {/* Button group */}