From eb6b3b8168f73dfa59d56ee1b15635586c8a7b00 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 18 Sep 2025 18:28:47 +1000
Subject: [PATCH] feat(ui): add unknown model base support in ui

---
 invokeai/frontend/web/src/features/modelManagerV2/models.ts | 3 +++
 invokeai/frontend/web/src/features/nodes/types/common.ts    | 1 +
 2 files changed, 4 insertions(+)

diff --git a/invokeai/frontend/web/src/features/modelManagerV2/models.ts b/invokeai/frontend/web/src/features/modelManagerV2/models.ts
index 76af8fa8db..2c0acb1d4d 100644
--- a/invokeai/frontend/web/src/features/modelManagerV2/models.ts
+++ b/invokeai/frontend/web/src/features/modelManagerV2/models.ts
@@ -148,6 +148,7 @@ export const MODEL_BASE_TO_COLOR: Record<BaseModelType, string> = {
   'gemini-2.5': 'pink',
   veo3: 'purple',
   runway: 'green',
+  unknown: 'red',
 };
 
 /**
@@ -169,6 +170,7 @@ export const MODEL_BASE_TO_LONG_NAME: Record<BaseModelType, string> = {
   'gemini-2.5': 'Gemini 2.5',
   veo3: 'Veo3',
   runway: 'Runway',
+  unknown: 'Unknown',
 };
 
 /**
@@ -190,6 +192,7 @@ export const MODEL_BASE_TO_SHORT_NAME: Record<BaseModelType, string> = {
   'gemini-2.5': 'Gemini 2.5',
   veo3: 'Veo3',
   runway: 'Runway',
+  unknown: 'Unknown',
 };
 
 /**
diff --git a/invokeai/frontend/web/src/features/nodes/types/common.ts b/invokeai/frontend/web/src/features/nodes/types/common.ts
index c9de796c33..d89ac6c823 100644
--- a/invokeai/frontend/web/src/features/nodes/types/common.ts
+++ b/invokeai/frontend/web/src/features/nodes/types/common.ts
@@ -89,6 +89,7 @@ export const zBaseModelType = z.enum([
   'gemini-2.5',
   'veo3',
   'runway',
+  'unknown',
 ]);
 export type BaseModelType = z.infer<typeof zBaseModelType>;
 export const zMainModelBase = z.enum([