Update main.py

main.py

@@ -5,7 +5,7 @@ import socket
 
 app = FastAPI()
 
-REAL_API_KEY = "sk-94NDKhKQkhKoYnY65mg4NAIFK5BqNiCtxo8u3PsDpb0IucZt"
+REAL_API_KEY = "sk-kwWVUQPDsLemvilLcyzSSWRzo8sctCzxzlbdN0ZC5ZUCCv0m"
 BASE_URL = "https://fast.typegpt.net"
 PUBLIC_AUTH_TOKEN = "TypeGPT-Free4ALL"
 
@@ -46,21 +46,23 @@ async def proxy(request: Request, path: str, authorization: str = Header(None)):
     async with httpx.AsyncClient(timeout=60) as client:
         try:
             if is_stream:
-                # For streaming, use .stream()
-                upstream_response = await client.stream(
+                async with client.stream(
                     method=request.method,
                     url=target_url,
                     headers=headers,
                     content=body
-                )
+                ) as upstream_response:
 
-                async def stream_generator():
-                    async for chunk in upstream_response.aiter_bytes():
-                        yield chunk
+                    async def stream_generator():
+                        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=upstream_response.status_code,
+                        media_type="text/event-stream"
+                    )
             else:
-                # For normal response, use .request()
                 response = await client.request(
                     method=request.method,
                     url=target_url,