Update main.py

main.py

@@ -20,6 +20,20 @@ class Payload(BaseModel):
     model: str
     messages: list
 
+@app.get("/")
+async def root():
+    return {"message": "Hi, I am API made by mr_leaderyt"}
+
+@app.get("/models")
+async def get_models():
+    try:
+        response = requests.get("https://lokiai.vercel.app/api/v1/models")
+        response.raise_for_status()
+        models = response.json()
+        return models
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Request failed: {e}")
+
 @app.post("/api/v1/chat/completions")
 async def get_completion(payload: Payload, request: Request):
     api_key = request.headers.get("Authorization")
@@ -51,6 +65,14 @@ async def get_completion(payload: Payload, request: Request):
     except ValueError:
         raise HTTPException(status_code=500, detail="Non-JSON response received")
 
+# Log the API endpoints
+@app.on_event("startup")
+async def startup_event():
+    print("API endpoints:")
+    print("GET /")
+    print("GET /models")
+    print("POST /api/v1/chat/completions")
+
 # Run the server with Uvicorn using the 'main' module
 if __name__ == "__main__":
     import uvicorn