Deprecate v1 routes in favor of v2 (#23)

awslabs · Jun 7, 2024 · 71446ff · 71446ff
1 parent 88a64e0
commit 71446ff
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lib/serve/rest-api/src/api/routes.py b/lib/serve/rest-api/src/api/routes.py
@@ -28,9 +28,13 @@
 security = OIDCHTTPBearer()
 router = APIRouter()
 
-router.include_router(models.router, prefix="/v1", tags=["models"], dependencies=[Depends(security)])
-router.include_router(embeddings.router, prefix="/v1", tags=["embeddings"], dependencies=[Depends(security)])
-router.include_router(generation.router, prefix="/v1", tags=["generation"], dependencies=[Depends(security)])
+router.include_router(models.router, prefix="/v1", tags=["models"], dependencies=[Depends(security)], deprecated=True)
+router.include_router(
+    embeddings.router, prefix="/v1", tags=["embeddings"], dependencies=[Depends(security)], deprecated=True
+)
+router.include_router(
+    generation.router, prefix="/v1", tags=["generation"], dependencies=[Depends(security)], deprecated=True
+)
 router.include_router(
     litellm_passthrough.router, prefix="/v2/serve", tags=["litellm_passthrough"], dependencies=[Depends(security)]
 )