@letta-ai/letta-agent-sdk 0.8.2 → 0.8.3

package/dist/client-entry.js

@@ -6614,6 +6614,7 @@ function turnSendOptions(turn) {
 
 // src/remote-turn-coordinator.ts
 var MAX_RECENTLY_SETTLED_RUN_IDS = 256;
+var TRAILING_USAGE_GRACE_MS = 100;
 
 class RemoteTurnCoordinator {
   label;
@@ -6656,6 +6657,9 @@ class RemoteTurnCoordinator {
       observedTurnEvidence: false,
       observedRequiresApprovalStop: false,
       pendingTerminal: null,
+      pendingTerminalTimeout: null,
+      deferredMessages: [],
+      deferredTurnEvidence: false,
       abortRequested: false,
       timeout: null
     };
@@ -6669,6 +6673,8 @@ class RemoteTurnCoordinator {
   removeTrackedTurn(turn) {
     if (turn.timeout)
       clearTimeout(turn.timeout);
+    if (turn.pendingTerminalTimeout)
+      clearTimeout(turn.pendingTerminalTimeout);
     if (this.activeTurn === turn) {
       this.activeTurn = null;
       return;
@@ -6699,6 +6705,29 @@ class RemoteTurnCoordinator {
       this.enqueue(sdkMessage2);
       return;
     }
+    const deferredDelta = streamDeltaRecord(message);
+    const deferredMessageType = deferredDelta ? streamDeltaMessageType(deferredDelta) : undefined;
+    const trailingUsageTurn = this.activeTurn?.pendingTerminalTimeout ? this.activeTurn : null;
+    if (trailingUsageTurn) {
+      const statusRunIds = message.type === "update_loop_status" ? loopStatusRunIds(message) : [];
+      const belongsToTerminalTurn = message.type === "update_loop_status" && statusRunIds.length > 0 && statusRunIds.every((runId) => trailingUsageTurn.runIds.has(runId));
+      if (belongsToTerminalTurn) {
+        this.handleLoopStatusMessage(message);
+        return;
+      }
+      if (message.type === "update_loop_status" && statusRunIds.length === 0) {
+        trailingUsageTurn.deferredMessages.push(message);
+        return;
+      }
+      const isTurnScoped = deferredDelta !== null && deferredMessageType !== "ping" || message.type === "update_loop_status" || turnFinishedRecord(message) !== null;
+      if (isTurnScoped && deferredMessageType !== "usage_statistics" || deferredMessageType === "usage_statistics" && trailingUsageTurn.deferredTurnEvidence) {
+        trailingUsageTurn.deferredMessages.push(message);
+        if (isTurnScoped && deferredMessageType !== "usage_statistics") {
+          trailingUsageTurn.deferredTurnEvidence = true;
+        }
+        return;
+      }
+    }
     if (message.type === "update_loop_status") {
       this.handleLoopStatusMessage(message);
       return;
@@ -6711,7 +6740,8 @@ class RemoteTurnCoordinator {
     const delta = streamDeltaRecord(message);
     if (!delta)
       return;
-    const active = this.activateNextTurnFromProtocol();
+    const messageType = streamDeltaMessageType(delta);
+    const active = messageType === "usage_statistics" || messageType === "stop_reason" ? this.activeTurn : this.activateNextTurnFromProtocol();
     if (active) {
       active.observedTurnEvidence = true;
       const runId = streamDeltaRunId(delta);
@@ -6736,13 +6766,18 @@ class RemoteTurnCoordinator {
   closeWithError(detail) {
     if (this.closed)
       return;
+    let completedCanonicalTurn = false;
+    while (this.activeTurn?.pendingTerminal && this.activeTurn.pendingTerminalTimeout) {
+      this.completeActiveTurn(this.activeTurn.pendingTerminal);
+      completedCanonicalTurn = true;
+    }
     const active = this.activeTurn;
     if (active) {
       this.failTurn(active, detail, {
         errorCode: "stream_closed",
         recoverable: true
       });
-    } else {
+    } else if (!completedCanonicalTurn) {
       this.enqueue({
         type: "error",
         message: detail,
@@ -6760,9 +6795,14 @@ class RemoteTurnCoordinator {
     this.closed = true;
     if (this.activeTurn?.timeout)
       clearTimeout(this.activeTurn.timeout);
+    if (this.activeTurn?.pendingTerminalTimeout) {
+      clearTimeout(this.activeTurn.pendingTerminalTimeout);
+    }
     for (const turn of this.pendingTurns) {
       if (turn.timeout)
         clearTimeout(turn.timeout);
+      if (turn.pendingTerminalTimeout)
+        clearTimeout(turn.pendingTerminalTimeout);
     }
     this.activeTurn = null;
     this.pendingTurns.length = 0;
@@ -6818,9 +6858,17 @@ class RemoteTurnCoordinator {
       clearTimeout(active.timeout);
       active.timeout = null;
     }
+    if (active.pendingTerminalTimeout) {
+      clearTimeout(active.pendingTerminalTimeout);
+      active.pendingTerminalTimeout = null;
+    }
+    const deferredMessages = active.deferredMessages.splice(0);
     this.rememberSettledRunIds(active.runIds);
     this.enqueue(this.resultFromTurn(turn, active));
     this.activeTurn = null;
+    for (const message of deferredMessages) {
+      this.handleProtocolMessage(message, active.runtime);
+    }
   }
   rememberSettledRunIds(runIds) {
     for (const runId of runIds) {
@@ -6885,11 +6933,13 @@ class RemoteTurnCoordinator {
     }
   }
   handleTurnFinished(finished) {
-    const active = this.activeTurn;
-    if (!active || !finished.runId)
+    if (!finished.runId)
       return;
     if (this.settledRunIds.has(finished.runId))
       return;
+    const active = this.activeTurn ?? this.activateNextTurnFromProtocol();
+    if (!active)
+      return;
     if (active.runIds.size > 0 && !active.runIds.has(finished.runId)) {
       return;
     }
@@ -6917,6 +6967,11 @@ class RemoteTurnCoordinator {
     };
     if (active.pendingTerminal) {
       active.pendingTerminal = terminal;
+      if (active.timeout) {
+        clearTimeout(active.timeout);
+        active.timeout = null;
+      }
+      this.schedulePendingTerminal(active);
       return;
     }
     this.completeActiveTurn(terminal);
@@ -6940,7 +6995,9 @@ class RemoteTurnCoordinator {
       return;
     }
     if (messageType === "usage_statistics" && active.pendingTerminal) {
-      this.completeActiveTurn(active.pendingTerminal);
+      if (active.pendingTerminalTimeout) {
+        this.completeActiveTurn(active.pendingTerminal);
+      }
       return;
     }
     if (sdkMessage?.type === "error") {
@@ -6954,6 +7011,16 @@ class RemoteTurnCoordinator {
       });
     }
   }
+  schedulePendingTerminal(active) {
+    if (active.pendingTerminalTimeout)
+      return;
+    active.pendingTerminalTimeout = setTimeout(() => {
+      if (this.activeTurn !== active || !active.pendingTerminal)
+        return;
+      this.completeActiveTurn(active.pendingTerminal);
+    }, TRAILING_USAGE_GRACE_MS);
+    active.pendingTerminalTimeout.unref?.();
+  }
   transformStreamDelta(delta) {
     const messageType = typeof delta.message_type === "string" ? delta.message_type : undefined;
     const runId = typeof delta.run_id === "string" ? delta.run_id : undefined;
@@ -8806,6 +8873,7 @@ var MAX_TTL_MINUTES = 60;
 var MAX_GITHUB_REPOSITORIES = 10;
 var GITHUB_OWNER_PATTERN = /^[A-Za-z0-9-]+$/;
 var GITHUB_REPOSITORY_PATTERN = /^[A-Za-z0-9._-]+$/;
+var GITHUB_COMMIT_PATTERN = /^[0-9a-fA-F]{40}$/;
 function validatePositiveInteger2(value, name) {
   if (value !== undefined && (!Number.isInteger(value) || value <= 0)) {
     throw new Error(`Invalid ${name}. Expected a positive integer.`);
@@ -8840,6 +8908,9 @@ function validateCloudSandboxOptions(options, name) {
       if (typeof repository.repo !== "string" || !GITHUB_REPOSITORY_PATTERN.test(repository.repo)) {
         throw new Error(`Invalid ${name}.githubRepositories[${index}].repo.`);
       }
+      if (repository.commit !== undefined && (typeof repository.commit !== "string" || !GITHUB_COMMIT_PATTERN.test(repository.commit))) {
+        throw new Error(`Invalid ${name}.githubRepositories[${index}].commit. Expected a full Git commit SHA.`);
+      }
     }
   }
   if (options.terminateOnClose !== undefined && typeof options.terminateOnClose !== "boolean") {
@@ -10928,4 +10999,4 @@ export {
   CloudManagedSandboxExpiredError
 };
 
-//# debugId=289B04BD4804EAF864756E2164756E21
+//# debugId=8CF5733E1C81C02264756E2164756E21