Update app.py

app.py

@@ -7,43 +7,26 @@ app = FastAPI()
 
 OR_BASE_URL = "https://api.openai.com/v1"
 
-# Создаем один асинхронный клиент на все приложение для переиспользования соединений
-# allow_redirects=False, т.к. мы хотим сами управлять редиректами как прокси
 client = httpx.AsyncClient(base_url=OR_BASE_URL, follow_redirects=False)
 
-# Список "hop-by-hop" заголовков, которые не должны проксироваться дальше.
-# Это стандартная практика для HTTP прокси.
+# "Hop-by-hop" заголовки, которые не должны проксироваться.
 HOP_BY_HOP_HEADERS = {
-    "connection",
-    "keep-alive",
-    "proxy-authenticate",
-    "proxy-authorization",
-    "te",
-    "trailers",
-    "transfer-encoding",
-    "upgrade",
+    "connection", "keep-alive", "proxy-authenticate", "proxy-authorization",
+    "te", "trailers", "transfer-encoding", "upgrade",
 }
 
-
 @app.get("/")
 async def home():
-    return {"status": "ok", "message": "Async Transparent OpenRouter proxy is working :3"}
+    return {"status": "ok", "message": "Robust Async OpenRouter proxy is working :3"}
 
 
 @app.api_route("/{endpoint:path}", methods=["GET", "POST", "PUT", "PATCH", "DELETE"])
 async def proxy(endpoint: str, request: Request):
     try:
-        # 1. Строим новый запрос для отправки на целевой сервер
         url = httpx.URL(path=f"/{endpoint}", query=request.url.query.encode("utf-8"))
-        
-        # Копируем заголовки от клиента, исключая "host"
         headers = {k: v for k, v in request.headers.items() if k.lower() != "host"}
-        
-        # Асинхронно читаем тело входящего запроса
         body = await request.body()
 
-        # 2. Отправляем запрос с потоковой передачей
-        # stream=True - критически важно для получения ответа по частям
         req = client.build_request(
             method=request.method,
             url=url,
@@ -52,28 +35,27 @@ async def proxy(endpoint: str, request: Request):
         )
         resp = await client.send(req, stream=True)
 
-        # 3. Копируем заголовки ответа от целевого сервера
-        # Исключаем только hop-by-hop заголовки. Все остальное (Content-Type, Content-Encoding,
-        # кастомные x-openai-*, и т.д.) будет скопировано.
+        # Заголовки, которые мы должны удалить, потому что мы сами обрабатываем контент.
+        # Content-Encoding: мы сами распаковываем.
+        # Content-Length: его значение для сжатых данных неверно для распакованных.
+        # Transfer-Encoding: управляется сервером, который отправляет ответ.
+        EXCLUDED_HEADERS = HOP_BY_HOP_HEADERS | {"content-encoding", "content-length", "transfer-encoding"}
+
         response_headers = [
             (key, value)
             for key, value in resp.headers.items()
-            if key.lower() not in HOP_BY_HOP_HEADERS
+            if key.lower() not in EXCLUDED_HEADERS
         ]
-
-        # 4. Возвращаем StreamingResponse
-        # resp.aiter_raw() - это "магия". Этот метод отдает сырые байты, как они приходят
-        # по сети, БЕЗ какой-либо обработки (например, без распаковки gzip).
-        # Это именно то, что нам нужно для идеального прокси.
+        
+        # Используем aiter_bytes(), который автоматически распаковывает контент (e.g. gzip).
+        # Это гарантирует, что мы отправляем дальше чистый, несжатый поток байт.
         return StreamingResponse(
-            resp.aiter_raw(),
+            resp.aiter_bytes(),
             status_code=resp.status_code,
             headers=response_headers,
         )
 
     except httpx.RequestError as e:
-        # Ошибки, связанные с сетью или HTTP
-        return {"error": f"Proxy request error: {str(e)}"}, 502 # 502 Bad Gateway - подходящий код
+        return {"error": f"Proxy request error: {str(e)}"}, 502
     except Exception as e:
-        # Другие непредвиденные ошибки на стороне прокси
         return {"error": f"Internal proxy error: {str(e)}"}, 500