diff --git a/src/components/Models/ModelDialog.jsx b/src/components/Models/ModelDialog.jsx
index b613734248..3de7370a26 100644
--- a/src/components/Models/ModelDialog.jsx
+++ b/src/components/Models/ModelDialog.jsx
@@ -16,7 +16,7 @@ import {
DialogTitle
} from '../ui/Dialog.tsx';
-export default function ModelDialog({ mutate }) {
+export default function ModelDialog({ mutate, modelMap }) {
const dispatch = useDispatch();
const [chatGptLabel, setChatGptLabel] = useState('');
const [promptPrefix, setPromptPrefix] = useState('');
@@ -50,7 +50,7 @@ export default function ModelDialog({ mutate }) {
updateCustomGpt.trigger({ value, chatGptLabel, promptPrefix });
mutate();
- setSaveText('Saved!');
+ setSaveText((prev) => prev + 'd!');
setTimeout(() => {
setSaveText('Save');
}, 2500);
@@ -60,6 +60,10 @@ export default function ModelDialog({ mutate }) {
// dispatch(setDisabled(false));
};
+ if (modelMap[chatGptLabel.toLowerCase()] && saveText === 'Save') {
+ setSaveText('Update');
+ }
+
const requiredProp = required ? { required: true } : {};
return (
diff --git a/src/components/Models/ModelMenu.jsx b/src/components/Models/ModelMenu.jsx
index c5f49dcb61..a53472f9fb 100644
--- a/src/components/Models/ModelMenu.jsx
+++ b/src/components/Models/ModelMenu.jsx
@@ -133,7 +133,7 @@ export default function ModelMenu() {
-
+
);
}