From 27ef9a5a7a9706742b3834c5e15abd536e460a63 Mon Sep 17 00:00:00 2001 From: bussyjd Date: Fri, 10 Apr 2026 18:12:06 +0900 Subject: [PATCH] style: apply black formatting to model management endpoints MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Run `black` (24.10.0, the version pinned in pyproject.toml) on the two files touched by this PR. No semantic changes — pure formatting to clear the LiteLLM Linting / lint check. --- .../model_management_endpoints.py | 16 ++--- .../test_model_management_endpoints.py | 60 +++++++++---------- 2 files changed, 31 insertions(+), 45 deletions(-) diff --git a/litellm/proxy/management_endpoints/model_management_endpoints.py b/litellm/proxy/management_endpoints/model_management_endpoints.py index ea279681292..00464945210 100644 --- a/litellm/proxy/management_endpoints/model_management_endpoints.py +++ b/litellm/proxy/management_endpoints/model_management_endpoints.py @@ -505,9 +505,7 @@ async def _update_model_in_config( if model_params.litellm_params is not None: existing_lp = entry.get("litellm_params", {}) - existing_lp.update( - model_params.litellm_params.model_dump(exclude_none=True) - ) + existing_lp.update(model_params.litellm_params.model_dump(exclude_none=True)) entry["litellm_params"] = existing_lp if model_params.model_info is not None: @@ -964,9 +962,7 @@ async def delete_model( if result is None: raise HTTPException( status_code=400, - detail={ - "error": f"Model with id={model_info.id} not found in db" - }, + detail={"error": f"Model with id={model_info.id} not found in db"}, ) ## DELETE FROM ROUTER ## @@ -1148,9 +1144,7 @@ async def add_new_model( ) # Team-scoped models require the DB for team management. if ( - getattr( - getattr(model_params, "model_info", None), "team_id", None - ) + getattr(getattr(model_params, "model_info", None), "team_id", None) is not None ): raise HTTPException( @@ -1405,9 +1399,7 @@ async def update_model( if updated is None: raise HTTPException( status_code=400, - detail={ - "error": f"Model with id={_model_id} not found in config" - }, + detail={"error": f"Model with id={_model_id} not found in config"}, ) return updated except Exception as e: diff --git a/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py index bd72490bfbe..eb5f03c91e2 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py @@ -64,11 +64,12 @@ class MockPrismaClient: # Support model_name startswith filter (used by _get_team_deployments) if where and "model_name" in where: model_name_filter = where["model_name"] - if isinstance(model_name_filter, dict) and "startswith" in model_name_filter: + if ( + isinstance(model_name_filter, dict) + and "startswith" in model_name_filter + ): prefix = model_name_filter["startswith"] - results = [ - d for d in results if d.model_name.startswith(prefix) - ] + results = [d for d in results if d.model_name.startswith(prefix)] return results @@ -1215,14 +1216,19 @@ class TestAddAndDeleteModelLifecycle: _PS = "litellm.proxy.proxy_server" _ENCRYPT = "litellm.proxy.management_endpoints.model_management_endpoints.encrypt_value_helper" - with patch(f"{_PS}.prisma_client", mock_prisma), \ - patch(f"{_PS}.store_model_in_db", True), \ - patch(f"{_PS}.proxy_config", mock_proxy_config), \ - patch(f"{_PS}.proxy_logging_obj", MagicMock()), \ - patch(f"{_PS}.general_settings", {}), \ - patch(f"{_PS}.premium_user", True), \ - patch(f"{_PS}.llm_router", mock_router), \ - patch(_ENCRYPT, side_effect=lambda value, **kwargs: value): + with patch(f"{_PS}.prisma_client", mock_prisma), patch( + f"{_PS}.store_model_in_db", True + ), patch(f"{_PS}.proxy_config", mock_proxy_config), patch( + f"{_PS}.proxy_logging_obj", MagicMock() + ), patch( + f"{_PS}.general_settings", {} + ), patch( + f"{_PS}.premium_user", True + ), patch( + f"{_PS}.llm_router", mock_router + ), patch( + _ENCRYPT, side_effect=lambda value, **kwargs: value + ): # --- ADD --- add_result = await add_new_model( @@ -1341,6 +1347,8 @@ class TestGetTeamDeployments: result = await _get_team_deployments(team_id, prisma_client) assert len(result) == 1 assert result[0] is dep1 + + ################################################################################ # Config-Only Model Management Tests ################################################################################ @@ -1409,9 +1417,7 @@ class TestConfigOnlyAddModel: with patch("litellm.proxy.proxy_server.prisma_client", None), patch( "litellm.proxy.proxy_server.store_model_in_db", False - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", mock_router ), patch( "litellm.proxy.proxy_server.general_settings", {} @@ -1457,9 +1463,7 @@ class TestConfigOnlyAddModel: with patch("litellm.proxy.proxy_server.prisma_client", None), patch( "litellm.proxy.proxy_server.store_model_in_db", False - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", mock_router ), patch( "litellm.proxy.proxy_server.general_settings", {} @@ -1542,9 +1546,7 @@ class TestConfigOnlyDeleteModel: with patch("litellm.proxy.proxy_server.prisma_client", None), patch( "litellm.proxy.proxy_server.store_model_in_db", False - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", mock_router ), patch( "litellm.proxy.proxy_server.premium_user", False @@ -1578,9 +1580,7 @@ class TestConfigOnlyDeleteModel: with patch("litellm.proxy.proxy_server.prisma_client", None), patch( "litellm.proxy.proxy_server.store_model_in_db", False - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", mock_router ), patch( "litellm.proxy.proxy_server.premium_user", False @@ -1633,9 +1633,7 @@ class TestConfigOnlyUpdateModel: with patch("litellm.proxy.proxy_server.prisma_client", None), patch( "litellm.proxy.proxy_server.store_model_in_db", False - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", mock_router ), patch( "litellm.proxy.proxy_server.premium_user", False @@ -1680,9 +1678,7 @@ class TestConfigOnlyUpdateModel: with patch("litellm.proxy.proxy_server.prisma_client", None), patch( "litellm.proxy.proxy_server.store_model_in_db", False - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", mock_router ), patch( "litellm.proxy.proxy_server.premium_user", False @@ -1737,9 +1733,7 @@ class TestDBModeUnaffected: with patch("litellm.proxy.proxy_server.prisma_client", mock_prisma), patch( "litellm.proxy.proxy_server.store_model_in_db", True - ), patch( - "litellm.proxy.proxy_server.proxy_config", mock_proxy_config - ), patch( + ), patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config), patch( "litellm.proxy.proxy_server.llm_router", MagicMock() ), patch( "litellm.proxy.proxy_server.general_settings", {}