ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-09-05 17:28:05 +04:00
parent 5b3eae3855
commit 1bc9711afd
1 changed files with 110 additions and 100 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
backend/open_webui/utils/middleware.py
View file

@ -1284,7 +1284,11 @@ async def process_chat_response(
# Non-streaming response # Non-streaming response
if not isinstance(response, StreamingResponse): if not isinstance(response, StreamingResponse):
if event_emitter: if event_emitter:
try:
if isinstance(response, dict) or isinstance(response, JSONResponse): if isinstance(response, dict) or isinstance(response, JSONResponse):
if isinstance(response, list) and len(response) == 1:
# If the response is a single-item list, unwrap it #17213
response = response[0]
if isinstance(response, JSONResponse) and isinstance( if isinstance(response, JSONResponse) and isinstance(
response.body, bytes response.body, bytes
@ -1292,7 +1296,9 @@ async def process_chat_response(
try: try:
response_data = json.loads(response.body.decode("utf-8")) response_data = json.loads(response.body.decode("utf-8"))
except json.JSONDecodeError: except json.JSONDecodeError:
response_data = {"error": {"detail": "Invalid JSON response"}} response_data = {
"error": {"detail": "Invalid JSON response"}
}
else: else:
response_data = response response_data = response
@ -1403,6 +1409,10 @@ async def process_chat_response(
status_code=response.status_code, status_code=response.status_code,
) )
except Exception as e:
log.debug(f"Error occurred while processing request: {e}")
pass
return response return response
else: else:
if events and isinstance(events, list) and isinstance(response, dict): if events and isinstance(events, list) and isinstance(response, dict):