@superatomai/sdk-node 0.0.54 → 0.0.56

package/dist/index.js

@@ -5405,6 +5405,19 @@ var LLM = class {
 // src/userResponse/base-llm.ts
 init_logger();
 
+// src/utils/datetime.ts
+function getCurrentDateTimeForPrompt() {
+  return (/* @__PURE__ */ new Date()).toLocaleString("en-US", {
+    weekday: "long",
+    year: "numeric",
+    month: "long",
+    day: "numeric",
+    hour: "2-digit",
+    minute: "2-digit",
+    timeZoneName: "short"
+  });
+}
+
 // src/userResponse/knowledge-base.ts
 init_logger();
 var getKnowledgeBase = async ({
@@ -5885,7 +5898,8 @@ ${fieldsText}`;
         SCHEMA_DOC: schemaDoc,
         DATABASE_RULES: databaseRules,
         DEFERRED_TOOLS: deferredToolsText,
-        EXECUTED_TOOLS: executedToolsText
+        EXECUTED_TOOLS: executedToolsText,
+        CURRENT_DATETIME: getCurrentDateTimeForPrompt()
       });
       logger.debug(`[${this.getProviderName()}] Loaded match-text-components prompts`);
       const systemPromptStr = Array.isArray(prompts.system) ? prompts.system.join("\n") : prompts.system;
@@ -6160,7 +6174,8 @@ ${executedToolsText}`);
         USER_PROMPT: userPrompt,
         CONVERSATION_HISTORY: conversationHistory || "No previous conversation",
         AVAILABLE_TOOLS: availableToolsDoc,
-        SCHEMA_DOC: schemaDoc || "No database schema available"
+        SCHEMA_DOC: schemaDoc || "No database schema available",
+        CURRENT_DATETIME: getCurrentDateTimeForPrompt()
       });
       const systemPrompt = Array.isArray(prompts.system) ? prompts.system.join("\n") : prompts.system;
       const userPromptText = Array.isArray(prompts.user) ? prompts.user.join("\n") : prompts.user;
@@ -6227,7 +6242,8 @@ ${executedToolsText}`);
         COMPONENT_PROPS: JSON.stringify(component.props, null, 2),
         CACHED_TEXT_RESPONSE: cachedTextResponse || "No cached text response available",
         SCHEMA_DOC: schemaDoc || "No schema available",
-        DATABASE_RULES: databaseRules
+        DATABASE_RULES: databaseRules,
+        CURRENT_DATETIME: getCurrentDateTimeForPrompt()
       });
       const result = await LLM.stream(
         {
@@ -6359,7 +6375,8 @@ ${executedToolsText}`);
         SCHEMA_DOC: schemaDoc,
         DATABASE_RULES: databaseRules,
         KNOWLEDGE_BASE_CONTEXT: knowledgeBaseContext || "No additional knowledge base context available.",
-        AVAILABLE_EXTERNAL_TOOLS: availableToolsDoc
+        AVAILABLE_EXTERNAL_TOOLS: availableToolsDoc,
+        CURRENT_DATETIME: getCurrentDateTimeForPrompt()
       });
       const sysPrompt = Array.isArray(prompts.system) ? prompts.system.join("\n") : prompts.system;
       const usrPrompt = Array.isArray(prompts.user) ? prompts.user.join("\n") : prompts.user;
@@ -6481,6 +6498,32 @@ ${executedToolsText}`);
         fullStreamedText += chunk;
         streamCallback(chunk);
       } : void 0;
+      const withProgressHeartbeat = async (operation, progressMessage, intervalMs = 1e3) => {
+        if (!wrappedStreamCallback) {
+          return operation();
+        }
+        const startTime = Date.now();
+        let dotCount = 0;
+        const maxDots = 3;
+        wrappedStreamCallback(`\u23F3 ${progressMessage}`);
+        const heartbeatInterval = setInterval(() => {
+          const elapsedSeconds = Math.floor((Date.now() - startTime) / 1e3);
+          dotCount = (dotCount + 1) % (maxDots + 1);
+          const dots = ".".repeat(dotCount || 1);
+          if (elapsedSeconds >= 2) {
+            wrappedStreamCallback(`${dots} (${elapsedSeconds}s)`);
+          } else {
+            wrappedStreamCallback(dots);
+          }
+        }, intervalMs);
+        try {
+          const result2 = await operation();
+          return result2;
+        } finally {
+          clearInterval(heartbeatInterval);
+          wrappedStreamCallback("\n\n");
+        }
+      };
       const toolHandler = async (toolName, toolInput) => {
         if (toolName === "execute_query") {
           let sql = toolInput.sql;
@@ -6534,9 +6577,6 @@ Please try rephrasing your question or simplifying your request.
 ${sql}
 \`\`\`${paramsDisplay}
 
-`);
-                wrappedStreamCallback(`\u26A1 **Executing query...**
-
 `);
               } else {
                 wrappedStreamCallback(`
@@ -6554,9 +6594,6 @@ ${sql}
 ${sql}
 \`\`\`${paramsDisplay}
 
-`);
-                wrappedStreamCallback(`\u26A1 **Executing query...**
-
 `);
               }
             }
@@ -6569,13 +6606,19 @@ ${sql}
               throw new Error("Database collection not registered. Please register database.execute collection to execute queries.");
             }
             const queryPayload = Object.keys(params).length > 0 ? { sql: JSON.stringify({ sql, values: params }) } : { sql };
-            const result2 = await collections["database"]["execute"](queryPayload);
+            const result2 = await withProgressHeartbeat(
+              () => collections["database"]["execute"](queryPayload),
+              "Executing database query",
+              800
+              // Send heartbeat every 800ms for responsive feedback
+            );
             const data = result2?.data || result2;
             const rowCount = result2?.count ?? (Array.isArray(data) ? data.length : "N/A");
             logger.info(`[${this.getProviderName()}] Query executed successfully, rows returned: ${rowCount}`);
             logCollector?.info(`Query successful, returned ${rowCount} rows`);
             if (wrappedStreamCallback) {
-              wrappedStreamCallback(`\u2705 **Query executed successfully!**
+              wrappedStreamCallback(`
+\u2705 **Query executed successfully!**
 
 `);
               if (Array.isArray(data) && data.length > 0) {
@@ -6662,7 +6705,12 @@ Please try rephrasing your request or contact support.
 `);
                 }
               }
-              const result2 = await externalTool.fn(toolInput);
+              const result2 = await withProgressHeartbeat(
+                () => externalTool.fn(toolInput),
+                `Running ${externalTool.name}`,
+                800
+                // Send heartbeat every 800ms
+              );
               logger.info(`[${this.getProviderName()}] External tool ${externalTool.name} executed successfully`);
               logCollector?.info(`\u2713 ${externalTool.name} executed successfully`);
               if (!executedToolsList.find((t) => t.id === externalTool.id)) {
@@ -6764,6 +6812,7 @@ ${errorMsg}
         }
       );
       if (wrappedStreamCallback && components && components.length > 0 && category !== "general") {
+        wrappedStreamCallback("\n\n\u{1F4CA} **Generating visualization components...**\n\n");
         wrappedStreamCallback("__TEXT_COMPLETE__COMPONENT_GENERATION_START__");
       }
       let matchedComponents = [];
@@ -7130,7 +7179,8 @@ ${errorMsg}
         ORIGINAL_USER_PROMPT: originalUserPrompt,
         COMPONENT_INFO: component_info,
         COMPONENT_DATA: component_data,
-        CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
+        CONVERSATION_HISTORY: conversationHistory || "No previous conversation",
+        CURRENT_DATETIME: getCurrentDateTimeForPrompt()
       });
       const result = await LLM.stream(
         {
@@ -11033,7 +11083,8 @@ async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApi
       AVAILABLE_COMPONENTS: availableComponentsText,
       SCHEMA_DOC: schemaDoc || "No database schema available",
       DATABASE_RULES: databaseRules,
-      AVAILABLE_TOOLS: availableToolsText
+      AVAILABLE_TOOLS: availableToolsText,
+      CURRENT_DATETIME: getCurrentDateTimeForPrompt()
     });
     logger.debug("[DASH_COMP_REQ] Loaded dash-comp-picker prompts with schema and tools");
     const { apiKey, model } = getApiKeyAndModel(
@@ -11182,7 +11233,8 @@ async function createFilterWithLLM(prompt, components, existingComponents, anthr
       EXISTING_COMPONENTS: formatExistingComponentsForPrompt(existingComponents),
       SCHEMA_DOC: schemaDoc || "No database schema available",
       DATABASE_RULES: databaseRules,
-      AVAILABLE_TOOLS: formatToolsForPrompt(tools)
+      AVAILABLE_TOOLS: formatToolsForPrompt(tools),
+      CURRENT_DATETIME: getCurrentDateTimeForPrompt()
     });
     logger.debug("[DASH_COMP_REQ:FILTER] Loaded dash-filter-picker prompts");
     const { apiKey, model } = getApiKeyAndModel(