UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-06 21:19:57 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

remove more nesting

Browse Source
This commit is contained in:
Michael Poluektov 2024-08-19 10:34:44 +01:00
parent 32874a816d
commit ce7a1a73ac
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/main.py
View File

@ -680,7 +680,9 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
] ]
response = await call_next(request) response = await call_next(request)
if isinstance(response, StreamingResponse): if not isinstance(response, StreamingResponse):
return response
content_type = response.headers["Content-Type"] content_type = response.headers["Content-Type"]
is_openai = "text/event-stream" in content_type is_openai = "text/event-stream" in content_type
is_ollama = "application/x-ndjson" in content_type is_ollama = "application/x-ndjson" in content_type
@ -699,8 +701,6 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
return StreamingResponse(stream_wrapper(response.body_iterator, data_items)) return StreamingResponse(stream_wrapper(response.body_iterator, data_items))
return response
async def _receive(self, body: bytes): async def _receive(self, body: bytes):
return {"type": "http.request", "body": body, "more_body": False} return {"type": "http.request", "body": body, "more_body": False}