Update app.py

app.py

@@ -3,6 +3,7 @@ from functools import partial
 from os import environ
 from typing import Callable, Coroutine
 from itertools import chain
+from asyncio import anext
 from anyio import create_task_group
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import HTMLResponse
@@ -101,19 +102,6 @@ class OverrideStreamResponse(StreamingResponse):
             await self.background()
 
 
-
-async def get_response_with_valid_api_key(request: Request):
-    for api_key in API_KEYS:
-        try:
-            response_generator = stream_api_response(api_key, request)
-            response = OverrideStreamResponse(response_generator)
-            return response
-        except AuthError:
-            print(f'ключ API {api_key} недействителен или превышен лимит отправки запросов\n всего ключей: {len(API_KEYS)}')
-            continue
-    raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')
-
-
 async def proxy_openai_api(request: Request):
     headers = {k: v for k, v in request.headers.items() if k not in {'host', 'content-length', 'x-forwarded-for', 'x-real-ip', 'connection'}}
 
@@ -147,7 +135,7 @@ async def proxy_openai_api(request: Request):
         for api_key in API_KEYS:
             response_generator = stream_api_response(api_key)
             try:
-                first_chunk = await response_generator.__anext__()
+                first_chunk = await anext(response_generator)
                 if first_chunk == 'auth_error':
                     print(f'ключ API {api_key} недействителен или превышен лимит отправки запросов')
                     continue