feat(ui): add main model trigger phrases

invokeai/frontend/web/public/locales/en.json

@@ -855,7 +855,8 @@
         "statusConverting": "Converting",
         "syncModels": "Sync Models",
         "syncModelsDesc": "If your models are out of sync with the backend, you can refresh them up using this option. This is generally handy in cases where you add models to the InvokeAI root folder or autoimport directory after the application has booted.",
-        "triggerPhrases": "Trigger Phrases",
+        "loraTriggerPhrases": "LoRA Trigger Phrases",
+        "mainModelTriggerPhrases": "Main Model Trigger Phrases",
         "typePhraseHere": "Type phrase here",
         "upcastAttention": "Upcast Attention",
         "uploadImage": "Upload Image",

invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/ModelView.tsx

@@ -64,7 +64,7 @@ export const ModelView = () => {
           <DefaultSettings />
         </Box>
       )}
-      {data.type === 'lora' && (
+      {(data.type === 'main' || data.type === 'lora') && (
         <Box layerStyle="second" borderRadius="base" p={4}>
           <TriggerPhrases />
         </Box>

invokeai/frontend/web/src/features/prompt/PromptTriggerSelect.tsx

@@ -1,9 +1,11 @@
 import type { ChakraProps, ComboboxOnChange, ComboboxOption } from '@invoke-ai/ui-library';
 import { Combobox, FormControl } from '@invoke-ai/ui-library';
+import { skipToken } from '@reduxjs/toolkit/query';
 import { createMemoizedSelector } from 'app/store/createMemoizedSelector';
 import { useAppSelector } from 'app/store/storeHooks';
 import type { GroupBase } from 'chakra-react-select';
 import { selectLoraSlice } from 'features/lora/store/loraSlice';
+import { selectGenerationSlice } from 'features/parameters/store/generationSlice';
 import type { PromptTriggerSelectProps } from 'features/prompt/types';
 import { t } from 'i18next';
 import { flatten, map } from 'lodash-es';
@@ -13,18 +15,23 @@ import {
   loraModelsAdapterSelectors,
   textualInversionModelsAdapterSelectors,
   useGetLoRAModelsQuery,
+  useGetModelConfigQuery,
   useGetTextualInversionModelsQuery,
 } from 'services/api/endpoints/models';
 
 const noOptionsMessage = () => t('prompt.noMatchingTriggers');
 
 const selectLoRAs = createMemoizedSelector(selectLoraSlice, (loras) => loras.loras);
+const selectMainModel = createMemoizedSelector(selectGenerationSlice, (generation) => generation.model);
 
 export const PromptTriggerSelect = memo(({ onSelect, onClose }: PromptTriggerSelectProps) => {
   const { t } = useTranslation();
 
-  const currentBaseModel = useAppSelector((s) => s.generation.model?.base);
+  const mainModel = useAppSelector(selectMainModel);
   const addedLoRAs = useAppSelector(selectLoRAs);
+  const { data: mainModelConfig, isLoading: isLoadingMainModelConfig } = useGetModelConfigQuery(
+    mainModel?.key ?? skipToken
+  );
   const { data: loraModels, isLoading: isLoadingLoRAs } = useGetLoRAModelsQuery();
   const { data: tiModels, isLoading: isLoadingTIs } = useGetTextualInversionModelsQuery();
 
@@ -46,7 +53,7 @@ export const PromptTriggerSelect = memo(({ onSelect, onClose }: PromptTriggerSel
     if (tiModels) {
       const embeddingOptions = textualInversionModelsAdapterSelectors
         .selectAll(tiModels)
-        .filter((ti) => ti.base === currentBaseModel)
+        .filter((ti) => ti.base === mainModelConfig?.base)
         .map((model) => ({ label: model.name, value: `<${model.name}>` }));
 
       if (embeddingOptions.length > 0) {
@@ -71,18 +78,33 @@ export const PromptTriggerSelect = memo(({ onSelect, onClose }: PromptTriggerSel
 
       if (triggerPhraseOptions.length > 0) {
         _options.push({
-          label: t('modelManager.triggerPhrases'),
+          label: t('modelManager.loraTriggerPhrases'),
           options: flatten(triggerPhraseOptions),
         });
       }
     }
+
+    if (mainModelConfig && mainModelConfig.trigger_phrases?.length) {
+      _options.push({
+        label: t('modelManager.mainModelTriggerPhrases'),
+        options: mainModelConfig.trigger_phrases.map((triggerPhrase) => ({
+          label: triggerPhrase,
+          value: triggerPhrase,
+        })),
+      });
+    }
+
     return _options;
-  }, [tiModels, loraModels, t, currentBaseModel, addedLoRAs]);
+  }, [tiModels, loraModels, mainModelConfig, t, addedLoRAs]);
 
   return (
     <FormControl>
       <Combobox
-        placeholder={isLoadingLoRAs || isLoadingTIs ? t('common.loading') : t('prompt.addPromptTrigger')}
+        placeholder={
+          isLoadingLoRAs || isLoadingTIs || isLoadingMainModelConfig
+            ? t('common.loading')
+            : t('prompt.addPromptTrigger')
+        }
         defaultMenuIsOpen
         autoFocus
         value={null}