@superatomai/sdk-node 0.0.22-mds → 0.0.23-mds

package/dist/index.js

@@ -5017,7 +5017,8 @@ var TOOL_TRACKING_MAX_ROWS = 5;
 var TOOL_TRACKING_MAX_CHARS = 200;
 var TOOL_TRACKING_SAMPLE_ROWS = 3;
 var DEFAULT_QUERY_LIMIT = 10;
-var MAX_COMPONENT_QUERY_LIMIT = 10;
+var MAX_AGENT_QUERY_LIMIT = 10;
+var MAX_COMPONENT_QUERY_LIMIT = 100;
 var EXACT_MATCH_SIMILARITY_THRESHOLD = 0.99;
 var DEFAULT_CONVERSATION_SIMILARITY_THRESHOLD = 0.8;
 var MAX_TOOL_CALLING_ITERATIONS = 20;
@@ -7806,11 +7807,11 @@ function validateAndCleanQuery(query, config) {
     }
   }
   if (typeof cleanedQuery === "string") {
-    const limitedQuery = ensureQueryLimit(cleanedQuery, config.defaultLimit, MAX_COMPONENT_QUERY_LIMIT);
+    const limitedQuery = ensureQueryLimit(cleanedQuery, config.defaultLimit, MAX_AGENT_QUERY_LIMIT);
     if (limitedQuery !== cleanedQuery) wasModified = true;
     cleanedQuery = limitedQuery;
   } else if (cleanedQuery?.sql) {
-    const limitedSql = ensureQueryLimit(cleanedQuery.sql, config.defaultLimit, MAX_COMPONENT_QUERY_LIMIT);
+    const limitedSql = ensureQueryLimit(cleanedQuery.sql, config.defaultLimit, MAX_AGENT_QUERY_LIMIT);
     if (limitedSql !== cleanedQuery.sql) wasModified = true;
     cleanedQuery = { ...cleanedQuery, sql: limitedSql };
   }
@@ -8207,7 +8208,7 @@ async function validateSingleExternalToolQuery(component, collections, executedT
   if (!toolId || !currentSql) return component;
   const executedTool = executedTools.find((t) => t.id === toolId);
   const dbType = executedTool?.sourceType || "postgresql";
-  currentSql = ensureQueryLimit(currentSql, 10, MAX_COMPONENT_QUERY_LIMIT, dbType);
+  currentSql = ensureQueryLimit(currentSql, 10, MAX_AGENT_QUERY_LIMIT, dbType);
   let attempts = 0;
   while (attempts < MAX_QUERY_VALIDATION_RETRIES) {
     attempts++;
@@ -8257,7 +8258,7 @@ async function validateSingleExternalToolQuery(component, collections, executedT
           apiKey
         );
         if (fixedSql && fixedSql !== currentSql) {
-          currentSql = ensureQueryLimit(fixedSql, 10, MAX_COMPONENT_QUERY_LIMIT, dbType);
+          currentSql = ensureQueryLimit(fixedSql, 10, MAX_AGENT_QUERY_LIMIT, dbType);
           logger.info(`[AgentComponentGen] LLM provided fix for ${component.name}, retrying...`);
         } else {
           logger.warn(`[AgentComponentGen] LLM returned same or empty query, stopping retries`);
@@ -8613,7 +8614,7 @@ Fixed SQL query:`;
           );
           if (fixedQueryStr && fixedQueryStr !== currentQueryStr) {
             logger.info(`[${this.config.providerName}] Received fixed query for ${component.name}, retrying...`);
-            const limitedFixedQuery = ensureQueryLimit(fixedQueryStr, this.config.defaultLimit, MAX_COMPONENT_QUERY_LIMIT);
+            const limitedFixedQuery = ensureQueryLimit(fixedQueryStr, this.config.defaultLimit, MAX_AGENT_QUERY_LIMIT);
             currentQueryStr = limitedFixedQuery;
             if (typeof currentQuery === "string") {
               currentQuery = limitedFixedQuery;
@@ -8745,7 +8746,7 @@ var ToolExecutorService = class {
     const params = toolInput.params || {};
     const reasoning = toolInput.reasoning;
     const { streamBuffer, collections, providerName } = this.config;
-    sql = ensureQueryLimit(sql, MAX_COMPONENT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT);
+    sql = ensureQueryLimit(sql, MAX_AGENT_QUERY_LIMIT, MAX_AGENT_QUERY_LIMIT);
     const queryKey = sql.toLowerCase().replace(/\s+/g, " ").trim();
     const attempts = (this.queryAttempts.get(queryKey) || 0) + 1;
     this.queryAttempts.set(queryKey, attempts);
@@ -8941,11 +8942,11 @@ Please try rephrasing your request or contact support.
         await streamDelay();
       }
       const cappedToolInput = { ...toolInput };
-      if (cappedToolInput.limit !== void 0 && cappedToolInput.limit > MAX_COMPONENT_QUERY_LIMIT) {
-        logger.info(`[${providerName}] Capping external tool limit from ${cappedToolInput.limit} to ${MAX_COMPONENT_QUERY_LIMIT}`);
-        cappedToolInput.limit = MAX_COMPONENT_QUERY_LIMIT;
+      if (cappedToolInput.limit !== void 0 && cappedToolInput.limit > MAX_AGENT_QUERY_LIMIT) {
+        logger.info(`[${providerName}] Capping external tool limit from ${cappedToolInput.limit} to ${MAX_AGENT_QUERY_LIMIT}`);
+        cappedToolInput.limit = MAX_AGENT_QUERY_LIMIT;
       } else if (cappedToolInput.limit === void 0) {
-        cappedToolInput.limit = MAX_COMPONENT_QUERY_LIMIT;
+        cappedToolInput.limit = MAX_AGENT_QUERY_LIMIT;
       }
       const result = await withProgressHeartbeat(
         () => externalTool.fn(cappedToolInput),
@@ -9266,10 +9267,10 @@ ${executedToolsText}`);
                   let answerQuery = answerComponent.props?.query;
                   if (answerQuery) {
                     if (typeof answerQuery === "string") {
-                      answerQuery = ensureQueryLimit(answerQuery, this.defaultLimit, MAX_COMPONENT_QUERY_LIMIT);
+                      answerQuery = ensureQueryLimit(answerQuery, this.defaultLimit, MAX_AGENT_QUERY_LIMIT);
                     } else if (answerQuery?.sql) {
                       const queryObj = answerQuery;
-                      answerQuery = { ...queryObj, sql: ensureQueryLimit(queryObj.sql, this.defaultLimit, MAX_COMPONENT_QUERY_LIMIT) };
+                      answerQuery = { ...queryObj, sql: ensureQueryLimit(queryObj.sql, this.defaultLimit, MAX_AGENT_QUERY_LIMIT) };
                     }
                     answerComponent.props.query = answerQuery;
                   }
@@ -9310,7 +9311,7 @@ ${executedToolsText}`);
                                 },
                                 apiKey
                               );
-                              const limitedFixedQuery = ensureQueryLimit(fixedQueryStr, this.defaultLimit, MAX_COMPONENT_QUERY_LIMIT);
+                              const limitedFixedQuery = ensureQueryLimit(fixedQueryStr, this.defaultLimit, MAX_AGENT_QUERY_LIMIT);
                               if (typeof currentQuery === "string") {
                                 currentQuery = limitedFixedQuery;
                               } else {