From 29aaab132cbb7eee17a81f5b8c87dbbff6ced61f Mon Sep 17 00:00:00 2001
From: Charis Kyriakou <charisk@users.noreply.github.com>
Date: Mon, 31 Jul 2023 07:38:51 +0000
Subject: [PATCH] Only show automodel button for Java

---
 .../data-extensions-editor/data-extensions-editor-view.ts    | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/extensions/ql-vscode/src/data-extensions-editor/data-extensions-editor-view.ts b/extensions/ql-vscode/src/data-extensions-editor/data-extensions-editor-view.ts
index e78796898a6..888893c94ca 100644
--- a/extensions/ql-vscode/src/data-extensions-editor/data-extensions-editor-view.ts
+++ b/extensions/ql-vscode/src/data-extensions-editor/data-extensions-editor-view.ts
@@ -190,12 +190,15 @@ export class DataExtensionsEditorView extends AbstractWebview<
   }
 
   private async setViewState(): Promise<void> {
+    const showLlmButton =
+      this.databaseItem.language === "java" && showLlmGeneration();
+
     await this.postMessage({
       t: "setDataExtensionEditorViewState",
       viewState: {
         extensionPack: this.extensionPack,
         enableFrameworkMode: enableFrameworkMode(),
-        showLlmButton: showLlmGeneration(),
+        showLlmButton,
         mode: this.mode,
       },
     });