Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ from datetime import datetime, timedelta
|
|
2 |
from functools import partial
|
3 |
from os import environ
|
4 |
from typing import Callable, Coroutine
|
5 |
-
|
6 |
from anyio import create_task_group
|
7 |
from fastapi import FastAPI, HTTPException, Request
|
8 |
from fastapi.responses import HTMLResponse
|
@@ -153,7 +153,7 @@ async def proxy_openai_api(request: Request):
|
|
153 |
continue
|
154 |
else:
|
155 |
headers_to_forward = {k: v for k, v in headers.items() if k.lower() not in {'content-length', 'content-encoding', 'alt-svc'}}
|
156 |
-
return OverrideStreamResponse(
|
157 |
except StopAsyncIteration:
|
158 |
continue
|
159 |
raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')
|
|
|
2 |
from functools import partial
|
3 |
from os import environ
|
4 |
from typing import Callable, Coroutine
|
5 |
+
from itertools import chain
|
6 |
from anyio import create_task_group
|
7 |
from fastapi import FastAPI, HTTPException, Request
|
8 |
from fastapi.responses import HTMLResponse
|
|
|
153 |
continue
|
154 |
else:
|
155 |
headers_to_forward = {k: v for k, v in headers.items() if k.lower() not in {'content-length', 'content-encoding', 'alt-svc'}}
|
156 |
+
return OverrideStreamResponse(chain([first_chunk], response_generator), headers=headers_to_forward)
|
157 |
except StopAsyncIteration:
|
158 |
continue
|
159 |
raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')
|