fix: reconcile output on save and fix continuation

Author: Algorithm5838

backend/open_webui/routers/chats.py

@@ -889,9 +889,7 @@ async def update_chat_by_id(
     if chat:
         updated_chat = {**chat.chat, **form_data.chat}
 
-        # Reconcile output with content for any edited assistant messages so
-        # the DB is always consistent and process_messages_with_output can use
-        # output as the source of truth without re-parsing content at read time.
+        # Reconcile output with any content edits so the DB stays consistent.
         messages = updated_chat.get('history', {}).get('messages', {})
         for mid, msg in messages.items():
             if msg.get('role') == 'assistant' and msg.get('output'):

backend/open_webui/utils/middleware.py

@@ -2076,7 +2076,7 @@ def load_messages_from_db(
     keeping only LLM-relevant fields (role, content, output).
 
     If require_done is True, returns None when the anchor message is not
-    marked done — prevents loading a partial mid-stream output as context.
+    marked done; prevents loading a partial mid-stream output as context.
     """
     messages_map = Chats.get_messages_map_by_chat_id(chat_id)
     if not messages_map:
@@ -2098,8 +2098,7 @@ def process_messages_with_output(messages: list[dict]) -> list[dict]:
 
     For assistant messages with 'output' field, produces properly formatted
     OpenAI-style messages (tool_calls + tool results). Strips 'output' before LLM.
-    The DB is kept consistent at write-time (update_chat_by_id), so output can
-    be used directly as the source of truth here.
+    output is reconciled at write-time; use it as-is.
     """
     processed = []
 
@@ -2163,10 +2162,9 @@ async def process_chat_payload(request, form_data, user, metadata, model):
     db_messages = None
 
     if chat_id and not chat_id.startswith('local:'):
-        # For "Continue", message_id is the existing assistant message already in the
-        # DB — load from it so the reconciled output is included in context.
-        # For a new generation, message_id is a fresh UUID not yet saved, so
-        # load_messages_from_db returns None and we fall through to parent_message_id.
+        # For "Continue", message_id already exists in the DB; load from it directly.
+        # For a new turn, message_id is a fresh UUID; load_messages_from_db returns None
+        # and falls back to parent_message_id.
         if message_id:
             db_messages = load_messages_from_db(chat_id, message_id, require_done=True)
         if db_messages is None and parent_message_id:

backend/open_webui/utils/misc.py

@@ -325,7 +325,7 @@ def reconcile_output_with_content(message: dict) -> dict:
     content = message.get('content', '')
     output = message.get('output', [])
 
-    # Extract plain text by stripping <details> blocks that serialize_output wrote
+    # Strip rendered <details> blocks to get the edited plain text
     edited_text = re.sub(r'<details\b[^>]*>.*?</details>', '', content, flags=re.S).strip()
 
     # Drop output items for <details> blocks the user removed from content
@@ -344,7 +344,7 @@ def reconcile_output_with_content(message: dict) -> dict:
         else:
             reconciled_output.append(item)
 
-    # If there was no 'message' item, append one so the text is not lost
+    # No message item yet; preserve the edited text
     if not text_applied and edited_text:
         reconciled_output.append({
             'type': 'message',

src/lib/components/chat/Messages/Markdown.svelte

@@ -1,19 +1,41 @@
-<script>
-	import { onDestroy } from 'svelte';
+<script context="module">
 	import { marked } from 'marked';
-	import { replaceTokens, processResponseContent } from '$lib/utils';
-	import { user } from '$lib/stores';
 
 	import markedExtension from '$lib/utils/marked/extension';
 	import markedKatexExtension from '$lib/utils/marked/katex-extension';
 	import { disableSingleTilde } from '$lib/utils/marked/strikethrough-extension';
 	import { mentionExtension } from '$lib/utils/marked/mention-extension';
 	import colonFenceExtension from '$lib/utils/marked/colon-fence-extension';
-
-	import MarkdownTokens from './Markdown/MarkdownTokens.svelte';
 	import footnoteExtension from '$lib/utils/marked/footnote-extension';
 	import citationExtension from '$lib/utils/marked/citation-extension';
 
+	const options = {
+		throwOnError: false,
+		breaks: true
+	};
+
+	marked.use(markedKatexExtension(options));
+	marked.use(markedExtension(options));
+	marked.use(citationExtension(options));
+	marked.use(footnoteExtension(options));
+	marked.use(colonFenceExtension(options));
+	marked.use(disableSingleTilde);
+	marked.use({
+		extensions: [
+			mentionExtension({ triggerChar: '@' }),
+			mentionExtension({ triggerChar: '#' }),
+			mentionExtension({ triggerChar: '$' })
+		]
+	});
+</script>
+
+<script>
+	import { onDestroy } from 'svelte';
+	import { replaceTokens, processResponseContent } from '$lib/utils';
+	import { user } from '$lib/stores';
+
+	import MarkdownTokens from './Markdown/MarkdownTokens.svelte';
+
 	export let id = '';
 	export let content;
 	export let done = true;
@@ -40,25 +62,6 @@
 	let lastContent = '';
 	let lastParsedContent = '';
 
-	const options = {
-		throwOnError: false,
-		breaks: true
-	};
-
-	marked.use(markedKatexExtension(options));
-	marked.use(markedExtension(options));
-	marked.use(citationExtension(options));
-	marked.use(footnoteExtension(options));
-	marked.use(colonFenceExtension(options));
-	marked.use(disableSingleTilde);
-	marked.use({
-		extensions: [
-			mentionExtension({ triggerChar: '@' }),
-			mentionExtension({ triggerChar: '#' }),
-			mentionExtension({ triggerChar: '$' })
-		]
-	});
-
 	const parseTokens = () => {
 		if (content === lastContent) return;
 		lastContent = content;