@paymanai/payman-ask-sdk 4.0.26 → 4.0.27

package/dist/index.native.js

@@ -1153,6 +1153,9 @@ function isUserActionExpired(prompt) {
   return left !== void 0 && left <= 0;
 }
 var EMPTY_USER_ACTION_STATE2 = { prompts: [], notifications: [] };
+function promptSlotKey(p) {
+  return p.toolCallId || p.userActionId;
+}
 function upsertPrompt(prompts, req) {
   const idx = prompts.findIndex(
     (p) => req.toolCallId ? p.toolCallId === req.toolCallId : p.userActionId === req.userActionId
@@ -1199,6 +1202,7 @@ function useChatV2(config, callbacks = {}) {
   configRef.current = config;
   const messagesRef = react.useRef(messages);
   messagesRef.current = messages;
+  const pendingSubmissionsRef = react.useRef(/* @__PURE__ */ new Map());
   const storeAwareSetMessages = react.useCallback(
     (updater) => {
       const streamUserId = streamUserIdRef.current;
@@ -1233,6 +1237,8 @@ function useChatV2(config, callbacks = {}) {
     if (!config.userId) return EMPTY_USER_ACTION_STATE2;
     return activeStreamStore.get(config.userId)?.userActionState ?? EMPTY_USER_ACTION_STATE2;
   });
+  const userActionStateRef = react.useRef(userActionState);
+  userActionStateRef.current = userActionState;
   const storeAwareSetUserActionState = react.useCallback(
     (updater) => {
       const streamUserId = streamUserIdRef.current;
@@ -1260,6 +1266,12 @@ function useChatV2(config, callbacks = {}) {
     onExecutionTraceClick: (data) => callbacksRef.current.onExecutionTraceClick?.(data),
     onSessionIdChange: (sessionId) => callbacksRef.current.onSessionIdChange?.(sessionId),
     onUserActionRequired: (request) => {
+      const requestSlotKey = promptSlotKey(request);
+      for (const [pendingId, slotKey] of pendingSubmissionsRef.current) {
+        if (slotKey === requestSlotKey) {
+          pendingSubmissionsRef.current.delete(pendingId);
+        }
+      }
       storeAwareSetUserActionState((prev) => ({
         ...prev,
         prompts: upsertPrompt(prev.prompts, request)
@@ -1271,6 +1283,28 @@ function useChatV2(config, callbacks = {}) {
         (prev) => prev.notifications.some((n) => n.id === notification.id) ? prev : { ...prev, notifications: [...prev.notifications, notification] }
       );
       callbacksRef.current.onUserNotification?.(notification);
+    },
+    // A submitted-but-unconfirmed prompt (see `submitUserAction`) is only
+    // safe to drop once the stream has demonstrably moved on. `onEvent`
+    // in useStreamManagerV2 calls `onStepsUpdate` for *every* processed
+    // stream event, unconditionally — so the first time this fires
+    // after a submission, the event in hand is either the rejection
+    // retry (a fresh `USER_ACTION_REQUIRED`, which the handler above
+    // already deleted this pending entry for, earlier in the very same
+    // event) or genuine forward progress (RUN_IN_PROGRESS, a tool call,
+    // content deltas, RUN_COMPLETED, ...). Anything still pending by
+    // the time we get here therefore means the run resumed normally —
+    // clear it.
+    onStepsUpdate: (steps) => {
+      if (pendingSubmissionsRef.current.size > 0) {
+        const resolved = [...pendingSubmissionsRef.current.keys()];
+        pendingSubmissionsRef.current.clear();
+        storeAwareSetUserActionState((prev) => ({
+          ...prev,
+          prompts: prev.prompts.filter((p) => !resolved.includes(p.userActionId))
+        }));
+      }
+      callbacksRef.current.onStepsUpdate?.(steps);
     }
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }), []);
@@ -1406,6 +1440,7 @@ function useChatV2(config, callbacks = {}) {
     [storeAwareSetUserActionState]
   );
   const removePrompt = react.useCallback((userActionId) => {
+    pendingSubmissionsRef.current.delete(userActionId);
     storeAwareSetUserActionState((prev) => ({
       ...prev,
       prompts: prev.prompts.filter((p) => p.userActionId !== userActionId)
@@ -1416,7 +1451,10 @@ function useChatV2(config, callbacks = {}) {
       setPromptStatus(userActionId, "submitting");
       try {
         await submitUserAction(configRef.current, userActionId, content);
-        removePrompt(userActionId);
+        const prompt = userActionStateRef.current.prompts.find(
+          (p) => p.userActionId === userActionId
+        );
+        pendingSubmissionsRef.current.set(userActionId, prompt ? promptSlotKey(prompt) : userActionId);
       } catch (error) {
         if (error instanceof UserActionStaleError) {
           setPromptStatus(userActionId, "stale");
@@ -1427,7 +1465,7 @@ function useChatV2(config, callbacks = {}) {
         throw error;
       }
     },
-    [removePrompt, setPromptStatus]
+    [setPromptStatus]
   );
   const cancelUserAction2 = react.useCallback(
     async (userActionId) => {