@querypanel/node-sdk 1.2.8 → 1.2.10

package/dist/index.cjs

@@ -749,6 +749,118 @@ function sanitize2(value) {
 
 // src/core/client.ts
 var import_node_crypto = __toESM(require("crypto"), 1);
+
+// src/errors.ts
+var QueryErrorCode = {
+  // Moderation errors
+  MODERATION_FAILED: "MODERATION_FAILED",
+  // Guardrail errors
+  RELEVANCE_CHECK_FAILED: "RELEVANCE_CHECK_FAILED",
+  SECURITY_CHECK_FAILED: "SECURITY_CHECK_FAILED",
+  // SQL generation errors
+  SQL_GENERATION_FAILED: "SQL_GENERATION_FAILED",
+  // SQL validation errors
+  SQL_VALIDATION_FAILED: "SQL_VALIDATION_FAILED",
+  // Context retrieval errors
+  CONTEXT_RETRIEVAL_FAILED: "CONTEXT_RETRIEVAL_FAILED",
+  // Clarification errors (v2)
+  CLARIFICATION_NEEDED: "CLARIFICATION_NEEDED",
+  // Bring-your-own-key / provider errors
+  AI_PROVIDER_AUTH_FAILED: "AI_PROVIDER_AUTH_FAILED",
+  AI_PROVIDER_RATE_LIMITED: "AI_PROVIDER_RATE_LIMITED",
+  AI_PROVIDER_UNAVAILABLE: "AI_PROVIDER_UNAVAILABLE",
+  // General errors
+  INTERNAL_ERROR: "INTERNAL_ERROR",
+  AUTHENTICATION_REQUIRED: "AUTHENTICATION_REQUIRED",
+  VALIDATION_ERROR: "VALIDATION_ERROR"
+};
+function isQueryErrorCode(code) {
+  return typeof code === "string" && Object.values(QueryErrorCode).includes(code);
+}
+function isAiProviderErrorCode(code) {
+  return code === QueryErrorCode.AI_PROVIDER_AUTH_FAILED || code === QueryErrorCode.AI_PROVIDER_RATE_LIMITED || code === QueryErrorCode.AI_PROVIDER_UNAVAILABLE;
+}
+function isAiProviderPipelineError(error) {
+  if (error instanceof QueryPipelineError) {
+    return isAiProviderErrorCode(error.code);
+  }
+  if (error && typeof error === "object" && "code" in error) {
+    return isAiProviderErrorCode(error.code);
+  }
+  return false;
+}
+function throwIfGenerationErrorResponse(response) {
+  if (!response || typeof response !== "object") {
+    return;
+  }
+  const record = response;
+  if (typeof record.error !== "string") {
+    return;
+  }
+  if (isQueryErrorCode(record.code)) {
+    throw new QueryPipelineError(record.error, record.code);
+  }
+  throw new Error(record.error);
+}
+var QueryPipelineError = class extends Error {
+  constructor(message, code, details, status) {
+    super(message);
+    this.code = code;
+    this.details = details;
+    this.status = status;
+    this.name = "QueryPipelineError";
+  }
+  /**
+   * Check if this is a moderation error
+   */
+  isModeration() {
+    return this.code === QueryErrorCode.MODERATION_FAILED;
+  }
+  /**
+   * Check if this is a relevance error (question not related to database)
+   */
+  isRelevanceError() {
+    return this.code === QueryErrorCode.RELEVANCE_CHECK_FAILED;
+  }
+  /**
+   * Check if this is a security error (SQL injection, prompt injection, etc.)
+   */
+  isSecurityError() {
+    return this.code === QueryErrorCode.SECURITY_CHECK_FAILED;
+  }
+  /**
+   * Check if this is any guardrail error (relevance or security)
+   */
+  isGuardrailError() {
+    return this.isRelevanceError() || this.isSecurityError();
+  }
+  /**
+   * Check if this is a clarification needed error (v2 pipeline)
+   */
+  isClarificationNeeded() {
+    return this.code === QueryErrorCode.CLARIFICATION_NEEDED;
+  }
+  /**
+   * Check if this is a BYOK / provider credential error
+   */
+  isAiProviderAuthError() {
+    return this.code === QueryErrorCode.AI_PROVIDER_AUTH_FAILED;
+  }
+  /**
+   * Check if this is a BYOK / provider rate-limit error
+   */
+  isAiProviderRateLimited() {
+    return this.code === QueryErrorCode.AI_PROVIDER_RATE_LIMITED;
+  }
+  /**
+   * Check if this is any BYOK / provider error (auth, rate limit, or unavailable)
+   */
+  isAiProviderError() {
+    return isAiProviderErrorCode(this.code);
+  }
+};
+
+// src/core/client.ts
 var ApiClient = class {
   constructor(baseUrl, privateKey, organizationId, options) {
     this.cryptoKey = null;
@@ -880,10 +992,18 @@ var ApiClient = class {
       json = void 0;
     }
     if (!response.ok) {
-      const error = new Error(
-        json?.error || response.statusText || "Request failed"
-      );
+      const message = json?.error || response.statusText || "Request failed";
+      if (isQueryErrorCode(json?.code)) {
+        throw new QueryPipelineError(
+          message,
+          json.code,
+          json?.details,
+          response.status
+        );
+      }
+      const error = new Error(message);
       error.status = response.status;
+      if (json?.code) error.code = json.code;
       if (json?.details) error.details = json.details;
       throw error;
     }
@@ -1212,65 +1332,6 @@ var QueryEngine = class {
   }
 };
 
-// src/errors.ts
-var QueryErrorCode = {
-  // Moderation errors
-  MODERATION_FAILED: "MODERATION_FAILED",
-  // Guardrail errors
-  RELEVANCE_CHECK_FAILED: "RELEVANCE_CHECK_FAILED",
-  SECURITY_CHECK_FAILED: "SECURITY_CHECK_FAILED",
-  // SQL generation errors
-  SQL_GENERATION_FAILED: "SQL_GENERATION_FAILED",
-  // SQL validation errors
-  SQL_VALIDATION_FAILED: "SQL_VALIDATION_FAILED",
-  // Context retrieval errors
-  CONTEXT_RETRIEVAL_FAILED: "CONTEXT_RETRIEVAL_FAILED",
-  // Clarification errors (v2)
-  CLARIFICATION_NEEDED: "CLARIFICATION_NEEDED",
-  // General errors
-  INTERNAL_ERROR: "INTERNAL_ERROR",
-  AUTHENTICATION_REQUIRED: "AUTHENTICATION_REQUIRED",
-  VALIDATION_ERROR: "VALIDATION_ERROR"
-};
-var QueryPipelineError = class extends Error {
-  constructor(message, code, details) {
-    super(message);
-    this.code = code;
-    this.details = details;
-    this.name = "QueryPipelineError";
-  }
-  /**
-   * Check if this is a moderation error
-   */
-  isModeration() {
-    return this.code === QueryErrorCode.MODERATION_FAILED;
-  }
-  /**
-   * Check if this is a relevance error (question not related to database)
-   */
-  isRelevanceError() {
-    return this.code === QueryErrorCode.RELEVANCE_CHECK_FAILED;
-  }
-  /**
-   * Check if this is a security error (SQL injection, prompt injection, etc.)
-   */
-  isSecurityError() {
-    return this.code === QueryErrorCode.SECURITY_CHECK_FAILED;
-  }
-  /**
-   * Check if this is any guardrail error (relevance or security)
-   */
-  isGuardrailError() {
-    return this.isRelevanceError() || this.isSecurityError();
-  }
-  /**
-   * Check if this is a clarification needed error (v2 pipeline)
-   */
-  isClarificationNeeded() {
-    return this.code === QueryErrorCode.CLARIFICATION_NEEDED;
-  }
-};
-
 // src/routes/charts.ts
 async function createChart(client, body, options, signal) {
   const tenantId = resolveTenantId(client, options?.tenantId);
@@ -1907,60 +1968,60 @@ async function ask(client, queryEngine, question, options, signal) {
         specType: chartType,
         notes: rows.length === 0 ? "Query returned no rows." : null
       };
-      if (rows.length > 0) {
-        if (chartType === "vizspec") {
-          const vizspecResponse = await client.post(
-            "/vizspec",
-            {
-              question,
-              sql,
-              rationale: queryResponse.data.rationale,
-              fields: execution.fields,
-              rows: anonymizeResults(rows),
-              max_retries: options.chartMaxRetries ?? 3,
-              query_id: queryResponse.data.queryId,
-              ...options.supportedChartTypes?.length ? {
-                supported_chart_types: options.supportedChartTypes
-              } : {}
-            },
-            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,
-              rationale: queryResponse.data.rationale,
-              fields: execution.fields,
-              rows: anonymizeResults(rows),
-              max_retries: options.chartMaxRetries ?? 3,
-              query_id: queryResponse.data.queryId
-            },
-            tenantId,
-            options.userId,
-            options.scopes,
-            signal,
-            sessionId
-          );
-          chart = {
-            vegaLiteSpec: chartResponse.chart ? {
-              ...chartResponse.chart,
-              data: { values: rows }
-            } : null,
-            specType: "vega-lite",
-            notes: chartResponse.notes
-          };
-        }
+      if (chartType === "vizspec") {
+        const vizspecResponse = await client.post(
+          "/vizspec",
+          {
+            question,
+            sql,
+            rationale: queryResponse.data.rationale,
+            fields: execution.fields,
+            rows: anonymizeResults(rows),
+            max_retries: options.chartMaxRetries ?? 3,
+            query_id: queryResponse.data.queryId,
+            ...options.supportedChartTypes?.length ? {
+              supported_chart_types: options.supportedChartTypes
+            } : {}
+          },
+          tenantId,
+          options.userId,
+          options.scopes,
+          signal,
+          sessionId
+        );
+        throwIfGenerationErrorResponse(vizspecResponse);
+        chart = {
+          vizSpec: vizspecResponse.spec,
+          specType: "vizspec",
+          notes: rows.length === 0 ? vizspecResponse.notes ?? "Query returned no rows." : vizspecResponse.notes
+        };
+      } else {
+        const chartResponse = await client.post(
+          "/chart",
+          {
+            question,
+            sql,
+            rationale: queryResponse.data.rationale,
+            fields: execution.fields,
+            rows: anonymizeResults(rows),
+            max_retries: options.chartMaxRetries ?? 3,
+            query_id: queryResponse.data.queryId
+          },
+          tenantId,
+          options.userId,
+          options.scopes,
+          signal,
+          sessionId
+        );
+        throwIfGenerationErrorResponse(chartResponse);
+        chart = {
+          vegaLiteSpec: chartResponse.chart ? {
+            ...chartResponse.chart,
+            data: { values: rows }
+          } : null,
+          specType: "vega-lite",
+          notes: rows.length === 0 ? chartResponse.notes ?? "Query returned no rows." : chartResponse.notes
+        };
       }
       return {
         sql,
@@ -1980,6 +2041,13 @@ async function ask(client, queryEngine, question, options, signal) {
         trace: queryResponse.data.trace
       };
     } catch (error) {
+      if (isAiProviderPipelineError(error)) {
+        if (error instanceof QueryPipelineError) {
+          throw error;
+        }
+        const providerError = error;
+        throw new QueryPipelineError(providerError.message, providerError.code);
+      }
       attempt++;
       if (attempt > maxRetry) {
         throw error;
@@ -2253,64 +2321,64 @@ async function modifyChart(client, queryEngine, input, options, signal) {
     specType: chartType,
     notes: rows.length === 0 ? "Query returned no rows." : null
   };
-  if (rows.length > 0) {
-    const vizHints = hasVizMods ? buildVizHints(input.vizModifications) : {};
-    const vizHintsForChart = chartType === "vizspec" ? vizHints : stripVizSpecOnlyHints(vizHints);
-    if (chartType === "vizspec") {
-      const vizspecResponse = await client.post(
-        "/vizspec",
-        {
-          question: finalQuestion,
-          sql: finalSql,
-          rationale,
-          fields: execution.fields,
-          rows: anonymizeResults(rows),
-          max_retries: options?.chartMaxRetries ?? 3,
-          query_id: queryId,
-          // Include viz hints for the chart generator
-          ...hasVizMods ? { encoding_hints: vizHintsForChart } : {},
-          ...options?.supportedChartTypes?.length ? { supported_chart_types: options.supportedChartTypes } : {}
-        },
-        tenantId,
-        options?.userId,
-        options?.scopes,
-        signal,
-        sessionId
-      );
-      chart = {
-        vizSpec: vizspecResponse.spec,
-        specType: "vizspec",
-        notes: vizspecResponse.notes
-      };
-    } else {
-      const chartResponse = await client.post(
-        "/chart",
-        {
-          question: finalQuestion,
-          sql: finalSql,
-          rationale,
-          fields: execution.fields,
-          rows: anonymizeResults(rows),
-          max_retries: options?.chartMaxRetries ?? 3,
-          query_id: queryId,
-          // Include viz hints for the chart generator
-          ...hasVizMods ? { encoding_hints: vizHintsForChart } : {}
-        },
-        tenantId,
-        options?.userId,
-        options?.scopes,
-        signal,
-        sessionId
-      );
-      chart = {
-        vegaLiteSpec: chartResponse.chart ? {
-          ...chartResponse.chart,
-          data: { values: rows }
-        } : null,
-        specType: "vega-lite",
-        notes: chartResponse.notes
-      };
-    }
+  const vizHints = hasVizMods ? buildVizHints(input.vizModifications) : {};
+  const vizHintsForChart = chartType === "vizspec" ? vizHints : stripVizSpecOnlyHints(vizHints);
+  if (chartType === "vizspec") {
+    const vizspecResponse = await client.post(
+      "/vizspec",
+      {
+        question: finalQuestion,
+        sql: finalSql,
+        rationale,
+        fields: execution.fields,
+        rows: anonymizeResults(rows),
+        max_retries: options?.chartMaxRetries ?? 3,
+        query_id: queryId,
+        // Include viz hints for the chart generator
+        ...hasVizMods ? { encoding_hints: vizHintsForChart } : {},
+        ...options?.supportedChartTypes?.length ? { supported_chart_types: options.supportedChartTypes } : {}
+      },
+      tenantId,
+      options?.userId,
+      options?.scopes,
+      signal,
+      sessionId
+    );
+    throwIfGenerationErrorResponse(vizspecResponse);
+    chart = {
+      vizSpec: vizspecResponse.spec,
+      specType: "vizspec",
+      notes: rows.length === 0 ? vizspecResponse.notes ?? "Query returned no rows." : vizspecResponse.notes
+    };
+  } else {
+    const chartResponse = await client.post(
+      "/chart",
+      {
+        question: finalQuestion,
+        sql: finalSql,
+        rationale,
+        fields: execution.fields,
+        rows: anonymizeResults(rows),
+        max_retries: options?.chartMaxRetries ?? 3,
+        query_id: queryId,
+        // Include viz hints for the chart generator
+        ...hasVizMods ? { encoding_hints: vizHintsForChart } : {}
+      },
+      tenantId,
+      options?.userId,
+      options?.scopes,
+      signal,
+      sessionId
+    );
+    throwIfGenerationErrorResponse(chartResponse);
+    chart = {
+      vegaLiteSpec: chartResponse.chart ? {
+        ...chartResponse.chart,
+        data: { values: rows }
+      } : null,
+      specType: "vega-lite",
+      notes: rows.length === 0 ? chartResponse.notes ?? "Query returned no rows." : chartResponse.notes
+    };
   }
   return {
     sql: finalSql,