From 1bed18c9973084a261b162c466dd3eb10edaa612 Mon Sep 17 00:00:00 2001
From: Praising Harris Ratnam <praisingharris@gmail.com>
Date: Sat, 17 Oct 2026 15:53:37 +0530
Subject: [PATCH 1/2] fix(management_helpers): exclude budget_id from entity
 budget settings to prevent duplicate kwarg (#46999)

---
 litellm/proxy/management_helpers/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/management_helpers/utils.py b/litellm/proxy/management_helpers/utils.py
index 30aaaacad04..33e0c1597b3 100644
--- a/litellm/proxy/management_helpers/utils.py
+++ b/litellm/proxy/management_helpers/utils.py
@@ -215,8 +215,8 @@ async def handle_budget_for_entity(
         update_budget,
     )
 
-    # Get all budget field names
-    budget_params: Final = LiteLLM_BudgetTable.model_fields.keys()
+    # Get all budget field names, excluding budget_id which selects the budget rather than a budget setting
+    budget_params: Final = set(LiteLLM_BudgetTable.model_fields.keys()) - {"budget_id"}
 
     # Extract budget fields from data
     _json_data: Final = _BUDGET_DATA_MAPPING.validate_python(

From 09ecdc9848eedf911ad165f964ad7b83089809c0 Mon Sep 17 00:00:00 2001
From: Praising Harris Ratnam <praisingharris@gmail.com>
Date: Sun, 18 Oct 2026 17:36:21 +0530
Subject: [PATCH 2/2] test(management_helpers): add regression test for entity
 budget update with existing budget_id (#46999)

Asserts that handle_budget_for_entity constructs BudgetNewRequest without a duplicate-kwarg TypeError when the update payload includes both budget_id and budget settings, and asserts no-op update is avoided when only the matching budget_id is passed.
---
 .../test_management_helpers_utils.py          | 66 +++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/tests/unit/proxy/management_helpers/test_management_helpers_utils.py b/tests/unit/proxy/management_helpers/test_management_helpers_utils.py
index 644f0656c7a..4b4d4e4f17c 100644
--- a/tests/unit/proxy/management_helpers/test_management_helpers_utils.py
+++ b/tests/unit/proxy/management_helpers/test_management_helpers_utils.py
@@ -309,6 +309,72 @@ async def test_handle_budget_for_entity_updates_only_cleared_or_supplied_fields(
     }
 
 
+@pytest.mark.asyncio
+async def test_handle_budget_for_entity_same_budget_id_and_settings_prevents_duplicate_kwarg():
+    """
+    Regression test for #46999:
+    Ensure that when an update payload carries both `budget_id` (matching `existing_budget_id`)
+    and budget settings (e.g. max_budget), BudgetNewRequest is constructed cleanly without a
+    duplicate-kwarg TypeError.
+    Also ensure that when an update payload carries ONLY `budget_id` (matching `existing_budget_id`)
+    without any settings, no no-op update_budget is executed.
+    """
+    from unittest.mock import AsyncMock, patch
+    from litellm.proxy._types import UserAPIKeyAuth
+    from litellm.proxy.management_helpers.utils import handle_budget_for_entity
+    from litellm.types.tag_management import TagUpdateRequest
+
+    mock_prisma_client = AsyncMock()
+
+    # 1. Update payload containing both matching budget_id AND budget setting(s)
+    tag_with_settings = TagUpdateRequest.model_validate(
+        {
+            "name": "tag-1",
+            "budget_id": "budget-123",
+            "max_budget": 150.0,
+        }
+    )
+
+    with patch(
+        "litellm.proxy.management_endpoints.budget_management_endpoints.update_budget",
+        new_callable=AsyncMock,
+    ) as mock_update_budget:
+        result = await handle_budget_for_entity(
+            data=tag_with_settings,
+            existing_budget_id="budget-123",
+            user_api_key_dict=UserAPIKeyAuth(user_id="admin"),
+            prisma_client=mock_prisma_client,
+            litellm_proxy_admin_name="admin",
+        )
+        assert result == "budget-123"
+        mock_update_budget.assert_awaited_once()
+        budget_req = mock_update_budget.await_args.kwargs["budget_obj"]
+        assert budget_req.budget_id == "budget-123"
+        assert budget_req.max_budget == 150.0
+
+    # 2. Update payload containing ONLY matching budget_id (no settings) -> no-op update avoided
+    tag_id_only = TagUpdateRequest.model_validate(
+        {
+            "name": "tag-1",
+            "budget_id": "budget-123",
+        }
+    )
+
+    with patch(
+        "litellm.proxy.management_endpoints.budget_management_endpoints.update_budget",
+        new_callable=AsyncMock,
+    ) as mock_update_budget:
+        result = await handle_budget_for_entity(
+            data=tag_id_only,
+            existing_budget_id="budget-123",
+            user_api_key_dict=UserAPIKeyAuth(user_id="admin"),
+            prisma_client=mock_prisma_client,
+            litellm_proxy_admin_name="admin",
+        )
+        assert result == "budget-123"
+        mock_update_budget.assert_not_awaited()
+
+
 @pytest.mark.asyncio
 async def test_add_new_member_no_budget_when_default_budget_row_is_missing():
     from litellm.proxy._types import LitellmUserRoles