@superatomai/sdk-node 0.0.27-mds → 0.0.28-mds

package/dist/index.d.mts

@@ -2637,8 +2637,10 @@ interface SourceAgentResult {
     data: any[];
     /** Metadata about the query execution */
     metadata: SourceAgentMetadata;
-    /** Tool execution info (reused for component generation) */
+    /** Tool execution info for the last successful query (backward compat) */
     executedTool: ExecutedToolInfo;
+    /** All successful tool executions (primary + follow-up queries) */
+    allExecutedTools?: ExecutedToolInfo[];
     /** Error message if failed */
     error?: string;
 }

package/dist/index.d.ts

@@ -2637,8 +2637,10 @@ interface SourceAgentResult {
     data: any[];
     /** Metadata about the query execution */
     metadata: SourceAgentMetadata;
-    /** Tool execution info (reused for component generation) */
+    /** Tool execution info for the last successful query (backward compat) */
     executedTool: ExecutedToolInfo;
+    /** All successful tool executions (primary + follow-up queries) */
+    allExecutedTools?: ExecutedToolInfo[];
     /** Error message if failed */
     error?: string;
 }

package/dist/index.js

@@ -1362,13 +1362,15 @@ var Thread = class {
         if (metadata.description) {
           parts.push(`Description: ${metadata.description}`);
         }
-        if (metadata.props) {
-          parts.push(`Props: ${JSON.stringify(metadata.props)}`);
+        const layoutTitle = metadata.props?.config?.title;
+        if (layoutTitle) {
+          parts.push(`Dashboard Title: ${layoutTitle}`);
         }
         responseParts.push(parts.join("\n"));
       }
       if (hasTextResponse) {
-        responseParts.push(textResponse);
+        const trimmedText = textResponse.length > 500 ? textResponse.substring(0, 500) + "...[truncated]" : textResponse;
+        responseParts.push(trimmedText);
       }
       if (responseParts.length > 0) {
         assistantResponse = responseParts.join("\n");
@@ -7056,6 +7058,7 @@ var SourceAgent = class {
       logger.logLLMPrompt(`sourceAgent:${this.tool.name}`, "user", prompts.user);
       const tools = this.buildToolDefinitions();
       let executedTool = null;
+      let allExecutedTools = [];
       let resultData = [];
       let queryExecuted;
       let totalRowsMatched = 0;
@@ -7184,6 +7187,7 @@ Analyze the error and try again with a corrected query.`;
             sourceSchema: this.tool.description,
             sourceType: this.extractSourceType()
           };
+          allExecutedTools.push(executedTool);
           const formatted = typeof formattedResult === "string" ? formattedResult : JSON.stringify(formattedResult);
           const followUpNote = successfulQueries < 2 ? "You may make ONE follow-up query if this data is incomplete for the intent. Otherwise, STOP." : "Maximum queries reached. Use the data you have.";
           return `\u2705 Query executed successfully. ${resultData.length} rows returned (${totalRowsMatched} total matched). ${followUpNote}
@@ -7255,7 +7259,8 @@ Analyze the error and try again with a corrected query.`;
           queryExecuted,
           executionTimeMs
         },
-        executedTool
+        executedTool,
+        allExecutedTools
       };
     } catch (error) {
       const executionTimeMs = Date.now() - startTime;
@@ -7511,7 +7516,11 @@ var MainAgent = class {
       const result = await sourceAgent.execute(sourceInput);
       sourceResults.push(result);
       if (result.success) {
-        executedTools.push(result.executedTool);
+        if (result.allExecutedTools && result.allExecutedTools.length > 0) {
+          executedTools.push(...result.allExecutedTools);
+        } else {
+          executedTools.push(result.executedTool);
+        }
       }
       return this.formatResultForMainAgent(result);
     };
@@ -8053,12 +8062,24 @@ function formatExecutedTools(executedTools) {
    Fields:
 ${fieldsText}`;
     }
+    let sampleDataText = "";
+    const sampleData = tool.result?._sampleData;
+    if (Array.isArray(sampleData) && sampleData.length > 0) {
+      const sampleFields = Object.keys(sampleData[0]);
+      sampleDataText = `
+   \u{1F511} RESULT FIELDS: ${sampleFields.join(", ")}`;
+      try {
+        sampleDataText += `
+   \u{1F4C4} SAMPLE ROW: ${JSON.stringify(sampleData[0])}`;
+      } catch {
+      }
+    }
     return `${idx + 1}. **${tool.name}**
    toolId: "${tool.id}"
    toolName: "${tool.name}"
    parameters: ${JSON.stringify(tool.params || {})}
    recordCount: ${recordCount} rows returned${metadataText}
-   outputSchema: ${outputSchemaText}${fieldNamesList}`;
+   outputSchema: ${outputSchemaText}${fieldNamesList}${sampleDataText}`;
   }).join("\n\n");
 }
 function tryStreamAnswerComponent(text, components, executedTools, collections, apiKey, callback) {
@@ -10453,7 +10474,7 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
       streamBuffer.hasCallback() ? (chunk) => streamBuffer.write(chunk) : void 0
     );
     const rawText = streamBuffer.getFullText() || agentResponse.text || "I apologize, but I was unable to generate a response.";
-    const textResponse = rawText.replace(/__SB_\w+_(?:START|MSG)__/g, "").replace(/__SB_END__/g, "").replace(/__QUERY_TIMER_START_[^_]*__/g, "").replace(/__QUERY_TIMER_DONE_[\d.]+__/g, "").replace(/__TEXT_COMPLETE__COMPONENT_GENERATION_START__/g, "");
+    const textResponse = rawText.replace(/_?_?SB_END_?_?/g, "").replace(/__SB_\w+_(?:START|MSG)_?_?/g, "").replace(/__QUERY_TIMER_START_[^_]*__/g, "").replace(/__QUERY_TIMER_DONE_[\d.]+__/g, "").replace(/__TEXT_COMPLETE__COMPONENT_GENERATION_START__/g, "").replace(/\[COMPLEXITY:\s*(?:simple|medium|complex)\]/gi, "");
     streamBuffer.flush();
     const hasExecutedTools = agentResponse.executedTools.length > 0;
     let matchedComponents = [];
@@ -10496,12 +10517,12 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
         streamBuffer.write(answerMarker);
         streamBuffer.flush();
       } : void 0;
-      const sanitizedTextResponse = textResponse.replace(
+      const cleanAnalysis = (agentResponse.text || textResponse).replace(/__SB_\w+_(?:START|MSG)_?_?/g, "").replace(/_?_?SB_END_?_?/g, "").replace(/__QUERY_TIMER_START_[^_]*__/g, "").replace(/__QUERY_TIMER_DONE_[\d.]+__/g, "").replace(/\[COMPLEXITY:\s*(?:simple|medium|complex)\]/gi, "").replace(
         /<DataTable>[\s\S]*?<\/DataTable>/g,
-        "<DataTable>[Data preview removed - for table components, REUSE the exact SQL query shown above (the one that returned these results). Do NOT write a new query or embed data in props.]</DataTable>"
+        "<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 matchResult = await generateAgentComponents({
-        analysisContent: sanitizedTextResponse,
+        analysisContent: cleanAnalysis,
         components,
         userPrompt: prompt,
         executedTools: agentResponse.executedTools,