From db5aba7eab31bca98efa9268173440fd889dfbae Mon Sep 17 00:00:00 2001
From: Diego Imbert <diego@windmill.dev>
Date: Tue, 8 Jul 2025 17:55:20 +0200
Subject: [PATCH] import windmill client when necessary

---
 frontend/src/lib/components/EditorBar.svelte | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/frontend/src/lib/components/EditorBar.svelte b/frontend/src/lib/components/EditorBar.svelte
index 2f9601bf01..80a192662c 100644
--- a/frontend/src/lib/components/EditorBar.svelte
+++ b/frontend/src/lib/components/EditorBar.svelte
@@ -655,8 +655,14 @@ JsonNode ${windmillPathToCamelCaseName(path)} = JsonNode.Parse(await client.GetS
 			if (s3obj.detail?.s3.endsWith('.parquet')) s = `read_parquet(${s})`
 			editor?.insertAtCursor(s)
 		} else if (lang === 'python3') {
+			if (!editor?.getCode().includes('import wmill')) {
+				editor?.insertAtBeginning('import wmill\n')
+			}
 			editor?.insertAtCursor(`wmill.load_s3_file(${s})`)
 		} else if (['javascript', 'typescript'].includes(scriptLangToEditorLang(lang))) {
+			if (!editor?.getCode().includes('import * as wmill from')) {
+				editor?.insertAtBeginning(`import * as wmill from "npm:windmill-client@1"\n`)
+			}
 			editor?.insertAtCursor(`wmill.loadS3File(${s})`)
 		}
 	}}