From cd4b188c85c6a337c1694a1948739d5bcaaf1742 Mon Sep 17 00:00:00 2001
From: aoright <102943475+aoright@users.noreply.github.com>
Date: Mon, 28 Sep 2026 15:07:00 +0800
Subject: [PATCH] test(proxy): match ProxyException code string type in
 add_new_model test

---
 .../management_endpoints/test_model_management_endpoints.py     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 aed28764f49..3935f742beb 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
@@ -5919,7 +5919,7 @@ class TestStrategyRouterWriteValidation:
                     ),
                     user_api_key_dict=admin,
                 )
-            assert exc_info.value.code == 400
+            assert str(exc_info.value.code) == "400"
             assert "not a supported or routable provider" in str(
                 exc_info.value.message
             ) or "not a routable provider" in str(exc_info.value.message)