From 89a7451cb9ee26ff9f642335714dcc6f449d1fc2 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 20 Jun 2025 22:42:30 -0700 Subject: [PATCH] fix: fix test --- litellm/proxy/management_helpers/utils.py | 2 +- tests/proxy_unit_tests/test_proxy_server.py | 9 +++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/management_helpers/utils.py b/litellm/proxy/management_helpers/utils.py index 170ece20488..246fa0b6390 100644 --- a/litellm/proxy/management_helpers/utils.py +++ b/litellm/proxy/management_helpers/utils.py @@ -135,7 +135,7 @@ async def add_new_member( else: _budget_id = default_team_budget_id - if returned_user is not None and returned_user.user_id is not None: + if _budget_id and returned_user is not None and returned_user.user_id is not None: _returned_team_membership = ( await prisma_client.db.litellm_teammembership.create( data={ diff --git a/tests/proxy_unit_tests/test_proxy_server.py b/tests/proxy_unit_tests/test_proxy_server.py index 878a6e1c172..a515f6bbaa3 100644 --- a/tests/proxy_unit_tests/test_proxy_server.py +++ b/tests/proxy_unit_tests/test_proxy_server.py @@ -1037,6 +1037,7 @@ async def test_create_team_member_add(prisma_client, new_member_method): return_value=LiteLLM_TeamTableCachedObj(team_id="1234") ) + print(f"team_member_add_request={team_member_add_request}") await team_member_add( data=team_member_add_request, user_api_key_dict=UserAPIKeyAuth(user_role="proxy_admin"), @@ -2207,7 +2208,9 @@ async def test_run_background_health_check_reflects_llm_model_list(monkeypatch): monkeypatch.setattr(proxy_server, "health_check_interval", 1) monkeypatch.setattr(proxy_server, "health_check_details", None) - monkeypatch.setattr(proxy_server, "llm_model_list", copy.deepcopy(test_model_list_1)) + monkeypatch.setattr( + proxy_server, "llm_model_list", copy.deepcopy(test_model_list_1) + ) monkeypatch.setattr(proxy_server, "perform_health_check", fake_perform_health_check) monkeypatch.setattr(proxy_server, "health_check_results", {}) @@ -2221,7 +2224,9 @@ async def test_run_background_health_check_reflects_llm_model_list(monkeypatch): except asyncio.CancelledError: pass - monkeypatch.setattr(proxy_server, "llm_model_list", copy.deepcopy(test_model_list_2)) + monkeypatch.setattr( + proxy_server, "llm_model_list", copy.deepcopy(test_model_list_2) + ) try: await proxy_server._run_background_health_check()