@querypanel/node-sdk 1.0.39 → 1.0.41

package/dist/index.cjs

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,6 +17,14 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
@@ -554,6 +564,7 @@ function sanitize2(value) {
 }
 
 // src/core/client.ts
+var import_node_crypto = __toESM(require("crypto"), 1);
 var ApiClient = class {
   baseUrl;
   privateKey;
@@ -711,7 +722,7 @@ var ApiClient = class {
     if (this.cryptoKey) {
       return this.cryptoKey;
     }
-    this.cryptoKey = await crypto.subtle.importKey(
+    this.cryptoKey = await import_node_crypto.default.subtle.importKey(
       "pkcs8",
       this.privateKeyToArrayBuffer(this.privateKey),
       {
@@ -753,7 +764,7 @@ var ApiClient = class {
     const data = `${encodedHeader}.${encodedPayload}`;
     const key = await this.getCryptoKey();
     const dataBytes = new TextEncoder().encode(data);
-    const signature = await crypto.subtle.sign(
+    const signature = await import_node_crypto.default.subtle.sign(
       {
         name: "RSASSA-PKCS1-v1_5"
       },
@@ -1092,6 +1103,7 @@ function resolveTenantId2(client, tenantId) {
 }
 
 // src/routes/ingest.ts
+var import_node_crypto2 = __toESM(require("crypto"), 1);
 async function syncSchema(client, queryEngine, databaseName, options, signal) {
   const tenantId = resolveTenantId3(client, options.tenantId);
   const adapter = queryEngine.getDatabase(databaseName);
@@ -1103,7 +1115,7 @@ async function syncSchema(client, queryEngine, databaseName, options, signal) {
   if (options.forceReindex) {
     payload.force_reindex = true;
   }
-  const sessionId = crypto.randomUUID();
+  const sessionId = import_node_crypto2.default.randomUUID();
   const response = await client.post(
     "/ingest",
     payload,
@@ -1152,9 +1164,10 @@ function buildSchemaRequest(databaseName, adapter, introspection, metadata) {
 }
 
 // src/routes/query.ts
+var import_node_crypto3 = __toESM(require("crypto"), 1);
 async function ask(client, queryEngine, question, options, signal) {
   const tenantId = resolveTenantId4(client, options.tenantId);
-  const sessionId = crypto.randomUUID();
+  const sessionId = import_node_crypto3.default.randomUUID();
   const maxRetry = options.maxRetry ?? 0;
   let attempt = 0;
   let lastError = options.lastError;
@@ -1191,35 +1204,62 @@ async function ask(client, queryEngine, question, options, signal) {
         tenantId
       );
       const rows = execution.rows ?? [];
+      const chartType = options.chartType ?? "vega-lite";
       let chart = {
-        vegaLiteSpec: null,
+        specType: chartType,
         notes: rows.length === 0 ? "Query returned no rows." : null
       };
       if (rows.length > 0) {
-        const chartResponse = await client.post(
-          "/chart",
-          {
-            question,
-            sql: queryResponse.sql,
-            rationale: queryResponse.rationale,
-            fields: execution.fields,
-            rows: anonymizeResults(rows),
-            max_retries: options.chartMaxRetries ?? 3,
-            query_id: queryResponse.queryId
-          },
-          tenantId,
-          options.userId,
-          options.scopes,
-          signal,
-          sessionId
-        );
-        chart = {
-          vegaLiteSpec: chartResponse.chart ? {
-            ...chartResponse.chart,
-            data: { values: rows }
-          } : null,
-          notes: chartResponse.notes
-        };
+        if (chartType === "vizspec") {
+          const vizspecResponse = await client.post(
+            "/vizspec",
+            {
+              question,
+              sql: queryResponse.sql,
+              rationale: queryResponse.rationale,
+              fields: execution.fields,
+              rows: anonymizeResults(rows),
+              max_retries: options.chartMaxRetries ?? 3,
+              query_id: queryResponse.queryId
+            },
+            tenantId,
+            options.userId,
+            options.scopes,
+            signal,
+            sessionId
+          );
+          chart = {
+            vizSpec: vizspecResponse.spec,
+            specType: "vizspec",
+            notes: vizspecResponse.notes
+          };
+        } else {
+          const chartResponse = await client.post(
+            "/chart",
+            {
+              question,
+              sql: queryResponse.sql,
+              rationale: queryResponse.rationale,
+              fields: execution.fields,
+              rows: anonymizeResults(rows),
+              max_retries: options.chartMaxRetries ?? 3,
+              query_id: queryResponse.queryId
+            },
+            tenantId,
+            options.userId,
+            options.scopes,
+            signal,
+            sessionId
+          );
+          chart = {
+            vegaLiteSpec: chartResponse.chart ? {
+              ...chartResponse.chart,
+              data: { values: rows }
+            } : null,
+            specType: "vega-lite",
+            notes: chartResponse.notes
+          };
+        }
       }
       return {
         sql: queryResponse.sql,
@@ -1271,6 +1311,40 @@ function anonymizeResults(rows) {
   });
 }
 
+// src/routes/vizspec.ts
+var import_node_crypto4 = __toESM(require("crypto"), 1);
+async function generateVizSpec(client, input, options, signal) {
+  const tenantId = resolveTenantId5(client, options?.tenantId);
+  const sessionId = import_node_crypto4.default.randomUUID();
+  const response = await client.post(
+    "/vizspec",
+    {
+      question: input.question,
+      sql: input.sql,
+      rationale: input.rationale,
+      fields: input.fields,
+      rows: input.rows,
+      max_retries: options?.maxRetries ?? input.max_retries ?? 3,
+      query_id: input.query_id
+    },
+    tenantId,
+    options?.userId,
+    options?.scopes,
+    signal,
+    sessionId
+  );
+  return response;
+}
+function resolveTenantId5(client, tenantId) {
+  const resolved = tenantId ?? client.getDefaultTenantId();
+  if (!resolved) {
+    throw new Error(
+      "tenantId is required. Provide it per request or via defaultTenantId option."
+    );
+  }
+  return resolved;
+}
+
 // src/index.ts
 var QueryPanelSdkAPI = class {
   client;
@@ -1336,6 +1410,15 @@ var QueryPanelSdkAPI = class {
       signal
     );
   }
+  // VizSpec generation
+  async generateVizSpec(input, options, signal) {
+    return await generateVizSpec(
+      this.client,
+      input,
+      options,
+      signal
+    );
+  }
   // Chart CRUD operations
   async createChart(body, options, signal) {
     return await createChart(this.client, body, options, signal);