@runtypelabs/persona 4.21.0 → 4.21.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runtypelabs/persona",
-  "version": "4.21.0",
+  "version": "4.21.1",
   "description": "Themeable, pluggable streaming agent widget for websites, in plain JS with support for voice input and reasoning / tool output.",
   "type": "module",
   "main": "dist/index.cjs",

package/src/client-visitor-history.test.ts

@@ -825,6 +825,137 @@ describe('controller wiring', () => {
     mount.remove();
   });
 
+  it('persists the latest durable cursor with the transcript before page exit', async () => {
+    await seedToken('cvt_stored');
+    installFetch([ok({ sessionId: 'sess_ui' })]);
+    const mount = document.createElement('div');
+    document.body.appendChild(mount);
+    const controller = createAgentExperience(mount, {
+      apiUrl: API_URL,
+      clientToken: CLIENT_TOKEN,
+      launcher: { enabled: false },
+    });
+    await vi.waitFor(() => expect(requests).toHaveLength(1));
+
+    let streamController!: ReadableStreamDefaultController<Uint8Array>;
+    const stream = new ReadableStream<Uint8Array>({
+      start(value) {
+        streamController = value;
+      },
+    });
+    const connected = controller.connectStream(stream);
+    const encoder = new TextEncoder();
+    streamController.enqueue(
+      encoder.encode(
+        'id: 7\nevent: text_delta\ndata: {"type":"text_delta","executionId":"exec_exit_latest","id":"text_exit_latest","delta":"Hi"}\n\n'
+      )
+    );
+    await vi.waitFor(() =>
+      expect(controller.getPersistentMetadata().durableResume).toEqual({
+        executionId: 'exec_exit_latest',
+        after: '7',
+      })
+    );
+
+    streamController.enqueue(
+      encoder.encode(
+        'id: 8\nevent: text_delta\ndata: {"type":"text_delta","executionId":"exec_exit_latest","id":"text_exit_latest","delta":" there"}\n\n'
+      )
+    );
+    await vi.waitFor(() =>
+      expect(controller.getMessages().find((message) => message.role === 'assistant')?.content).toBe(
+        'Hi there'
+      )
+    );
+    await vi.waitFor(() =>
+      expect(controller.getPersistentMetadata().durableResume).toEqual({
+        executionId: 'exec_exit_latest',
+        after: '8',
+      })
+    );
+
+    window.dispatchEvent(new Event('pagehide'));
+
+    streamController.close();
+    await connected;
+    controller.destroy();
+    mount.remove();
+  });
+
+  it('keeps async transcript snapshots aligned with the durable cursor', async () => {
+    await seedToken('cvt_stored');
+    installFetch([ok({ sessionId: 'sess_ui' })]);
+    const saves: AgentWidgetStoredState[] = [];
+    const mount = document.createElement('div');
+    document.body.appendChild(mount);
+    const controller = createAgentExperience(mount, {
+      apiUrl: API_URL,
+      clientToken: CLIENT_TOKEN,
+      launcher: { enabled: false },
+      storageAdapter: {
+        save: async (state) => {
+          await Promise.resolve();
+          saves.push(state);
+        },
+      },
+    });
+    await vi.waitFor(() => expect(requests).toHaveLength(1));
+
+    let streamController!: ReadableStreamDefaultController<Uint8Array>;
+    const stream = new ReadableStream<Uint8Array>({
+      start(value) {
+        streamController = value;
+      },
+    });
+    const connected = controller.connectStream(stream);
+    const encoder = new TextEncoder();
+    streamController.enqueue(
+      encoder.encode(
+        'id: 7\nevent: text_delta\ndata: {"type":"text_delta","executionId":"exec_async","id":"text_async","delta":"Hi"}\n\n'
+      )
+    );
+    await vi.waitFor(() =>
+      expect(controller.getPersistentMetadata().durableResume).toEqual({
+        executionId: 'exec_async',
+        after: '7',
+      })
+    );
+
+    streamController.enqueue(
+      encoder.encode(
+        'id: 8\nevent: text_delta\ndata: {"type":"text_delta","executionId":"exec_async","id":"text_async","delta":" there"}\n\n'
+      )
+    );
+    let firstCompleteTranscript: AgentWidgetStoredState | undefined;
+    await vi.waitFor(() => {
+      firstCompleteTranscript = saves.find((state) =>
+        state.messages?.some(
+          (message) => message.role === 'assistant' && message.content === 'Hi there'
+        )
+      );
+      expect(firstCompleteTranscript).toBeDefined();
+    });
+    expect(firstCompleteTranscript?.metadata?.durableResume).toEqual({
+      executionId: 'exec_async',
+      after: '8',
+    });
+    await Promise.resolve();
+    const savesBeforeExit = saves.length;
+    window.dispatchEvent(new Event('pagehide'));
+    await Promise.resolve();
+    expect(saves).toHaveLength(savesBeforeExit);
+
+    streamController.enqueue(
+      encoder.encode(
+        'id: 9\nevent: execution_complete\ndata: {"type":"execution_complete","executionId":"exec_async","kind":"agent","success":true}\n\n'
+      )
+    );
+    streamController.close();
+    await connected;
+    controller.destroy();
+    mount.remove();
+  });
+
   it('uses Persona-owned stored recovery state to reopen the durable conversation', async () => {
     await seedToken('cvt_stored');
     global.fetch = vi.fn(async (url: string | URL | Request, options?: RequestInit) => {

package/src/ui.ts

@@ -1054,6 +1054,8 @@ export const createAgentExperience = (
     messagePersistenceDisabled
       ? null
       : (config.storageAdapter ?? createLocalStorageAdapter());
+  const personaOwnsClientTokenRecovery =
+    Boolean(config.clientToken) && typeof config.reconnectStream !== "function";
   let persistentMetadata: Record<string, unknown> = {};
   let pendingStoredState: Promise<AgentWidgetStoredState | null> | null = null;
   // Resolves once stored state has been applied (or its load failed). Every
@@ -5313,6 +5315,17 @@ export const createAgentExperience = (
         ? getMessagesForPersistence()
         : [];
 
+    const durableHandle =
+      personaOwnsClientTokenRecovery && session?.getResumableHandle();
+    if (durableHandle) {
+      persistentMetadata = {
+        ...persistentMetadata,
+        durableResume: {
+          executionId: durableHandle.executionId,
+          after: durableHandle.lastEventId,
+        },
+      };
+    }
     const payload: AgentWidgetStoredState = {
       messages,
       metadata: persistentMetadata,
@@ -8384,8 +8397,6 @@ export const createAgentExperience = (
       ? { executionId: candidate.executionId, after: candidate.after }
       : null;
   };
-  const personaOwnsClientTokenRecovery =
-    Boolean(config.clientToken) && typeof config.reconnectStream !== "function";
   if (config.clientToken) {
     config = {
       ...config,
@@ -8498,17 +8509,11 @@ export const createAgentExperience = (
     const hostExecutionState = config.onExecutionState;
     const recoveryWindow = mount.ownerDocument.defaultView;
     let pageExitInProgress = false;
-    const markPageExit = () => {
-      pageExitInProgress = true;
-    };
-    const markPageActive = () => {
-      pageExitInProgress = false;
-    };
-    recoveryWindow?.addEventListener("beforeunload", markPageExit);
+    const markPageExit = () => (pageExitInProgress = true);
+    const markPageActive = () => (pageExitInProgress = false);
     recoveryWindow?.addEventListener("pagehide", markPageExit);
     recoveryWindow?.addEventListener("pageshow", markPageActive);
     destroyCallbacks.push(() => {
-      recoveryWindow?.removeEventListener("beforeunload", markPageExit);
       recoveryWindow?.removeEventListener("pagehide", markPageExit);
       recoveryWindow?.removeEventListener("pageshow", markPageActive);
     });
@@ -8638,6 +8643,18 @@ export const createAgentExperience = (
   let activeStreamingTextCandidate: { index: number; id: string } | null = null;
   let pendingStreamingTextMessages: AgentWidgetMessage[] | null = null;
   let streamingTextRAF: number | null = null;
+  let pendingDurablePersistenceMessages: AgentWidgetMessage[] | null = null;
+
+  const persistMessages = (messages: AgentWidgetMessage[]) => {
+    if (!personaOwnsClientTokenRecovery) return persistState(messages);
+    pendingDurablePersistenceMessages = messages;
+    queueMicrotask(() => {
+      if (pendingDurablePersistenceMessages === messages) {
+        pendingDurablePersistenceMessages = null;
+        persistState(messages);
+      }
+    });
+  };
 
   const applyMessagesChanged = (messages: AgentWidgetMessage[]) => {
     lastAppliedMessages = messages;
@@ -8698,7 +8715,7 @@ export const createAgentExperience = (
     }
 
     voiceState.lastUserMessageWasVoice = Boolean(lastUserMessage?.viaVoice);
-    persistState(messages);
+    persistMessages(messages);
     syncComposerBarPeek();
   };