Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-08-29 10:09:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-08-25 14:56:23 -04:00
parent a21c8d15ee
commit 684111715f
1 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
backend/open_webui/utils/middleware.py
View File

@@ -6266,14 +6266,26 @@ async def streaming_chat_response_handler(response, ctx):
yield wrap_item(JSONCodec.dumps(event)) yield wrap_item(JSONCodec.dumps(event))
async for data in original_generator: async for data in original_generator:
data, _ = await process_filter_functions( if filter_functions:
request=request, line = data.decode('utf-8', 'replace') if isinstance(data, bytes) else data
filter_context=filter_context, if isinstance(line, str) and line.startswith('data:'):
filter_functions=filter_functions, payload = line.removeprefix('data:').strip()
filter_type='stream', if payload and payload != '[DONE]':
form_data=data, try:
extra_params=extra_params, event = JSONCodec.loads(payload)
) except JSONCodec.JSONDecodeError:
event = None
if isinstance(event, dict):
event, _ = await process_filter_functions(
request=request,
filter_context=filter_context,
filter_functions=filter_functions,
filter_type='stream',
form_data=event,
extra_params=extra_params,
)
data = wrap_item(JSONCodec.dumps(event)) if event else None
if data: if data:
if has_api_outlet_filters: if has_api_outlet_filters: