Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Request, Response
+from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse
 import httpx
 import json
@@ -13,7 +13,7 @@ app = FastAPI()
 async def chat(request: Request):
     """
     Handle chat completion requests.
-    This function processes incoming POST requests to the '/chat/completions' endpoint.
+    This function processes incoming POST requests to the '/ai/v1/chat/completions' endpoint.
     It prepares the payload for the LLM API, generates a JWT for authentication,
     and streams the response from the LLM API back to the client.
     Returns:
@@ -68,6 +68,10 @@ async def chat(request: Request):
 
     return StreamingResponse(generate(), media_type='application/octet-stream')
 
+@app.get("/")
+async def root():
+    return {"message": "Welcome to the AI Chat Completions API"}
+
 if __name__ == '__main__':
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)