From cc1133d0cb5ffbcb74c0ac084113c53011e4455c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 28 Dec 2023 13:13:04 +0530 Subject: [PATCH] (docs) proxy - swagger docs, add tags for model/info --- litellm/proxy/proxy_server.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 425b042b39..891e735e89 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -984,9 +984,11 @@ async def startup_event(): #### API ENDPOINTS #### -@router.get("/v1/models", dependencies=[Depends(user_api_key_auth)]) @router.get( - "/models", dependencies=[Depends(user_api_key_auth)] + "/v1/models", dependencies=[Depends(user_api_key_auth)], tags=["model management"] +) +@router.get( + "/models", dependencies=[Depends(user_api_key_auth)], tags=["model management"] ) # if project requires model list def model_list(): global llm_model_list, general_settings @@ -2026,7 +2028,9 @@ async def async_queue_request( ) -@router.get("/ollama_logs", dependencies=[Depends(user_api_key_auth)]) +@router.get( + "/ollama_logs", dependencies=[Depends(user_api_key_auth)], tags=["experimental"] +) async def retrieve_server_log(request: Request): filepath = os.path.expanduser("~/.ollama/logs/server.log") return FileResponse(filepath)