Update main.py

main.py

@@ -5,6 +5,7 @@ import socket
 
 app = FastAPI()
 
+# Replace this with your actual backend API key
 REAL_API_KEY = "sk-kwWVUQPDsLemvilLcyzSSWRzo8sctCzxzlbdN0ZC5ZUCCv0m"
 BASE_URL = "https://fast.typegpt.net"
 PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
@@ -21,14 +22,16 @@ async def startup_event():
 
 @app.api_route("/{path:path}", methods=["GET", "POST"])
 async def proxy(request: Request, path: str, authorization: str = Header(None)):
+    # Validate 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'.")
 
     target_url = f"{BASE_URL}/{path}"
+
     headers = {
         "Authorization": f"Bearer {REAL_API_KEY}",
         "Content-Type": request.headers.get("content-type", "application/json"),
@@ -46,22 +49,21 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
     async with httpx.AsyncClient(timeout=60) as client:
         try:
             if is_stream:
-                async with client.stream(
-                    method=request.method,
-                    url=target_url,
-                    headers=headers,
-                    content=body
-                ) as upstream_response:
-
-                    async def stream_generator():
+                async def stream_generator():
+                    async with client.stream(
+                        method=request.method,
+                        url=target_url,
+                        headers=headers,
+                        content=body
+                    ) as upstream_response:
                         async for chunk in upstream_response.aiter_bytes():
                             yield chunk
 
-                    return StreamingResponse(
-                        stream_generator(),
-                        status_code=upstream_response.status_code,
-                        media_type="text/event-stream"
-                    )
+                return StreamingResponse(
+                    stream_generator(),
+                    status_code=200,
+                    media_type="text/event-stream"
+                )
             else:
                 response = await client.request(
                     method=request.method,