Merge pull request #2793 from github/koesie10/feature-flag-flow-generation

Add feature flag for flow generation

extensions/ql-vscode/src/config.ts

@@ -706,9 +706,14 @@ export function showQueriesPanel(): boolean {
 }
 
 const MODEL_SETTING = new Setting("model", ROOT_SETTING);
+const FLOW_GENERATION = new Setting("flowGeneration", MODEL_SETTING);
 const LLM_GENERATION = new Setting("llmGeneration", MODEL_SETTING);
 const EXTENSIONS_DIRECTORY = new Setting("extensionsDirectory", MODEL_SETTING);
 
+export function showFlowGeneration(): boolean {
+  return !!FLOW_GENERATION.getValue<boolean>();
+}
+
 export function showLlmGeneration(): boolean {
   return !!LLM_GENERATION.getValue<boolean>();
 }

extensions/ql-vscode/src/model-editor/model-editor-view.ts

@@ -34,7 +34,7 @@ import {
 import { Method, Usage } from "./method";
 import { ModeledMethod } from "./modeled-method";
 import { ExtensionPack } from "./shared/extension-pack";
-import { showLlmGeneration } from "../config";
+import { showFlowGeneration, showLlmGeneration } from "../config";
 import { Mode } from "./shared/mode";
 import { loadModeledMethods, saveModeledMethods } from "./modeled-method-fs";
 import { join } from "path";
@@ -322,6 +322,7 @@ export class ModelEditorView extends AbstractWebview<
       t: "setModelEditorViewState",
       viewState: {
         extensionPack: this.extensionPack,
+        showFlowGeneration: showFlowGeneration(),
         showLlmButton,
         mode: this.mode,
       },

extensions/ql-vscode/src/model-editor/shared/view-state.ts

@@ -3,6 +3,7 @@ import { Mode } from "./mode";
 
 export interface ModelEditorViewState {
   extensionPack: ExtensionPack;
+  showFlowGeneration: boolean;
   showLlmButton: boolean;
   mode: Mode;
 }

extensions/ql-vscode/src/stories/model-editor/ModelEditor.stories.tsx

@@ -28,6 +28,7 @@ ModelEditor.args = {
       extensionTargets: {},
       dataExtensions: [],
     },
+    showFlowGeneration: true,
     showLlmButton: true,
     mode: Mode.Application,
   },

extensions/ql-vscode/src/view/model-editor/LibraryRow.tsx

@@ -213,12 +213,13 @@ export const LibraryRow = ({
              Stop
           </VSCodeButton>
         )}
-        {viewState.mode === Mode.Application && (
-          <VSCodeButton appearance="icon" onClick={handleModelFromSource}>
-            <Codicon name="code" label="Model from source" />
-            &nbsp;Model from source
-          </VSCodeButton>
-        )}
+        {viewState.showFlowGeneration &&
+          viewState.mode === Mode.Application && (
+            <VSCodeButton appearance="icon" onClick={handleModelFromSource}>
+              <Codicon name="code" label="Model from source" />
+              &nbsp;Model from source
+            </VSCodeButton>
+          )}
         {viewState.mode === Mode.Application && (
           <VSCodeButton appearance="icon" onClick={handleModelDependency}>
             <Codicon name="references" label="Model dependency" />

extensions/ql-vscode/src/view/model-editor/ModelEditor.tsx

@@ -345,11 +345,12 @@ export function ModelEditor({
           <VSCodeButton appearance="secondary" onClick={onRefreshClick}>
             Refresh
           </VSCodeButton>
-          {viewState.mode === Mode.Framework && (
-            <VSCodeButton onClick={onGenerateFromSourceClick}>
-              Generate
-            </VSCodeButton>
-          )}
+          {viewState.showFlowGeneration &&
+            viewState.mode === Mode.Framework && (
+              <VSCodeButton onClick={onGenerateFromSourceClick}>
+                Generate
+              </VSCodeButton>
+            )}
         </ButtonsContainer>
         <ModeledMethodsList
           methods={methods}