@superatomai/sdk-node 0.0.29-mds → 0.0.30-mds

package/dist/index.js

@@ -7487,7 +7487,13 @@ Even if a table appears in the detailed schema above, search_schema returns samp
    */
   extractSourceType() {
     const match = this.tool.id.match(/^(\w+)-[a-f0-9]+_/);
-    return match ? match[1] : "unknown";
+    if (!match) return "unknown";
+    const raw = match[1].toLowerCase();
+    if (raw.startsWith("mssql")) return "mssql";
+    if (raw.startsWith("postgres")) return "postgres";
+    if (raw.startsWith("mysql")) return "mysql";
+    if (raw.startsWith("snowflake")) return "snowflake";
+    return raw;
   }
   /**
    * Build an empty ExecutedToolInfo for error cases.
@@ -8282,8 +8288,10 @@ async function validateSingleExternalToolQuery(component, collections, executedT
   let currentSql = compProps?.externalTool?.parameters?.sql;
   if (!toolId || !currentSql) return component;
   const executedTool = executedTools.find((t) => t.id === toolId);
-  const dbType = executedTool?.sourceType || "postgresql";
-  currentSql = ensureQueryLimit(currentSql, 10, MAX_AGENT_QUERY_LIMIT, dbType);
+  const dbType = executedTool?.sourceType;
+  if (dbType) {
+    currentSql = ensureQueryLimit(currentSql, 10, MAX_AGENT_QUERY_LIMIT, dbType);
+  }
   let attempts = 0;
   while (attempts < MAX_QUERY_VALIDATION_RETRIES) {
     attempts++;
@@ -8333,7 +8341,7 @@ async function validateSingleExternalToolQuery(component, collections, executedT
           apiKey
         );
         if (fixedSql && fixedSql !== currentSql) {
-          currentSql = ensureQueryLimit(fixedSql, 10, MAX_AGENT_QUERY_LIMIT, dbType);
+          currentSql = dbType ? ensureQueryLimit(fixedSql, 10, MAX_AGENT_QUERY_LIMIT, dbType) : fixedSql;
           logger.info(`[AgentComponentGen] LLM provided fix for ${component.name}, retrying...`);
         } else {
           logger.warn(`[AgentComponentGen] LLM returned same or empty query, stopping retries`);