Update app.py

app.py

@@ -8,47 +8,72 @@ app = FastAPI()
 OR_BASE_URL = "https://api.openai.com/v1"
 
 # Создаем один асинхронный клиент на все приложение для переиспользования соединений
-client = httpx.AsyncClient(base_url=OR_BASE_URL)
+# allow_redirects=False, т.к. мы хотим сами управлять редиректами как прокси
+client = httpx.AsyncClient(base_url=OR_BASE_URL, follow_redirects=False)
+
+# Список "hop-by-hop" заголовков, которые не должны проксироваться дальше.
+# Это стандартная практика для HTTP прокси.
+HOP_BY_HOP_HEADERS = {
+    "connection",
+    "keep-alive",
+    "proxy-authenticate",
+    "proxy-authorization",
+    "te",
+    "trailers",
+    "transfer-encoding",
+    "upgrade",
+}
+
 
 @app.get("/")
 async def home():
-    return {"status": "ok", "message": "Async OpenRouter proxy is working :3"}
+    return {"status": "ok", "message": "Async Transparent OpenRouter proxy is working :3"}
 
 
 @app.api_route("/{endpoint:path}", methods=["GET", "POST", "PUT", "PATCH", "DELETE"])
 async def proxy(endpoint: str, request: Request):
-    # Копируем заголовки, кроме тех, что не должны пересылаться
-    headers = {k: v for k, v in request.headers.items() if k.lower() not in ["host", "content-length", "content-encoding"]}
-    
     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=f"/{endpoint}",
+            url=url,
             headers=headers,
-            params=request.query_params,
-            content=await request.body() # Асинхронно читаем тело входящего запроса
+            content=body
         )
-        
-        # `stream=True` здесь работает по-настоящему асинхронно
         resp = await client.send(req, stream=True)
 
-        # Асинхронный генератор для потоковой передачи ответа
-        async def async_generator():
-            async for chunk in resp.aiter_bytes():
-                yield chunk
-
-        # Копируем заголовки ответа
-        response_headers = {k: v for k, v in resp.headers.items() if k.lower() not in ["content-encoding", "transfer-encoding", "connection"]}
+        # 3. Копируем заголовки ответа от целевого сервера
+        # Исключаем только hop-by-hop заголовки. Все остальное (Content-Type, Content-Encoding,
+        # кастомные x-openai-*, и т.д.) будет скопировано.
+        response_headers = [
+            (key, value)
+            for key, value in resp.headers.items()
+            if key.lower() not in HOP_BY_HOP_HEADERS
+        ]
 
+        # 4. Возвращаем StreamingResponse
+        # resp.aiter_raw() - это "магия". Этот метод отдает сырые байты, как они приходят
+        # по сети, БЕЗ какой-либо обработки (например, без распаковки gzip).
+        # Это именно то, что нам нужно для идеального прокси.
         return StreamingResponse(
-            async_generator(),
+            resp.aiter_raw(),
             status_code=resp.status_code,
             headers=response_headers,
         )
+
     except httpx.RequestError as e:
-        return {"error": f"HTTPX error: {str(e)}"}, 500
+        # Ошибки, связанные с сетью или HTTP
+        return {"error": f"Proxy request error: {str(e)}"}, 502 # 502 Bad Gateway - подходящий код
     except Exception as e:
-        return {"error": str(e)}, 500
-
-# Для запуска: uvicorn your_file_name:app --host 0.0.0.0 --port 7860
+        # Другие непредвиденные ошибки на стороне прокси
+        return {"error": f"Internal proxy error: {str(e)}"}, 500