Update app.py

app.py

@@ -10,7 +10,7 @@ from fastapi.responses import StreamingResponse
 from httpx import AsyncClient, RequestError, Timeout
 from starlette.types import Receive, Scope, Send
 
-API_KEYS = [line for line in environ['API_KEYS'].strip().split('\n') if line and line.startswith('sk-')]
+API_KEYS = ['bad_key'] + [line for line in environ['API_KEYS'].strip().split('\n') if line and line.startswith('sk-')]
 COMPLETIONS_URL = 'https://openrouter.ai/api/v1/chat/completions'
 app = FastAPI(title='reverse-proxy')
 
@@ -120,27 +120,31 @@ async def proxy_openai_api(request: Request):
                     raise AuthError('ключ API недействителен или превышен лимит отправки запросов')
                 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'}})
-
-                content = bytearray()
-                async for chunk in stream_response.aiter_bytes():
+                
+                # Инициализация заголовков должна быть здесь, используя объект stream_response
+                headers_to_send = {k: v for k, v in stream_response.headers.items() if k not in {'content-length', 'content-encoding', 'alt-svc'}}
+                response = OverrideStreamResponse(stream_response.aiter_bytes(), headers=headers_to_send)
+                
+                async for chunk in response.body_iterator:
                     if chunk.strip():
                         yield chunk.strip()
-                        content.extend(chunk.strip())
-
+    
         except RequestError as exc:
             raise HTTPException(status_code=500, detail=f'произошла ошибка при запросе: {exc}')
 
+
     async def try_api_keys():
         for api_key in API_KEYS:
             try:
-                response_generator = stream_api_response(api_key)
-                response = OverrideStreamResponse(response_generator)
-                return response
+                async for chunk in stream_api_response(api_key):
+                    yield chunk
+                break  # Если получили данные, выходим из цикла
             except AuthError:
                 print(f'ключ API {api_key} недействителен или превышен лимит отправки запросов')
                 continue
-        raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')
+        else:
+            raise HTTPException(status_code=401, detail='все ключи API использованы, доступ запрещен.')
+
 
     return await try_api_keys()