feat(ui): add basic metadata to imagen3/chatgpt-4o graphs

Author: psychedelicious

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildChatGPT4oGraph.ts

@@ -39,7 +39,7 @@ export const buildChatGPT4oGraph = async (state: RootState, manager: CanvasManag
   const { positivePrompt } = selectPresetModifiedPrompts(state);
 
   assert(model, 'No model found in state');
-  assert(model.base === 'chatgpt-4o', 'Model is not a FLUX model');
+  assert(model.base === 'chatgpt-4o', 'Model is not a ChatGPT 4o model');
 
   assert(isChatGPT4oAspectRatioID(bbox.aspectRatio.id), 'ChatGPT 4o does not support this aspect ratio');
 
@@ -77,6 +77,12 @@ export const buildChatGPT4oGraph = async (state: RootState, manager: CanvasManag
       is_intermediate,
       board,
     });
+    g.upsertMetadata({
+      positive_prompt: positivePrompt,
+      model: Graph.getModelMetadataField(model),
+      width: bbox.rect.width,
+      height: bbox.rect.height,
+    });
     return {
       g,
       positivePromptFieldIdentifier: { nodeId: gptImage.id, fieldName: 'positive_prompt' },
@@ -102,6 +108,12 @@ export const buildChatGPT4oGraph = async (state: RootState, manager: CanvasManag
       is_intermediate,
       board,
     });
+    g.upsertMetadata({
+      positive_prompt: positivePrompt,
+      model: Graph.getModelMetadataField(model),
+      width: bbox.rect.width,
+      height: bbox.rect.height,
+    });
     return {
       g,
       positivePromptFieldIdentifier: { nodeId: gptImage.id, fieldName: 'positive_prompt' },

invokeai/frontend/web/src/features/nodes/util/graph/generation/buildImagen3Graph.ts

@@ -13,6 +13,7 @@ import {
 } from 'features/nodes/util/graph/graphBuilderUtils';
 import { type GraphBuilderReturn, UnsupportedGenerationModeError } from 'features/nodes/util/graph/types';
 import { t } from 'i18next';
+import { selectMainModelConfig } from 'services/api/endpoints/models';
 import type { Equals } from 'tsafe';
 import { assert } from 'tsafe';
 
@@ -32,7 +33,9 @@ export const buildImagen3Graph = async (state: RootState, manager: CanvasManager
 
   const { bbox } = canvas;
   const { positivePrompt, negativePrompt } = selectPresetModifiedPrompts(state);
+  const model = selectMainModelConfig(state);
 
+  assert(model, 'No model found for Imagen3 graph');
   assert(isImagen3AspectRatioID(bbox.aspectRatio.id), 'Imagen3 does not support this aspect ratio');
   assert(positivePrompt.length > 0, 'Imagen3 requires positive prompt to have at least one character');
 
@@ -54,6 +57,13 @@ export const buildImagen3Graph = async (state: RootState, manager: CanvasManager
       is_intermediate,
       board,
     });
+    g.upsertMetadata({
+      positive_prompt: positivePrompt,
+      negative_prompt: negativePrompt,
+      width: bbox.rect.width,
+      height: bbox.rect.height,
+      model: Graph.getModelMetadataField(model),
+    });
     return {
       g,
       seedFieldIdentifier: { nodeId: imagen3.id, fieldName: 'seed' },