Wimmboo commited on
Commit
eeb5b0e
·
1 Parent(s): 10aa100

revert to /v1 prefix with APIRouter

Browse files
Files changed (1) hide show
  1. proxy/router.py +6 -6
proxy/router.py CHANGED
@@ -12,8 +12,8 @@ from proxy.client import forward_request
12
  from proxy.providers import PROVIDERS
13
 
14
  logger = logging.getLogger(__name__)
 
15
 
16
- api = APIRouter()
17
 
18
 
19
  MODELS = {
@@ -36,7 +36,7 @@ MODELS = {
36
  }
37
 
38
 
39
- @api.get("/api/v1/models")
40
  async def list_models() -> JSONResponse:
41
  data = {
42
  "object": "list",
@@ -53,7 +53,7 @@ async def list_models() -> JSONResponse:
53
  return JSONResponse(content=data)
54
 
55
 
56
- @api.post("/api/v1/chat/completions")
57
  async def chat_completions(request: Request) -> Response:
58
  provider = resolve_provider(request)
59
  real_key = get_real_key(provider)
@@ -80,12 +80,12 @@ async def chat_completions(request: Request) -> Response:
80
  )
81
 
82
 
83
- @api.get("/api/v1/health")
84
  async def health() -> JSONResponse:
85
  return JSONResponse({"status": "ok", "providers": list(PROVIDERS.keys())})
86
 
87
 
88
- @api.get("/api/v1/keys")
89
  async def proxy_keys() -> JSONResponse:
90
  masked = {}
91
  for name, key in PROVIDER_KEYS.items():
@@ -101,7 +101,7 @@ def create_app() -> FastAPI:
101
 
102
  @app.get("/")
103
  async def root():
104
- return {"message": "OpenAI API Proxy", "docs": "/api/v1/health"}
105
 
106
  @app.exception_handler(StarletteHTTPException)
107
  async def http_exception_handler(_request: Request, exc: StarletteHTTPException) -> JSONResponse:
 
12
  from proxy.providers import PROVIDERS
13
 
14
  logger = logging.getLogger(__name__)
15
+ api = APIRouter(prefix="/v1")
16
 
 
17
 
18
 
19
  MODELS = {
 
36
  }
37
 
38
 
39
+ @api.get("/models")
40
  async def list_models() -> JSONResponse:
41
  data = {
42
  "object": "list",
 
53
  return JSONResponse(content=data)
54
 
55
 
56
+ @api.post("/chat/completions")
57
  async def chat_completions(request: Request) -> Response:
58
  provider = resolve_provider(request)
59
  real_key = get_real_key(provider)
 
80
  )
81
 
82
 
83
+ @api.get("/health")
84
  async def health() -> JSONResponse:
85
  return JSONResponse({"status": "ok", "providers": list(PROVIDERS.keys())})
86
 
87
 
88
+ @api.get("/keys")
89
  async def proxy_keys() -> JSONResponse:
90
  masked = {}
91
  for name, key in PROVIDER_KEYS.items():
 
101
 
102
  @app.get("/")
103
  async def root():
104
+ return {"message": "OpenAI API Proxy", "docs": "/v1/health"}
105
 
106
  @app.exception_handler(StarletteHTTPException)
107
  async def http_exception_handler(_request: Request, exc: StarletteHTTPException) -> JSONResponse: