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} /> ))}