test: fix raise

Author: krrishdholakia

litellm/proxy/management_endpoints/key_management_endpoints.py

@@ -923,8 +923,7 @@ async def generate_key_fn(
             litellm_changed_by=litellm_changed_by,
             team_table=team_table,
         )
-    except HTTPException as e:
-        raise e
+
     except Exception as e:
         verbose_proxy_logger.exception(
             "litellm.proxy.proxy_server.generate_key_fn(): Exception occured - {}".format(

tests/proxy_unit_tests/test_key_generate_prisma.py

@@ -1511,7 +1511,10 @@ async def test():
         asyncio.run(test())
     except Exception as e:
         print("Got Exception", e)
-        print(e.message)
+        if hasattr(e, "message"):
+            print(e.message)
+        else:
+            print(e)
         pytest.fail(f"An exception occurred - {str(e)}")
 
 

tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py

@@ -18,6 +18,7 @@
     LiteLLM_TeamTableCachedObj,
     LiteLLM_VerificationToken,
     LitellmUserRoles,
+    ProxyException,
     UpdateKeyRequest,
 )
 from litellm.proxy.auth.user_api_key_auth import UserAPIKeyAuth