@superatomai/sdk-node 0.0.68 → 0.0.70

package/dist/index.mjs

@@ -3499,7 +3499,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;
   }
@@ -6131,7 +6131,7 @@ Fixed SQL query:`;
       {
         model: this.getModelForTask("simple"),
         maxTokens: 2048,
-        temperature: 0.1,
+        temperature: 0,
         apiKey: this.getApiKey(apiKey)
       }
     );
@@ -6377,12 +6377,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);
@@ -6418,7 +6419,7 @@ ${executedToolsText}`);
         {
           model: this.getModelForTask("complex"),
           maxTokens: 8192,
-          temperature: 0.2,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey),
           partial: partialCallback
         },
@@ -6561,6 +6562,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`);
         }
@@ -6681,11 +6702,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`);
@@ -6813,7 +6835,7 @@ ${executedToolsText}`);
         {
           model: this.getModelForTask("simple"),
           maxTokens: 1500,
-          temperature: 0.2,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey)
         },
         true
@@ -6886,7 +6908,7 @@ ${executedToolsText}`);
         {
           model: this.getModelForTask("complex"),
           maxTokens: 8192,
-          temperature: 0.2,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey)
         },
         true
@@ -7207,7 +7229,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);
@@ -7471,7 +7493,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
@@ -7881,7 +7903,7 @@ ${errorMsg}
         {
           model: this.getModelForTask("simple"),
           maxTokens: 1200,
-          temperature: 0.7,
+          temperature: 0,
           apiKey: this.getApiKey(apiKey)
         },
         true