From 5b65e08fdf2ece890a6c2f86b81ec8a08ef3b458 Mon Sep 17 00:00:00 2001
From: Robert <robertbrignull@github.com>
Date: Tue, 11 Jul 2023 17:22:25 +0100
Subject: [PATCH] Don't show model from source button in framework mode

---
 .../src/view/data-extensions-editor/LibraryRow.tsx     | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/extensions/ql-vscode/src/view/data-extensions-editor/LibraryRow.tsx b/extensions/ql-vscode/src/view/data-extensions-editor/LibraryRow.tsx
index 89e50a30c..48d8deb65 100644
--- a/extensions/ql-vscode/src/view/data-extensions-editor/LibraryRow.tsx
+++ b/extensions/ql-vscode/src/view/data-extensions-editor/LibraryRow.tsx
@@ -170,10 +170,12 @@ export const LibraryRow = ({
             &nbsp;Model with AI
           </VSCodeButton>
         )}
-        <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={handleModelFromSource}>
+            <Codicon name="code" label="Model from source" />
+            &nbsp;Model from source
+          </VSCodeButton>
+        )}
         {viewState.enableFrameworkMode &&
           viewState.mode === Mode.Application && (
             <VSCodeButton appearance="icon" onClick={handleModelDependency}>