improvement: auto-install sqlglot when using mo.sql in wasm (#3900)

auto install sqlglot when using sql cells
marimo-team · Feb 24, 2025 · 21d3156 · 21d3156
1 parent d0eef44
commit 21d3156
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/frontend/src/core/islands/worker/worker.tsx b/frontend/src/core/islands/worker/worker.tsx
@@ -89,6 +89,7 @@ const requestHandler = createRPCRequestHandler({
       // Add pandas and duckdb to the code
       code = `import pandas\n${code}`;
       code = `import duckdb\n${code}`;
+      code = `import sqlglot\n${code}`;
     }
 
     await self.pyodide.loadPackagesFromImports(code, {

diff --git a/frontend/src/core/wasm/worker/bootstrap.ts b/frontend/src/core/wasm/worker/bootstrap.ts
@@ -165,6 +165,7 @@ export class DefaultWasmController implements WasmController {
       // We need pandas and duckdb for mo.sql
       code = `import pandas\n${code}`;
       code = `import duckdb\n${code}`;
+      code = `import sqlglot\n${code}`;
     }
 
     // Add:

diff --git a/frontend/src/core/wasm/worker/worker.ts b/frontend/src/core/wasm/worker/worker.ts
@@ -140,6 +140,7 @@ const requestHandler = createRPCRequestHandler({
       // Add pandas and duckdb to the code
       code = `import pandas\n${code}`;
       code = `import duckdb\n${code}`;
+      code = `import sqlglot\n${code}`;
     }
 
     await self.pyodide.loadPackagesFromImports(code, {