fix(image-generator): preserve saved catalog settings

Author: waleedlatif1

apps/sim/blocks/blocks/image_generator.ts

@@ -314,12 +314,13 @@ export const ImageGeneratorV2Block: BlockConfig = {
         id: 'model',
         title: 'Model',
         type: 'dropdown',
-        options: imageModels
-          .filter((model) => model.provider === provider)
-          .map((model) => ({
-            label: model.label,
-            id: model.id,
-          })),
+        options: (params) =>
+          imageModels
+            .filter((model) => !params || model.provider === provider)
+            .flatMap((model) => [
+              { label: model.label, id: model.id },
+              ...(model.aliases ?? []).map((id) => ({ id, label: model.label, hidden: true })),
+            ]),
         value: () => imageModelDefaults[provider],
         condition: { field: 'provider', value: provider },
         dependsOn: ['provider'],
@@ -358,9 +359,10 @@ export const ImageGeneratorV2Block: BlockConfig = {
         }),
         options: (params) => [
           { id: 'default', label: 'Model default' },
-          ...(
-            getImageModel(params?.values.provider, params?.values.model)?.options?.[key]?.values ??
-            []
+          ...(params
+            ? (getImageModel(params.values.provider, params.values.model)?.options?.[key]?.values ??
+              [])
+            : [...new Set(imageModels.flatMap((model) => model.options?.[key]?.values ?? []))]
           ).map((value) => ({
             id: String(value),
             label: String(value),
@@ -372,8 +374,8 @@ export const ImageGeneratorV2Block: BlockConfig = {
       (key): SubBlockConfig => ({
         id: key,
         title: key === 'enableSafetyChecker' ? 'Safety Checker' : 'Web Search',
-        type: 'dropdown',
-        value: () => 'default',
+        type: 'switch',
+        defaultValue: key === 'enableSafetyChecker',
         mode: 'advanced',
         dependsOn: ['provider', 'model'],
         condition: (values) => ({
@@ -382,11 +384,6 @@ export const ImageGeneratorV2Block: BlockConfig = {
             ? [String(values?.model)]
             : [],
         }),
-        options: [
-          { id: 'default', label: 'Model default' },
-          { id: 'true', label: 'On' },
-          { id: 'false', label: 'Off' },
-        ],
       })
     ),
     {

apps/sim/lib/media/image-catalog.ts

@@ -17,6 +17,7 @@ interface ImageModel {
   provider: 'openai' | 'gemini' | 'falai'
   id: string
   label: string
+  aliases?: readonly string[]
   endpoint?: string
   output?: string
   options?: Partial<Record<ImageOption, { field: string; values: readonly (string | number)[] }>>
@@ -90,6 +91,9 @@ export const imageModels: readonly ImageModel[] = [
       provider: 'gemini',
       id,
       label,
+      aliases: ['gemini-3-pro-image', 'gemini-3.1-flash-image'].includes(id)
+        ? [`${id}-preview`]
+        : [],
       options: {
         aspectRatio: {
           field: 'aspectRatio',
@@ -113,21 +117,17 @@ export const imageModels: readonly ImageModel[] = [
       provider: 'falai',
       endpoint,
       id: 'legacyId' in spec ? spec.legacyId : endpoint,
+      aliases: 'legacyId' in spec ? [endpoint] : [],
     })
   ),
 ]
 
 const byId = new Map(imageModels.map((model) => [`${model.provider}:${model.id}`, model]))
 for (const model of imageModels) {
-  if (model.endpoint) byId.set(`falai:${model.endpoint}`, model)
-}
-const geminiAliases: Record<string, string> = {
-  'gemini-3.1-flash-image-preview': 'gemini-3.1-flash-image',
-  'gemini-3-pro-image-preview': 'gemini-3-pro-image',
+  for (const alias of model.aliases ?? []) byId.set(`${model.provider}:${alias}`, model)
 }
 
 export function getImageModel(provider: unknown, id: unknown) {
   if (typeof provider !== 'string' || typeof id !== 'string') return undefined
-  const modelId = provider === 'gemini' ? (geminiAliases[id] ?? id) : id
-  return byId.get(`${provider}:${modelId}`)
+  return byId.get(`${provider}:${id}`)
 }

apps/sim/lib/workflows/application/replace-workflow-state.test.ts

@@ -54,6 +54,7 @@ import { replaceWorkflowState } from '@/lib/workflows/application/replace-workfl
 import { validateInputsForBlock } from '@/lib/workflows/editing/validation'
 import { AgentBlock } from '@/blocks/blocks/agent'
 import { ExaBlock } from '@/blocks/blocks/exa'
+import { ImageGeneratorV2Block } from '@/blocks/blocks/image_generator'
 import { getBlock } from '@/blocks/registry'
 
 const mockNeedsRedeployment = workflowDeploymentStatusMockFns.mockCheckNeedsRedeployment
@@ -93,7 +94,13 @@ const input = { workflowId: 'workflow-1', blocks: { 'block-1': BLOCK }, edges: [
 describe('replaceWorkflowState', () => {
   beforeEach(() => {
     vi.mocked(getBlock).mockImplementation((type) =>
-      type === 'exa' ? ExaBlock : type === 'agent' ? AgentBlock : defaultGetBlock?.(type)
+      type === 'image_generator_v2'
+        ? ImageGeneratorV2Block
+        : type === 'exa'
+          ? ExaBlock
+          : type === 'agent'
+            ? AgentBlock
+            : defaultGetBlock?.(type)
     )
     mockResolveContext.mockResolvedValue(context)
     mockResolvePermission.mockResolvedValue('write')
@@ -113,6 +120,38 @@ describe('replaceWorkflowState', () => {
     mockLoadNormalized.mockResolvedValue({ blocks: {}, edges: [], loops: {}, parallels: {} })
   })
 
+  it.each([
+    { provider: 'openai', model: 'gpt-image-2', size: '1024x1024', quality: 'high' },
+    {
+      provider: 'gemini',
+      model: 'gemini-3-pro-image-preview',
+      resolution: '2K',
+      aspectRatio: '16:9',
+    },
+    {
+      provider: 'falai',
+      model: 'nano-banana-2',
+      resolution: '1K',
+      aspectRatio: '1:1',
+      enableSafetyChecker: true,
+      enableWebSearch: false,
+    },
+  ])('saves configured $provider image workflows through the state API', async (settings) => {
+    const block = {
+      ...BLOCK,
+      type: 'image_generator_v2',
+      subBlocks: Object.fromEntries(
+        Object.entries(settings).map(([id, value]) => [id, { id, type: 'dropdown', value }])
+      ),
+    }
+    await expect(
+      replaceWorkflowState.execute({
+        principal: sessionPrincipal,
+        input: { ...input, blocks: { 'block-1': block } },
+      })
+    ).resolves.toMatchObject({ workflowId: 'workflow-1' })
+  })
+
   /**
    * Two things this pins that a same-shape input and output cannot: the write
    * carries the **sanitized** graph, not the caller's body, and the reported

apps/sim/scripts/validate-fal-image-models.ts

@@ -110,7 +110,11 @@ for (const model of imageModels.filter((item) => item.provider === 'falai')) {
       const wireKey = option?.field ?? model.booleans?.[field.id]
       if (!wireKey) continue
       const options =
-        typeof field.options === 'function' ? field.options({ values: base }) : field.options
+        field.type === 'switch'
+          ? [{ id: 'true' }, { id: 'false' }]
+          : typeof field.options === 'function'
+            ? field.options({ values: base })
+            : field.options
       assert(options?.length)
       for (const value of options) check({ [field.id]: value.id }, wireKey, value.id)
     }