@superatomai/sdk-node 0.0.32-mds → 0.0.33-mds

package/dist/index.js

@@ -7903,12 +7903,13 @@ function validateAndCleanQuery(query, config) {
       cleanedQuery = { ...cleanedQuery, sql: fixedQuery };
     }
   }
+  const maxLimit = config.maxLimit ?? MAX_AGENT_QUERY_LIMIT;
   if (typeof cleanedQuery === "string") {
-    const limitedQuery = ensureQueryLimit(cleanedQuery, config.defaultLimit, MAX_AGENT_QUERY_LIMIT);
+    const limitedQuery = ensureQueryLimit(cleanedQuery, config.defaultLimit, maxLimit);
     if (limitedQuery !== cleanedQuery) wasModified = true;
     cleanedQuery = limitedQuery;
   } else if (cleanedQuery?.sql) {
-    const limitedSql = ensureQueryLimit(cleanedQuery.sql, config.defaultLimit, MAX_AGENT_QUERY_LIMIT);
+    const limitedSql = ensureQueryLimit(cleanedQuery.sql, config.defaultLimit, maxLimit);
     if (limitedSql !== cleanedQuery.sql) wasModified = true;
     cleanedQuery = { ...cleanedQuery, sql: limitedSql };
   }
@@ -8064,7 +8065,7 @@ ${executedToolsText}`);
       const cleanedProps = processComponentProps(
         mc.props,
         executedTools,
-        { providerName: "AgentComponentGen", defaultLimit: 10 }
+        { providerName: "AgentComponentGen", defaultLimit: DEFAULT_QUERY_LIMIT, maxLimit: MAX_COMPONENT_QUERY_LIMIT }
       );
       return {
         ...originalComponent,
@@ -8203,10 +8204,16 @@ function tryStreamAnswerComponent(text, components, executedTools, collections,
       props: { ...original.props, ...answerData.props }
     };
     const answerProps = answerComponent.props;
-    const sql = answerProps?.externalTool?.parameters?.sql;
+    let sql = answerProps?.externalTool?.parameters?.sql;
     if (sql && collections?.["external-tools"]?.["execute"]) {
       const toolId = answerProps.externalTool.toolId;
       const toolName = answerProps.externalTool.toolName;
+      const executedTool = executedTools.find((t) => t.id === toolId);
+      const dbType = executedTool?.sourceType;
+      if (dbType) {
+        sql = ensureQueryLimit(sql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, dbType);
+        answerProps.externalTool.parameters.sql = sql;
+      }
       (async () => {
         try {
           const result = await collections["external-tools"]["execute"]({
@@ -8309,7 +8316,7 @@ async function validateSingleExternalToolQuery(component, collections, executedT
   const executedTool = executedTools.find((t) => t.id === toolId);
   const dbType = executedTool?.sourceType;
   if (dbType) {
-    currentSql = ensureQueryLimit(currentSql, 10, MAX_AGENT_QUERY_LIMIT, dbType);
+    currentSql = ensureQueryLimit(currentSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, dbType);
   }
   let attempts = 0;
   while (attempts < MAX_QUERY_VALIDATION_RETRIES) {
@@ -8360,7 +8367,7 @@ async function validateSingleExternalToolQuery(component, collections, executedT
           apiKey
         );
         if (fixedSql && fixedSql !== currentSql) {
-          currentSql = dbType ? ensureQueryLimit(fixedSql, 10, MAX_AGENT_QUERY_LIMIT, dbType) : fixedSql;
+          currentSql = dbType ? ensureQueryLimit(fixedSql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_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`);