@marimo-team/frontend 0.23.9-dev20 → 0.23.9-dev21

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (70) hide show
  1. package/dist/assets/{CellStatus-cST_KROR.js → CellStatus-BqCsCfF0.js} +1 -1
  2. package/dist/assets/{JsonOutput-BwgMLuPn.js → JsonOutput-7sNcAFVa.js} +1 -1
  3. package/dist/assets/{MarimoErrorOutput-DIbgMg0T.js → MarimoErrorOutput-BOi3ico3.js} +1 -1
  4. package/dist/assets/{RenderHTML-CH8QrLVl.js → RenderHTML-CUf8D25v.js} +1 -1
  5. package/dist/assets/{add-cell-with-ai-DDmHtapY.js → add-cell-with-ai-DoLVEWe2.js} +1 -1
  6. package/dist/assets/{add-connection-dialog-CRbrpdQu.js → add-connection-dialog-C9ehEIhe.js} +1 -1
  7. package/dist/assets/{agent-panel-Ci6kmMWq.js → agent-panel-DfGsNTLw.js} +1 -1
  8. package/dist/assets/{ai-model-dropdown-CNobct-x.js → ai-model-dropdown-BWdck8rF.js} +1 -1
  9. package/dist/assets/{app-config-button-DrlCOjyd.js → app-config-button-BJhb3-gi.js} +1 -1
  10. package/dist/assets/{cell-editor-Bhvnqdc-.js → cell-editor-ZKy9HWcX.js} +1 -1
  11. package/dist/assets/{cell-link-rh-JpkCu.js → cell-link-DLb_k4Kf.js} +1 -1
  12. package/dist/assets/{cells-Cwex_Zk1.js → cells-yL2qMkYn.js} +41 -41
  13. package/dist/assets/{chat-display-DzBxD2g1.js → chat-display-UA8v27kw.js} +1 -1
  14. package/dist/assets/{chat-panel-ipOkNc7y.js → chat-panel-BGtfb6Wo.js} +1 -1
  15. package/dist/assets/{chat-ui-DqHdyg0V.js → chat-ui-CfYDiSmX.js} +1 -1
  16. package/dist/assets/{column-preview-WtwlsutV.js → column-preview-D1138kdA.js} +1 -1
  17. package/dist/assets/{command-palette-C5YuwxOo.js → command-palette-D9JIsy_G.js} +1 -1
  18. package/dist/assets/{common-B3E9c-gU.js → common-DqZ15mGE.js} +1 -1
  19. package/dist/assets/{components-DFkvRpxg.js → components-BK3xAU_5.js} +1 -1
  20. package/dist/assets/{components-wm4TXPI6.js → components-DXfCmhfc.js} +1 -1
  21. package/dist/assets/{datasource-rDrNZEmo.js → datasource-CJC51Mx7.js} +1 -1
  22. package/dist/assets/{dependency-graph-panel-BWe9yDlP.js → dependency-graph-panel-avUgiyMA.js} +1 -1
  23. package/dist/assets/{documentation-panel-BAxMFFrB.js → documentation-panel-BX7dDKRB.js} +1 -1
  24. package/dist/assets/{download-NRqwJhRT.js → download-DucsG72p.js} +1 -1
  25. package/dist/assets/{edit-page-CNuf6Bpw.js → edit-page-BG3s4KtX.js} +3 -3
  26. package/dist/assets/{error-panel-CByNONxp.js → error-panel-BqoOUjfw.js} +1 -1
  27. package/dist/assets/{file-explorer-panel-DcCFx-Mu.js → file-explorer-panel-8hhN0YIE.js} +1 -1
  28. package/dist/assets/{file-icons-CV0odlbf.js → file-icons-CNUmy2Dl.js} +1 -1
  29. package/dist/assets/{floating-outline-DmddQK6b.js → floating-outline-B_uWfcys.js} +1 -1
  30. package/dist/assets/{focus-DLnEejrn.js → focus-Cyz1EKwA.js} +1 -1
  31. package/dist/assets/{form-Dnx_5kyZ.js → form-Dyt6aN_7.js} +1 -1
  32. package/dist/assets/{home-page-ffvNN1h7.js → home-page-CGoMMRBf.js} +1 -1
  33. package/dist/assets/{hooks-DuHQuYPv.js → hooks-CQnd87Km.js} +1 -1
  34. package/dist/assets/{html-to-image-CzpeYelq.js → html-to-image-CKzwKb6F.js} +1 -1
  35. package/dist/assets/{index-BN0lrf4C.js → index-vrfD6Tng.js} +3 -3
  36. package/dist/assets/{kiosk-mode-C6QWFFID.js → kiosk-mode-z2bbZ6_w.js} +1 -1
  37. package/dist/assets/{layout-dxaLq-eR.js → layout-Brpm-l5C.js} +3 -3
  38. package/dist/assets/{logs-panel-9W91k1zJ.js → logs-panel-CNV8j7by.js} +1 -1
  39. package/dist/assets/{markdown-renderer-B-iPeXPA.js → markdown-renderer-3mlKZ6Wa.js} +1 -1
  40. package/dist/assets/{name-cell-input-uFfmqWYT.js → name-cell-input-BnL86ka3.js} +1 -1
  41. package/dist/assets/{outline-panel-BAWIWb20.js → outline-panel-k7h8ZSSH.js} +1 -1
  42. package/dist/assets/{packages-panel-DncGPBWC.js → packages-panel-CHPpObHs.js} +1 -1
  43. package/dist/assets/{panels-ByGgkdwq.js → panels-DlR3fy59.js} +1 -1
  44. package/dist/assets/{process-output-D06o1mmH.js → process-output-Bi7Z_Agd.js} +1 -1
  45. package/dist/assets/{radio-group-DUTOVun5.js → radio-group-CdBpUP4h.js} +1 -1
  46. package/dist/assets/{readonly-python-code-BnaYQo0K.js → readonly-python-code-beWODwjc.js} +1 -1
  47. package/dist/assets/{reveal-component-BXfhZjrl.js → reveal-component-2Gg2VcRe.js} +1 -1
  48. package/dist/assets/{run-page-Um3pVJx9.js → run-page-BJKTdA2B.js} +1 -1
  49. package/dist/assets/{scratchpad-panel-D8kI0OKl.js → scratchpad-panel-D1N-Kxhd.js} +1 -1
  50. package/dist/assets/{session-panel-07dCccS4.js → session-panel-C5xz5D-N.js} +1 -1
  51. package/dist/assets/{snippets-panel-s3hpmPTY.js → snippets-panel-rCxyqna5.js} +1 -1
  52. package/dist/assets/{state-DIFGjBXT.js → state-B6DTPcu0.js} +1 -1
  53. package/dist/assets/{state-B5L8iIIr.js → state-vUGPvNzI.js} +1 -1
  54. package/dist/assets/{textarea-D0VOSGcU.js → textarea-DRcRrFRn.js} +1 -1
  55. package/dist/assets/{tracing-CSRhEdtZ.js → tracing-DKMmxWD9.js} +1 -1
  56. package/dist/assets/{tracing-panel-DuaSjEhP.js → tracing-panel-KG023vH2.js} +2 -2
  57. package/dist/assets/{useCellActionButton-CtygwcqH.js → useCellActionButton-ZXUUkx8s.js} +1 -1
  58. package/dist/assets/{useDeleteCell-DeZ7T5A3.js → useDeleteCell-wNHy5lEM.js} +1 -1
  59. package/dist/assets/{useDependencyPanelTab-Bpa86UXQ.js → useDependencyPanelTab-B7Qj36AL.js} +1 -1
  60. package/dist/assets/{useNotebookActions-DuMI8tMy.js → useNotebookActions-CXlvqZKD.js} +1 -1
  61. package/dist/assets/{useRunCells-CScbBaxj.js → useRunCells-CJadPmdw.js} +1 -1
  62. package/dist/assets/{useSplitCell-CBlrUeHM.js → useSplitCell-C04l3e__.js} +1 -1
  63. package/dist/index.html +22 -22
  64. package/package.json +1 -1
  65. package/src/components/databases/display.tsx +2 -0
  66. package/src/components/datasources/__tests__/utils.test.ts +82 -0
  67. package/src/components/datasources/utils.ts +16 -15
  68. package/src/core/codemirror/format.ts +1 -0
  69. package/src/core/codemirror/language/languages/sql/sql.ts +1 -0
  70. package/src/core/codemirror/language/languages/sql/utils.ts +2 -0
package/dist/index.html CHANGED
@@ -66,7 +66,7 @@
66
66
  <marimo-server-token data-token="{{ server_token }}" hidden></marimo-server-token>
67
67
  <!-- /TODO -->
68
68
  <title>{{ title }}</title>
69
- <script type="module" crossorigin src="./assets/index-BN0lrf4C.js"></script>
69
+ <script type="module" crossorigin src="./assets/index-vrfD6Tng.js"></script>
70
70
  <link rel="modulepreload" crossorigin href="./assets/preload-helper-BPPi7vOr.js">
71
71
  <link rel="modulepreload" crossorigin href="./assets/chunk-LvLJmgfZ.js">
72
72
  <link rel="modulepreload" crossorigin href="./assets/react-Bj1aDYRI.js">
@@ -133,7 +133,7 @@
133
133
  <link rel="modulepreload" crossorigin href="./assets/debounce-DhnxH9Rh.js">
134
134
  <link rel="modulepreload" crossorigin href="./assets/database-zap-kIkTfzTX.js">
135
135
  <link rel="modulepreload" crossorigin href="./assets/main-B0OX4z33.js">
136
- <link rel="modulepreload" crossorigin href="./assets/cells-Cwex_Zk1.js">
136
+ <link rel="modulepreload" crossorigin href="./assets/cells-yL2qMkYn.js">
137
137
  <link rel="modulepreload" crossorigin href="./assets/ErrorBoundary-DyYDV0HI.js">
138
138
  <link rel="modulepreload" crossorigin href="./assets/kbd-CTUAEnEx.js">
139
139
  <link rel="modulepreload" crossorigin href="./assets/useInstallPackage-DUF4IRRI.js">
@@ -147,35 +147,35 @@
147
147
  <link rel="modulepreload" crossorigin href="./assets/usePress-DQ_tAz5W.js">
148
148
  <link rel="modulepreload" crossorigin href="./assets/input-C3Hrdlqq.js">
149
149
  <link rel="modulepreload" crossorigin href="./assets/ImperativeModal-B3Th7k4R.js">
150
- <link rel="modulepreload" crossorigin href="./assets/cell-link-rh-JpkCu.js">
150
+ <link rel="modulepreload" crossorigin href="./assets/cell-link-DLb_k4Kf.js">
151
151
  <link rel="modulepreload" crossorigin href="./assets/multi-map-CUuNtzHt.js">
152
152
  <link rel="modulepreload" crossorigin href="./assets/alert-yTS3WpF4.js">
153
153
  <link rel="modulepreload" crossorigin href="./assets/chevron-right-CG5QYXYk.js">
154
154
  <link rel="modulepreload" crossorigin href="./assets/dropdown-menu-CR7cnzLX.js">
155
155
  <link rel="modulepreload" crossorigin href="./assets/links-D1JoyKTt.js">
156
- <link rel="modulepreload" crossorigin href="./assets/useRunCells-CScbBaxj.js">
156
+ <link rel="modulepreload" crossorigin href="./assets/useRunCells-CJadPmdw.js">
157
157
  <link rel="modulepreload" crossorigin href="./assets/copy-LK56fFow.js">
158
158
  <link rel="modulepreload" crossorigin href="./assets/copy-BwrPA9zQ.js">
159
159
  <link rel="modulepreload" crossorigin href="./assets/copy-icon-OjtDb4gO.js">
160
- <link rel="modulepreload" crossorigin href="./assets/RenderHTML-CH8QrLVl.js">
161
- <link rel="modulepreload" crossorigin href="./assets/datasource-rDrNZEmo.js">
162
- <link rel="modulepreload" crossorigin href="./assets/state-DIFGjBXT.js">
160
+ <link rel="modulepreload" crossorigin href="./assets/RenderHTML-CUf8D25v.js">
161
+ <link rel="modulepreload" crossorigin href="./assets/datasource-CJC51Mx7.js">
162
+ <link rel="modulepreload" crossorigin href="./assets/state-B6DTPcu0.js">
163
163
  <link rel="modulepreload" crossorigin href="./assets/package-Tv6ztuzw.js">
164
164
  <link rel="modulepreload" crossorigin href="./assets/sparkles-lWUAsPhp.js">
165
- <link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-DIbgMg0T.js">
165
+ <link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-BOi3ico3.js">
166
166
  <link rel="modulepreload" crossorigin href="./assets/spinner-Bhir8k53.js">
167
- <link rel="modulepreload" crossorigin href="./assets/html-to-image-CzpeYelq.js">
168
- <link rel="modulepreload" crossorigin href="./assets/focus-DLnEejrn.js">
167
+ <link rel="modulepreload" crossorigin href="./assets/html-to-image-CKzwKb6F.js">
168
+ <link rel="modulepreload" crossorigin href="./assets/focus-Cyz1EKwA.js">
169
169
  <link rel="modulepreload" crossorigin href="./assets/useAsyncData-bgszE9F0.js">
170
170
  <link rel="modulepreload" crossorigin href="./assets/LazyAnyLanguageCodeMirror-DFv3yT20.js">
171
171
  <link rel="modulepreload" crossorigin href="./assets/micromark-factory-space-BygYYKhs.js">
172
172
  <link rel="modulepreload" crossorigin href="./assets/chunk-5FQGJX7Z-BzXnQz5s.js">
173
- <link rel="modulepreload" crossorigin href="./assets/markdown-renderer-B-iPeXPA.js">
173
+ <link rel="modulepreload" crossorigin href="./assets/markdown-renderer-3mlKZ6Wa.js">
174
174
  <link rel="modulepreload" crossorigin href="./assets/command-KARR7KMq.js">
175
175
  <link rel="modulepreload" crossorigin href="./assets/field-zLmMOSA4.js">
176
176
  <link rel="modulepreload" crossorigin href="./assets/popover-Bz_0Vkyf.js">
177
177
  <link rel="modulepreload" crossorigin href="./assets/errors-vr57w7Ul.js">
178
- <link rel="modulepreload" crossorigin href="./assets/download-NRqwJhRT.js">
178
+ <link rel="modulepreload" crossorigin href="./assets/download-DucsG72p.js">
179
179
  <link rel="modulepreload" crossorigin href="./assets/table-BGPSHfig.js">
180
180
  <link rel="modulepreload" crossorigin href="./assets/useIframeCapabilities-CcI1zSdn.js">
181
181
  <link rel="modulepreload" crossorigin href="./assets/error-banner-LdWZDbqd.js">
@@ -196,25 +196,25 @@
196
196
  <link rel="modulepreload" crossorigin href="./assets/plus-BgB18UzY.js">
197
197
  <link rel="modulepreload" crossorigin href="./assets/trash-2-rVklqqFF.js">
198
198
  <link rel="modulepreload" crossorigin href="./assets/react-resizable-panels.browser.esm-CV8-hvjx.js">
199
- <link rel="modulepreload" crossorigin href="./assets/JsonOutput-BwgMLuPn.js">
199
+ <link rel="modulepreload" crossorigin href="./assets/JsonOutput-7sNcAFVa.js">
200
200
  <link rel="modulepreload" crossorigin href="./assets/chart-no-axes-column-nqk474t8.js">
201
201
  <link rel="modulepreload" crossorigin href="./assets/square-function-uY_yJr5g.js">
202
202
  <link rel="modulepreload" crossorigin href="./assets/spec-CPQR_o92.js">
203
203
  <link rel="modulepreload" crossorigin href="./assets/ellipsis-vertical-CkwWkOQL.js">
204
204
  <link rel="modulepreload" crossorigin href="./assets/refresh-cw-DHwG4Mac.js">
205
205
  <link rel="modulepreload" crossorigin href="./assets/tree-actions-BM_EJr3E.js">
206
- <link rel="modulepreload" crossorigin href="./assets/components-wm4TXPI6.js">
207
- <link rel="modulepreload" crossorigin href="./assets/column-preview-WtwlsutV.js">
206
+ <link rel="modulepreload" crossorigin href="./assets/components-DXfCmhfc.js">
207
+ <link rel="modulepreload" crossorigin href="./assets/column-preview-D1138kdA.js">
208
208
  <link rel="modulepreload" crossorigin href="./assets/icons-Ol38nIbL.js">
209
- <link rel="modulepreload" crossorigin href="./assets/radio-group-DUTOVun5.js">
210
- <link rel="modulepreload" crossorigin href="./assets/floating-outline-DmddQK6b.js">
209
+ <link rel="modulepreload" crossorigin href="./assets/radio-group-CdBpUP4h.js">
210
+ <link rel="modulepreload" crossorigin href="./assets/floating-outline-B_uWfcys.js">
211
211
  <link rel="modulepreload" crossorigin href="./assets/objectWithoutPropertiesLoose-DfWeGRFv.js">
212
212
  <link rel="modulepreload" crossorigin href="./assets/esm-CqWdmSnV.js">
213
- <link rel="modulepreload" crossorigin href="./assets/readonly-python-code-BnaYQo0K.js">
213
+ <link rel="modulepreload" crossorigin href="./assets/readonly-python-code-beWODwjc.js">
214
214
  <link rel="modulepreload" crossorigin href="./assets/file-headphone-BrQspHac.js">
215
215
  <link rel="modulepreload" crossorigin href="./assets/file-BrdxGLRX.js">
216
216
  <link rel="modulepreload" crossorigin href="./assets/image-DQHXdEQn.js">
217
- <link rel="modulepreload" crossorigin href="./assets/file-icons-CV0odlbf.js">
217
+ <link rel="modulepreload" crossorigin href="./assets/file-icons-CNUmy2Dl.js">
218
218
  <link rel="modulepreload" crossorigin href="./assets/switch-BrbMxTq7.js">
219
219
  <link rel="modulepreload" crossorigin href="./assets/events-CPoJAfgx.js">
220
220
  <link rel="modulepreload" crossorigin href="./assets/globals-CDDbRvHE.js">
@@ -223,11 +223,11 @@
223
223
  <link rel="modulepreload" crossorigin href="./assets/memoize-Tp7rARFe.js">
224
224
  <link rel="modulepreload" crossorigin href="./assets/get-C-qh_et5.js">
225
225
  <link rel="modulepreload" crossorigin href="./assets/_baseSet-CxV9N1bc.js">
226
- <link rel="modulepreload" crossorigin href="./assets/state-B5L8iIIr.js">
226
+ <link rel="modulepreload" crossorigin href="./assets/state-vUGPvNzI.js">
227
227
  <link rel="modulepreload" crossorigin href="./assets/label-xHqFtfdz.js">
228
- <link rel="modulepreload" crossorigin href="./assets/textarea-D0VOSGcU.js">
228
+ <link rel="modulepreload" crossorigin href="./assets/textarea-DRcRrFRn.js">
229
229
  <link rel="modulepreload" crossorigin href="./assets/refresh-ccw-C-n2VFP5.js">
230
- <link rel="modulepreload" crossorigin href="./assets/form-Dnx_5kyZ.js">
230
+ <link rel="modulepreload" crossorigin href="./assets/form-Dyt6aN_7.js">
231
231
  <link rel="modulepreload" crossorigin href="./assets/renderShortcut-BrHz5aiT.js">
232
232
  <link rel="modulepreload" crossorigin href="./assets/useBoolean-CtP63TnG.js">
233
233
  <link rel="modulepreload" crossorigin href="./assets/useDeepCompareMemoize-zUHU--0D.js">
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@marimo-team/frontend",
3
- "version": "0.23.9-dev20",
3
+ "version": "0.23.9-dev21",
4
4
  "main": "dist/main.js",
5
5
  "types": "dist/index.d.ts",
6
6
  "type": "module",
@@ -53,6 +53,8 @@ export function dbDisplayName(name: string) {
53
53
  return "MongoDB";
54
54
  case "iceberg":
55
55
  return "Apache Iceberg";
56
+ case "dremio":
57
+ return "Dremio";
56
58
  default:
57
59
  return name;
58
60
  }
@@ -313,6 +313,26 @@ describe("sqlCode", () => {
313
313
  );
314
314
  });
315
315
 
316
+ it("should preserve dots inside quoted schema names", () => {
317
+ const sqlTableContext: SQLTableContext = {
318
+ engine: "postgres",
319
+ schema: "analytics.events",
320
+ defaultSchema: "public",
321
+ defaultDatabase: "mydb",
322
+ database: "remote",
323
+ dialect: "postgres",
324
+ };
325
+
326
+ const result = sqlCode({
327
+ table: mockTable,
328
+ columnName: mockColumn.name,
329
+ sqlTableContext,
330
+ });
331
+ expect(result).toBe(
332
+ '_df = mo.sql(f"""\nSELECT "email" FROM "remote"."analytics.events"."users" LIMIT 100\n""", engine=postgres)',
333
+ );
334
+ });
335
+
316
336
  it("should not quote * column name", () => {
317
337
  const sqlTableContext: SQLTableContext = {
318
338
  engine: "postgres",
@@ -334,6 +354,68 @@ describe("sqlCode", () => {
334
354
  });
335
355
  });
336
356
 
357
+ describe("Dremio dialect", () => {
358
+ it("should quote reserved column names and table path parts", () => {
359
+ const sqlTableContext: SQLTableContext = {
360
+ engine: "dremio_conn",
361
+ schema: "operations",
362
+ defaultSchema: "",
363
+ defaultDatabase: "",
364
+ database: "lakehouse",
365
+ dialect: "dremio",
366
+ };
367
+
368
+ const result = sqlCode({
369
+ table: { ...mockTable, name: "shipments" as const },
370
+ columnName: "order",
371
+ sqlTableContext,
372
+ });
373
+ expect(result).toBe(
374
+ '_df = mo.sql(f"""\nSELECT "order" FROM "lakehouse"."operations"."shipments" LIMIT 100\n""", engine=dremio_conn)',
375
+ );
376
+ });
377
+
378
+ it("should not quote * column name", () => {
379
+ const sqlTableContext: SQLTableContext = {
380
+ engine: "dremio_conn",
381
+ schema: "operations",
382
+ defaultSchema: "",
383
+ defaultDatabase: "",
384
+ database: "lakehouse",
385
+ dialect: "dremio",
386
+ };
387
+
388
+ const result = sqlCode({
389
+ table: { ...mockTable, name: "customers" as const },
390
+ columnName: "*",
391
+ sqlTableContext,
392
+ });
393
+ expect(result).toBe(
394
+ '_df = mo.sql(f"""\nSELECT * FROM "lakehouse"."operations"."customers" LIMIT 100\n""", engine=dremio_conn)',
395
+ );
396
+ });
397
+
398
+ it("should preserve dots inside quoted schema names", () => {
399
+ const sqlTableContext: SQLTableContext = {
400
+ engine: "dremio_conn",
401
+ schema: "samples.dremio.com",
402
+ defaultSchema: "",
403
+ defaultDatabase: "",
404
+ database: "Samples",
405
+ dialect: "dremio",
406
+ };
407
+
408
+ const result = sqlCode({
409
+ table: { ...mockTable, name: "airlines" as const },
410
+ columnName: "*",
411
+ sqlTableContext,
412
+ });
413
+ expect(result).toBe(
414
+ '_df = mo.sql(f"""\nSELECT * FROM "Samples"."samples.dremio.com"."airlines" LIMIT 100\n""", engine=dremio_conn)',
415
+ );
416
+ });
417
+ });
418
+
337
419
  describe("fallback behavior", () => {
338
420
  it("should use default formatter for unknown dialect", () => {
339
421
  const sqlTableContext: SQLTableContext = {
@@ -15,9 +15,9 @@ export function isSchemaless(schemaName: string) {
15
15
 
16
16
  interface SqlCodeFormatter {
17
17
  /**
18
- * Format the table name based on dialect-specific rules
18
+ * Format the table path based on dialect-specific rules
19
19
  */
20
- formatTableName: (tableName: string) => string;
20
+ formatTablePath: (tablePath: string[]) => string;
21
21
  /**
22
22
  * Format the SELECT clause
23
23
  */
@@ -25,7 +25,7 @@ interface SqlCodeFormatter {
25
25
  }
26
26
 
27
27
  const defaultFormatter: SqlCodeFormatter = {
28
- formatTableName: (tableName: string) => tableName,
28
+ formatTablePath: (tablePath: string[]) => tablePath.join("."),
29
29
  formatSelectClause: (columnName: string, tableName: string) =>
30
30
  `SELECT ${columnName} FROM ${tableName} LIMIT 100`,
31
31
  };
@@ -41,7 +41,8 @@ function getFormatter(dialect: string): SqlCodeFormatter {
41
41
  const quote = BigQueryDialect.spec.identifierQuotes;
42
42
  return {
43
43
  // BigQuery uses backticks for identifiers
44
- formatTableName: (tableName: string) => `${quote}${tableName}${quote}`,
44
+ formatTablePath: (tablePath: string[]) =>
45
+ `${quote}${tablePath.join(".")}${quote}`,
45
46
  formatSelectClause: defaultFormatter.formatSelectClause,
46
47
  };
47
48
  }
@@ -49,7 +50,7 @@ function getFormatter(dialect: string): SqlCodeFormatter {
49
50
  case "sqlserver":
50
51
  case "microsoft sql server":
51
52
  return {
52
- formatTableName: defaultFormatter.formatTableName,
53
+ formatTablePath: defaultFormatter.formatTablePath,
53
54
  formatSelectClause: (columnName: string, tableName: string) =>
54
55
  `SELECT TOP 100 ${columnName} FROM ${tableName}`,
55
56
  };
@@ -57,12 +58,11 @@ function getFormatter(dialect: string): SqlCodeFormatter {
57
58
  case "postgres":
58
59
  case "postgresql":
59
60
  case "duckdb":
61
+ case "dremio":
60
62
  // Quote column and table names to avoid raising errors on weird characters
61
63
  return {
62
- formatTableName: (tableName: string) => {
63
- const parts = tableName.split(".");
64
- return parts.map((part) => `"${part}"`).join(".");
65
- },
64
+ formatTablePath: (tablePath: string[]) =>
65
+ tablePath.map((part) => `"${part}"`).join("."),
66
66
  formatSelectClause: (columnName: string, tableName: string) =>
67
67
  `SELECT ${columnName === "*" ? "*" : `"${columnName}"`} FROM ${tableName} LIMIT 100`,
68
68
  };
@@ -114,26 +114,27 @@ export function sqlCode({
114
114
  database,
115
115
  dialect,
116
116
  } = sqlTableContext;
117
- let tableName = table.name;
117
+ const tablePath = [table.name];
118
118
 
119
119
  // Set the fully qualified table name based on schema and database
120
120
  if (isSchemaless(schema)) {
121
- tableName =
122
- database === defaultDatabase ? tableName : `${database}.${tableName}`;
121
+ if (database !== defaultDatabase) {
122
+ tablePath.unshift(database);
123
+ }
123
124
  } else {
124
125
  // Include schema if it's not the default schema
125
126
  if (schema !== defaultSchema) {
126
- tableName = `${schema}.${tableName}`;
127
+ tablePath.unshift(schema);
127
128
  }
128
129
 
129
130
  // Include database if it's not the default database
130
131
  if (database !== defaultDatabase) {
131
- tableName = `${database}.${tableName}`;
132
+ tablePath.unshift(database);
132
133
  }
133
134
  }
134
135
 
135
136
  const formatter = getFormatter(dialect);
136
- const formattedTableName = formatter.formatTableName(tableName);
137
+ const formattedTableName = formatter.formatTablePath(tablePath);
137
138
  const selectClause = formatter.formatSelectClause(
138
139
  columnName,
139
140
  formattedTableName,
@@ -205,6 +205,7 @@ async function getSqlFormatterDialect(
205
205
  case "mongodb":
206
206
  case "timescaledb":
207
207
  case "datafusion":
208
+ case "dremio":
208
209
  return sql;
209
210
  case "databricks":
210
211
  return spark;
@@ -410,6 +410,7 @@ function connectionNameToParserDialect(
410
410
  case "spark":
411
411
  case "databricks":
412
412
  case "datafusion":
413
+ case "dremio":
413
414
  Logger.debug("Unsupported dialect", { dialect });
414
415
  return null;
415
416
  default:
@@ -52,6 +52,7 @@ const KNOWN_DIALECTS_ARRAY = [
52
52
  "databricks",
53
53
  "datafusion",
54
54
  "microsoft sql server",
55
+ "dremio",
55
56
  ] as const;
56
57
  const KNOWN_DIALECTS: ReadonlySet<string> = new Set(KNOWN_DIALECTS_ARRAY);
57
58
  type KnownDialect = (typeof KNOWN_DIALECTS_ARRAY)[number];
@@ -115,6 +116,7 @@ export function guessDialect(
115
116
  case "spark":
116
117
  case "databricks":
117
118
  case "datafusion":
119
+ case "dremio":
118
120
  Logger.debug("Unsupported dialect", { dialect });
119
121
  return ModifiedStandardSQL;
120
122
  default: