@superatomai/sdk-node 0.0.18-mds → 0.0.19-mds

package/dist/index.js

@@ -4270,6 +4270,15 @@ var promptLoader = new PromptLoader({
 function getDatabaseType() {
   return promptLoader.getDatabaseType();
 }
+function getDbTypeFromToolId(toolId) {
+  if (!toolId) return getDatabaseType();
+  const prefix = toolId.split("-")[0]?.toLowerCase();
+  if (prefix === "mssql") return "mssql";
+  if (prefix === "postgres" || prefix === "postgresql") return "postgresql";
+  if (prefix === "mysql") return "mysql";
+  if (prefix === "snowflake") return "snowflake";
+  return getDatabaseType();
+}
 function convertTopToLimit(query) {
   if (!query || query.trim().length === 0) {
     return query;
@@ -10457,10 +10466,12 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
       if (streamBuffer.hasCallback()) {
         streamBuffer.write("\n\n\u{1F4CA} **Generating visualization components...**\n\n");
         streamBuffer.write("__TEXT_COMPLETE__COMPONENT_GENERATION_START__");
+        streamBuffer.flush();
       }
       const componentStreamCallback = streamBuffer.hasCallback() ? (component) => {
         const answerMarker = `__ANSWER_COMPONENT_START__${JSON.stringify(component)}__ANSWER_COMPONENT_END__`;
         streamBuffer.write(answerMarker);
+        streamBuffer.flush();
       } : void 0;
       const sanitizedTextResponse = textResponse.replace(
         /<DataTable>[\s\S]*?<\/DataTable>/g,
@@ -10802,7 +10813,10 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
       }
     }
   }
-  const finalTextResponse = responseMode === "text" && accumulatedStreamResponse ? accumulatedStreamResponse : textResponse;
+  let finalTextResponse = responseMode === "text" && accumulatedStreamResponse ? accumulatedStreamResponse : textResponse;
+  if (finalTextResponse) {
+    finalTextResponse = finalTextResponse.replace(/__TEXT_COMPLETE__[\s\S]*/g, "").replace(/__ANSWER_COMPONENT_START__[\s\S]*?__ANSWER_COMPONENT_END__/g, "").replace(/__COMPONENT_GENERATION_START__/g, "").trim();
+  }
   const uiBlock = new UIBlock(
     prompt,
     {},
@@ -14339,7 +14353,9 @@ Using this data, select the appropriate component and respond with ONLY the JSON
     }
     const etSql = finalComponent.props?.externalTool?.parameters?.sql;
     if (etSql && typeof etSql === "string") {
-      finalComponent.props.externalTool.parameters.sql = ensureQueryLimit(etSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT);
+      const etToolId = finalComponent.props?.externalTool?.toolId;
+      const etDbType = getDbTypeFromToolId(etToolId);
+      finalComponent.props.externalTool.parameters.sql = ensureQueryLimit(etSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, etDbType);
     }
     logger.info(`[DASH_COMP_REQ] Successfully picked component: ${finalComponent.name} (${finalComponent.type})`);
     if (finalComponent.props?.query && collections?.["database"]?.["execute"]) {
@@ -14477,7 +14493,7 @@ Fixed SQL query:`;
             fixedSql = fixedSql.replace(/^```\s*/i, "").replace(/\s*```$/i, "");
             const { query: validatedSql } = validateAndFixSqlQuery(fixedSql);
             if (validatedSql && validatedSql !== extToolSql) {
-              extToolSql = ensureQueryLimit(validatedSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT);
+              extToolSql = ensureQueryLimit(validatedSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, getDbTypeFromToolId(extToolId));
               logger.info(`[DASH_COMP_REQ] LLM provided SQL fix, retrying...`);
             } else {
               logger.warn(`[DASH_COMP_REQ] LLM returned same or empty query, stopping retries`);
@@ -15209,7 +15225,8 @@ Using this data, generate the complete report with multiple components. Respond
       };
       const etSql = finalComp.props?.externalTool?.parameters?.sql;
       if (etSql && typeof etSql === "string") {
-        finalComp.props.externalTool.parameters.sql = ensureQueryLimit(etSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT);
+        const etToolId = finalComp.props?.externalTool?.toolId;
+        finalComp.props.externalTool.parameters.sql = ensureQueryLimit(etSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, getDbTypeFromToolId(etToolId));
       }
       finalComponents.push(finalComp);
     }
@@ -15381,7 +15398,7 @@ Fixed SQL query:`;
           fixedSql = fixedSql.replace(/^```\s*/i, "").replace(/\s*```$/i, "");
           const { query: validatedSql } = validateAndFixSqlQuery(fixedSql);
           if (validatedSql && validatedSql !== extToolSql) {
-            extToolSql = ensureQueryLimit(validatedSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT);
+            extToolSql = ensureQueryLimit(validatedSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, getDbTypeFromToolId(extToolId));
             logger.info(`[REPORT_COMP_REQ] LLM provided SQL fix for ${comp.name}, retrying...`);
           } else {
             logger.warn(`[REPORT_COMP_REQ] LLM returned same or empty query for ${comp.name}`);