Upload 3 files

Dockerfile

@@ -0,0 +1,18 @@
+# Указываем базовый образ Python
+FROM python:3.9-slim
+
+# Устанавливаем рабочую директорию внутри контейнера
+WORKDIR /code
+
+# Копируем файл с зависимостями
+COPY ./requirements.txt /code/requirements.txt
+
+# Устанавливаем зависимости
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Копируем остальной код приложения
+COPY ./ /code/
+
+# Указываем команду для запуска приложения
+# Она будет исполнена, когда контейнер запустится
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,43 +1,67 @@
 import os
-import requests
-from flask import Flask, request, Response
+import httpx
+from fastapi import FastAPI, Request, Response
 
-app = Flask(__name__)
+app = FastAPI(
+    title="Simple OpenAI Proxy",
+    description="A non-streaming proxy that waits for the full response from OpenAI."
+)
 
-OR_BASE_URL = "https://openrouter.ai/api"
+OR_BASE_URL = "https://api.openai.com/v1"
 
+# Создаем один асинхронный клиент на все приложение
+# Увеличим таймаут по умолчанию, т.к. генерация может быть долгой
+client = httpx.AsyncClient(base_url=OR_BASE_URL, timeout=120.0)
 
-@app.route("/", methods=["GET"])
-def home():
-    return {"status": "ok", "message": "OpenRouter proxy is working :3"}
+# "Hop-by-hop" заголовки, которые не должны проксироваться.
+# Они относятся к конкретному соединению, а не к сообщению в целом.
+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": "Simple, non-streaming OpenAI proxy is working."}
 
-@app.route("/<path:endpoint>", methods=["GET", "POST", "PUT", "PATCH", "DELETE"])
-def proxy(endpoint):
-    url = f"{OR_BASE_URL}/{endpoint}"
-
-    # Копируем все заголовки, кроме host и content-length
-    headers = {k: v for k, v in request.headers if k.lower() not in ["host", "content-length"]}
 
+@app.api_route("/{endpoint:path}", methods=["GET", "POST", "PUT", "PATCH", "DELETE"])
+async def proxy(endpoint: str, request: Request):
     try:
-        resp = requests.request(
+        # 1. Формируем URL для запроса к OpenAI
+        url = httpx.URL(path=f"/{endpoint}", query=request.url.query.encode("utf-8"))
+        
+        # 2. Копируем заголовки и тело из входящего запроса
+        headers = {k: v for k, v in request.headers.items() if k.lower() != "host"}
+        body = await request.body()
+
+        # 3. Отправляем запрос и ЖДЕМ ПОЛНОГО ОТВЕТА
+        # Никакого stream=True. Мы ждем, пока `httpx` полностью загрузит ответ.
+        resp_openai = await client.request(
             method=request.method,
             url=url,
             headers=headers,
-            params=request.args,
-            data=request.get_data(),
-            allow_redirects=False,
-            stream=True,
+            content=body
         )
 
-        excluded_headers = ["content-encoding", "transfer-encoding", "connection"]
-        response_headers = [(k, v) for k, v in resp.raw.headers.items() if k.lower() not in excluded_headers]
+        # 4. Фильтруем "hop-by-hop" заголовки из ответа OpenAI
+        # Все остальные заголовки (Content-Type, openai-*, и т.д.) будут сохранены
+        response_headers = {
+            k: v for k, v in resp_openai.headers.items() 
+            if k.lower() not in HOP_BY_HOP_HEADERS
+        }
 
-        return Response(resp.content, resp.status_code, response_headers)
+        # 5. Возвращаем обычный `Response` со всеми данными от OpenAI
+        # Это не стриминг. Мы возвращаем готовый, полный ответ.
+        return Response(
+            content=resp_openai.content,
+            status_code=resp_openai.status_code,
+            headers=response_headers
+        )
 
+    except httpx.RequestError as e:
+        # Ошибка сети при обращении к OpenAI
+        return Response(content=f"Error connecting to OpenAI API: {e}", status_code=502) # 502 Bad Gateway
     except Exception as e:
-        return {"error": str(e)}, 500
-
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+        # Внутренняя ошибка в самом прокси-сервере
+        return Response(content=f"Internal proxy error: {e}", status_code=500)

requirements.txt

@@ -1,2 +1,3 @@
-flask
-requests
+fastapi
+uvicorn
+httpx