Update app.py

app.py

@@ -131,7 +131,7 @@ async def proxy_openai_api(request: Request):
             streaming = client.stream(request.method, COMPLETIONS_URL, headers=headers, params=request.query_params, json=request_body)
             async with streaming as stream_response:
                 if stream_response.status_code in {401, 402, 429}:
-                    return 'auth_error'
+                    return
                 if stream_response.status_code == 403:
                     raise CensoredError('отклонено по цензуре')  # это специфичная ошибка именно для опенроутера!
                 response.init_headers({k: v for k, v in stream_response.headers.items() if k not in {'content-length', 'content-encoding', 'alt-svc'}})
@@ -147,13 +147,12 @@ async def proxy_openai_api(request: Request):
 
     for api_key in API_KEYS:
         response_generator = stream_api_response(api_key)
-        response = OverrideStreamResponse(response_generator)
-        response_body = await response.body_iterator.__anext__()
-        if response_body == 'auth_error':
+        try:
+            response = OverrideStreamResponse(response_generator)
+            return response
+        except StopAsyncIteration:
             print(f'ключ API {api_key} недействителен или превышен лимит отправки запросов')
             continue
-        else:
-            return OverrideStreamResponse(itertools.chain([response_body], response.body_iterator))
     raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')