From 8f1f9cb426a7191aef73a866d4424cb0730b66a7 Mon Sep 17 00:00:00 2001
From: Shivam Mishra <scm.mymail@gmail.com>
Date: Tue, 23 Dec 2025 13:59:08 +0530
Subject: [PATCH] feat: add preferred badge

---
 .../dashboard/i18n/locale/en/general.json     |  3 +-
 .../captain/components/ModelSelector.vue      | 36 ++++++++++++-------
 2 files changed, 26 insertions(+), 13 deletions(-)

diff --git a/app/javascript/dashboard/i18n/locale/en/general.json b/app/javascript/dashboard/i18n/locale/en/general.json
index 283cf79c4..98f724223 100644
--- a/app/javascript/dashboard/i18n/locale/en/general.json
+++ b/app/javascript/dashboard/i18n/locale/en/general.json
@@ -7,6 +7,7 @@
     },
     "CLOSE": "Close",
     "BETA": "Beta",
-    "BETA_DESCRIPTION": "This feature is in beta and may change as we improve it."
+    "BETA_DESCRIPTION": "This feature is in beta and may change as we improve it.",
+    "PREFERRED": "Preferred"
   }
 }
diff --git a/app/javascript/dashboard/routes/dashboard/settings/captain/components/ModelSelector.vue b/app/javascript/dashboard/routes/dashboard/settings/captain/components/ModelSelector.vue
index 8d828cbe3..adc8bd0b1 100644
--- a/app/javascript/dashboard/routes/dashboard/settings/captain/components/ModelSelector.vue
+++ b/app/javascript/dashboard/routes/dashboard/settings/captain/components/ModelSelector.vue
@@ -1,7 +1,7 @@
 <script setup>
 import { ref, computed, watch } from 'vue';
 import { useI18n } from 'vue-i18n';
-import { useMapGetter } from 'dashboard/composables/store';
+import { useFunctionGetter } from 'dashboard/composables/store';
 import Icon from 'dashboard/components-next/icon/Icon.vue';
 import DropdownBody from 'dashboard/components-next/dropdown-menu/base/DropdownBody.vue';
 import DropdownItem from 'dashboard/components-next/dropdown-menu/base/DropdownItem.vue';
@@ -38,17 +38,21 @@ const iconForModel = model => {
 const { t } = useI18n();
 const isOpen = ref(false);
 
-const getModelsForFeature = useMapGetter('captainConfig/getModelsForFeature');
-const getSelectedModelForFeature = useMapGetter(
-  'captainConfig/getSelectedModelForFeature'
+const availableModels = useFunctionGetter(
+  'captainConfig/getModelsForFeature',
+  props.featureKey
 );
 
-const availableModels = computed(() =>
-  getModelsForFeature.value(props.featureKey)
+const recommendedModelId = useFunctionGetter(
+  'captainConfig/getDefaultModelForFeature',
+  props.featureKey
 );
-const selectedModel = computed(() =>
-  getSelectedModelForFeature.value(props.featureKey)
+
+const selectedModel = useFunctionGetter(
+  'captainConfig/getSelectedModelForFeature',
+  props.featureKey
 );
+
 const selectedModelId = ref(null);
 
 watch(
@@ -124,7 +128,7 @@ const selectModel = model => {
       </button>
       <DropdownBody
         v-if="isOpen"
-        class="absolute right-0 top-full mt-1 min-w-56 z-50"
+        class="absolute right-0 top-full mt-1 min-w-64 z-50"
       >
         <DropdownItem
           v-for="model in availableModels"
@@ -133,15 +137,23 @@ const selectModel = model => {
           class="rounded-lg hover:bg-n-alpha-1"
           :class="{ 'bg-n-alpha-2': selectedModelId === model.id }"
         >
-          <div class="flex gap-2">
+          <div class="flex gap-2 w-full">
             <Icon
               :icon="iconForModel(model)"
               class="size-4 flex-shrink-0 text-n-slate-11"
             />
             <div class="flex flex-col w-full text-left gap-1">
-              <span class="text-sm font-medium leading-none text-n-slate-12">
+              <div
+                class="text-sm w-full font-medium leading-none text-n-slate-12 flex items-baseline justify-between"
+              >
                 {{ model.display_name }}
-              </span>
+                <span
+                  v-if="model.id === recommendedModelId"
+                  class="text-[10px] uppercase text-n-iris-11 border border-1 border-n-iris-10 leading-none rounded-lg px-1 py-0.5"
+                >
+                  {{ t('GENERAL.PREFERRED') }}
+                </span>
+              </div>
               <span class="text-xs text-n-slate-11">
                 {{ getCreditLabel(model) }}
               </span>