From 98a321939ea1ba551025b16dc09f13e7a5e980ca Mon Sep 17 00:00:00 2001
From: Sahil <issahil404@gmail.com>
Date: Fri, 21 Mar 2025 15:52:18 +0530
Subject: [PATCH] fix: add checkmark icon to indicate selected model in
 dropdown

---
 gui/src/components/modelSelection/ModelSelect.tsx | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/gui/src/components/modelSelection/ModelSelect.tsx b/gui/src/components/modelSelection/ModelSelect.tsx
index 7cf854e4d07..3bda3b95ec7 100644
--- a/gui/src/components/modelSelection/ModelSelect.tsx
+++ b/gui/src/components/modelSelection/ModelSelect.tsx
@@ -5,6 +5,7 @@ import {
   CubeIcon,
   PlusIcon,
   TrashIcon,
+  CheckIcon,
 } from "@heroicons/react/24/outline";
 import { useContext, useEffect, useRef, useState } from "react";
 import { useDispatch } from "react-redux";
@@ -29,6 +30,7 @@ interface ModelOptionProps {
   idx: number;
   showMissingApiKeyMsg: boolean;
   showDelete?: boolean;
+  isSelected?: boolean;
 }
 
 interface Option {
@@ -132,6 +134,7 @@ function ModelOption({
   idx,
   showDelete,
   showMissingApiKeyMsg,
+  isSelected,
 }: ModelOptionProps) {
   const ideMessenger = useContext(IdeMessengerContext);
 
@@ -199,6 +202,9 @@ function ModelOption({
             {showDelete && (
               <StyledTrashIcon $hovered={hovered} onClick={onClickDelete} />
             )}
+            {isSelected && (
+              <CheckIcon className="ml-1 h-4 w-4" />
+            )}
           </div>
         </div>
       </div>
@@ -338,6 +344,7 @@ function ModelSelect() {
                 key={idx}
                 showDelete={options.length > 1}
                 showMissingApiKeyMsg={option.apiKey === ""}
+                isSelected={option.value === defaultModel?.title}
               />
             ))}
           </div>