Spaces:
Running
Running
File size: 2,595 Bytes
08a3216 949dc99 08a3216 9ebd969 08a3216 9ebd969 949dc99 9ebd969 5a86b8e 08a3216 9ebd969 08a3216 5a86b8e 9ebd969 b07bcdd 9ebd969 b07bcdd 5a86b8e 08a3216 9ebd969 08a3216 9ebd969 08a3216 5a86b8e b07bcdd 08a3216 9ebd969 08a3216 9ebd969 08a3216 949dc99 9ebd969 949dc99 9ebd969 949dc99 9ebd969 949dc99 08a3216 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
from fastapi import FastAPI, Request, Header, HTTPException
from fastapi.responses import JSONResponse, Response
import httpx
import socket
app = FastAPI()
# Your internal real API key (DO NOT expose this to users)
REAL_API_KEY = "sk-qO9N6kQEEULMWtF4YGVlTTSjIPllEm1h1wfEBzSmnSbxiXwe"
BASE_URL = "https://fast.typegpt.net"
# Public token that users must use
PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
# Print server IP address on startup
@app.on_event("startup")
async def startup_event():
try:
hostname = socket.gethostname()
server_ip = socket.gethostbyname(hostname)
print(f"===== Server Started =====")
print(f"📡 Server IP: {server_ip}")
except Exception as e:
print(f"⚠️ Could not get server IP: {e}")
@app.api_route("/{path:path}", methods=["GET", "POST"])
async def proxy(request: Request, path: str, authorization: str = Header(None)):
# Validate Authorization token
if not authorization or not authorization.startswith("Bearer "):
raise HTTPException(status_code=401, detail="Missing or malformed Authorization header.")
token = authorization.replace("Bearer ", "").strip()
if token != PUBLIC_AUTH_TOKEN:
raise HTTPException(status_code=401, detail="Invalid Authorization token. Use 'TypeGPT-Free4ALL'.")
# Prepare the backend target URL
target_url = f"{BASE_URL}/{path}"
# Clone headers, replace Authorization
headers = dict(request.headers)
headers["Authorization"] = f"Bearer {REAL_API_KEY}"
headers.pop("host", None)
# Read request body
body = await request.body()
# Forward request to backend
async with httpx.AsyncClient() as client:
try:
response = await client.request(
method=request.method,
url=target_url,
content=body,
headers=headers,
timeout=60
)
except httpx.RequestError as e:
raise HTTPException(status_code=502, detail=f"Request to backend failed: {e}")
# Debug log
print(f"🔄 Forwarded to: {target_url}")
print(f"↩️ TypeGPT Status: {response.status_code}")
print(f"🧾 TypeGPT Response (first 200 chars): {response.text[:200]}")
# Try to return JSON or fallback
try:
return JSONResponse(content=response.json(), status_code=response.status_code)
except Exception:
return Response(
content=response.content,
status_code=response.status_code,
media_type=response.headers.get("content-type", "text/plain")
)
|