@superatomai/sdk-node 0.0.35-mds → 0.0.36-mds

package/dist/index.js

@@ -10611,11 +10611,33 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
         /<DataTable>[\s\S]*?<\/DataTable>/g,
         "<DataTable>[Data preview removed - for table components, REUSE the exact SQL from EXECUTED_TOOLS parameters. Do NOT write a new query or embed data in props.]</DataTable>"
       );
+      const componentExecutedTools = [...agentResponse.executedTools];
+      const federationTool = agentTools.find((t) => t.id === "federation_query");
+      if (federationTool && agentResponse.executedTools.length >= 2) {
+        const sourceToolIds = new Set(agentResponse.executedTools.map((t) => t.id));
+        const hasMultipleSources = sourceToolIds.size >= 2;
+        if (hasMultipleSources) {
+          componentExecutedTools.push({
+            id: "federation_query",
+            name: "Cross-Source Query",
+            params: { sql: "DuckDB SQL with schema-qualified table names" },
+            result: {
+              _totalRecords: 0,
+              _recordsShown: 0,
+              _sampleData: []
+            },
+            outputSchema: federationTool.outputSchema,
+            sourceSchema: federationTool.fullSchema || federationTool.description,
+            sourceType: "duckdb"
+          });
+          logger.info(`[AgentFlow] Injected federation_query tool for cross-source component generation`);
+        }
+      }
       const matchResult = await generateAgentComponents({
         analysisContent: cleanAnalysis,
         components,
         userPrompt: prompt,
-        executedTools: agentResponse.executedTools,
+        executedTools: componentExecutedTools,
         collections,
         apiKey,
         componentStreamCallback,