Upload 2 files

app.py

@@ -0,0 +1,93 @@
+import os
+import asyncio
+import aiohttp
+import json
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import StreamingResponse, JSONResponse
+
+TARGET_BASE_URL = "https://api.deepinfra.com/v1/openai"
+REAL_API_KEY = os.environ["REAL_API_KEY"]        # 从 Space secrets 读取
+FAKE_KEY = os.environ.get("FAKE_KEY", "123456")  # 本地访问时的鉴权
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    app.state.session = aiohttp.ClientSession()
+    yield
+    await app.state.session.close()
+
+app = FastAPI(lifespan=lifespan)
+
+@app.api_route("/v1/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS"])
+async def proxy_endpoint(path: str = "", request: Request = None):
+    if request.method == "OPTIONS":
+        return Response(status_code=200, headers={
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Methods": "GET, POST, PUT, DELETE, OPTIONS",
+            "Access-Control-Allow-Headers": "Authorization, Content-Type"
+        })
+
+    # 鉴权：只允许携带正确 FAKE_KEY 的请求
+    auth_header = request.headers.get("authorization", "")
+    if auth_header.lower() != f"bearer {FAKE_KEY}":
+        return JSONResponse(status_code=401, content={"error": "Invalid fake key"})
+
+    target_url = f"{TARGET_BASE_URL.rstrip('/')}/{path}" if path else TARGET_BASE_URL.rstrip('/')
+
+    headers = {}
+    for key, value in request.headers.items():
+        if key.lower() in ["host", "content-length", "connection", "authorization", "transfer-encoding"]:
+            continue
+        headers[key] = value
+    headers["Authorization"] = f"Bearer {REAL_API_KEY}"
+
+    body = await request.body()
+    query_params = dict(request.query_params)
+
+    is_stream = False
+    if body:
+        try:
+            is_stream = json.loads(body).get("stream", False)
+        except:
+            pass
+
+    session = request.app.state.session
+    timeout = aiohttp.ClientTimeout(total=None, connect=10, sock_read=60)
+
+    resp = await session.request(
+        method=request.method,
+        url=target_url,
+        headers=headers,
+        data=body,
+        params=query_params,
+        timeout=timeout
+    )
+
+    if is_stream or "text/event-stream" in resp.headers.get("content-type", ""):
+        async def stream_generator():
+            try:
+                async for chunk in resp.content.iter_any():
+                    yield chunk
+            except:
+                pass
+            finally:
+                resp.close()
+        return StreamingResponse(
+            stream_generator(),
+            media_type=resp.headers.get("content-type", "text/event-stream"),
+            headers={
+                "cache-control": "no-cache",
+                "connection": "keep-alive",
+                "access-control-allow-origin": "*"
+            }
+        )
+    else:
+        content = await resp.read()
+        resp.close()
+        return Response(
+            content=content,
+            status_code=resp.status,
+            headers={
+                "content-type": resp.headers.get("content-type", "application/json"),
+                "access-control-allow-origin": "*"
+            }
+        )

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+aiohttp