Shyamnath commited on
Commit
2375a0e
·
1 Parent(s): c40c75a

Fix app.py imports to use proxy server instead of playground

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,11 +1,12 @@
1
  from fastapi import FastAPI
2
- from litellm.playground_server import add_playground_routes
 
3
 
4
  app = FastAPI()
5
 
6
- # Add LiteLLM Playground UI at `/docs`
7
- add_playground_routes(app)
8
 
9
  @app.get("/")
10
  def home():
11
- return {"message": "LiteLLM UI is running! Visit /docs"}
 
1
  from fastapi import FastAPI
2
+ from litellm.proxy.proxy_server import ProxyServer
3
+ from litellm.proxy.proxy_server import app as proxy_app
4
 
5
  app = FastAPI()
6
 
7
+ # Mount the LiteLLM Proxy server
8
+ app.mount("/proxy", proxy_app)
9
 
10
  @app.get("/")
11
  def home():
12
+ return {"message": "LiteLLM Server is running! Visit /proxy/docs for the API documentation"}