@letta-ai/letta-agent-sdk 0.8.81 → 0.8.82

package/dist/client-entry.js

@@ -15680,9 +15680,12 @@ function messageEventData(event) {
   }
   return null;
 }
-function channelUrl(url, channel) {
+function channelUrl(url, channel, query = {}) {
   const parsed = new URL(url);
   parsed.searchParams.set("channel", channel);
+  for (const [key, value] of Object.entries(query)) {
+    parsed.searchParams.set(key, value);
+  }
   return parsed.toString();
 }
 
@@ -15702,8 +15705,8 @@ class CloudStatusTransport {
   constructor(options) {
     this.options = options;
     const socketOptions = options.headers ? { headers: options.headers } : undefined;
-    this.controlSocket = new options.WebSocket(channelUrl(options.url, "control"), socketOptions);
-    this.streamSocket = new options.WebSocket(channelUrl(options.url, "stream"), socketOptions);
+    this.controlSocket = new options.WebSocket(channelUrl(options.url, "control", options.channelQuery?.control), socketOptions);
+    this.streamSocket = new options.WebSocket(channelUrl(options.url, "stream", options.channelQuery?.stream), socketOptions);
     this.bindSocket("control", this.controlSocket);
     this.bindSocket("stream", this.streamSocket);
     this.startPing();
@@ -16034,8 +16037,8 @@ function validateCloudClientOptions(options) {
       throw new Error("Letta Cloud clients cannot specify both registerComputer and sandbox options.");
     }
   }
-  if (options.webSocketAuth !== undefined && options.webSocketAuth !== "header" && options.webSocketAuth !== "query") {
-    throw new Error("Invalid webSocketAuth. Valid values: header, query.");
+  if (options.webSocketAuth !== undefined && options.webSocketAuth !== "header" && options.webSocketAuth !== "ticket" && options.webSocketAuth !== "query") {
+    throw new Error("Invalid webSocketAuth. Valid values: header, ticket, query.");
   }
 }
 function environmentToRemoteTarget(environment) {
@@ -16222,33 +16225,63 @@ class CloudEnvironmentSession extends RemoteClientSessionCore {
       throw error;
     }
   }
-  createCloudTransportClient(runtime, connectionId) {
+  createCloudTransportClient(runtime, connectionId, channelTickets) {
     const apiKey = getCloudApiKey(this.cloudOptions);
+    const authMode = this.cloudOptions.webSocketAuth ?? "header";
     const url = buildCloudStatusWebSocketUrl({
       apiBaseUrl: this.cloudOptions.apiBaseUrl,
       connectionId,
       agentId: runtime.agent_id,
       conversationId: runtime.conversation_id,
       apiKey,
-      authMode: this.cloudOptions.webSocketAuth ?? "header"
+      authMode: authMode === "query" && !channelTickets ? "query" : "header"
     });
     return applyUniqueRequestIds(createAppServerClient({
       url,
       WebSocket: createCloudStatusTransportConstructor({
         url,
         WebSocket: getWebSocketConstructor(this.cloudOptions.WebSocket),
-        ...(this.cloudOptions.webSocketAuth ?? "header") === "header" ? { headers: cloudWebSocketHeaders(this.cloudOptions) } : {},
+        ...authMode === "header" ? { headers: cloudWebSocketHeaders(this.cloudOptions) } : {},
+        ...channelTickets ? { channelQuery: channelTickets } : {},
         pingIntervalMs: this.cloudOptions.pingIntervalMs ?? DEFAULT_PING_INTERVAL_MS,
         runtime
       }),
       requestTimeoutMs: this.cloudOptions.requestTimeoutMs ?? DEFAULT_TURN_TIMEOUT_MS
     }));
   }
+  async issueStatusWebSocketTickets(connectionId) {
+    const authMode = this.cloudOptions.webSocketAuth ?? "header";
+    if (authMode !== "ticket" && authMode !== "query")
+      return null;
+    const path2 = `/v1/environments/${encodeURIComponent(connectionId)}/status/ws-ticket`;
+    const issue = async () => {
+      const body = await this.apiClient.post(path2, { body: {} });
+      const ticket = body?.ticket;
+      if (typeof ticket !== "string" || !ticket) {
+        throw new Error("Cloud status WebSocket ticket response did not include a ticket.");
+      }
+      return ticket;
+    };
+    try {
+      const [control, stream] = await Promise.all([issue(), issue()]);
+      return { control: { ticket: control }, stream: { ticket: stream } };
+    } catch (error) {
+      if (authMode === "query" && isNotFound2(error)) {
+        console.warn(JSON.stringify({
+          event: "cloud_status_websocket_query_auth_fallback",
+          connection_id: connectionId
+        }));
+        return null;
+      }
+      throw error;
+    }
+  }
   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);
+      const channelTickets = await this.issueStatusWebSocketTickets(connectionId);
+      const client = this.createCloudTransportClient(runtime, connectionId, channelTickets);
       this.installTransportHandlers(client);
       try {
         await client.connect();
@@ -18350,4 +18383,4 @@ export {
   CloudManagedSandboxExpiredError
 };
 
-//# debugId=56DE4B3DCF03FA0964756E2164756E21
+//# debugId=184FD5599A36662B64756E2164756E21