@finos/legend-application-studio 26.1.9 → 26.1.11

package/src/components/editor/editor-group/service-editor/ServiceExecutionQueryEditor.tsx

@@ -31,7 +31,6 @@ import {
   ModalBody,
   ModalFooter,
   ModalHeader,
-  MoreVerticalIcon,
 } from '@finos/legend-art';
 import { assertErrorThrown } from '@finos/legend-shared';
 import { flowResult } from 'mobx';
@@ -287,24 +286,24 @@ export const ServiceExecutionQueryEditor = observer(
             </div>
           </div>
           <div className="panel__header__actions">
-            <div className="service-editor__execution__action-btn">
+            <div className="btn__dropdown-combo btn__dropdown-combo--primary">
               <button
-                className="service-editor__execution__action-btn__label service-editor__execution__action-btn__label--primary"
+                className="btn__dropdown-combo__label"
                 onClick={editWithQueryBuilder()}
                 title="Edit Query"
                 tabIndex={-1}
               >
-                <PencilIcon className="service-editor__execution__action-btn__label__icon" />
-                <div className="service-editor__execution__action-btn__label__title">
+                <PencilIcon className="btn__dropdown-combo__label__icon" />
+                <div className="btn__dropdown-combo__label__title">
                   Edit Query
                 </div>
               </button>
               <DropdownMenu
-                className="service-editor__execution__action-btn__dropdown-btn service-editor__execution__action-btn__dropdown-btn--primary"
+                className="btn__dropdown-combo__dropdown-btn"
                 content={
                   <MenuContent>
                     <MenuContentItem
-                      className="service-editor__execution__action-btn__option"
+                      className="btn__dropdown-combo__option"
                       onClick={editWithQueryBuilder(true)}
                     >
                       Text Mode
@@ -319,96 +318,97 @@ export const ServiceExecutionQueryEditor = observer(
                 <CaretDownIcon />
               </DropdownMenu>
             </div>
-            {executionState.isRunningQuery ? (
-              <button
-                className="service-editor__execution__stop-btn"
-                onClick={cancelQuery}
-                tabIndex={-1}
-              >
-                <div className="btn--dark btn--caution service-editor__execution__stop-btn__label">
-                  <PauseCircleIcon className="service-editor__execution__stop-btn__label__icon" />
-                  <div className="service-editor__execution__stop-btn__label__title">
-                    Stop
-                  </div>
-                </div>
-              </button>
-            ) : (
-              <div className="service-editor__execution__action-btn">
+            <div className="btn__dropdown-combo btn__dropdown-combo--primary">
+              {executionState.isRunningQuery ? (
                 <button
-                  className="service-editor__execution__action-btn__label"
-                  onClick={runQuery}
-                  title="Run Query"
-                  disabled={executionIsRunning}
+                  className="btn__dropdown-combo__canceler"
+                  onClick={cancelQuery}
                   tabIndex={-1}
                 >
-                  <PlayIcon className="service-editor__execution__action-btn__label__icon" />
-                  <div className="service-editor__execution__action-btn__label__title">
-                    Run Query
+                  <div className="btn--dark btn--caution btn__dropdown-combo__canceler__label">
+                    <PauseCircleIcon className="btn__dropdown-combo__canceler__label__icon" />
+                    <div className="btn__dropdown-combo__canceler__label__title">
+                      Stop
+                    </div>
                   </div>
                 </button>
-                <DropdownMenu
-                  className="service-editor__execution__action-btn__dropdown-btn"
-                  disabled={executionIsRunning}
-                  content={
-                    <MenuContent>
-                      <MenuContentItem
-                        className="service-editor__execution__action-btn__option"
-                        onClick={generatePlan}
-                      >
-                        Generate Plan
-                      </MenuContentItem>
-                      <MenuContentItem
-                        className="service-editor__execution__action-btn__option"
-                        onClick={debugPlanGeneration}
-                      >
-                        Debug
-                      </MenuContentItem>
-                    </MenuContent>
-                  }
-                  menuProps={{
-                    anchorOrigin: { vertical: 'bottom', horizontal: 'right' },
-                    transformOrigin: { vertical: 'top', horizontal: 'right' },
-                  }}
-                >
-                  <CaretDownIcon />
-                </DropdownMenu>
-              </div>
-            )}
-            <div>
-              <DropdownMenu
-                className="service-editor__execution__advanced-btn"
-                disabled={executionIsRunning}
-                content={
-                  <MenuContent>
-                    <MenuContentItem
-                      className="service-editor__execution__advanced-btn__option"
-                      onClick={importQuery}
-                    >
-                      Import Query
-                    </MenuContentItem>
-                    <MenuContentItem
-                      className="service-editor__execution__advanced-btn__option"
-                      onClick={openQueryInLegendQuery}
-                      disabled={!applicationStore.config.queryApplicationUrl}
-                    >
-                      Create an Ad-hoc Query
-                    </MenuContentItem>
-                  </MenuContent>
-                }
-                menuProps={{
-                  anchorOrigin: { vertical: 'bottom', horizontal: 'right' },
-                  transformOrigin: { vertical: 'top', horizontal: 'right' },
-                }}
-              >
-                <div className="service-editor__execution__advanced-btn__label">
-                  <MoreVerticalIcon className="service-editor__execution__advanced-btn__label__icon" />
+              ) : (
+                <>
+                  <button
+                    className="btn__dropdown-combo__label"
+                    onClick={runQuery}
+                    title="Run Query"
+                    disabled={executionIsRunning}
+                    tabIndex={-1}
+                  >
+                    <PlayIcon className="btn__dropdown-combo__label__icon" />
+                    <div className="btn__dropdown-combo__label__title">
+                      Run Query
+                    </div>
+                  </button>
+                  <DropdownMenu
+                    className="btn__dropdown-combo__dropdown-btn"
+                    disabled={executionIsRunning}
+                    content={
+                      <MenuContent>
+                        <MenuContentItem
+                          className="btn__dropdown-combo__option"
+                          onClick={generatePlan}
+                        >
+                          Generate Plan
+                        </MenuContentItem>
+                        <MenuContentItem
+                          className="btn__dropdown-combo__option"
+                          onClick={debugPlanGeneration}
+                        >
+                          Debug
+                        </MenuContentItem>
+                      </MenuContent>
+                    }
+                    menuProps={{
+                      anchorOrigin: { vertical: 'bottom', horizontal: 'right' },
+                      transformOrigin: { vertical: 'top', horizontal: 'right' },
+                    }}
+                  >
+                    <CaretDownIcon />
+                  </DropdownMenu>
+                </>
+              )}
+            </div>
+            <DropdownMenu
+              className="btn__dropdown-combo"
+              disabled={executionIsRunning}
+              content={
+                <MenuContent>
+                  <MenuContentItem
+                    className="btn__dropdown-combo__option"
+                    onClick={importQuery}
+                  >
+                    Import Query
+                  </MenuContentItem>
+                  <MenuContentItem
+                    className="btn__dropdown-combo__option"
+                    onClick={openQueryInLegendQuery}
+                    disabled={!applicationStore.config.queryApplicationUrl}
+                  >
+                    Create an Ad-hoc Query
+                  </MenuContentItem>
+                </MenuContent>
+              }
+              menuProps={{
+                anchorOrigin: { vertical: 'bottom', horizontal: 'right' },
+                transformOrigin: { vertical: 'top', horizontal: 'right' },
+              }}
+            >
+              <div className="btn__dropdown-combo__label">
+                <div className="btn__dropdown-combo__label__title">
                   Advanced
                 </div>
-                <div className="service-editor__execution__advanced-btn__icon">
-                  <CaretDownIcon />
-                </div>
-              </DropdownMenu>
-            </div>
+              </div>
+              <div className="btn__dropdown-combo__dropdown-btn">
+                <CaretDownIcon />
+              </div>
+            </DropdownMenu>
           </div>
         </div>
         <div className="panel__content property-mapping-editor__entry__container">
@@ -417,15 +417,7 @@ export const ServiceExecutionQueryEditor = observer(
               executionState.isOpeningQueryEditor || executionIsRunning
             }
           />
-          <div
-            className="service-execution-query-editor__content"
-            title="Double click to edit in query builder"
-            onDoubleClick={(event) => {
-              event.preventDefault();
-              event.stopPropagation();
-              editWithQueryBuilder()();
-            }}
-          >
+          <div className="service-execution-query-editor__content">
             <CodeEditor
               inputValue={queryState.lambdaString}
               isReadOnly={true}

package/src/components/editor/panel-group/DevToolPanel.tsx

@@ -52,6 +52,11 @@ export const DevToolPanel = observer(() => {
       editorStore.graphState.enableStrictMode,
     );
   };
+  const toggleArtifactGeneration = (): void => {
+    editorStore.graphState.graphGenerationState.setEnableArtifactGeneration(
+      !editorStore.graphState.graphGenerationState.enableArtifactGeneration,
+    );
+  };
 
   return (
     <Panel>
@@ -116,6 +121,15 @@ export const DevToolPanel = observer(() => {
           isReadOnly={false}
           update={toggleStrictMode}
         />
+        <PanelFormBooleanField
+          name="Generate Artifact Generations"
+          prompt="Include generation of artifacet extensions during generation action (F10)"
+          value={
+            editorStore.graphState.graphGenerationState.enableArtifactGeneration
+          }
+          isReadOnly={false}
+          update={toggleArtifactGeneration}
+        />
       </PanelForm>
     </Panel>
   );

package/src/components/editor/panel-group/SQLPlaygroundPanel.tsx

@@ -248,6 +248,9 @@ const buildRelationalDatabaseConnectionOption = (
         <div className="sql-playground__config__connection-selector__option__type">
           {connectionValue.type}
         </div>
+        <div className="sql-playground__config__connection-selector__option__path">
+          {connection.path}
+        </div>
       </div>
     ),
   };

package/src/components/editor/side-bar/CreateNewElementModal.tsx

@@ -38,7 +38,6 @@ import {
   type Class,
   ELEMENT_PATH_DELIMITER,
 } from '@finos/legend-graph';
-import type { FileGenerationTypeOption } from '../../../stores/editor/editor-state/GraphGenerationState.js';
 import { flowResult } from 'mobx';
 import { useApplicationStore } from '@finos/legend-application';
 import {
@@ -51,6 +50,7 @@ import type { DSL_Data_LegendStudioApplicationPlugin_Extension } from '../../../
 import { PACKAGEABLE_ELEMENT_TYPE } from '../../../stores/editor/utils/ModelClassifierUtils.js';
 import { EmbeddedDataType } from '../../../stores/editor/editor-state/ExternalFormatState.js';
 import type { DSL_Mapping_LegendStudioApplicationPlugin_Extension } from '../../../stores/extensions/DSL_Mapping_LegendStudioApplicationPlugin_Extension.js';
+import type { GenerationTypeOption } from '../../../stores/editor/editor-state/GraphGenerationState.js';
 
 export const getElementTypeLabel = (
   editorStore: EditorStore,
@@ -379,11 +379,9 @@ const NewFileGenerationDriverEditor = observer(() => {
     NewFileGenerationDriver,
   );
   const options =
-    editorStore.graphState.graphGenerationState
+    editorStore.graphState.graphGenerationState.globalFileGenerationState
       .fileGenerationConfigurationOptions;
-  const onTypeSelectionChange = (
-    val: FileGenerationTypeOption | null,
-  ): void => {
+  const onTypeSelectionChange = (val: GenerationTypeOption | null): void => {
     if (!val) {
       newConnectionDriver.setTypeOption(undefined);
     } else {

package/src/components/editor/side-bar/Explorer.tsx

@@ -14,7 +14,13 @@
  * limitations under the License.
  */
 
-import { Fragment, useRef, useEffect, useState, forwardRef } from 'react';
+import React, {
+  Fragment,
+  useRef,
+  useEffect,
+  useState,
+  forwardRef,
+} from 'react';
 import { observer } from 'mobx-react-lite';
 import {
   type TreeNodeContainerProps,
@@ -61,6 +67,7 @@ import { LEGEND_STUDIO_TEST_ID } from '../../../__lib__/LegendStudioTesting.js';
 import {
   ACTIVITY_MODE,
   GRAPH_EDITOR_MODE,
+  PANEL_MODE,
 } from '../../../stores/editor/EditorConfig.js';
 import { getTreeChildNodes } from '../../../stores/editor/utils/PackageTreeUtils.js';
 import type { PackageTreeNodeData } from '../../../stores/editor/utils/TreeUtils.js';
@@ -76,6 +83,7 @@ import {
 import {
   guaranteeNonEmptyString,
   guaranteeNonNullable,
+  guaranteeType,
   isNonNullable,
   toTitleCase,
 } from '@finos/legend-shared';
@@ -102,6 +110,7 @@ import {
   RelationalDatabaseConnection,
   guaranteeRelationalDatabaseConnection,
   extractDependencyGACoordinateFromRootPackageName,
+  type FunctionActivatorConfiguration,
 } from '@finos/legend-graph';
 import { useApplicationStore } from '@finos/legend-application';
 import {
@@ -125,6 +134,7 @@ import {
   CodeEditor,
 } from '@finos/legend-lego/code-editor';
 import { DatabaseBuilder } from '../editor-group/connection-editor/DatabaseBuilder.js';
+import { FunctionEditorState } from '../../../stores/editor/editor-state/element-editor-state/FunctionEditorState.js';
 
 const ElementRenamer = observer(() => {
   const editorStore = useEditorStore();
@@ -513,6 +523,15 @@ const ExplorerContextMenu = observer(
         }
       },
     );
+    const openSQLPlayground = (): void => {
+      if (isRelationalDatabaseConnection(node?.packageableElement)) {
+        editorStore.panelGroupDisplayState.open();
+        editorStore.setActivePanelMode(PANEL_MODE.SQL_PLAYGROUND);
+        editorStore.sqlPlaygroundState.setConnection(
+          guaranteeType(node?.packageableElement, PackageableConnection),
+        );
+      }
+    };
     const removeElement = (): void => {
       if (node) {
         flowResult(
@@ -663,6 +682,50 @@ const ExplorerContextMenu = observer(
         ).catch(applicationStore.alertUnhandledError);
       }
     };
+    const activateFunction = (): void => {
+      if (node?.packageableElement instanceof ConcreteFunctionDefinition) {
+        editorStore.setQuickInputState({
+          title: 'Activate function',
+          placeholder: 'Select an activation...',
+          options: editorStore.graphState.functionActivatorConfigurations.map(
+            (config) => ({
+              value: config,
+              label: (
+                <div
+                  className="function-editor__activator__selector__option"
+                  title={config.description}
+                >
+                  <div className="function-editor__activator__selector__option__name">
+                    {config.name}
+                  </div>
+                  <div className="function-editor__activator__selector__option__description">
+                    {config.description}
+                  </div>
+                </div>
+              ),
+            }),
+          ),
+          getSearchValue: (option: {
+            value: FunctionActivatorConfiguration;
+            label: React.ReactNode;
+          }): string => option.value.name,
+          onSelect: (option: {
+            value: FunctionActivatorConfiguration;
+            label: React.ReactNode;
+          }) => {
+            editorStore.graphEditorMode.openElement(node.packageableElement);
+            editorStore.tabManagerState
+              .getCurrentEditorState(FunctionEditorState)
+              .activatorBuilderState.setCurrentActivatorConfiguration(
+                option.value,
+              );
+          },
+          customization: {
+            rowHeight: 70,
+          },
+        });
+      }
+    };
 
     if (isDependencyProjectRoot()) {
       return (
@@ -722,8 +785,27 @@ const ExplorerContextMenu = observer(
             <MenuContentDivider />
           </>
         )}
+        {node.packageableElement instanceof ConcreteFunctionDefinition && (
+          <>
+            {editorStore.applicationStore.config.options
+              .TEMPORARY__enableFunctionActivatorSupport && (
+              <>
+                <MenuContentItem onClick={activateFunction}>
+                  Activate...
+                </MenuContentItem>
+                <MenuContentDivider />
+              </>
+            )}
+          </>
+        )}
         {isRelationalDatabaseConnection(node.packageableElement) && (
           <>
+            {editorStore.applicationStore.config.options
+              .TEMPORARY__enableRawSQLExecutor && (
+              <MenuContentItem onClick={openSQLPlayground}>
+                Execute SQL...
+              </MenuContentItem>
+            )}
             <MenuContentItem onClick={buildDatabase}>
               Build Database...
             </MenuContentItem>
@@ -977,7 +1059,7 @@ const ExplorerTrees = observer(() => {
 
   // Generated Files Tree
   const generationFileTreeData =
-    editorStore.explorerTreeState.getFileGenerationTreeData();
+    editorStore.explorerTreeState.getArtifactsGenerationTreeData();
   const onGenerationFileTreeNodeSelect = (node: FileSystemTreeNodeData): void =>
     editorStore.graphState.graphGenerationState.onTreeNodeSelect(
       node,

package/src/stores/editor/EditorGraphState.ts

@@ -98,6 +98,7 @@ import { LEGEND_STUDIO_APP_EVENT } from '../../__lib__/LegendStudioEvent.js';
 import { LEGEND_STUDIO_SETTING_KEY } from '../../__lib__/LegendStudioSetting.js';
 import type { TabState } from '@finos/legend-lego/application';
 import { LegendStudioTelemetryHelper } from '../../__lib__/LegendStudioTelemetryHelper.js';
+import { ArtifactGenerationViewerState } from './editor-state/ArtifactGenerationViewerState.js';
 
 export enum GraphBuilderStatus {
   SUCCEEDED = 'SUCCEEDED',
@@ -583,15 +584,15 @@ export class EditorGraphState {
           openedTabEditorPaths.push(state.elementPath);
         }
       });
+      const currentTab = this.editorStore.tabManagerState.currentTab;
       const currentTabState =
-        this.editorStore.tabManagerState.currentTab instanceof
-        ElementEditorState
+        currentTab instanceof ElementEditorState ||
+        currentTab instanceof ArtifactGenerationViewerState
           ? undefined
-          : this.editorStore.tabManagerState.currentTab;
+          : currentTab;
       const currentTabElementPath =
-        this.editorStore.tabManagerState.currentTab instanceof
-        ElementEditorState
-          ? this.editorStore.tabManagerState.currentTab.elementPath
+        currentTab instanceof ElementEditorState
+          ? currentTab.elementPath
           : undefined;
       this.editorStore.tabManagerState.closeAllTabs();
 

package/src/stores/editor/EditorStore.ts

@@ -671,6 +671,9 @@ export class EditorStore implements CommandRegistrar {
         },
         {
           tracerService: this.applicationStore.tracerService,
+          TEMPORARY__enableNewServiceRegistrationInputCollectorMechanism:
+            this.applicationStore.config.options
+              .TEMPORARY__enableNewServiceRegistrationInputCollectorMechanism,
         },
       ),
     ]);
@@ -716,7 +719,7 @@ export class EditorStore implements CommandRegistrar {
       this.workspaceReviewState.fetchCurrentWorkspaceReview(),
       this.workspaceUpdaterState.fetchLatestCommittedReviews(),
       this.projectConfigurationEditorState.fetchLatestProjectStructureVersion(),
-      this.graphState.graphGenerationState.fetchAvailableFileGenerationDescriptions(),
+      this.graphState.graphGenerationState.globalFileGenerationState.fetchAvailableFileGenerationDescriptions(),
       this.graphState.graphGenerationState.externalFormatState.fetchExternalFormatDescriptions(),
       this.graphState.fetchAvailableFunctionActivatorConfigurations(),
       this.sdlcState.fetchProjectVersions(),
@@ -757,7 +760,7 @@ export class EditorStore implements CommandRegistrar {
       this.conflictResolutionState.initialize(),
       this.sdlcState.checkIfWorkspaceIsOutdated(),
       this.projectConfigurationEditorState.fetchLatestProjectStructureVersion(),
-      this.graphState.graphGenerationState.fetchAvailableFileGenerationDescriptions(),
+      this.graphState.graphGenerationState.globalFileGenerationState.fetchAvailableFileGenerationDescriptions(),
       this.graphState.graphGenerationState.externalFormatState.fetchExternalFormatDescriptions(),
       this.graphState.fetchAvailableFunctionActivatorConfigurations(),
       this.sdlcState.fetchProjectVersions(),

package/src/stores/editor/ExplorerTreeState.ts

@@ -71,7 +71,7 @@ export class ExplorerTreeState {
   legalTreeData?: TreeData<PackageTreeNodeData> | undefined;
   dependencyTreeData?: TreeData<PackageTreeNodeData> | undefined;
   selectedNode?: PackageTreeNodeData | undefined;
-  fileGenerationTreeData?: TreeData<FileSystemTreeNodeData> | undefined;
+  artifactsGenerationTreeData?: TreeData<FileSystemTreeNodeData> | undefined;
 
   elementToRename?: PackageableElement | undefined;
   classToGenerateSampleData?: Class | undefined;
@@ -84,7 +84,7 @@ export class ExplorerTreeState {
       systemTreeData: observable.ref,
       legalTreeData: observable.ref,
       dependencyTreeData: observable.ref,
-      fileGenerationTreeData: observable.ref,
+      artifactsGenerationTreeData: observable.ref,
       selectedNode: observable.ref,
       elementToRename: observable,
       classToGenerateSampleData: observable,
@@ -94,7 +94,7 @@ export class ExplorerTreeState {
       setSystemTreeData: action,
       setLegalTreeData: action,
       setDependencyTreeData: action,
-      setFileGenerationTreeData: action,
+      setArtifactsGenerationTreeData: action,
       setSelectedNode: action,
       setElementToRename: action,
       setClassToGenerateSampleData: action,
@@ -166,8 +166,8 @@ export class ExplorerTreeState {
   setDependencyTreeData(data: TreeData<PackageTreeNodeData>): void {
     this.dependencyTreeData = data;
   }
-  setFileGenerationTreeData(data: TreeData<FileSystemTreeNodeData>): void {
-    this.fileGenerationTreeData = data;
+  setArtifactsGenerationTreeData(data: TreeData<FileSystemTreeNodeData>): void {
+    this.artifactsGenerationTreeData = data;
   }
   setElementToRename(val: PackageableElement | undefined): void {
     this.elementToRename = val;
@@ -214,7 +214,7 @@ export class ExplorerTreeState {
       this.editorStore.graphManagerState.graph.generationModel.root,
       ExplorerTreeRootPackageLabel.MODEL_GENERATION,
     );
-    this.fileGenerationTreeData = getFileSystemTreeData(
+    this.artifactsGenerationTreeData = getFileSystemTreeData(
       this.editorStore.graphState.graphGenerationState.rootFileDirectory,
       ExplorerTreeRootPackageLabel.FILE_GENERATION,
     );
@@ -248,7 +248,7 @@ export class ExplorerTreeState {
       this.editorStore.graphManagerState.graph.generationModel.root,
       ExplorerTreeRootPackageLabel.MODEL_GENERATION,
     );
-    this.fileGenerationTreeData = getFileSystemTreeData(
+    this.artifactsGenerationTreeData = getFileSystemTreeData(
       this.editorStore.graphState.graphGenerationState.rootFileDirectory,
       ExplorerTreeRootPackageLabel.FILE_GENERATION,
     );
@@ -438,7 +438,7 @@ export class ExplorerTreeState {
     }
     // File generation tree
     // TODO: fix this so it does proper reprocessing, right now it just rebuilds
-    this.fileGenerationTreeData = getFileSystemTreeData(
+    this.artifactsGenerationTreeData = getFileSystemTreeData(
       this.editorStore.graphState.graphGenerationState.rootFileDirectory,
       ExplorerTreeRootPackageLabel.FILE_GENERATION,
     );
@@ -558,9 +558,9 @@ export class ExplorerTreeState {
     }
   }
 
-  getFileGenerationTreeData(): TreeData<FileSystemTreeNodeData> {
+  getArtifactsGenerationTreeData(): TreeData<FileSystemTreeNodeData> {
     return guaranteeNonNullable(
-      this.fileGenerationTreeData,
+      this.artifactsGenerationTreeData,
       'File generation tree data has not been initialized',
     );
   }

package/src/stores/editor/NewElementState.ts

@@ -34,7 +34,7 @@ import {
 import { decorateRuntimeWithNewMapping } from './editor-state/element-editor-state/RuntimeEditorState.js';
 import type { DSL_LegendStudioApplicationPlugin_Extension } from '../LegendStudioApplicationPlugin.js';
 import {
-  type FileGenerationTypeOption,
+  type GenerationTypeOption,
   DEFAULT_GENERATION_SPECIFICATION_NAME,
 } from './editor-state/GraphGenerationState.js';
 import {
@@ -523,7 +523,7 @@ export class NewServiceDriver extends NewElementDriver<Service> {
 }
 
 export class NewFileGenerationDriver extends NewElementDriver<FileGenerationSpecification> {
-  typeOption?: FileGenerationTypeOption | undefined;
+  typeOption?: GenerationTypeOption | undefined;
 
   constructor(editorStore: EditorStore) {
     super(editorStore);
@@ -534,13 +534,13 @@ export class NewFileGenerationDriver extends NewElementDriver<FileGenerationSpec
     });
 
     this.typeOption = editorStore.graphState.graphGenerationState
-      .fileGenerationConfigurationOptions.length
-      ? editorStore.graphState.graphGenerationState
+      .globalFileGenerationState.fileGenerationConfigurationOptions.length
+      ? editorStore.graphState.graphGenerationState.globalFileGenerationState
           .fileGenerationConfigurationOptions[0]
       : undefined;
   }
 
-  setTypeOption(typeOption: FileGenerationTypeOption | undefined): void {
+  setTypeOption(typeOption: GenerationTypeOption | undefined): void {
     this.typeOption = typeOption;
   }
 

package/src/stores/editor/QuickInputState.ts

@@ -21,4 +21,7 @@ export type QuickInputState<T> = {
   options: QuickInputOption<T>[];
   getSearchValue: (option: QuickInputOption<T>) => string;
   onSelect: (option: QuickInputOption<T>) => void;
+  customization?: {
+    rowHeight?: number | undefined;
+  };
 };