From aadab2f480a8c17a9265a244d262947da23ddc79 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Sun, 26 Jul 2026 22:32:06 -0400
Subject: [PATCH] refac

---
 backend/open_webui/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py
index 3ede68c5f7..409bbb6640 100644
--- a/backend/open_webui/main.py
+++ b/backend/open_webui/main.py
@@ -1698,6 +1698,7 @@ async def chat_completion(
             per_model_metadata = {
                 **metadata,
                 'message_id': assistant_message_id,
+                'task_id': str(uuid4()),
             }
 
             # Per-model form_data: own model
@@ -1733,8 +1734,8 @@ async def chat_completion(
                 request.app.state.redis,
                 process,
                 id=chat_id,
+                task_id=per_model_metadata['task_id'],
             )
-            per_model_metadata['task_id'] = task_id
             task_ids.append(task_id)
 
         if is_internal: