@paymanai/payman-typescript-ask-sdk 4.0.26 → 4.0.27

package/dist/index.mjs

@@ -1146,6 +1146,9 @@ function isUserActionExpired(prompt) {
 
 // src/hooks/useChatV2.ts
 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
@@ -1192,6 +1195,7 @@ function useChatV2(config, callbacks = {}) {
   configRef.current = config;
   const messagesRef = useRef(messages);
   messagesRef.current = messages;
+  const pendingSubmissionsRef = useRef(/* @__PURE__ */ new Map());
   const storeAwareSetMessages = useCallback(
     (updater) => {
       const streamUserId = streamUserIdRef.current;
@@ -1226,6 +1230,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 = useRef(userActionState);
+  userActionStateRef.current = userActionState;
   const storeAwareSetUserActionState = useCallback(
     (updater) => {
       const streamUserId = streamUserIdRef.current;
@@ -1253,6 +1259,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)
@@ -1264,6 +1276,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
   }), []);
@@ -1399,6 +1433,7 @@ function useChatV2(config, callbacks = {}) {
     [storeAwareSetUserActionState]
   );
   const removePrompt = useCallback((userActionId) => {
+    pendingSubmissionsRef.current.delete(userActionId);
     storeAwareSetUserActionState((prev) => ({
       ...prev,
       prompts: prev.prompts.filter((p) => p.userActionId !== userActionId)
@@ -1409,7 +1444,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");
@@ -1420,7 +1458,7 @@ function useChatV2(config, callbacks = {}) {
         throw error;
       }
     },
-    [removePrompt, setPromptStatus]
+    [setPromptStatus]
   );
   const cancelUserAction2 = useCallback(
     async (userActionId) => {