Update main.py

main.py

@@ -110,15 +110,17 @@ async def get_models():
 async def fetch_models():
     return await get_models()
 
-@app.post(["/chat/completions", "/v1/chat/completions"])
+@app.post("/chat/completions")
+@app.post("/v1/chat/completions")
 async def get_completion(payload: Payload, request: Request):
     # Use the correct endpoint depending on the model type (no authentication now 😉)
     endpoint = secret_api_endpoint_2 if payload.model in alternate_models else secret_api_endpoint
-    
+
     # Use the payload directly as it includes stream and other user data
     payload_dict = payload.dict()
-    print(payload_dict) # coz i m curious af heheheh :)
-    #data is kept to me only so dont worry
+    print(payload_dict)  # coz I m curious af heheheh :)
+    # data is kept to me only don't worry :)
+
     async def stream_generator(payload_dict):
         async with httpx.AsyncClient() as client:
             try:
@@ -133,8 +135,9 @@ async def get_completion(payload: Payload, request: Request):
                 raise HTTPException(status_code=500, detail=f"Streaming failed: {req_err}")
             except Exception as e:
                 raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {e}")
+
+    return StreamingResponse(stream_generator(payload_dict), media_type="application/json")
     
-    return StreamingResponse(stream_generator(openai_payload), media_type="application/json")
 @app.on_event("startup")
 async def startup_event():
     print("API endpoints:")