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-04-17 13:10:06 +09:00
parent 34d569d564
commit e695d854f2
1 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/middleware.py
+35
View File
@@ -3755,6 +3755,40 @@ async def streaming_chat_response_handler(response, ctx):
elif data.get('type', '').startswith('response.'): elif data.get('type', '').startswith('response.'):
output, response_metadata = handle_responses_streaming_event(data, output) output, response_metadata = handle_responses_streaming_event(data, output)
# Emit citation sources from finalized output items
# (mirrors Chat Completions annotation handling at delta level)
if data.get('type') == 'response.output_item.done':
item = data.get('item', {})
if item.get('type') == 'message':
for part in item.get('content', []):
for annotation in part.get('annotations', []):
if annotation.get('type') == 'url_citation':
# Handle both flat (Responses API) and nested (Chat Completions) formats
url_citation = annotation.get('url_citation', annotation)
url = url_citation.get('url', '')
title = url_citation.get('title', url)
if url:
await event_emitter(
{
'type': 'source',
'data': {
'source': {
'name': title,
'url': url,
},
'document': [title],
'metadata': [
{
'source': url,
'name': title,
}
],
},
}
)
processed_data = { processed_data = {
'output': full_output(), 'output': full_output(),
'content': serialize_output(full_output()), 'content': serialize_output(full_output()),
@@ -4187,6 +4221,7 @@ async def streaming_chat_response_handler(response, ctx):
) )
reasoning_item['status'] = 'completed' reasoning_item['status'] = 'completed'
if response_tool_calls: if response_tool_calls:
tool_calls.append(_split_tool_calls(response_tool_calls)) tool_calls.append(_split_tool_calls(response_tool_calls))