Update app.py

app.py

@@ -3,7 +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
@@ -135,16 +135,17 @@ async def proxy_openai_api(request: Request):
         for api_key in API_KEYS:
             response_generator = stream_api_response(api_key)
             try:
-                first_chunk = await anext(response_generator)
-                if first_chunk == 'auth_error':
-                    print(f'ключ API {api_key} недействителен или превышен лимит отправки запросов')
-                    continue
-                else:
-                    headers_to_forward = {k: v for k, v in headers.items() if k.lower() not in {'content-length', 'content-encoding', 'alt-svc'}}
-                    return OverrideStreamResponse(chain([first_chunk], response_generator), headers=headers_to_forward)
+                async for first_chunk in response_generator:
+                    if first_chunk == 'auth_error':
+                        print(f'ключ API {api_key} недействителен или превышен лимит отправки запросов')
+                        break
+                    else:
+                        headers_to_forward = {k: v for k, v in headers.items() if k.lower() not in {'content-length', 'content-encoding', 'alt-svc'}}
+                        return OverrideStreamResponse(chain([first_chunk], response_generator), headers=headers_to_forward)
             except StopAsyncIteration:
                 continue
         raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')
+
     
     return await get_response()