@agents24/react 0.2.27 → 0.2.28

package/dist/index.js

@@ -929,11 +929,11 @@ function reconcileModelTurnParts(parts, event) {
   const terminal = eventName === "run.completed" || eventName === "run.failed" || eventName === "run.cancelled";
   const modelTurnId = typeof payload.model_turn_id === "string" ? payload.model_turn_id : null;
   let next = parts;
-  if (eventName === "run.snapshot" || eventName === "assistant.completed" || terminal) {
+  if (["run.snapshot", "run.waiting", "run.interrupted", "run.paused", "assistant.completed"].includes(eventName) || terminal) {
     next = next.filter((part) => part.kind !== "model-turn");
   } else if (eventName === "assistant.delta" && typeof payload.content === "string" && payload.content) {
     next = next.filter((part) => part.kind !== "model-turn" || part.turnKind !== "agent");
-  } else if (eventName.startsWith("tool.") || eventName === "node.status.updated") {
+  } else if (eventName === "tool.started") {
     next = next.filter((part) => part.kind !== "model-turn" || part.turnKind !== "agent");
   } else if ((eventName === "model.turn.failed" || eventName === "model.turn.completed" && payload.outcome !== "tool_calls") && modelTurnId) {
     next = next.filter((part) => part.kind !== "model-turn" || part.modelTurnId !== modelTurnId);
@@ -984,12 +984,24 @@ function applyCanonicalRuntimeEvent(state, event) {
   const distinctSegment = Boolean(segmentId && segmentId !== `${runId}:0`);
   const messageId = distinctSegment ? `${runId || "run"}-assistant-${segmentId}` : state.streamingMessageId || `${runId || "run"}-assistant`;
   const terminal = eventName === "run.completed" || eventName === "run.failed" || eventName === "run.cancelled";
-  const paused = eventName === "run.paused" || projection.parts.some((part) => part.kind === "hitl" && part.status === "pending");
-  const suspension = eventName === "run.waiting" ? "waiting" : eventName === "run.interrupted" ? "interrupted" : null;
+  const snapshotStatus = eventName === "run.snapshot" ? event.payload?.status : null;
+  const paused = eventName === "run.paused" || snapshotStatus === "paused" || projection.parts.some((part) => part.kind === "hitl" && part.status === "pending");
+  const retainedSuspension = !eventName.startsWith("run.") && !["model.turn.started", "assistant.delta", "tool.started"].includes(eventName) && (state.runState === "waiting" || state.runState === "interrupted") ? state.runState : null;
+  const suspension = eventName === "run.waiting" || snapshotStatus === "waiting" ? "waiting" : eventName === "run.interrupted" || snapshotStatus === "interrupted" ? "interrupted" : retainedSuspension;
   const cancelling = eventName === "run.cancelling" || String(event.payload?.status || "") === "cancelling";
   const existing = state.messages.find((message) => message.role === "assistant" && message.id === messageId) || (!distinctSegment ? state.messages.find((message) => message.role === "assistant" && Boolean(runId && message.runId === runId)) : void 0);
   const sealedAfterTerminal = responseCompleted && !state.activeRunId && ["completed", "failed", "cancelled"].includes(state.runState) && existing?.runId === runId;
   const payload = event.payload || {};
+  const wait = payload.capacity_wait;
+  const capacityWait = wait && typeof wait.reason === "string" ? {
+    reason: wait.reason,
+    ...typeof wait.retry_at === "string" ? { retryAt: wait.retry_at } : {},
+    ...typeof wait.deadline === "string" ? { deadline: wait.deadline } : {}
+  } : state.runState === "waiting" ? existing?.activity?.capacityWait : void 0;
+  const activity = terminal || paused || responseCompleted || existing?.responseCompletion ? void 0 : {
+    status: suspension || (cancelling ? "cancelling" : "running"),
+    ...suspension === "waiting" && capacityWait ? { capacityWait } : {}
+  };
   const eventThreadId = typeof payload.thread_id === "string" && payload.thread_id.trim() ? payload.thread_id.trim() : null;
   const contextApplies = (!state.activeThreadId || !eventThreadId || eventThreadId === state.activeThreadId) && (!state.activeRunId || !runId || runId === state.activeRunId);
   const contextPayload = payload.context_window && typeof payload.context_window === "object" ? { ...payload.context_window, run_id: runId } : payload.context_window;
@@ -1024,6 +1036,7 @@ function applyCanonicalRuntimeEvent(state, event) {
       content,
       createdAt: (/* @__PURE__ */ new Date()).toISOString(),
       isFinal: terminal || responseCompleted,
+      activity,
       responseCompletion: completion,
       parts
     }),
@@ -1033,6 +1046,7 @@ function applyCanonicalRuntimeEvent(state, event) {
       content,
       parts,
       isFinal: terminal || responseCompleted || message.isFinal,
+      activity,
       responseCompletion: completion || message.responseCompletion
     })
   });
@@ -1042,10 +1056,12 @@ function applyCanonicalRuntimeEvent(state, event) {
     return {
       ...message,
       isFinal: true,
+      activity: void 0,
       responseCompletion: completion,
       ...segment ? { durableId: segment.message_id } : {}
     };
   });
+  if (terminal || distinctSegment) messages = messages.map((message) => message.role === "assistant" && message.runId === runId && message.activity && (terminal || message.id !== (existing?.id || messageId)) ? { ...message, activity: void 0 } : message);
   const activityStatus = suspension || (paused ? "paused" : cancelling ? "cancelling" : "running");
   const terminalThread = runId && !sealedAfterTerminal ? {
     active_run: terminal ? null : {