From dd86b984bd508cf2841f08dae90411dfb5fe407f Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Mon, 27 Jul 2026 00:55:16 -0400
Subject: [PATCH] refac

---
 backend/open_webui/utils/middleware.py | 11 +++++--
 backend/open_webui/utils/misc.py       | 42 ++++++++++++++++++++++++--
 2 files changed, 49 insertions(+), 4 deletions(-)

diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index d964469e19..98444d1263 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -2076,6 +2076,7 @@ def process_messages_with_output(
                 message['output'],
                 raw=True,
                 reasoning_format=reasoning_format,
+                flatten_tool_images=True,
             )
             if output_messages:
                 processed.extend(output_messages)
@@ -5094,7 +5095,10 @@ async def streaming_chat_response_handler(response, ctx):
                             new_form_data['previous_response_id'] = last_response_id
                         else:
                             tool_messages = convert_output_to_messages(
-                                output, raw=True, reasoning_format=get_reasoning_format(model)
+                                output,
+                                raw=True,
+                                reasoning_format=get_reasoning_format(model),
+                                flatten_tool_images=True,
                             )
 
                             # Chat Completions providers don't support multimodal
@@ -5335,7 +5339,10 @@ async def streaming_chat_response_handler(response, ctx):
                                 'messages': [
                                     *form_data['messages'],
                                     *convert_output_to_messages(
-                                        output, raw=True, reasoning_format=get_reasoning_format(model)
+                                        output,
+                                        raw=True,
+                                        reasoning_format=get_reasoning_format(model),
+                                        flatten_tool_images=True,
                                     ),
                                 ],
                             }
diff --git a/backend/open_webui/utils/misc.py b/backend/open_webui/utils/misc.py
index 8e84a1a47c..ddef7b91c2 100644
--- a/backend/open_webui/utils/misc.py
+++ b/backend/open_webui/utils/misc.py
@@ -258,6 +258,7 @@ def convert_output_to_messages(
     output: list,
     raw: bool = False,
     reasoning_format: str | None = None,
+    flatten_tool_images: bool = False,
 ) -> list[dict]:
     """
     Convert OR-aligned output items to OpenAI Chat Completion-format messages.
@@ -276,6 +277,8 @@ def convert_output_to_messages(
               (for Ollama, which expects reasoning as tagged content).
             - ``'reasoning_content'``: set as ``reasoning_content`` top-level field
               (for llama.cpp, which routes it via the chat template).
+        flatten_tool_images: Move tool output images into a following user
+            message for Chat Completions providers.
     """
     if not output or not isinstance(output, list):
         return []
@@ -285,6 +288,10 @@ def convert_output_to_messages(
     pending_content = []
     pending_reasoning = []  # Only populated when reasoning_format == 'reasoning_content'
     pending_reasoning_details = []
+    pending_tool_image_urls = []
+    function_call_ids = {
+        item.get('call_id') for item in output if item.get('type') == 'function_call' and item.get('call_id')
+    }
 
     def flush_pending():
         nonlocal pending_content, pending_tool_calls, pending_reasoning, pending_reasoning_details
@@ -309,8 +316,29 @@ def convert_output_to_messages(
         pending_reasoning = []
         pending_reasoning_details = []
 
+    def flush_tool_images():
+        nonlocal pending_tool_image_urls
+        if not pending_tool_image_urls:
+            return
+
+        messages.append(
+            {
+                'role': 'user',
+                'content': [
+                    {
+                        'type': 'text',
+                        'text': 'Here are the images from the tool results above. Please analyze them.',
+                    },
+                    *[{'type': 'image_url', 'image_url': {'url': url}} for url in pending_tool_image_urls],
+                ],
+            }
+        )
+        pending_tool_image_urls = []
+
     for item in output:
         item_type = item.get('type', '')
+        if item_type != 'function_call_output':
+            flush_tool_images()
 
         if item_type == 'message':
             # Extract text from output_text content parts
@@ -356,8 +384,17 @@ def convert_output_to_messages(
                     if url:
                         image_urls.append(url)
 
-            if image_urls:
-                # Multimodal tool content with image(s)
+            if flatten_tool_images:
+                messages.append(
+                    {
+                        'role': 'tool',
+                        'tool_call_id': item.get('call_id', ''),
+                        'content': content,
+                    }
+                )
+                if item.get('call_id') in function_call_ids:
+                    pending_tool_image_urls.extend(image_urls)
+            elif image_urls:
                 messages.append(
                     {
                         'role': 'tool',
@@ -429,6 +466,7 @@ def convert_output_to_messages(
             pass
 
     # Flush remaining content/tool_calls
+    flush_tool_images()
     flush_pending()
 
     return reconcile_tool_pairs(messages)