@querypanel/node-sdk 1.2.9 → 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);
@@ -1928,6 +1989,7 @@ async function ask(client, queryEngine, question, options, signal) {
           signal,
           sessionId
         );
+        throwIfGenerationErrorResponse(vizspecResponse);
         chart = {
           vizSpec: vizspecResponse.spec,
           specType: "vizspec",
@@ -1951,6 +2013,7 @@ async function ask(client, queryEngine, question, options, signal) {
           signal,
           sessionId
         );
+        throwIfGenerationErrorResponse(chartResponse);
         chart = {
           vegaLiteSpec: chartResponse.chart ? {
             ...chartResponse.chart,
@@ -1978,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;
@@ -2274,6 +2344,7 @@ async function modifyChart(client, queryEngine, input, options, signal) {
       signal,
       sessionId
     );
+    throwIfGenerationErrorResponse(vizspecResponse);
     chart = {
       vizSpec: vizspecResponse.spec,
       specType: "vizspec",
@@ -2299,6 +2370,7 @@ async function modifyChart(client, queryEngine, input, options, signal) {
       signal,
       sessionId
     );
+    throwIfGenerationErrorResponse(chartResponse);
     chart = {
       vegaLiteSpec: chartResponse.chart ? {
         ...chartResponse.chart,