from fastapi import FastAPI
from litellm.playground_server import add_playground_routes

app = FastAPI()

# Add LiteLLM Playground UI at `/docs`
add_playground_routes(app)

@app.get("/")
def home():
    return {"message": "LiteLLM UI is running! Visit /docs"}