@superatomai/sdk-node 0.0.6-mds → 0.0.7-mds

package/dist/index.mjs

@@ -6777,6 +6777,7 @@ ${formatted}`;
 Analyze the error and try again with a corrected query.`;
         }
       };
+      const maxIterations = this.config.maxRetries + 2;
       await LLM.streamWithTools(
         { sys: prompts.system, user: prompts.user },
         [llmTool],
@@ -6787,7 +6788,7 @@ Analyze the error and try again with a corrected query.`;
           temperature: 0,
           apiKey: this.config.apiKey
         },
-        this.config.maxRetries
+        maxIterations
       );
       const executionTimeMs = Date.now() - startTime;
       if (!executedTool) {
@@ -10170,21 +10171,25 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
   if (responseMode === "text") {
     streamCallback = (chunk) => {
       accumulatedStreamResponse += chunk;
-      const streamMessage = {
-        id: `stream_${existingUiBlockId}`,
-        // Different ID pattern for streaming
-        type: "USER_PROMPT_STREAM",
-        from: { type: "data-agent" },
-        to: {
-          type: "runtime",
-          id: wsId
-        },
-        payload: {
-          uiBlockId: existingUiBlockId,
-          chunk
-        }
-      };
-      sendMessage(streamMessage);
+      try {
+        const streamMessage = {
+          id: `stream_${existingUiBlockId}`,
+          // Different ID pattern for streaming
+          type: "USER_PROMPT_STREAM",
+          from: { type: "data-agent" },
+          to: {
+            type: "runtime",
+            id: wsId
+          },
+          payload: {
+            uiBlockId: existingUiBlockId,
+            chunk
+          }
+        };
+        sendMessage(streamMessage);
+      } catch (err) {
+        logger.warn(`[UserPromptRequest] Stream send failed (WebSocket may be closed): ${err instanceof Error ? err.message : String(err)}`);
+      }
     };
   }
   const userResponse = await get_agent_user_response(
@@ -15419,17 +15424,27 @@ var SuperatomSDK = class {
     }
   }
   /**
-   * Send a message to the Superatom service
+   * Send a message to the Superatom service.
+   * Returns true if the message was sent, false if the WebSocket is not connected.
+   * Does NOT throw on closed connections — callers can check the return value if needed.
    */
   send(message) {
     if (!this.ws || !this.connected) {
-      throw new Error("WebSocket is not connected. Call connect() first.");
+      logger.warn("WebSocket is not connected, message not sent:", message.type);
+      return false;
     }
     if (this.ws.readyState !== this.ws.OPEN) {
-      throw new Error("WebSocket is not ready to send messages.");
+      logger.warn("WebSocket is not in OPEN state, message not sent:", message.type);
+      return false;
+    }
+    try {
+      const payload = JSON.stringify(message);
+      this.ws.send(payload);
+      return true;
+    } catch (error) {
+      logger.warn(`Failed to send WebSocket message (${message.type}):`, error instanceof Error ? error.message : String(error));
+      return false;
     }
-    const payload = JSON.stringify(message);
-    this.ws.send(payload);
   }
   /**
    * Register a message handler to receive all messages