@letta-ai/letta-agent-sdk 0.7.7 → 0.7.8

package/dist/client-entry.js

@@ -11016,6 +11016,7 @@ var DEFAULT_SANDBOX_READY_TIMEOUT_MS = 120000;
 var DEFAULT_SANDBOX_READY_POLL_INTERVAL_MS = 1000;
 var DEFAULT_REPOSITORY_ATTACH_TIMEOUT_MS = 1e4;
 var DEFAULT_REPOSITORY_ATTACH_POLL_INTERVAL_MS = 250;
+var INITIAL_CLOUD_TRANSPORT_ATTEMPTS = 2;
 var SDK_AGENT_ORIGIN = "@letta-ai/letta-agent-sdk";
 
 class CloudManagedSandboxOwnershipError extends Error {
@@ -11206,30 +11207,9 @@ class CloudEnvironmentSession extends RemoteClientSessionCore {
     }
   }
   async openCloudTransport(runtime, connectionId) {
-    const apiKey = getCloudApiKey(this.cloudOptions);
-    const url = buildCloudStatusWebSocketUrl({
-      apiBaseUrl: this.cloudOptions.apiBaseUrl,
-      connectionId,
-      agentId: runtime.agent_id,
-      conversationId: runtime.conversation_id,
-      apiKey,
-      authMode: this.cloudOptions.webSocketAuth ?? "header"
-    });
-    const client = applyUniqueRequestIds(createAppServerClient({
-      url,
-      WebSocket: createCloudStatusTransportConstructor({
-        url,
-        WebSocket: getWebSocketConstructor(this.cloudOptions.WebSocket),
-        ...(this.cloudOptions.webSocketAuth ?? "header") === "header" ? { headers: cloudWebSocketHeaders(this.cloudOptions) } : {},
-        pingIntervalMs: this.cloudOptions.pingIntervalMs ?? DEFAULT_PING_INTERVAL_MS,
-        runtime
-      }),
-      requestTimeoutMs: this.cloudOptions.requestTimeoutMs ?? DEFAULT_TURN_TIMEOUT_MS
-    }));
+    const client = await this.connectCloudTransport(runtime, connectionId);
     const options = this.currentOptions();
-    this.installTransportHandlers(client);
     try {
-      await client.connect();
       const response = await this.startCloudRuntime(client, runtime);
       if (!response.success || !response.runtime) {
         throw new Error(response.error ?? "Failed to start Cloud status runtime");
@@ -11257,6 +11237,62 @@ class CloudEnvironmentSession extends RemoteClientSessionCore {
       throw error;
     }
   }
+  createCloudTransportClient(runtime, connectionId) {
+    const apiKey = getCloudApiKey(this.cloudOptions);
+    const url = buildCloudStatusWebSocketUrl({
+      apiBaseUrl: this.cloudOptions.apiBaseUrl,
+      connectionId,
+      agentId: runtime.agent_id,
+      conversationId: runtime.conversation_id,
+      apiKey,
+      authMode: this.cloudOptions.webSocketAuth ?? "header"
+    });
+    return applyUniqueRequestIds(createAppServerClient({
+      url,
+      WebSocket: createCloudStatusTransportConstructor({
+        url,
+        WebSocket: getWebSocketConstructor(this.cloudOptions.WebSocket),
+        ...(this.cloudOptions.webSocketAuth ?? "header") === "header" ? { headers: cloudWebSocketHeaders(this.cloudOptions) } : {},
+        pingIntervalMs: this.cloudOptions.pingIntervalMs ?? DEFAULT_PING_INTERVAL_MS,
+        runtime
+      }),
+      requestTimeoutMs: this.cloudOptions.requestTimeoutMs ?? DEFAULT_TURN_TIMEOUT_MS
+    }));
+  }
+  async connectCloudTransport(runtime, initialConnectionId) {
+    let connectionId = initialConnectionId;
+    let lastError;
+    for (let attempt = 0;attempt < INITIAL_CLOUD_TRANSPORT_ATTEMPTS; attempt++) {
+      const client = this.createCloudTransportClient(runtime, connectionId);
+      this.installTransportHandlers(client);
+      try {
+        await client.connect();
+        this.connectionId = connectionId;
+        return client;
+      } catch (error) {
+        lastError = error;
+        console.warn(JSON.stringify({
+          event: "cloud_status_transport_connection_failed",
+          attempt: attempt + 1,
+          max_attempts: INITIAL_CLOUD_TRANSPORT_ATTEMPTS,
+          will_retry: attempt + 1 < INITIAL_CLOUD_TRANSPORT_ATTEMPTS,
+          connection_id: connectionId,
+          agent_id: runtime.agent_id,
+          conversation_id: runtime.conversation_id,
+          error_name: error instanceof Error ? error.name : null,
+          error_message: error instanceof Error ? error.message : String(error)
+        }));
+        this.cleanupTransportHandlers();
+        client.close();
+      }
+      if (attempt + 1 < INITIAL_CLOUD_TRANSPORT_ATTEMPTS) {
+        const resolved = await this.resolveRecoveryConnection(runtime);
+        connectionId = resolved.connectionId;
+      }
+    }
+    const detail = lastError instanceof Error ? lastError.message : String(lastError);
+    throw new Error(`Cloud status WebSocket failed to open for connection ${connectionId}: ${detail}`, { cause: lastError });
+  }
   installTransportHandlers(client) {
     this.cleanupTransportHandlers();
     this.removeControlRequestHandler = registerAppServerControlRequestHandler({
@@ -12858,4 +12894,4 @@ export {
   CloudManagedSandboxExpiredError
 };
 
-//# debugId=4D2DEBF58B35A6BA64756E2164756E21
+//# debugId=02750F72F2B016D564756E2164756E21