@superatomai/sdk-node 0.0.68 → 0.0.70

package/dist/index.js

@@ -3557,7 +3557,7 @@ function convertTopToLimit(query) {
   let modifiedQuery = query.replace(/\bSELECT\s+TOP\s+\(?\d+\)?\s*/gi, "SELECT ");
   return modifiedQuery;
 }
-function ensureQueryLimit(query, defaultLimit = 10, maxLimit = 100, dbType) {
+function ensureQueryLimit(query, defaultLimit = 10, maxLimit = 10, dbType) {
   if (!query || query.trim().length === 0) {
     return query;
   }
@@ -6189,7 +6189,7 @@ Fixed SQL query:`;
       {
         model: this.getModelForTask("simple"),
         maxTokens: 2048,
-        temperature: 0.1,
+        temperature: 0,
         apiKey: this.getApiKey(apiKey)
       }
     );
@@ -6435,12 +6435,13 @@ ${executedToolsText}`);
                                 },
                                 apiKey
                               );
+                              const limitedFixedQuery = ensureQueryLimit(fixedQueryStr, this.defaultLimit, 10);
                               if (typeof currentQuery === "string") {
-                                currentQuery = fixedQueryStr;
+                                currentQuery = limitedFixedQuery;
                               } else {
-                                currentQuery = { ...currentQuery, sql: fixedQueryStr };
+                                currentQuery = { ...currentQuery, sql: limitedFixedQuery };
                               }
-                              currentQueryStr = fixedQueryStr;
+                              currentQueryStr = limitedFixedQuery;
                               logger.info(`[${this.getProviderName()}] LLM provided fixed query for answer component, retrying...`);
                             } catch (fixError) {
                               const fixErrorMsg = fixError instanceof Error ? fixError.message : String(fixError);
@@ -6476,7 +6477,7 @@ ${executedToolsText}`);
         {
           model: this.getModelForTask("complex"),
           maxTokens: 8192,
-          temperature: 0.2,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey),
           partial: partialCallback
         },
@@ -6619,6 +6620,26 @@ ${executedToolsText}`);
             }
           }
         }
+        if (cleanedProps.query) {
+          if (typeof cleanedProps.query === "string") {
+            cleanedProps.query = ensureQueryLimit(
+              cleanedProps.query,
+              this.defaultLimit,
+              10
+              // maxLimit - enforce maximum of 10 rows for component queries
+            );
+          } else if (cleanedProps.query?.sql) {
+            cleanedProps.query = {
+              ...cleanedProps.query,
+              sql: ensureQueryLimit(
+                cleanedProps.query.sql,
+                this.defaultLimit,
+                10
+                // maxLimit - enforce maximum of 10 rows for component queries
+              )
+            };
+          }
+        }
         if (cleanedProps.query && cleanedProps.externalTool) {
           logger.info(`[${this.getProviderName()}] Both query and externalTool exist, keeping both - frontend will decide`);
         }
@@ -6739,11 +6760,12 @@ ${executedToolsText}`);
           );
           if (fixedQueryStr && fixedQueryStr !== currentQueryStr) {
             logger.info(`[${this.getProviderName()}] Received fixed query for ${component.name}, retrying...`);
-            currentQueryStr = fixedQueryStr;
+            const limitedFixedQuery = ensureQueryLimit(fixedQueryStr, this.defaultLimit, 10);
+            currentQueryStr = limitedFixedQuery;
             if (typeof currentQuery === "string") {
-              currentQuery = fixedQueryStr;
+              currentQuery = limitedFixedQuery;
             } else {
-              currentQuery = { ...currentQuery, sql: fixedQueryStr };
+              currentQuery = { ...currentQuery, sql: limitedFixedQuery };
             }
           } else {
             logger.warn(`[${this.getProviderName()}] LLM returned same or empty query, stopping retries`);
@@ -6871,7 +6893,7 @@ ${executedToolsText}`);
         {
           model: this.getModelForTask("simple"),
           maxTokens: 1500,
-          temperature: 0.2,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey)
         },
         true
@@ -6944,7 +6966,7 @@ ${executedToolsText}`);
         {
           model: this.getModelForTask("complex"),
           maxTokens: 8192,
-          temperature: 0.2,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey)
         },
         true
@@ -7265,7 +7287,7 @@ ${executedToolsText}`);
           let sql = toolInput.sql;
           const params = toolInput.params || {};
           const reasoning = toolInput.reasoning;
-          sql = ensureQueryLimit(sql, 10, 100);
+          sql = ensureQueryLimit(sql, 10, 10);
           const queryKey = sql.toLowerCase().replace(/\s+/g, " ").trim();
           const attempts = (queryAttempts.get(queryKey) || 0) + 1;
           queryAttempts.set(queryKey, attempts);
@@ -7529,7 +7551,7 @@ ${errorMsg}
         {
           model: this.getModelForTask("complex"),
           maxTokens: 4e3,
-          temperature: 0.7,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey),
           partial: wrappedStreamCallback
           // Pass the wrapped streaming callback to LLM
@@ -7939,7 +7961,7 @@ ${errorMsg}
         {
           model: this.getModelForTask("simple"),
           maxTokens: 1200,
-          temperature: 0.7,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey)
         },
         true