@assistant-ui/react 0.11.24 → 0.11.25

package/dist/legacy-runtime/hooks/AttachmentContext.d.ts

@@ -278,8 +278,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -292,8 +292,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -307,8 +307,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -321,8 +321,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -336,8 +336,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -350,8 +350,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -364,8 +364,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -378,8 +378,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -395,8 +395,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -409,8 +409,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -426,8 +426,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -440,8 +440,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -457,8 +457,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-            file: File;
+            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+            content: import("../..").ThreadUserMessagePart[];
         } & {
             readonly source: "edit-composer";
         } & {
@@ -471,8 +471,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-            content: import("../..").ThreadUserMessagePart[];
+            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+            file: File;
         } & {
             readonly source: "edit-composer";
         } & {
@@ -489,8 +489,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-            file: File;
+            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+            content: import("../..").ThreadUserMessagePart[];
         } & {
             readonly source: "edit-composer";
         } & {
@@ -503,8 +503,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-            content: import("../..").ThreadUserMessagePart[];
+            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+            file: File;
         } & {
             readonly source: "edit-composer";
         } & {
@@ -518,8 +518,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -532,8 +532,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {
@@ -549,8 +549,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-            file: File;
+            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+            content: import("../..").ThreadUserMessagePart[];
         } & {
             readonly source: "edit-composer";
         } & {
@@ -563,8 +563,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-            content: import("../..").ThreadUserMessagePart[];
+            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+            file: File;
         } & {
             readonly source: "edit-composer";
         } & {
@@ -581,8 +581,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-            file: File;
+            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+            content: import("../..").ThreadUserMessagePart[];
         } & {
             readonly source: "edit-composer";
         } & {
@@ -595,8 +595,8 @@ export declare const useEditComposerAttachment: {
             file?: File;
             content?: import("../..").ThreadUserMessagePart[];
         } & {
-            status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-            content: import("../..").ThreadUserMessagePart[];
+            status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+            file: File;
         } & {
             readonly source: "edit-composer";
         } & {
@@ -610,8 +610,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
-        file: File;
+        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
+        content: import("../..").ThreadUserMessagePart[];
     } & {
         readonly source: "edit-composer";
     } & {
@@ -624,8 +624,8 @@ export declare const useEditComposerAttachment: {
         file?: File;
         content?: import("../..").ThreadUserMessagePart[];
     } & {
-        status: import("../../types/AttachmentTypes").CompleteAttachmentStatus;
-        content: import("../..").ThreadUserMessagePart[];
+        status: import("../../types/AttachmentTypes").PendingAttachmentStatus;
+        file: File;
     } & {
         readonly source: "edit-composer";
     } & {

package/dist/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useToolInvocations.d.ts","sourceRoot":"","sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.ts"],"names":[],"mappings":"AACA,OAAO,EAKL,KAAK,IAAI,EACV,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EACV,yBAAyB,EACzB,uBAAuB,EACxB,MAAM,SAAS,CAAC;AAejB,KAAK,wBAAwB,GAAG;IAC9B,KAAK,EAAE,uBAAuB,CAAC;IAC/B,QAAQ,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,SAAS,CAAC;IACjD,QAAQ,EAAE,CAAC,OAAO,EAAE,yBAAyB,KAAK,IAAI,CAAC;IACvD,eAAe,EAAE,CACf,OAAO,EACH,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,GACnC,CAAC,CACC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,KACtC,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC,KAC1C,IAAI,CAAC;CACX,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAC3B;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,GACrB;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,OAAO,EAAE;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAA;CAAE,CAAC;AAExE,wBAAgB,kBAAkB,CAAC,EACjC,KAAK,EACL,QAAQ,EACR,QAAQ,EACR,eAAe,GAChB,EAAE,wBAAwB;;;yBAmLF,MAAM,WAAW,OAAO;EAiBhD"}
+{"version":3,"file":"useToolInvocations.d.ts","sourceRoot":"","sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.ts"],"names":[],"mappings":"AACA,OAAO,EAKL,KAAK,IAAI,EACV,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EACV,yBAAyB,EACzB,uBAAuB,EACxB,MAAM,SAAS,CAAC;AAejB,KAAK,wBAAwB,GAAG;IAC9B,KAAK,EAAE,uBAAuB,CAAC;IAC/B,QAAQ,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,SAAS,CAAC;IACjD,QAAQ,EAAE,CAAC,OAAO,EAAE,yBAAyB,KAAK,IAAI,CAAC;IACvD,eAAe,EAAE,CACf,OAAO,EACH,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,GACnC,CAAC,CACC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,KACtC,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC,KAC1C,IAAI,CAAC;CACX,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAC3B;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,GACrB;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,OAAO,EAAE;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAA;CAAE,CAAC;AAExE,wBAAgB,kBAAkB,CAAC,EACjC,KAAK,EACL,QAAQ,EACR,QAAQ,EACR,eAAe,GAChB,EAAE,wBAAwB;;;yBA2LF,MAAM,WAAW,OAAO;EAiBhD"}

package/dist/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.js

@@ -77,72 +77,74 @@ function useToolInvocations({
   const ignoredToolIds = useRef(/* @__PURE__ */ new Set());
   const isInititialState = useRef(true);
   useEffect(() => {
-    if (isInititialState.current) {
-      state.messages.forEach((message) => {
-        message.content.forEach((content) => {
-          if (content.type === "tool-call") {
-            ignoredToolIds.current.add(content.toolCallId);
-          }
-        });
-      });
-      isInititialState.current = false;
-    } else {
-      state.messages.forEach((message) => {
+    const processMessages = (messages) => {
+      messages.forEach((message) => {
         message.content.forEach((content) => {
           if (content.type === "tool-call") {
-            if (ignoredToolIds.current.has(content.toolCallId)) {
-              return;
-            }
-            let lastState = lastToolStates.current[content.toolCallId];
-            if (!lastState) {
-              const toolCallController = controller.addToolCallPart({
-                toolName: content.toolName,
-                toolCallId: content.toolCallId
-              });
-              lastState = {
-                argsText: "",
-                hasResult: false,
-                controller: toolCallController
-              };
-              lastToolStates.current[content.toolCallId] = lastState;
-            }
-            if (content.argsText !== lastState.argsText) {
-              if (!content.argsText.startsWith(lastState.argsText)) {
-                throw new Error(
-                  `Tool call argsText can only be appended, not updated: ${content.argsText} does not start with ${lastState.argsText}`
+            if (isInititialState.current) {
+              ignoredToolIds.current.add(content.toolCallId);
+            } else {
+              if (ignoredToolIds.current.has(content.toolCallId)) {
+                return;
+              }
+              let lastState = lastToolStates.current[content.toolCallId];
+              if (!lastState) {
+                const toolCallController = controller.addToolCallPart({
+                  toolName: content.toolName,
+                  toolCallId: content.toolCallId
+                });
+                lastState = {
+                  argsText: "",
+                  hasResult: false,
+                  controller: toolCallController
+                };
+                lastToolStates.current[content.toolCallId] = lastState;
+              }
+              if (content.argsText !== lastState.argsText) {
+                if (!content.argsText.startsWith(lastState.argsText)) {
+                  throw new Error(
+                    `Tool call argsText can only be appended, not updated: ${content.argsText} does not start with ${lastState.argsText}`
+                  );
+                }
+                const argsTextDelta = content.argsText.slice(
+                  lastState.argsText.length
                 );
+                lastState.controller.argsText.append(argsTextDelta);
+                if (isArgsTextComplete(content.argsText)) {
+                  lastState.controller.argsText.close();
+                }
+                lastToolStates.current[content.toolCallId] = {
+                  argsText: content.argsText,
+                  hasResult: lastState.hasResult,
+                  controller: lastState.controller
+                };
               }
-              const argsTextDelta = content.argsText.slice(
-                lastState.argsText.length
-              );
-              lastState.controller.argsText.append(argsTextDelta);
-              if (isArgsTextComplete(content.argsText)) {
-                lastState.controller.argsText.close();
+              if (content.result !== void 0 && !lastState.hasResult) {
+                lastState.controller.setResponse(
+                  new ToolResponse({
+                    result: content.result,
+                    artifact: content.artifact,
+                    isError: content.isError
+                  })
+                );
+                lastState.controller.close();
+                lastToolStates.current[content.toolCallId] = {
+                  hasResult: true,
+                  argsText: lastState.argsText,
+                  controller: lastState.controller
+                };
               }
-              lastToolStates.current[content.toolCallId] = {
-                argsText: content.argsText,
-                hasResult: lastState.hasResult,
-                controller: lastState.controller
-              };
             }
-            if (content.result !== void 0 && !lastState.hasResult) {
-              lastState.controller.setResponse(
-                new ToolResponse({
-                  result: content.result,
-                  artifact: content.artifact,
-                  isError: content.isError
-                })
-              );
-              lastState.controller.close();
-              lastToolStates.current[content.toolCallId] = {
-                hasResult: true,
-                argsText: lastState.argsText,
-                controller: lastState.controller
-              };
+            if (content.messages) {
+              processMessages(content.messages);
             }
           }
         });
       });
+    };
+    processMessages(state.messages);
+    if (isInititialState.current) {
+      isInititialState.current = false;
     }
   }, [state, controller, onResult]);
   const abort = () => {

package/dist/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\nimport {\n  createAssistantStreamController,\n  ToolCallStreamController,\n  ToolResponse,\n  unstable_toolResultStream,\n  type Tool,\n} from \"assistant-stream\";\nimport type {\n  AssistantTransportCommand,\n  AssistantTransportState,\n} from \"./types\";\nimport {\n  AssistantMetaTransformStream,\n  type ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\n\nconst isArgsTextComplete = (argsText: string) => {\n  try {\n    JSON.parse(argsText);\n    return true;\n  } catch {\n    return false;\n  }\n};\n\ntype UseToolInvocationsParams = {\n  state: AssistantTransportState;\n  getTools: () => Record<string, Tool> | undefined;\n  onResult: (command: AssistantTransportCommand) => void;\n  setToolStatuses: (\n    updater:\n      | Record<string, ToolExecutionStatus>\n      | ((\n          prev: Record<string, ToolExecutionStatus>,\n        ) => Record<string, ToolExecutionStatus>),\n  ) => void;\n};\n\nexport type ToolExecutionStatus =\n  | { type: \"executing\" }\n  | { type: \"interrupt\"; payload: { type: \"human\"; payload: unknown } };\n\nexport function useToolInvocations({\n  state,\n  getTools,\n  onResult,\n  setToolStatuses,\n}: UseToolInvocationsParams) {\n  const lastToolStates = useRef<\n    Record<\n      string,\n      {\n        argsText: string;\n        hasResult: boolean;\n        controller: ToolCallStreamController;\n      }\n    >\n  >({});\n\n  const humanInputRef = useRef<\n    Map<\n      string,\n      {\n        resolve: (payload: unknown) => void;\n        reject: (reason: unknown) => void;\n      }\n    >\n  >(new Map());\n\n  const acRef = useRef<AbortController>(new AbortController());\n  const [controller] = useState(() => {\n    const [stream, controller] = createAssistantStreamController();\n    const transform = unstable_toolResultStream(\n      getTools,\n      () => acRef.current?.signal ?? new AbortController().signal,\n      (toolCallId: string, payload: unknown) => {\n        return new Promise<unknown>((resolve, reject) => {\n          // Reject previous human input request if it exists\n          const previous = humanInputRef.current.get(toolCallId);\n          if (previous) {\n            previous.reject(\n              new Error(\"Human input request was superseded by a new request\"),\n            );\n          }\n\n          humanInputRef.current.set(toolCallId, { resolve, reject });\n          setToolStatuses((prev) => ({\n            ...prev,\n            [toolCallId]: {\n              type: \"interrupt\",\n              payload: { type: \"human\", payload },\n            },\n          }));\n        });\n      },\n    );\n    stream\n      .pipeThrough(transform)\n      .pipeThrough(new AssistantMetaTransformStream())\n      .pipeTo(\n        new WritableStream({\n          write(chunk) {\n            if (chunk.type === \"result\") {\n              // the tool call result was already set by the backend\n              if (lastToolStates.current[chunk.meta.toolCallId]?.hasResult)\n                return;\n\n              onResult({\n                type: \"add-tool-result\",\n                toolCallId: chunk.meta.toolCallId,\n                toolName: chunk.meta.toolName,\n                result: chunk.result,\n                isError: chunk.isError,\n                ...(chunk.artifact && { artifact: chunk.artifact }),\n              });\n\n              // Clear status when result is set\n              setToolStatuses((prev) => {\n                const next = { ...prev };\n                delete next[chunk.meta.toolCallId];\n                return next;\n              });\n            }\n          },\n        }),\n      );\n\n    return controller;\n  });\n\n  const ignoredToolIds = useRef<Set<string>>(new Set());\n  const isInititialState = useRef(true);\n\n  useEffect(() => {\n    if (isInititialState.current) {\n      state.messages.forEach((message) => {\n        message.content.forEach((content) => {\n          if (content.type === \"tool-call\") {\n            ignoredToolIds.current.add(content.toolCallId);\n          }\n        });\n      });\n      isInititialState.current = false;\n    } else {\n      state.messages.forEach((message) => {\n        message.content.forEach((content) => {\n          if (content.type === \"tool-call\") {\n            if (ignoredToolIds.current.has(content.toolCallId)) {\n              return;\n            }\n            let lastState = lastToolStates.current[content.toolCallId];\n            if (!lastState) {\n              const toolCallController = controller.addToolCallPart({\n                toolName: content.toolName,\n                toolCallId: content.toolCallId,\n              });\n              lastState = {\n                argsText: \"\",\n                hasResult: false,\n                controller: toolCallController,\n              };\n              lastToolStates.current[content.toolCallId] = lastState;\n            }\n\n            if (content.argsText !== lastState.argsText) {\n              if (!content.argsText.startsWith(lastState.argsText)) {\n                throw new Error(\n                  `Tool call argsText can only be appended, not updated: ${content.argsText} does not start with ${lastState.argsText}`,\n                );\n              }\n\n              const argsTextDelta = content.argsText.slice(\n                lastState.argsText.length,\n              );\n              lastState.controller.argsText.append(argsTextDelta);\n\n              if (isArgsTextComplete(content.argsText)) {\n                lastState.controller.argsText.close();\n              }\n\n              lastToolStates.current[content.toolCallId] = {\n                argsText: content.argsText,\n                hasResult: lastState.hasResult,\n                controller: lastState.controller,\n              };\n            }\n\n            if (content.result !== undefined && !lastState.hasResult) {\n              lastState.controller.setResponse(\n                new ToolResponse({\n                  result: content.result as ReadonlyJSONValue,\n                  artifact: content.artifact as ReadonlyJSONValue | undefined,\n                  isError: content.isError,\n                }),\n              );\n              lastState.controller.close();\n\n              lastToolStates.current[content.toolCallId] = {\n                hasResult: true,\n                argsText: lastState.argsText,\n                controller: lastState.controller,\n              };\n            }\n          }\n        });\n      });\n    }\n  }, [state, controller, onResult]);\n\n  const abort = () => {\n    humanInputRef.current.forEach(({ reject }) => {\n      reject(new Error(\"Tool execution aborted\"));\n    });\n    humanInputRef.current.clear();\n    setToolStatuses({});\n\n    acRef.current.abort();\n    acRef.current = new AbortController();\n  };\n\n  return {\n    reset: () => {\n      abort();\n      isInititialState.current = true;\n    },\n    abort,\n    resume: (toolCallId: string, payload: unknown) => {\n      const handlers = humanInputRef.current.get(toolCallId);\n      if (handlers) {\n        humanInputRef.current.delete(toolCallId);\n        setToolStatuses((prev) => {\n          const next = { ...prev };\n          delete next[toolCallId];\n          return next;\n        });\n        handlers.resolve(payload);\n      } else {\n        throw new Error(\n          `Tool call ${toolCallId} is not waiting for human input`,\n        );\n      }\n    },\n  };\n}\n"],"mappings":";AAAA,SAAS,WAAW,QAAQ,gBAAgB;AAC5C;AAAA,EACE;AAAA,EAEA;AAAA,EACA;AAAA,OAEK;AAKP;AAAA,EACE;AAAA,OAEK;AAEP,IAAM,qBAAqB,CAAC,aAAqB;AAC/C,MAAI;AACF,SAAK,MAAM,QAAQ;AACnB,WAAO;AAAA,EACT,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAmBO,SAAS,mBAAmB;AAAA,EACjC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAA6B;AAC3B,QAAM,iBAAiB,OASrB,CAAC,CAAC;AAEJ,QAAM,gBAAgB,OAQpB,oBAAI,IAAI,CAAC;AAEX,QAAM,QAAQ,OAAwB,IAAI,gBAAgB,CAAC;AAC3D,QAAM,CAAC,UAAU,IAAI,SAAS,MAAM;AAClC,UAAM,CAAC,QAAQA,WAAU,IAAI,gCAAgC;AAC7D,UAAM,YAAY;AAAA,MAChB;AAAA,MACA,MAAM,MAAM,SAAS,UAAU,IAAI,gBAAgB,EAAE;AAAA,MACrD,CAAC,YAAoB,YAAqB;AACxC,eAAO,IAAI,QAAiB,CAAC,SAAS,WAAW;AAE/C,gBAAM,WAAW,cAAc,QAAQ,IAAI,UAAU;AACrD,cAAI,UAAU;AACZ,qBAAS;AAAA,cACP,IAAI,MAAM,qDAAqD;AAAA,YACjE;AAAA,UACF;AAEA,wBAAc,QAAQ,IAAI,YAAY,EAAE,SAAS,OAAO,CAAC;AACzD,0BAAgB,CAAC,UAAU;AAAA,YACzB,GAAG;AAAA,YACH,CAAC,UAAU,GAAG;AAAA,cACZ,MAAM;AAAA,cACN,SAAS,EAAE,MAAM,SAAS,QAAQ;AAAA,YACpC;AAAA,UACF,EAAE;AAAA,QACJ,CAAC;AAAA,MACH;AAAA,IACF;AACA,WACG,YAAY,SAAS,EACrB,YAAY,IAAI,6BAA6B,CAAC,EAC9C;AAAA,MACC,IAAI,eAAe;AAAA,QACjB,MAAM,OAAO;AACX,cAAI,MAAM,SAAS,UAAU;AAE3B,gBAAI,eAAe,QAAQ,MAAM,KAAK,UAAU,GAAG;AACjD;AAEF,qBAAS;AAAA,cACP,MAAM;AAAA,cACN,YAAY,MAAM,KAAK;AAAA,cACvB,UAAU,MAAM,KAAK;AAAA,cACrB,QAAQ,MAAM;AAAA,cACd,SAAS,MAAM;AAAA,cACf,GAAI,MAAM,YAAY,EAAE,UAAU,MAAM,SAAS;AAAA,YACnD,CAAC;AAGD,4BAAgB,CAAC,SAAS;AACxB,oBAAM,OAAO,EAAE,GAAG,KAAK;AACvB,qBAAO,KAAK,MAAM,KAAK,UAAU;AACjC,qBAAO;AAAA,YACT,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAEF,WAAOA;AAAA,EACT,CAAC;AAED,QAAM,iBAAiB,OAAoB,oBAAI,IAAI,CAAC;AACpD,QAAM,mBAAmB,OAAO,IAAI;AAEpC,YAAU,MAAM;AACd,QAAI,iBAAiB,SAAS;AAC5B,YAAM,SAAS,QAAQ,CAAC,YAAY;AAClC,gBAAQ,QAAQ,QAAQ,CAAC,YAAY;AACnC,cAAI,QAAQ,SAAS,aAAa;AAChC,2BAAe,QAAQ,IAAI,QAAQ,UAAU;AAAA,UAC/C;AAAA,QACF,CAAC;AAAA,MACH,CAAC;AACD,uBAAiB,UAAU;AAAA,IAC7B,OAAO;AACL,YAAM,SAAS,QAAQ,CAAC,YAAY;AAClC,gBAAQ,QAAQ,QAAQ,CAAC,YAAY;AACnC,cAAI,QAAQ,SAAS,aAAa;AAChC,gBAAI,eAAe,QAAQ,IAAI,QAAQ,UAAU,GAAG;AAClD;AAAA,YACF;AACA,gBAAI,YAAY,eAAe,QAAQ,QAAQ,UAAU;AACzD,gBAAI,CAAC,WAAW;AACd,oBAAM,qBAAqB,WAAW,gBAAgB;AAAA,gBACpD,UAAU,QAAQ;AAAA,gBAClB,YAAY,QAAQ;AAAA,cACtB,CAAC;AACD,0BAAY;AAAA,gBACV,UAAU;AAAA,gBACV,WAAW;AAAA,gBACX,YAAY;AAAA,cACd;AACA,6BAAe,QAAQ,QAAQ,UAAU,IAAI;AAAA,YAC/C;AAEA,gBAAI,QAAQ,aAAa,UAAU,UAAU;AAC3C,kBAAI,CAAC,QAAQ,SAAS,WAAW,UAAU,QAAQ,GAAG;AACpD,sBAAM,IAAI;AAAA,kBACR,yDAAyD,QAAQ,QAAQ,wBAAwB,UAAU,QAAQ;AAAA,gBACrH;AAAA,cACF;AAEA,oBAAM,gBAAgB,QAAQ,SAAS;AAAA,gBACrC,UAAU,SAAS;AAAA,cACrB;AACA,wBAAU,WAAW,SAAS,OAAO,aAAa;AAElD,kBAAI,mBAAmB,QAAQ,QAAQ,GAAG;AACxC,0BAAU,WAAW,SAAS,MAAM;AAAA,cACtC;AAEA,6BAAe,QAAQ,QAAQ,UAAU,IAAI;AAAA,gBAC3C,UAAU,QAAQ;AAAA,gBAClB,WAAW,UAAU;AAAA,gBACrB,YAAY,UAAU;AAAA,cACxB;AAAA,YACF;AAEA,gBAAI,QAAQ,WAAW,UAAa,CAAC,UAAU,WAAW;AACxD,wBAAU,WAAW;AAAA,gBACnB,IAAI,aAAa;AAAA,kBACf,QAAQ,QAAQ;AAAA,kBAChB,UAAU,QAAQ;AAAA,kBAClB,SAAS,QAAQ;AAAA,gBACnB,CAAC;AAAA,cACH;AACA,wBAAU,WAAW,MAAM;AAE3B,6BAAe,QAAQ,QAAQ,UAAU,IAAI;AAAA,gBAC3C,WAAW;AAAA,gBACX,UAAU,UAAU;AAAA,gBACpB,YAAY,UAAU;AAAA,cACxB;AAAA,YACF;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,OAAO,YAAY,QAAQ,CAAC;AAEhC,QAAM,QAAQ,MAAM;AAClB,kBAAc,QAAQ,QAAQ,CAAC,EAAE,OAAO,MAAM;AAC5C,aAAO,IAAI,MAAM,wBAAwB,CAAC;AAAA,IAC5C,CAAC;AACD,kBAAc,QAAQ,MAAM;AAC5B,oBAAgB,CAAC,CAAC;AAElB,UAAM,QAAQ,MAAM;AACpB,UAAM,UAAU,IAAI,gBAAgB;AAAA,EACtC;AAEA,SAAO;AAAA,IACL,OAAO,MAAM;AACX,YAAM;AACN,uBAAiB,UAAU;AAAA,IAC7B;AAAA,IACA;AAAA,IACA,QAAQ,CAAC,YAAoB,YAAqB;AAChD,YAAM,WAAW,cAAc,QAAQ,IAAI,UAAU;AACrD,UAAI,UAAU;AACZ,sBAAc,QAAQ,OAAO,UAAU;AACvC,wBAAgB,CAAC,SAAS;AACxB,gBAAM,OAAO,EAAE,GAAG,KAAK;AACvB,iBAAO,KAAK,UAAU;AACtB,iBAAO;AAAA,QACT,CAAC;AACD,iBAAS,QAAQ,OAAO;AAAA,MAC1B,OAAO;AACL,cAAM,IAAI;AAAA,UACR,aAAa,UAAU;AAAA,QACzB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["controller"]}
+{"version":3,"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\nimport {\n  createAssistantStreamController,\n  ToolCallStreamController,\n  ToolResponse,\n  unstable_toolResultStream,\n  type Tool,\n} from \"assistant-stream\";\nimport type {\n  AssistantTransportCommand,\n  AssistantTransportState,\n} from \"./types\";\nimport {\n  AssistantMetaTransformStream,\n  type ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\n\nconst isArgsTextComplete = (argsText: string) => {\n  try {\n    JSON.parse(argsText);\n    return true;\n  } catch {\n    return false;\n  }\n};\n\ntype UseToolInvocationsParams = {\n  state: AssistantTransportState;\n  getTools: () => Record<string, Tool> | undefined;\n  onResult: (command: AssistantTransportCommand) => void;\n  setToolStatuses: (\n    updater:\n      | Record<string, ToolExecutionStatus>\n      | ((\n          prev: Record<string, ToolExecutionStatus>,\n        ) => Record<string, ToolExecutionStatus>),\n  ) => void;\n};\n\nexport type ToolExecutionStatus =\n  | { type: \"executing\" }\n  | { type: \"interrupt\"; payload: { type: \"human\"; payload: unknown } };\n\nexport function useToolInvocations({\n  state,\n  getTools,\n  onResult,\n  setToolStatuses,\n}: UseToolInvocationsParams) {\n  const lastToolStates = useRef<\n    Record<\n      string,\n      {\n        argsText: string;\n        hasResult: boolean;\n        controller: ToolCallStreamController;\n      }\n    >\n  >({});\n\n  const humanInputRef = useRef<\n    Map<\n      string,\n      {\n        resolve: (payload: unknown) => void;\n        reject: (reason: unknown) => void;\n      }\n    >\n  >(new Map());\n\n  const acRef = useRef<AbortController>(new AbortController());\n  const [controller] = useState(() => {\n    const [stream, controller] = createAssistantStreamController();\n    const transform = unstable_toolResultStream(\n      getTools,\n      () => acRef.current?.signal ?? new AbortController().signal,\n      (toolCallId: string, payload: unknown) => {\n        return new Promise<unknown>((resolve, reject) => {\n          // Reject previous human input request if it exists\n          const previous = humanInputRef.current.get(toolCallId);\n          if (previous) {\n            previous.reject(\n              new Error(\"Human input request was superseded by a new request\"),\n            );\n          }\n\n          humanInputRef.current.set(toolCallId, { resolve, reject });\n          setToolStatuses((prev) => ({\n            ...prev,\n            [toolCallId]: {\n              type: \"interrupt\",\n              payload: { type: \"human\", payload },\n            },\n          }));\n        });\n      },\n    );\n    stream\n      .pipeThrough(transform)\n      .pipeThrough(new AssistantMetaTransformStream())\n      .pipeTo(\n        new WritableStream({\n          write(chunk) {\n            if (chunk.type === \"result\") {\n              // the tool call result was already set by the backend\n              if (lastToolStates.current[chunk.meta.toolCallId]?.hasResult)\n                return;\n\n              onResult({\n                type: \"add-tool-result\",\n                toolCallId: chunk.meta.toolCallId,\n                toolName: chunk.meta.toolName,\n                result: chunk.result,\n                isError: chunk.isError,\n                ...(chunk.artifact && { artifact: chunk.artifact }),\n              });\n\n              // Clear status when result is set\n              setToolStatuses((prev) => {\n                const next = { ...prev };\n                delete next[chunk.meta.toolCallId];\n                return next;\n              });\n            }\n          },\n        }),\n      );\n\n    return controller;\n  });\n\n  const ignoredToolIds = useRef<Set<string>>(new Set());\n  const isInititialState = useRef(true);\n\n  useEffect(() => {\n    const processMessages = (\n      messages: readonly (typeof state.messages)[number][],\n    ) => {\n      messages.forEach((message) => {\n        message.content.forEach((content) => {\n          if (content.type === \"tool-call\") {\n            if (isInititialState.current) {\n              ignoredToolIds.current.add(content.toolCallId);\n            } else {\n              if (ignoredToolIds.current.has(content.toolCallId)) {\n                return;\n              }\n              let lastState = lastToolStates.current[content.toolCallId];\n              if (!lastState) {\n                const toolCallController = controller.addToolCallPart({\n                  toolName: content.toolName,\n                  toolCallId: content.toolCallId,\n                });\n                lastState = {\n                  argsText: \"\",\n                  hasResult: false,\n                  controller: toolCallController,\n                };\n                lastToolStates.current[content.toolCallId] = lastState;\n              }\n\n              if (content.argsText !== lastState.argsText) {\n                if (!content.argsText.startsWith(lastState.argsText)) {\n                  throw new Error(\n                    `Tool call argsText can only be appended, not updated: ${content.argsText} does not start with ${lastState.argsText}`,\n                  );\n                }\n\n                const argsTextDelta = content.argsText.slice(\n                  lastState.argsText.length,\n                );\n                lastState.controller.argsText.append(argsTextDelta);\n\n                if (isArgsTextComplete(content.argsText)) {\n                  lastState.controller.argsText.close();\n                }\n\n                lastToolStates.current[content.toolCallId] = {\n                  argsText: content.argsText,\n                  hasResult: lastState.hasResult,\n                  controller: lastState.controller,\n                };\n              }\n\n              if (content.result !== undefined && !lastState.hasResult) {\n                lastState.controller.setResponse(\n                  new ToolResponse({\n                    result: content.result as ReadonlyJSONValue,\n                    artifact: content.artifact as ReadonlyJSONValue | undefined,\n                    isError: content.isError,\n                  }),\n                );\n                lastState.controller.close();\n\n                lastToolStates.current[content.toolCallId] = {\n                  hasResult: true,\n                  argsText: lastState.argsText,\n                  controller: lastState.controller,\n                };\n              }\n            }\n\n            // Recursively process nested messages\n            if (content.messages) {\n              processMessages(content.messages);\n            }\n          }\n        });\n      });\n    };\n\n    processMessages(state.messages);\n\n    if (isInititialState.current) {\n      isInititialState.current = false;\n    }\n  }, [state, controller, onResult]);\n\n  const abort = () => {\n    humanInputRef.current.forEach(({ reject }) => {\n      reject(new Error(\"Tool execution aborted\"));\n    });\n    humanInputRef.current.clear();\n    setToolStatuses({});\n\n    acRef.current.abort();\n    acRef.current = new AbortController();\n  };\n\n  return {\n    reset: () => {\n      abort();\n      isInititialState.current = true;\n    },\n    abort,\n    resume: (toolCallId: string, payload: unknown) => {\n      const handlers = humanInputRef.current.get(toolCallId);\n      if (handlers) {\n        humanInputRef.current.delete(toolCallId);\n        setToolStatuses((prev) => {\n          const next = { ...prev };\n          delete next[toolCallId];\n          return next;\n        });\n        handlers.resolve(payload);\n      } else {\n        throw new Error(\n          `Tool call ${toolCallId} is not waiting for human input`,\n        );\n      }\n    },\n  };\n}\n"],"mappings":";AAAA,SAAS,WAAW,QAAQ,gBAAgB;AAC5C;AAAA,EACE;AAAA,EAEA;AAAA,EACA;AAAA,OAEK;AAKP;AAAA,EACE;AAAA,OAEK;AAEP,IAAM,qBAAqB,CAAC,aAAqB;AAC/C,MAAI;AACF,SAAK,MAAM,QAAQ;AACnB,WAAO;AAAA,EACT,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAmBO,SAAS,mBAAmB;AAAA,EACjC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAA6B;AAC3B,QAAM,iBAAiB,OASrB,CAAC,CAAC;AAEJ,QAAM,gBAAgB,OAQpB,oBAAI,IAAI,CAAC;AAEX,QAAM,QAAQ,OAAwB,IAAI,gBAAgB,CAAC;AAC3D,QAAM,CAAC,UAAU,IAAI,SAAS,MAAM;AAClC,UAAM,CAAC,QAAQA,WAAU,IAAI,gCAAgC;AAC7D,UAAM,YAAY;AAAA,MAChB;AAAA,MACA,MAAM,MAAM,SAAS,UAAU,IAAI,gBAAgB,EAAE;AAAA,MACrD,CAAC,YAAoB,YAAqB;AACxC,eAAO,IAAI,QAAiB,CAAC,SAAS,WAAW;AAE/C,gBAAM,WAAW,cAAc,QAAQ,IAAI,UAAU;AACrD,cAAI,UAAU;AACZ,qBAAS;AAAA,cACP,IAAI,MAAM,qDAAqD;AAAA,YACjE;AAAA,UACF;AAEA,wBAAc,QAAQ,IAAI,YAAY,EAAE,SAAS,OAAO,CAAC;AACzD,0BAAgB,CAAC,UAAU;AAAA,YACzB,GAAG;AAAA,YACH,CAAC,UAAU,GAAG;AAAA,cACZ,MAAM;AAAA,cACN,SAAS,EAAE,MAAM,SAAS,QAAQ;AAAA,YACpC;AAAA,UACF,EAAE;AAAA,QACJ,CAAC;AAAA,MACH;AAAA,IACF;AACA,WACG,YAAY,SAAS,EACrB,YAAY,IAAI,6BAA6B,CAAC,EAC9C;AAAA,MACC,IAAI,eAAe;AAAA,QACjB,MAAM,OAAO;AACX,cAAI,MAAM,SAAS,UAAU;AAE3B,gBAAI,eAAe,QAAQ,MAAM,KAAK,UAAU,GAAG;AACjD;AAEF,qBAAS;AAAA,cACP,MAAM;AAAA,cACN,YAAY,MAAM,KAAK;AAAA,cACvB,UAAU,MAAM,KAAK;AAAA,cACrB,QAAQ,MAAM;AAAA,cACd,SAAS,MAAM;AAAA,cACf,GAAI,MAAM,YAAY,EAAE,UAAU,MAAM,SAAS;AAAA,YACnD,CAAC;AAGD,4BAAgB,CAAC,SAAS;AACxB,oBAAM,OAAO,EAAE,GAAG,KAAK;AACvB,qBAAO,KAAK,MAAM,KAAK,UAAU;AACjC,qBAAO;AAAA,YACT,CAAC;AAAA,UACH;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAEF,WAAOA;AAAA,EACT,CAAC;AAED,QAAM,iBAAiB,OAAoB,oBAAI,IAAI,CAAC;AACpD,QAAM,mBAAmB,OAAO,IAAI;AAEpC,YAAU,MAAM;AACd,UAAM,kBAAkB,CACtB,aACG;AACH,eAAS,QAAQ,CAAC,YAAY;AAC5B,gBAAQ,QAAQ,QAAQ,CAAC,YAAY;AACnC,cAAI,QAAQ,SAAS,aAAa;AAChC,gBAAI,iBAAiB,SAAS;AAC5B,6BAAe,QAAQ,IAAI,QAAQ,UAAU;AAAA,YAC/C,OAAO;AACL,kBAAI,eAAe,QAAQ,IAAI,QAAQ,UAAU,GAAG;AAClD;AAAA,cACF;AACA,kBAAI,YAAY,eAAe,QAAQ,QAAQ,UAAU;AACzD,kBAAI,CAAC,WAAW;AACd,sBAAM,qBAAqB,WAAW,gBAAgB;AAAA,kBACpD,UAAU,QAAQ;AAAA,kBAClB,YAAY,QAAQ;AAAA,gBACtB,CAAC;AACD,4BAAY;AAAA,kBACV,UAAU;AAAA,kBACV,WAAW;AAAA,kBACX,YAAY;AAAA,gBACd;AACA,+BAAe,QAAQ,QAAQ,UAAU,IAAI;AAAA,cAC/C;AAEA,kBAAI,QAAQ,aAAa,UAAU,UAAU;AAC3C,oBAAI,CAAC,QAAQ,SAAS,WAAW,UAAU,QAAQ,GAAG;AACpD,wBAAM,IAAI;AAAA,oBACR,yDAAyD,QAAQ,QAAQ,wBAAwB,UAAU,QAAQ;AAAA,kBACrH;AAAA,gBACF;AAEA,sBAAM,gBAAgB,QAAQ,SAAS;AAAA,kBACrC,UAAU,SAAS;AAAA,gBACrB;AACA,0BAAU,WAAW,SAAS,OAAO,aAAa;AAElD,oBAAI,mBAAmB,QAAQ,QAAQ,GAAG;AACxC,4BAAU,WAAW,SAAS,MAAM;AAAA,gBACtC;AAEA,+BAAe,QAAQ,QAAQ,UAAU,IAAI;AAAA,kBAC3C,UAAU,QAAQ;AAAA,kBAClB,WAAW,UAAU;AAAA,kBACrB,YAAY,UAAU;AAAA,gBACxB;AAAA,cACF;AAEA,kBAAI,QAAQ,WAAW,UAAa,CAAC,UAAU,WAAW;AACxD,0BAAU,WAAW;AAAA,kBACnB,IAAI,aAAa;AAAA,oBACf,QAAQ,QAAQ;AAAA,oBAChB,UAAU,QAAQ;AAAA,oBAClB,SAAS,QAAQ;AAAA,kBACnB,CAAC;AAAA,gBACH;AACA,0BAAU,WAAW,MAAM;AAE3B,+BAAe,QAAQ,QAAQ,UAAU,IAAI;AAAA,kBAC3C,WAAW;AAAA,kBACX,UAAU,UAAU;AAAA,kBACpB,YAAY,UAAU;AAAA,gBACxB;AAAA,cACF;AAAA,YACF;AAGA,gBAAI,QAAQ,UAAU;AACpB,8BAAgB,QAAQ,QAAQ;AAAA,YAClC;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAEA,oBAAgB,MAAM,QAAQ;AAE9B,QAAI,iBAAiB,SAAS;AAC5B,uBAAiB,UAAU;AAAA,IAC7B;AAAA,EACF,GAAG,CAAC,OAAO,YAAY,QAAQ,CAAC;AAEhC,QAAM,QAAQ,MAAM;AAClB,kBAAc,QAAQ,QAAQ,CAAC,EAAE,OAAO,MAAM;AAC5C,aAAO,IAAI,MAAM,wBAAwB,CAAC;AAAA,IAC5C,CAAC;AACD,kBAAc,QAAQ,MAAM;AAC5B,oBAAgB,CAAC,CAAC;AAElB,UAAM,QAAQ,MAAM;AACpB,UAAM,UAAU,IAAI,gBAAgB;AAAA,EACtC;AAEA,SAAO;AAAA,IACL,OAAO,MAAM;AACX,YAAM;AACN,uBAAiB,UAAU;AAAA,IAC7B;AAAA,IACA;AAAA,IACA,QAAQ,CAAC,YAAoB,YAAqB;AAChD,YAAM,WAAW,cAAc,QAAQ,IAAI,UAAU;AACrD,UAAI,UAAU;AACZ,sBAAc,QAAQ,OAAO,UAAU;AACvC,wBAAgB,CAAC,SAAS;AACxB,gBAAM,OAAO,EAAE,GAAG,KAAK;AACvB,iBAAO,KAAK,UAAU;AACtB,iBAAO;AAAA,QACT,CAAC;AACD,iBAAS,QAAQ,OAAO;AAAA,MAC1B,OAAO;AACL,cAAM,IAAI;AAAA,UACR,aAAa,UAAU;AAAA,QACzB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["controller"]}

package/dist/types/MessagePartTypes.d.ts

@@ -1,4 +1,5 @@
 import { ReadonlyJSONObject } from "assistant-stream/utils";
+import type { ThreadMessage } from "./AssistantTypes";
 export type TextMessagePart = {
     readonly type: "text";
     readonly text: string;
@@ -49,6 +50,7 @@ export type ToolCallMessagePart<TArgs = ReadonlyJSONObject, TResult = unknown> =
         payload: unknown;
     };
     readonly parentId?: string;
+    readonly messages?: readonly ThreadMessage[];
 };
 export type ThreadUserMessagePart = TextMessagePart | ImageMessagePart | FileMessagePart | Unstable_AudioMessagePart;
 export type ThreadAssistantMessagePart = TextMessagePart | ReasoningMessagePart | ToolCallMessagePart | SourceMessagePart | FileMessagePart | ImageMessagePart;

package/dist/types/MessagePartTypes.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessagePartTypes.d.ts","sourceRoot":"","sources":["../../src/types/MessagePartTypes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,MAAM,MAAM,eAAe,GAAG;IAC5B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC;IAC3B,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE;QACd,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;QACtB,QAAQ,CAAC,MAAM,EAAE,KAAK,GAAG,KAAK,CAAC;KAChC,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,mBAAmB,CAC7B,KAAK,GAAG,kBAAkB,EAC1B,OAAO,GAAG,OAAO,IACf;IACF,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;IAC3B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;IACrB,QAAQ,CAAC,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACtC,QAAQ,CAAC,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACvC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,SAAS,CAAC,EAAE;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IACzD,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAC7B,eAAe,GACf,gBAAgB,GAChB,eAAe,GACf,yBAAyB,CAAC;AAE9B,MAAM,MAAM,0BAA0B,GAClC,eAAe,GACf,oBAAoB,GACpB,mBAAmB,GACnB,iBAAiB,GACjB,eAAe,GACf,gBAAgB,CAAC"}
+{"version":3,"file":"MessagePartTypes.d.ts","sourceRoot":"","sources":["../../src/types/MessagePartTypes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEtD,MAAM,MAAM,eAAe,GAAG;IAC5B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC;IAC3B,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IACvB,QAAQ,CAAC,KAAK,EAAE;QACd,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;QACtB,QAAQ,CAAC,MAAM,EAAE,KAAK,GAAG,KAAK,CAAC;KAChC,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,mBAAmB,CAC7B,KAAK,GAAG,kBAAkB,EAC1B,OAAO,GAAG,OAAO,IACf;IACF,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAC;IAC3B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;IACrB,QAAQ,CAAC,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACtC,QAAQ,CAAC,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACvC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,SAAS,CAAC,EAAE;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IACzD,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,QAAQ,CAAC,EAAE,SAAS,aAAa,EAAE,CAAC;CAC9C,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAC7B,eAAe,GACf,gBAAgB,GAChB,eAAe,GACf,yBAAyB,CAAC;AAE9B,MAAM,MAAM,0BAA0B,GAClC,eAAe,GACf,oBAAoB,GACpB,mBAAmB,GACnB,iBAAiB,GACjB,eAAe,GACf,gBAAgB,CAAC"}

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.11.24",
+  "version": "0.11.25",
   "license": "MIT",
   "type": "module",
   "exports": {

package/src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.ts

@@ -133,78 +133,86 @@ export function useToolInvocations({
   const isInititialState = useRef(true);
 
   useEffect(() => {
-    if (isInititialState.current) {
-      state.messages.forEach((message) => {
+    const processMessages = (
+      messages: readonly (typeof state.messages)[number][],
+    ) => {
+      messages.forEach((message) => {
         message.content.forEach((content) => {
           if (content.type === "tool-call") {
-            ignoredToolIds.current.add(content.toolCallId);
-          }
-        });
-      });
-      isInititialState.current = false;
-    } else {
-      state.messages.forEach((message) => {
-        message.content.forEach((content) => {
-          if (content.type === "tool-call") {
-            if (ignoredToolIds.current.has(content.toolCallId)) {
-              return;
-            }
-            let lastState = lastToolStates.current[content.toolCallId];
-            if (!lastState) {
-              const toolCallController = controller.addToolCallPart({
-                toolName: content.toolName,
-                toolCallId: content.toolCallId,
-              });
-              lastState = {
-                argsText: "",
-                hasResult: false,
-                controller: toolCallController,
-              };
-              lastToolStates.current[content.toolCallId] = lastState;
-            }
+            if (isInititialState.current) {
+              ignoredToolIds.current.add(content.toolCallId);
+            } else {
+              if (ignoredToolIds.current.has(content.toolCallId)) {
+                return;
+              }
+              let lastState = lastToolStates.current[content.toolCallId];
+              if (!lastState) {
+                const toolCallController = controller.addToolCallPart({
+                  toolName: content.toolName,
+                  toolCallId: content.toolCallId,
+                });
+                lastState = {
+                  argsText: "",
+                  hasResult: false,
+                  controller: toolCallController,
+                };
+                lastToolStates.current[content.toolCallId] = lastState;
+              }
+
+              if (content.argsText !== lastState.argsText) {
+                if (!content.argsText.startsWith(lastState.argsText)) {
+                  throw new Error(
+                    `Tool call argsText can only be appended, not updated: ${content.argsText} does not start with ${lastState.argsText}`,
+                  );
+                }
 
-            if (content.argsText !== lastState.argsText) {
-              if (!content.argsText.startsWith(lastState.argsText)) {
-                throw new Error(
-                  `Tool call argsText can only be appended, not updated: ${content.argsText} does not start with ${lastState.argsText}`,
+                const argsTextDelta = content.argsText.slice(
+                  lastState.argsText.length,
                 );
-              }
+                lastState.controller.argsText.append(argsTextDelta);
 
-              const argsTextDelta = content.argsText.slice(
-                lastState.argsText.length,
-              );
-              lastState.controller.argsText.append(argsTextDelta);
+                if (isArgsTextComplete(content.argsText)) {
+                  lastState.controller.argsText.close();
+                }
 
-              if (isArgsTextComplete(content.argsText)) {
-                lastState.controller.argsText.close();
+                lastToolStates.current[content.toolCallId] = {
+                  argsText: content.argsText,
+                  hasResult: lastState.hasResult,
+                  controller: lastState.controller,
+                };
               }
 
-              lastToolStates.current[content.toolCallId] = {
-                argsText: content.argsText,
-                hasResult: lastState.hasResult,
-                controller: lastState.controller,
-              };
+              if (content.result !== undefined && !lastState.hasResult) {
+                lastState.controller.setResponse(
+                  new ToolResponse({
+                    result: content.result as ReadonlyJSONValue,
+                    artifact: content.artifact as ReadonlyJSONValue | undefined,
+                    isError: content.isError,
+                  }),
+                );
+                lastState.controller.close();
+
+                lastToolStates.current[content.toolCallId] = {
+                  hasResult: true,
+                  argsText: lastState.argsText,
+                  controller: lastState.controller,
+                };
+              }
             }
 
-            if (content.result !== undefined && !lastState.hasResult) {
-              lastState.controller.setResponse(
-                new ToolResponse({
-                  result: content.result as ReadonlyJSONValue,
-                  artifact: content.artifact as ReadonlyJSONValue | undefined,
-                  isError: content.isError,
-                }),
-              );
-              lastState.controller.close();
-
-              lastToolStates.current[content.toolCallId] = {
-                hasResult: true,
-                argsText: lastState.argsText,
-                controller: lastState.controller,
-              };
+            // Recursively process nested messages
+            if (content.messages) {
+              processMessages(content.messages);
             }
           }
         });
       });
+    };
+
+    processMessages(state.messages);
+
+    if (isInititialState.current) {
+      isInititialState.current = false;
     }
   }, [state, controller, onResult]);
 

package/src/types/MessagePartTypes.ts

@@ -1,4 +1,5 @@
 import { ReadonlyJSONObject } from "assistant-stream/utils";
+import type { ThreadMessage } from "./AssistantTypes";
 
 export type TextMessagePart = {
   readonly type: "text";
@@ -56,6 +57,7 @@ export type ToolCallMessagePart<
   readonly artifact?: unknown;
   readonly interrupt?: { type: "human"; payload: unknown };
   readonly parentId?: string;
+  readonly messages?: readonly ThreadMessage[];
 };
 
 export type ThreadUserMessagePart =