Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-22 17:51:08 -06:00
parent 342aa84bbe
commit becac2b2b7
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/middleware.py
+15 -10
View File
@@ -3981,6 +3981,8 @@ async def streaming_chat_response_handler(response, ctx):
tool_call_retries = 0 tool_call_retries = 0
tool_call_sources = [] # Track citation sources from tool results tool_call_sources = [] # Track citation sources from tool results
all_tool_call_sources = [] # Accumulated sources across all iterations
user_message = get_last_user_message(form_data["messages"])
while ( while (
len(tool_calls) > 0 len(tool_calls) > 0
@@ -4217,16 +4219,19 @@ async def streaming_chat_response_handler(response, ctx):
await event_emitter({"type": "source", "data": source}) await event_emitter({"type": "source", "data": source})
# Apply source context to messages for model # Apply source context to messages for model
if tool_call_sources: all_tool_call_sources.extend(tool_call_sources)
user_msg = get_last_user_message(form_data["messages"]) if all_tool_call_sources and user_message:
if user_msg: # Restore original user message before re-applying to avoid recursive nesting
form_data["messages"] = apply_source_context_to_messages( form_data["messages"] = add_or_update_user_message(
request, user_message, form_data["messages"], append=False
form_data["messages"], )
tool_call_sources, form_data["messages"] = apply_source_context_to_messages(
user_msg, request,
) form_data["messages"],
tool_call_sources.clear() all_tool_call_sources,
user_message,
)
tool_call_sources.clear()
await event_emitter( await event_emitter(
{ {