From c2c6b7cc31654c31fdea228eea90d4a1a30a916e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 20 Dec 2023 09:04:56 +0530 Subject: [PATCH] (docs) swagger - add embedding tag --- litellm/proxy/proxy_server.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 154dd57240..944bfe62da 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -872,9 +872,9 @@ def model_list(): object="list", ) -@router.post("/v1/completions", dependencies=[Depends(user_api_key_auth)]) -@router.post("/completions", dependencies=[Depends(user_api_key_auth)]) -@router.post("/engines/{model:path}/completions", dependencies=[Depends(user_api_key_auth)]) +@router.post("/v1/completions", dependencies=[Depends(user_api_key_auth)], tags=["completions"]) +@router.post("/completions", dependencies=[Depends(user_api_key_auth)], tags=["completions"]) +@router.post("/engines/{model:path}/completions", dependencies=[Depends(user_api_key_auth)], tags=["completions"]) async def completion(request: Request, model: Optional[str] = None, user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), background_tasks: BackgroundTasks = BackgroundTasks()): global user_temperature, user_request_timeout, user_max_tokens, user_api_base try: