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

package/dist/index.mjs

@@ -274,6 +274,8 @@ function getEventMessage(event) {
     }
     case "RUN_IN_PROGRESS":
       return event.partialText || "Thinking...";
+    case "LLM_CALL_STARTED":
+      return event.description || "";
     case "RUN_COMPLETED":
       return "Agent run completed";
     case "RUN_FAILED":
@@ -411,7 +413,8 @@ function createInitialV2State() {
     finalData: void 0,
     steps: [],
     stepCounter: 0,
-    currentExecutingStepId: void 0
+    currentExecutingStepId: void 0,
+    trailingActivity: false
   };
 }
 function upsertUserAction(state, req) {
@@ -433,6 +436,7 @@ function processStreamEventV2(rawEvent, state) {
   if (typeof eventType === "string" && eventType.toUpperCase() === "KEEP_ALIVE") {
     if (event.executionId) state.executionId = event.executionId;
     if (event.sessionId) state.sessionId = event.sessionId;
+    if (state.finalResponse) state.trailingActivity = true;
     const description = typeof event.description === "string" ? event.description : "";
     if (description) {
       for (let i = state.steps.length - 1; i >= 0; i--) {
@@ -467,7 +471,12 @@ function processStreamEventV2(rawEvent, state) {
       state.lastEventType = eventType;
       break;
     }
+    case "LLM_CALL_STARTED":
+      if (state.finalResponse) state.trailingActivity = true;
+      state.lastEventType = eventType;
+      break;
     case "INTENT_STARTED": {
+      if (state.finalResponse) state.trailingActivity = true;
       const stepId = `step-${state.stepCounter++}`;
       state.steps.push({
         id: stepId,
@@ -483,6 +492,7 @@ function processStreamEventV2(rawEvent, state) {
       break;
     }
     case "INTENT_COMPLETED": {
+      if (state.finalResponse) state.trailingActivity = true;
       const intentStep = state.steps.find((s) => s.eventType === "INTENT_STARTED" && s.status === "in_progress");
       if (intentStep) {
         intentStep.status = "completed";
@@ -514,6 +524,7 @@ function processStreamEventV2(rawEvent, state) {
           step.status = "completed";
         }
       });
+      state.trailingActivity = false;
       state.lastEventType = eventType;
       break;
     }
@@ -585,6 +596,7 @@ function processStreamEventV2(rawEvent, state) {
     case "WORKFLOW_ERROR":
       state.hasError = true;
       state.errorMessage = event.errorMessage || event.message || "Workflow error";
+      state.trailingActivity = false;
       state.lastEventType = eventType;
       break;
     // ---- K2 pipeline stage lifecycle events ----
@@ -879,7 +891,9 @@ function useStreamManagerV2(config, callbacks, setMessages, setIsWaitingForRespo
             const latestUsefulStep = [...state.steps].reverse().find(
               (s) => s.message && !isBlandStatus(s.message)
             );
-            const currentMessage = useful(activeStep?.message) ?? useful(lastInProgressStep?.message) ?? latestUsefulStep?.message ?? useful(getEventMessage(event)) ?? // Last-resort: every candidate is bland (very first event,
+            const eventTypeUpper = typeof event.eventType === "string" ? event.eventType.toUpperCase() : "";
+            const liveEventLabel = eventTypeUpper === "KEEP_ALIVE" || eventTypeUpper === "LLM_CALL_STARTED" ? useful(getEventMessage(event)) : void 0;
+            const currentMessage = useful(activeStep?.message) ?? useful(lastInProgressStep?.message) ?? liveEventLabel ?? latestUsefulStep?.message ?? useful(getEventMessage(event)) ?? // Last-resort: every candidate is bland (very first event,
             // nothing useful seen yet). Render the bland label so the
             // bubble isn't blank.
             activeStep?.message ?? lastInProgressStep?.message ?? getEventMessage(event);
@@ -904,6 +918,7 @@ function useStreamManagerV2(config, callbacks, setMessages, setIsWaitingForRespo
                 streamingContent: state.finalResponse,
                 content: "",
                 currentMessage,
+                hasTrailingActivity: state.trailingActivity,
                 streamProgress: "processing",
                 isError: false,
                 steps: [...state.steps],
@@ -1131,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
@@ -1177,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;
@@ -1211,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;
@@ -1238,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)
@@ -1249,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
   }), []);
@@ -1384,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)
@@ -1394,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");
@@ -1405,7 +1458,7 @@ function useChatV2(config, callbacks = {}) {
         throw error;
       }
     },
-    [removePrompt, setPromptStatus]
+    [setPromptStatus]
   );
   const cancelUserAction2 = useCallback(
     async (userActionId) => {