@querypanel/node-sdk 1.0.50 → 1.0.52

package/dist/index.cjs

@@ -929,6 +929,8 @@ var QueryErrorCode = {
   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",
@@ -965,6 +967,12 @@ var QueryPipelineError = class extends Error {
   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
@@ -1276,6 +1284,7 @@ async function ask(client, queryEngine, question, options, signal) {
   let attempt = 0;
   let lastError = options.lastError;
   let previousSql = options.previousSql;
+  const queryEndpoint = options.pipeline === "v2" ? "/v2/query" : "/query";
   while (attempt <= maxRetry) {
     console.log({ lastError, previousSql });
     const databaseName = options.database ?? queryEngine.getDefaultDatabase();
@@ -1289,7 +1298,7 @@ async function ask(client, queryEngine, question, options, signal) {
       };
     }
     const queryResponse = await client.postWithHeaders(
-      "/query",
+      queryEndpoint,
       {
         question,
         ...querypanelSessionId ? { session_id: querypanelSessionId } : {},
@@ -1405,7 +1414,9 @@ async function ask(client, queryEngine, question, options, signal) {
         context: queryResponse.data.context,
         attempts: attempt + 1,
         target_db: dbName,
-        querypanelSessionId: responseSessionId ?? void 0
+        querypanelSessionId: responseSessionId ?? void 0,
+        intent: queryResponse.data.intent,
+        trace: queryResponse.data.trace
       };
     } catch (error) {
       attempt++;
@@ -1517,6 +1528,7 @@ async function modifyChart(client, queryEngine, input, options, signal) {
   const hasSqlMods = !!input.sqlModifications;
   const hasVizMods = !!input.vizModifications;
   const hasCustomSql = !!input.sqlModifications?.customSql;
+  const queryEndpoint = options?.pipeline === "v2" ? "/v2/query" : "/query";
   let finalSql = input.sql;
   let finalParams = input.params ?? {};
   let paramMetadata = [];
@@ -1549,7 +1561,7 @@ async function modifyChart(client, queryEngine, input, options, signal) {
       input.sqlModifications
     );
     const queryResponse = await client.post(
-      "/query",
+      queryEndpoint,
       {
         question: modifiedQuestion,
         previous_sql: input.sql,