@querypanel/node-sdk 1.0.38 → 1.0.40

package/dist/index.cjs

@@ -1191,35 +1191,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 +1298,39 @@ function anonymizeResults(rows) {
   });
 }
 
+// src/routes/vizspec.ts
+async function generateVizSpec(client, input, options, signal) {
+  const tenantId = resolveTenantId5(client, options?.tenantId);
+  const sessionId = crypto.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 +1396,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);