From 1b271aa95ce922f1b8a96ff6aac2ab3bf3727df6 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 15 Jun 2024 12:22:16 -0700
Subject: [PATCH] fix swagger

---
 litellm/proxy/proxy_server.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 9709648e41..938661f307 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -10688,7 +10688,11 @@ async def get_routes():
 
 
 #### TEST ENDPOINTS ####
-@router.get("/token/generate", dependencies=[Depends(user_api_key_auth)])
+@router.get(
+    "/token/generate",
+    dependencies=[Depends(user_api_key_auth)],
+    include_in_schema=False,
+)
 async def token_generate():
     """
     Test endpoint. Admin-only access. Meant for generating admin tokens with specific claims and testing if they work for creating keys, etc.