diff --git a/backend/main.py b/backend/main.py index 4b91cdc84..7a4100fe2 100644 --- a/backend/main.py +++ b/backend/main.py @@ -628,7 +628,10 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware): async for data in original_generator: yield data - return StreamingResponse(stream_wrapper(response.body_iterator, data_items)) + return StreamingResponse( + stream_wrapper(response.body_iterator, data_items), + headers=dict(response.headers), + ) async def _receive(self, body: bytes): return {"type": "http.request", "body": body, "more_body": False}