@dudousxd/nestjs-agent-react 0.6.4 → 0.7.1

package/dist/index.cjs

@@ -1057,7 +1057,7 @@ __name(isAwaitingApproval, "isAwaitingApproval");
 function buildToolCall(part, options) {
   const toolCallId = part.toolCallId;
   const awaiting = isAwaitingApproval(part);
-  const isSubmitting = awaiting && options?.isSubmitting(toolCallId) === true;
+  const sending = awaiting ? options?.submitting(toolCallId) ?? null : null;
   return {
     part,
     toolCallId,
@@ -1065,12 +1065,12 @@ function buildToolCall(part, options) {
     isAwaitingApproval: awaiting,
     approve: {
       available: awaiting && options?.canApprove === true,
-      isSubmitting,
+      isSubmitting: sending === "approve",
       run: /* @__PURE__ */ __name(() => options?.approve(toolCallId), "run")
     },
     reject: {
       available: awaiting && options?.canReject === true,
-      isSubmitting,
+      isSubmitting: sending === "reject",
       run: /* @__PURE__ */ __name(() => options?.reject(toolCallId), "run")
     },
     error: options?.errorOf(toolCallId) ?? null
@@ -1168,7 +1168,7 @@ function buildElicitationBlock(key, part, request, options) {
       }))
     };
   });
-  const isSubmitting = isPending && options.isSubmitting(toolCallId);
+  const sending = isPending ? options.submitting(toolCallId) : null;
   return {
     kind: "elicitation",
     key,
@@ -1181,12 +1181,12 @@ function buildElicitationBlock(key, part, request, options) {
     error: options.errorOf(toolCallId),
     answer: {
       available: isPending && options.canAnswer,
-      isSubmitting,
+      isSubmitting: sending === "answer",
       run: /* @__PURE__ */ __name(() => options.answer(toolCallId), "run")
     },
     skip: {
       available: isPending && options.canSkip,
-      isSubmitting,
+      isSubmitting: sending === "skip",
       run: /* @__PURE__ */ __name(() => options.skip(toolCallId), "run")
     }
   };
@@ -1475,7 +1475,7 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
   const [copied, setCopied] = (0, import_react4.useState)(() => /* @__PURE__ */ new Set());
   const [openReasoning, setOpenReasoning] = (0, import_react4.useState)(() => /* @__PURE__ */ new Map());
   const [picks, setPicks] = (0, import_react4.useState)(() => /* @__PURE__ */ new Map());
-  const [settling, setSettling] = (0, import_react4.useState)(() => /* @__PURE__ */ new Set());
+  const [settling, setSettling] = (0, import_react4.useState)(() => /* @__PURE__ */ new Map());
   const [settleErrors, setSettleErrors] = (0, import_react4.useState)(() => /* @__PURE__ */ new Map());
   const latest = (0, import_react4.useRef)({
     options,
@@ -1518,7 +1518,7 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
   const pick = (0, import_react4.useCallback)((toolCallId, questionId, values) => {
     setPicks((current) => new Map(current).set(pickKey(toolCallId, questionId), values));
   }, []);
-  const settle = (0, import_react4.useCallback)((toolCallId, send) => {
+  const settle = (0, import_react4.useCallback)((toolCallId, action, send) => {
     setSettleErrors((current) => {
       if (!current.has(toolCallId)) {
         return current;
@@ -1527,10 +1527,10 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
       next.delete(toolCallId);
       return next;
     });
-    setSettling((current) => new Set(current).add(toolCallId));
+    setSettling((current) => new Map(current).set(toolCallId, action));
     const failed = /* @__PURE__ */ __name((error) => {
       setSettling((current) => {
-        const next = new Set(current);
+        const next = new Map(current);
         next.delete(toolCallId);
         return next;
       });
@@ -1548,14 +1548,14 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
     if (!onAnswer) {
       return;
     }
-    settle(toolCallId, () => onAnswer(toolCallId, answersFor(current.picks, toolCallId)));
+    settle(toolCallId, "answer", () => onAnswer(toolCallId, answersFor(current.picks, toolCallId)));
   }, [
     settle
   ]);
   const skip = (0, import_react4.useCallback)((toolCallId) => {
     const onSkip = latest.current.options.onSkip;
     if (onSkip) {
-      settle(toolCallId, () => onSkip(toolCallId));
+      settle(toolCallId, "skip", () => onSkip(toolCallId));
     }
   }, [
     settle
@@ -1563,7 +1563,7 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
   const approve = (0, import_react4.useCallback)((toolCallId) => {
     const onApprove = latest.current.options.onApprove;
     if (onApprove) {
-      settle(toolCallId, () => onApprove(toolCallId));
+      settle(toolCallId, "approve", () => onApprove(toolCallId));
     }
   }, [
     settle
@@ -1571,7 +1571,7 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
   const reject = (0, import_react4.useCallback)((toolCallId) => {
     const onReject = latest.current.options.onReject;
     if (onReject) {
-      settle(toolCallId, () => onReject(toolCallId));
+      settle(toolCallId, "reject", () => onReject(toolCallId));
     }
   }, [
     settle
@@ -1696,7 +1696,7 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
             canSkip: options.onSkip !== void 0,
             answer,
             skip,
-            isSubmitting: /* @__PURE__ */ __name((toolCallId) => settling.has(toolCallId), "isSubmitting"),
+            submitting: /* @__PURE__ */ __name((toolCallId) => settling.get(toolCallId) ?? null, "submitting"),
             errorOf: /* @__PURE__ */ __name((toolCallId) => settleErrors.get(toolCallId) ?? null, "errorOf")
           }
         } : {},
@@ -1706,7 +1706,7 @@ function useTranscriptItems({ messages, visibleStart, streamingMessageId, option
             canReject: options.onReject !== void 0,
             approve,
             reject,
-            isSubmitting: /* @__PURE__ */ __name((toolCallId) => settling.has(toolCallId), "isSubmitting"),
+            submitting: /* @__PURE__ */ __name((toolCallId) => settling.get(toolCallId) ?? null, "submitting"),
             errorOf: /* @__PURE__ */ __name((toolCallId) => settleErrors.get(toolCallId) ?? null, "errorOf")
           }
         } : {}