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

package/dist/index.js

@@ -14597,7 +14597,7 @@ Fixed SQL query:`;
 }
 
 // src/dashComp/create-filter.ts
-async function createFilterWithLLM(prompt, components, existingComponents, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, tools, dashCompModels) {
+async function createFilterWithLLM(prompt, components, existingComponents, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, tools, dashCompModels, collections) {
   const errors = [];
   try {
     const filterComponents = components.filter((c) => c.type.startsWith("Filter"));
@@ -14723,11 +14723,50 @@ async function createFilterWithLLM(prompt, components, existingComponents, anthr
     if (executedTools.length > 0) {
       logger.info(`[DASH_COMP_REQ:FILTER] Executed ${executedTools.length} external tool(s): ${executedTools.map((t) => t.name).join(", ")}`);
     }
+    const updatedComponents = result.updatedComponents || [];
+    for (const comp of updatedComponents) {
+      const extTool = comp.props?.externalTool;
+      if (!extTool?.parameters?.sql || !extTool?.toolId) continue;
+      let sql = extTool.parameters.sql;
+      const defaultParams = extTool.parameters.params || {};
+      const toolId = extTool.toolId;
+      const toolName = extTool.toolName;
+      sql = ensureQueryLimit(sql, DEFAULT_QUERY_LIMIT, MAX_COMPONENT_QUERY_LIMIT, getDbTypeFromToolId(toolId));
+      extTool.parameters.sql = sql;
+      if (collections?.["external-tools"]?.["execute"]) {
+        try {
+          const valResult = await collections["external-tools"]["execute"]({
+            toolId,
+            toolName,
+            sql,
+            ...defaultParams,
+            data: {}
+          });
+          if (valResult?.success === false || valResult?.error) {
+            const errMsg = valResult?.error || "Unknown error";
+            logger.warn(`[DASH_COMP_REQ:FILTER] SQL validation failed for component ${comp.id}: ${typeof errMsg === "string" ? errMsg : JSON.stringify(errMsg)}`);
+            continue;
+          }
+          const cacheKey2 = `${toolId}:${sql}`;
+          queryCache.set(cacheKey2, valResult?.data ?? valResult);
+          logger.info(`[DASH_COMP_REQ:FILTER] SQL validated and cached for component: ${comp.id}`);
+        } catch (err) {
+          const errMsg = err instanceof Error ? err.message : String(err);
+          logger.warn(`[DASH_COMP_REQ:FILTER] SQL validation error for component ${comp.id}: ${errMsg}`);
+        }
+      }
+      const { sql: _sql, ...restParams } = extTool.parameters;
+      const cacheKey = `${toolId}:${sql}`;
+      const cachedData = queryCache.get(cacheKey);
+      const queryId = queryCache.storeQuery(sql, cachedData);
+      extTool.parameters = { queryId, ...restParams };
+      logger.info(`[DASH_COMP_REQ:FILTER] Replaced SQL with queryId: ${queryId} for component: ${comp.id}`);
+    }
     return {
       success: true,
       data: {
         filterComponent: result.filterComponent,
-        updatedComponents: result.updatedComponents || [],
+        updatedComponents,
         filterBindings: result.filterBindings || {},
         reasoning: result.reasoning || "Filter created based on user prompt",
         executedTools: executedTools.length > 0 ? executedTools : void 0
@@ -14923,7 +14962,8 @@ var processDashCompRequest = async (data, components, _sendMessage, anthropicApi
       openaiApiKey,
       llmProviders,
       tools,
-      dashCompModels
+      dashCompModels,
+      collections
     );
   } else {
     llmResponse = await pickComponentWithLLM(