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 = ({
              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}>