From edc84b6b6d03d74605ddaeed09438b6fe8599982 Mon Sep 17 00:00:00 2001
From: tanjiro <56165694+NANDINI-star@users.noreply.github.com>
Date: Sun, 29 Jun 2025 00:06:01 +0900
Subject: [PATCH] Update conditional_public_model_name.tsx (#11713)

---
 .../add_model/conditional_public_model_name.tsx     | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/add_model/conditional_public_model_name.tsx b/ui/litellm-dashboard/src/components/add_model/conditional_public_model_name.tsx
index 0f79401753e..40a124264a7 100644
--- a/ui/litellm-dashboard/src/components/add_model/conditional_public_model_name.tsx
+++ b/ui/litellm-dashboard/src/components/add_model/conditional_public_model_name.tsx
@@ -40,11 +40,14 @@ const ConditionalPublicModelName: React.FC = () => {
       const currentMappings = form.getFieldValue('model_mappings') || [];
       
       // Only update if the mappings don't exist or don't match the selected models
-      const shouldUpdateMappings = currentMappings.length !== selectedModels.length || 
-        !selectedModels.every(model => 
-          currentMappings.some((mapping: { public_name: string; litellm_model: string }) => 
-            mapping.public_name === model || 
-            (model === 'custom' && mapping.public_name === customModelName)));
+      const shouldUpdateMappings = currentMappings.length !== selectedModels.length ||
+        !selectedModels.every(model =>
+          currentMappings.some((mapping: { public_name: string; litellm_model: string }) => {
+            if (model === 'custom') {
+              return mapping.litellm_model === 'custom' || mapping.litellm_model === customModelName;
+            }
+            return mapping.litellm_model === model;
+          }));
       
       if (shouldUpdateMappings) {
         const mappings = selectedModels.map((model: string) => {