gnilets commited on
Commit
1b9c7aa
·
verified ·
1 Parent(s): f86d874

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(itertools.chain([first_chunk], response_generator), headers=headers_to_forward)
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 использованы, доступ запрещен.')