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 && (
)}
+ {isSelected && (
+
+ )}
@@ -338,6 +344,7 @@ function ModelSelect() {
key={idx}
showDelete={options.length > 1}
showMissingApiKeyMsg={option.apiKey === ""}
+ isSelected={option.value === defaultModel?.title}
/>
))}