@superatomai/sdk-node 0.0.25-mds → 0.0.26-mds

package/dist/index.mjs

@@ -7057,11 +7057,12 @@ ${queryDisplay}
           }
           await streamDelay();
         }
+        let queryStartTime = Date.now();
         try {
           if (this.streamBuffer.hasCallback()) {
             this.streamBuffer.write(`__QUERY_TIMER_START_Executing query__`);
           }
-          const queryStartTime = Date.now();
+          queryStartTime = Date.now();
           const result = await this.tool.fn(cappedInput);
           if (this.streamBuffer.hasCallback()) {
             const querySeconds = ((Date.now() - queryStartTime) / 1e3).toFixed(1);
@@ -7131,6 +7132,10 @@ ${formatted}`;
           const errorMsg = execError instanceof Error ? execError.message : typeof execError === "object" && execError !== null ? execError.message || execError.error || JSON.stringify(execError) : String(execError);
           logger.warn(`[SourceAgent:${this.tool.name}] Tool execution failed (attempt ${this.attempts}/${this.config.maxRetries}): ${errorMsg}`);
           if (this.streamBuffer.hasCallback()) {
+            const querySeconds = ((Date.now() - queryStartTime) / 1e3).toFixed(1);
+            this.streamBuffer.write(`__QUERY_TIMER_DONE_${querySeconds}__
+
+`);
             this.streamBuffer.write(`\u274C **Query failed:** ${errorMsg}
 
 `);
@@ -7141,10 +7146,11 @@ Analyze the error and try again with a corrected query.`;
         }
       };
       const hasSchemaSearch = !!schemaSearchFn;
-      const queryIterations = this.config.maxRetries + 1;
-      const schemaIterations = hasSchemaSearch ? 3 : 0;
-      const responseIterations = 2;
-      const maxIterations = queryIterations + schemaIterations + responseIterations;
+      const totalQueryAttempts = this.config.maxRetries + 2;
+      const perQueryCost = hasSchemaSearch ? 2 : 1;
+      const extraSearches = hasSchemaSearch ? 2 : 0;
+      const responseIterations = 1;
+      const maxIterations = totalQueryAttempts * perQueryCost + extraSearches + responseIterations;
       await LLM.streamWithTools(
         { sys: prompts.system, user: prompts.user },
         tools,
@@ -7243,12 +7249,18 @@ Analyze the error and try again with a corrected query.`;
     if (hasSchemaSearch && schemaTier === "very_large") {
       schemaSearchInstructions = `## Schema Search \u2014 REQUIRED
 This source has a very large schema. The schema above shows only table names \u2014 no column details.
-**You MUST use search_schema BEFORE writing any query.** Do NOT guess column names.
+**You MUST use search_schema BEFORE writing any query.** Do NOT guess or fabricate column names.
 1. FIRST: Call search_schema with keywords related to the requested data
 2. Review the returned column details, types, and sample values carefully
 3. THEN write your SQL query using the EXACT column names from the search results
-You may search multiple times with different keywords if the first search doesn't find what you need.
-**Never write a query without searching first \u2014 guessing column names wastes time on failed queries.**`;
+
+**CRITICAL: Column names in this database may NOT follow predictable patterns.**
+- Do NOT assume column name prefixes based on table name abbreviations
+- Do NOT invent column names by combining a table alias with a guessed suffix
+- ONLY use column names that appeared in search_schema results \u2014 copy them exactly
+- If you need columns you haven't searched for, call search_schema again before writing the query
+
+You may search multiple times with different keywords if the first search doesn't find what you need.`;
     } else if (hasSchemaSearch) {
       schemaSearchInstructions = `## Schema Search \u2014 REQUIRED
 This source has a large schema. The top tables are shown in detail above, but not all tables have full column details.
@@ -7704,8 +7716,8 @@ var DEFAULT_AGENT_CONFIG = {
   // will use the provider's default model
   sourceAgentModel: "",
   // will use the provider's default model
-  maxRetries: 1,
-  // 1 retry = 2 total query attempts (1 initial + 1 retry for SQL errors)
+  maxRetries: 2,
+  // 2 retries = 3 total query attempts (1 initial + 2 retries for SQL errors)
   maxIterations: 8
   // schema search (2-3) + query attempts (2) + LLM responses + final
 };
@@ -10388,7 +10400,8 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
       conversationHistory,
       streamBuffer.hasCallback() ? (chunk) => streamBuffer.write(chunk) : void 0
     );
-    const textResponse = streamBuffer.getFullText() || agentResponse.text || "I apologize, but I was unable to generate a response.";
+    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, "");
     streamBuffer.flush();
     const hasExecutedTools = agentResponse.executedTools.length > 0;
     let matchedComponents = [];
@@ -10397,16 +10410,17 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
     let actions = [];
     if (!hasExecutedTools) {
       logger.info(`[AgentFlow] No tools executed \u2014 general question, wrapping in DynamicMarkdownBlock`);
+      const mainAgentText = agentResponse.text || textResponse;
       const nextQuestions = await llmInstance.generateNextQuestions(
         prompt,
         null,
         void 0,
         apiKey,
         conversationHistory,
-        textResponse
+        mainAgentText
       );
       actions = convertQuestionsToActions(nextQuestions);
-      const markdownContent = textResponse.replace(/<DashboardComponents>[\s\S]*?<\/DashboardComponents>/g, "").trim();
+      const markdownContent = mainAgentText.replace(/<DashboardComponents>[\s\S]*?<\/DashboardComponents>/g, "").trim();
       matchedComponents = [{
         id: "dynamic-markdown-block",
         name: "DynamicMarkdownBlock",