@querypanel/node-sdk 1.0.43 → 1.0.44

package/dist/index.cjs

@@ -32,7 +32,9 @@ var index_exports = {};
 __export(index_exports, {
   ClickHouseAdapter: () => ClickHouseAdapter,
   PostgresAdapter: () => PostgresAdapter,
+  QueryErrorCode: () => QueryErrorCode,
   QueryPanelSdkAPI: () => QueryPanelSdkAPI,
+  QueryPipelineError: () => QueryPipelineError,
   anonymizeResults: () => anonymizeResults
 });
 module.exports = __toCommonJS(index_exports);
@@ -881,6 +883,57 @@ 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",
+  // 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();
+  }
+};
+
 // src/routes/charts.ts
 async function createChart(client, body, options, signal) {
   const tenantId = resolveTenantId(client, options?.tenantId);
@@ -917,15 +970,10 @@ async function listCharts(client, queryEngine, options, signal) {
   );
   if (options?.includeData) {
     response.data = await Promise.all(
-      response.data.map(async (chart) => ({
-        ...chart,
-        vega_lite_spec: {
-          ...chart.vega_lite_spec,
-          data: {
-            values: await executeChartQuery(queryEngine, chart, tenantId)
-          }
-        }
-      }))
+      response.data.map(async (chart) => {
+        const rows = await executeChartQuery(queryEngine, chart, tenantId);
+        return hydrateChartWithData(chart, rows);
+      })
     );
   }
   return response;
@@ -939,15 +987,8 @@ async function getChart(client, queryEngine, id, options, signal) {
     options?.scopes,
     signal
   );
-  return {
-    ...chart,
-    vega_lite_spec: {
-      ...chart.vega_lite_spec,
-      data: {
-        values: await executeChartQuery(queryEngine, chart, tenantId)
-      }
-    }
-  };
+  const rows = await executeChartQuery(queryEngine, chart, tenantId);
+  return hydrateChartWithData(chart, rows);
 }
 async function updateChart(client, id, body, options, signal) {
   const tenantId = resolveTenantId(client, options?.tenantId);
@@ -998,6 +1039,31 @@ async function executeChartQuery(queryEngine, chart, tenantId) {
     return [];
   }
 }
+function hydrateChartWithData(chart, rows) {
+  const spec = chart.vega_lite_spec;
+  if (chart.spec_type === "vizspec") {
+    const existingData = spec.data ?? {};
+    return {
+      ...chart,
+      vega_lite_spec: {
+        ...spec,
+        data: {
+          ...existingData,
+          values: rows
+        }
+      }
+    };
+  }
+  return {
+    ...chart,
+    vega_lite_spec: {
+      ...spec,
+      data: {
+        values: rows
+      }
+    }
+  };
+}
 
 // src/routes/active-charts.ts
 async function createActiveChart(client, body, options, signal) {
@@ -1205,6 +1271,13 @@ async function ask(client, queryEngine, question, options, signal) {
       signal,
       sessionId
     );
+    if (!queryResponse.success) {
+      throw new QueryPipelineError(
+        queryResponse.error || "Query generation failed",
+        queryResponse.code || "INTERNAL_ERROR",
+        queryResponse.details
+      );
+    }
     const dbName = queryResponse.database ?? options.database ?? queryEngine.getDefaultDatabase();
     if (!dbName) {
       throw new Error(
@@ -2049,7 +2122,9 @@ var QueryPanelSdkAPI = class {
 0 && (module.exports = {
   ClickHouseAdapter,
   PostgresAdapter,
+  QueryErrorCode,
   QueryPanelSdkAPI,
+  QueryPipelineError,
   anonymizeResults
 });
 //# sourceMappingURL=index.cjs.map