assistant-stream 0.3.26 → 0.3.28

package/dist/core/utils/stream/merge.js.map

@@ -1 +1 @@
-{"version":3,"file":"merge.js","names":[],"sources":["../../../../src/core/utils/stream/merge.ts"],"sourcesContent":["import type { AssistantStreamChunk } from \"../../AssistantStreamChunk\";\nimport { promiseWithResolvers } from \"../../../utils/promiseWithResolvers\";\n\ntype MergeStreamItem = {\n  reader: ReadableStreamDefaultReader<AssistantStreamChunk>;\n  promise?: Promise<unknown> | undefined;\n};\n\nexport const createMergeStream = () => {\n  const list: MergeStreamItem[] = [];\n  let sealed = false;\n  let controller: ReadableStreamDefaultController<AssistantStreamChunk>;\n  let currentPull: ReturnType<typeof promiseWithResolvers<void>> | undefined;\n\n  const handlePull = (item: MergeStreamItem) => {\n    if (!item.promise) {\n      // TODO for most streams, we can directly pipeTo to avoid the microTask queue\n      // add an option to eagerly pipe the stream to the merge stream\n      // ideally, using assistant-stream w sync run method + piping to a sync WritableStream runs in the same microtask\n      // this is useful because we often use AssistantStreams internally as a serialization utility, e. g. AssistantTransformStream\n      // idea: avoid reader.read() by instead using a WritableStream & if (!hasPendingPull) await waitForPull()?\n      item.promise = item.reader\n        .read()\n        .then(({ done, value }) => {\n          item.promise = undefined;\n          if (done) {\n            list.splice(list.indexOf(item), 1);\n            if (sealed && list.length === 0) {\n              controller.close();\n            }\n          } else {\n            controller.enqueue(value);\n          }\n\n          currentPull?.resolve();\n          currentPull = undefined;\n        })\n        .catch((e) => {\n          console.error(e);\n\n          list.forEach((item) => {\n            item.reader.cancel();\n          });\n          list.length = 0;\n\n          controller.error(e);\n\n          currentPull?.reject(e);\n          currentPull = undefined;\n        });\n    }\n  };\n\n  const readable = new ReadableStream<AssistantStreamChunk>({\n    start(c) {\n      controller = c;\n    },\n    pull() {\n      currentPull = promiseWithResolvers();\n      list.forEach((item) => {\n        handlePull(item);\n      });\n\n      return currentPull.promise;\n    },\n    cancel() {\n      list.forEach((item) => {\n        item.reader.cancel();\n      });\n      list.length = 0;\n    },\n  });\n\n  return {\n    readable,\n    isSealed() {\n      return sealed;\n    },\n    seal() {\n      sealed = true;\n      if (list.length === 0) controller.close();\n    },\n    addStream(stream: ReadableStream<AssistantStreamChunk>) {\n      if (sealed)\n        throw new Error(\n          \"Cannot add streams after the run callback has settled.\",\n        );\n\n      const item = { reader: stream.getReader() };\n      list.push(item);\n      handlePull(item);\n    },\n    enqueue(chunk: AssistantStreamChunk) {\n      this.addStream(\n        new ReadableStream({\n          start(c) {\n            c.enqueue(chunk);\n            c.close();\n          },\n        }),\n      );\n    },\n  };\n};\n\n// TODO\n// export class SpanContainerMerger {\n//   public get isSealed() {\n//     return this.mergeStream.isSealed();\n//   }\n\n//   public get readable() {\n//     return this.mergeStream.readable;\n//   }\n\n//   private subAllocator = new Counter();\n//   private mergeStream = createMergeStream();\n\n//   constructor() {\n//     // id 0 is auto allocated\n//     this.subAllocator.up();\n//   }\n\n//   add(stream: ReadableStream<AssistantStreamChunk>) {\n//     this.mergeStream.addStream(\n//       stream.pipeThrough(new SpanParentEncoder(this.subAllocator)),\n//     );\n//   }\n\n//   enqueue(chunk: AssistantStreamChunk & { parentId: 0 }) {\n//     this.mergeStream.addStream(\n//       new ReadableStream({\n//         start(c) {\n//           c.enqueue(chunk);\n//           c.close();\n//         },\n//       }),\n//     );\n//   }\n\n//   seal() {\n//     this.mergeStream.seal();\n//   }\n// }\n\n// export class SpanContainerSplitter {\n//   public writable;\n\n//   private isSealed = false;\n//   private writers = new Map<\n//     number,\n//     WritableStreamDefaultWriter<AssistantStreamChunk>\n//   >();\n\n//   private closeTasks: Promise<void>[] = [];\n\n//   private allocator = new Counter();\n//   private subAllocator = new Counter();\n\n//   constructor() {\n//     // id 0 is auto-allocated\n//     this.allocator.up();\n\n//     this.writable = new WritableStream({\n//       write: (chunk) => {\n//         const { type, parentId } = chunk;\n\n//         const writer = this.writers.get(parentId);\n//         if (writer === undefined) throw new Error(\"Parent id not found\");\n\n//         writer.write(chunk);\n\n//         if (type === \"span\") {\n//           // allocate a new span id\n//           this.writers.set(this.allocator.up(), writer);\n//         }\n//         if (type === \"finish\") {\n//           this.writers.delete(parentId);\n//           writer.close();\n\n//           if (this.writers.size === 0) {\n//             const closeTask = this.writable.close();\n//             this.closeTasks.push(closeTask);\n//             closeTask.then(() => {\n//               this.closeTasks.splice(this.closeTasks.indexOf(closeTask), 1);\n//             });\n//           }\n//         }\n//       },\n//       close: async () => {\n//         if (this.writers.size > 0) throw new Error(\"Not all writers closed\");\n\n//         // await and throw on any errors\n//         await Promise.all(this.closeTasks);\n//       },\n//     });\n//   }\n\n//   add(stream: WritableStream<AssistantStreamChunk>) {\n//     if (this.isSealed) throw new Error(\"Cannot add streams after sealing\");\n\n//     const decoder = new SpanParentDecoder(this.subAllocator);\n//     decoder.readable.pipeTo(stream);\n\n//     this.writers.set(this.allocator.up(), decoder.writable.getWriter());\n//   }\n\n//   seal() {\n//     this.isSealed = true;\n//     if (this.writers.size === 0) this.writable.close();\n//   }\n// }\n"],"mappings":";;AAQA,MAAa,0BAA0B;CACrC,MAAM,OAA0B,CAAC;CACjC,IAAI,SAAS;CACb,IAAI;CACJ,IAAI;CAEJ,MAAM,cAAc,SAA0B;EAC5C,IAAI,CAAC,KAAK,SAMR,KAAK,UAAU,KAAK,OACjB,KAAK,CAAC,CACN,MAAM,EAAE,MAAM,YAAY;GACzB,KAAK,UAAU,KAAA;GACf,IAAI,MAAM;IACR,KAAK,OAAO,KAAK,QAAQ,IAAI,GAAG,CAAC;IACjC,IAAI,UAAU,KAAK,WAAW,GAC5B,WAAW,MAAM;GAErB,OACE,WAAW,QAAQ,KAAK;GAG1B,aAAa,QAAQ;GACrB,cAAc,KAAA;EAChB,CAAC,CAAC,CACD,OAAO,MAAM;GACZ,QAAQ,MAAM,CAAC;GAEf,KAAK,SAAS,SAAS;IACrB,KAAK,OAAO,OAAO;GACrB,CAAC;GACD,KAAK,SAAS;GAEd,WAAW,MAAM,CAAC;GAElB,aAAa,OAAO,CAAC;GACrB,cAAc,KAAA;EAChB,CAAC;CAEP;CAsBA,OAAO;EACL,UAAA,IArBmB,eAAqC;GACxD,MAAM,GAAG;IACP,aAAa;GACf;GACA,OAAO;IACL,cAAc,qBAAqB;IACnC,KAAK,SAAS,SAAS;KACrB,WAAW,IAAI;IACjB,CAAC;IAED,OAAO,YAAY;GACrB;GACA,SAAS;IACP,KAAK,SAAS,SAAS;KACrB,KAAK,OAAO,OAAO;IACrB,CAAC;IACD,KAAK,SAAS;GAChB;EACF,CAGS;EACP,WAAW;GACT,OAAO;EACT;EACA,OAAO;GACL,SAAS;GACT,IAAI,KAAK,WAAW,GAAG,WAAW,MAAM;EAC1C;EACA,UAAU,QAA8C;GACtD,IAAI,QACF,MAAM,IAAI,MACR,wDACF;GAEF,MAAM,OAAO,EAAE,QAAQ,OAAO,UAAU,EAAE;GAC1C,KAAK,KAAK,IAAI;GACd,WAAW,IAAI;EACjB;EACA,QAAQ,OAA6B;GACnC,KAAK,UACH,IAAI,eAAe,EACjB,MAAM,GAAG;IACP,EAAE,QAAQ,KAAK;IACf,EAAE,MAAM;GACV,EACF,CAAC,CACH;EACF;CACF;AACF"}
+{"version":3,"file":"merge.js","names":[],"sources":["../../../../src/core/utils/stream/merge.ts"],"sourcesContent":["import type { AssistantStreamChunk } from \"../../AssistantStreamChunk\";\nimport { promiseWithResolvers } from \"../../../utils/promiseWithResolvers\";\n\ntype MergeStreamItem = {\n  reader: ReadableStreamDefaultReader<AssistantStreamChunk>;\n  promise?: Promise<unknown> | undefined;\n};\n\nexport const createMergeStream = () => {\n  const list: MergeStreamItem[] = [];\n  let sealed = false;\n  let cancelled = false;\n  let errored = false;\n  let controller: ReadableStreamDefaultController<AssistantStreamChunk>;\n  let currentPull: ReturnType<typeof promiseWithResolvers<void>> | undefined;\n\n  const cancelAllReaders = () => {\n    list.forEach((item) => {\n      void item.reader.cancel().catch(() => undefined);\n    });\n    list.length = 0;\n  };\n\n  const handlePull = (item: MergeStreamItem) => {\n    if (!item.promise) {\n      // TODO for most streams, we can directly pipeTo to avoid the microTask queue\n      // add an option to eagerly pipe the stream to the merge stream\n      // ideally, using assistant-stream w sync run method + piping to a sync WritableStream runs in the same microtask\n      // this is useful because we often use AssistantStreams internally as a serialization utility, e. g. AssistantTransformStream\n      // idea: avoid reader.read() by instead using a WritableStream & if (!hasPendingPull) await waitForPull()?\n      item.promise = item.reader\n        .read()\n        .then(({ done, value }) => {\n          item.promise = undefined;\n          if (cancelled || errored) return;\n\n          if (done) {\n            list.splice(list.indexOf(item), 1);\n            if (sealed && list.length === 0) {\n              controller.close();\n            }\n          } else {\n            controller.enqueue(value);\n          }\n\n          currentPull?.resolve();\n          currentPull = undefined;\n        })\n        .catch((e) => {\n          if (cancelled || errored) return;\n\n          errored = true;\n          console.error(e);\n          cancelAllReaders();\n\n          controller.error(e);\n\n          currentPull?.reject(e);\n          currentPull = undefined;\n        });\n    }\n  };\n\n  const readable = new ReadableStream<AssistantStreamChunk>({\n    start(c) {\n      controller = c;\n    },\n    pull() {\n      currentPull = promiseWithResolvers();\n      list.forEach((item) => {\n        handlePull(item);\n      });\n\n      return currentPull.promise;\n    },\n    cancel() {\n      cancelled = true;\n      cancelAllReaders();\n      currentPull?.resolve();\n      currentPull = undefined;\n    },\n  });\n\n  return {\n    readable,\n    isSealed() {\n      return sealed;\n    },\n    isCancelled() {\n      return cancelled;\n    },\n    isErrored() {\n      return errored;\n    },\n    seal() {\n      if (cancelled || errored) return;\n      sealed = true;\n      if (list.length === 0) controller.close();\n    },\n    addStream(stream: ReadableStream<AssistantStreamChunk>) {\n      if (cancelled || errored) {\n        void stream.cancel().catch(() => undefined);\n        return;\n      }\n\n      if (sealed)\n        throw new Error(\n          \"Cannot add streams after the run callback has settled.\",\n        );\n\n      const item = { reader: stream.getReader() };\n      list.push(item);\n      handlePull(item);\n    },\n    enqueue(chunk: AssistantStreamChunk) {\n      this.addStream(\n        new ReadableStream({\n          start(c) {\n            c.enqueue(chunk);\n            c.close();\n          },\n        }),\n      );\n    },\n  };\n};\n\n// TODO\n// export class SpanContainerMerger {\n//   public get isSealed() {\n//     return this.mergeStream.isSealed();\n//   }\n\n//   public get readable() {\n//     return this.mergeStream.readable;\n//   }\n\n//   private subAllocator = new Counter();\n//   private mergeStream = createMergeStream();\n\n//   constructor() {\n//     // id 0 is auto allocated\n//     this.subAllocator.up();\n//   }\n\n//   add(stream: ReadableStream<AssistantStreamChunk>) {\n//     this.mergeStream.addStream(\n//       stream.pipeThrough(new SpanParentEncoder(this.subAllocator)),\n//     );\n//   }\n\n//   enqueue(chunk: AssistantStreamChunk & { parentId: 0 }) {\n//     this.mergeStream.addStream(\n//       new ReadableStream({\n//         start(c) {\n//           c.enqueue(chunk);\n//           c.close();\n//         },\n//       }),\n//     );\n//   }\n\n//   seal() {\n//     this.mergeStream.seal();\n//   }\n// }\n\n// export class SpanContainerSplitter {\n//   public writable;\n\n//   private isSealed = false;\n//   private writers = new Map<\n//     number,\n//     WritableStreamDefaultWriter<AssistantStreamChunk>\n//   >();\n\n//   private closeTasks: Promise<void>[] = [];\n\n//   private allocator = new Counter();\n//   private subAllocator = new Counter();\n\n//   constructor() {\n//     // id 0 is auto-allocated\n//     this.allocator.up();\n\n//     this.writable = new WritableStream({\n//       write: (chunk) => {\n//         const { type, parentId } = chunk;\n\n//         const writer = this.writers.get(parentId);\n//         if (writer === undefined) throw new Error(\"Parent id not found\");\n\n//         writer.write(chunk);\n\n//         if (type === \"span\") {\n//           // allocate a new span id\n//           this.writers.set(this.allocator.up(), writer);\n//         }\n//         if (type === \"finish\") {\n//           this.writers.delete(parentId);\n//           writer.close();\n\n//           if (this.writers.size === 0) {\n//             const closeTask = this.writable.close();\n//             this.closeTasks.push(closeTask);\n//             closeTask.then(() => {\n//               this.closeTasks.splice(this.closeTasks.indexOf(closeTask), 1);\n//             });\n//           }\n//         }\n//       },\n//       close: async () => {\n//         if (this.writers.size > 0) throw new Error(\"Not all writers closed\");\n\n//         // await and throw on any errors\n//         await Promise.all(this.closeTasks);\n//       },\n//     });\n//   }\n\n//   add(stream: WritableStream<AssistantStreamChunk>) {\n//     if (this.isSealed) throw new Error(\"Cannot add streams after sealing\");\n\n//     const decoder = new SpanParentDecoder(this.subAllocator);\n//     decoder.readable.pipeTo(stream);\n\n//     this.writers.set(this.allocator.up(), decoder.writable.getWriter());\n//   }\n\n//   seal() {\n//     this.isSealed = true;\n//     if (this.writers.size === 0) this.writable.close();\n//   }\n// }\n"],"mappings":";;AAQA,MAAa,0BAA0B;CACrC,MAAM,OAA0B,CAAC;CACjC,IAAI,SAAS;CACb,IAAI,YAAY;CAChB,IAAI,UAAU;CACd,IAAI;CACJ,IAAI;CAEJ,MAAM,yBAAyB;EAC7B,KAAK,SAAS,SAAS;GACrB,KAAU,OAAO,OAAO,CAAC,CAAC,YAAY,KAAA,CAAS;EACjD,CAAC;EACD,KAAK,SAAS;CAChB;CAEA,MAAM,cAAc,SAA0B;EAC5C,IAAI,CAAC,KAAK,SAMR,KAAK,UAAU,KAAK,OACjB,KAAK,CAAC,CACN,MAAM,EAAE,MAAM,YAAY;GACzB,KAAK,UAAU,KAAA;GACf,IAAI,aAAa,SAAS;GAE1B,IAAI,MAAM;IACR,KAAK,OAAO,KAAK,QAAQ,IAAI,GAAG,CAAC;IACjC,IAAI,UAAU,KAAK,WAAW,GAC5B,WAAW,MAAM;GAErB,OACE,WAAW,QAAQ,KAAK;GAG1B,aAAa,QAAQ;GACrB,cAAc,KAAA;EAChB,CAAC,CAAC,CACD,OAAO,MAAM;GACZ,IAAI,aAAa,SAAS;GAE1B,UAAU;GACV,QAAQ,MAAM,CAAC;GACf,iBAAiB;GAEjB,WAAW,MAAM,CAAC;GAElB,aAAa,OAAO,CAAC;GACrB,cAAc,KAAA;EAChB,CAAC;CAEP;CAsBA,OAAO;EACL,UAAA,IArBmB,eAAqC;GACxD,MAAM,GAAG;IACP,aAAa;GACf;GACA,OAAO;IACL,cAAc,qBAAqB;IACnC,KAAK,SAAS,SAAS;KACrB,WAAW,IAAI;IACjB,CAAC;IAED,OAAO,YAAY;GACrB;GACA,SAAS;IACP,YAAY;IACZ,iBAAiB;IACjB,aAAa,QAAQ;IACrB,cAAc,KAAA;GAChB;EACF,CAGS;EACP,WAAW;GACT,OAAO;EACT;EACA,cAAc;GACZ,OAAO;EACT;EACA,YAAY;GACV,OAAO;EACT;EACA,OAAO;GACL,IAAI,aAAa,SAAS;GAC1B,SAAS;GACT,IAAI,KAAK,WAAW,GAAG,WAAW,MAAM;EAC1C;EACA,UAAU,QAA8C;GACtD,IAAI,aAAa,SAAS;IACxB,OAAY,OAAO,CAAC,CAAC,YAAY,KAAA,CAAS;IAC1C;GACF;GAEA,IAAI,QACF,MAAM,IAAI,MACR,wDACF;GAEF,MAAM,OAAO,EAAE,QAAQ,OAAO,UAAU,EAAE;GAC1C,KAAK,KAAK,IAAI;GACd,WAAW,IAAI;EACjB;EACA,QAAQ,OAA6B;GACnC,KAAK,UACH,IAAI,eAAe,EACjB,MAAM,GAAG;IACP,EAAE,QAAQ,KAAK;IACf,EAAE,MAAM;GACV,EACF,CAAC,CACH;EACF;CACF;AACF"}

package/dist/core/utils/stream/path-utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"path-utils.d.ts","names":[],"sources":["../../../../src/core/utils/stream/path-utils.ts"],"mappings":";;;cAGa,0BAA0B,gBACrC,sBACA;cAEY;;cAYD,0BAA0B,gBACrC,sBACA;cAEY;;cAmBD,yBAAyB,gBACpC,sBACA;cAEY,SAAS"}
+{"version":3,"file":"path-utils.d.ts","names":[],"sources":["../../../../src/core/utils/stream/path-utils.ts"],"mappings":";;;cAGa,0BAA0B,gBACrC,sBACA;EAEY,YAAA;;cAYD,0BAA0B,gBACrC,sBACA;EAEY,YAAA;;cAmBD,yBAAyB,gBACpC,sBACA;EAEY,YAAA,SAAS"}

package/dist/index.d.ts

@@ -1,17 +1,19 @@
+import { ReadonlyJSONValue } from "./utils/json/json-value.js";
 import { TextStreamController } from "./core/modules/text.js";
 import { McpServerConfig, ProviderOptions, Tool, ToolCallReader, ToolDeclaration, ToolModelContentPart, ToolModelOutputFunction } from "./core/tool/tool-types.js";
 import { ToolResponse, ToolResponseLike } from "./core/tool/ToolResponse.js";
 import { ToolCallStreamController } from "./core/modules/tool-call.js";
 import { AssistantMessage, AssistantMessageTiming, DataPart, ToolCallTiming } from "./core/utils/types.js";
 import { AssistantStreamController, createAssistantStream, createAssistantStreamController, createAssistantStreamResponse } from "./core/modules/assistant-stream.js";
-import { ObjectStreamChunk } from "./core/object/types.js";
+import "./utils.js";
+import { AssistantTransportStateOperation, GorpStreamChunk, GorpStreamOperation } from "./core/gorp/types.js";
 import { AssistantStreamChunk } from "./core/AssistantStreamChunk.js";
 import { AssistantStream } from "./core/AssistantStream.js";
 import { AssistantMessageStream } from "./core/accumulators/AssistantMessageStream.js";
 import { AssistantMessageAccumulator, createInitialMessage } from "./core/accumulators/assistant-message-accumulator.js";
 import { GenericAssistantMessage, GenericFilePart, GenericMessage, GenericSystemMessage, GenericTextPart, GenericToolCallPart, GenericToolMessage, GenericToolResultPart, GenericUserMessage, toGenericMessages } from "./core/converters/toGenericMessages.js";
-import { ObjectStreamResponse, fromObjectStreamResponse } from "./core/object/ObjectStreamResponse.js";
-import { createObjectStream } from "./core/object/createObjectStream.js";
+import { GorpStreamDeltaTracker } from "./core/gorp/GorpStreamDeltaTracker.js";
+import { GorpStreamResponse } from "./core/gorp/GorpStreamResponse.js";
 import { PlainTextDecoder, PlainTextEncoder } from "./core/serialization/PlainText.js";
 import { AssistantTransportDecoder, AssistantTransportEncoder } from "./core/serialization/assistant-transport/AssistantTransport.js";
 import { DataStreamDecoder, DataStreamEncoder } from "./core/serialization/data-stream/DataStream.js";
@@ -20,4 +22,23 @@ import { UIMessageStreamDecoder, UIMessageStreamDecoderOptions } from "./core/se
 import { ToolExecutionStream } from "./core/tool/ToolExecutionStream.js";
 import { ToToolsJSONSchemaOptions, ToolJSONSchema, toJSONSchema, toPartialJSONSchema, toToolsJSONSchema } from "./core/tool/schema-utils.js";
 import { ToolResultStreamOptions, toolResultStream, unstable_runPendingTools } from "./core/tool/toolResultStream.js";
-export { type AssistantMessage, AssistantMessageAccumulator, AssistantMessageStream, type AssistantMessageTiming, AssistantStream, type AssistantStreamChunk, type AssistantStreamController, AssistantTransportDecoder, AssistantTransportEncoder, type DataPart, DataStreamDecoder, DataStreamEncoder, type GenericAssistantMessage, type GenericFilePart, type GenericMessage, type GenericSystemMessage, type GenericTextPart, type GenericToolCallPart, type GenericToolMessage, type GenericToolResultPart, type GenericUserMessage, type McpServerConfig, type ObjectStreamChunk, ObjectStreamResponse, PlainTextDecoder, PlainTextEncoder, type ProviderOptions, type TextStreamController, type ToToolsJSONSchemaOptions, type Tool, type ToolCallReader, type ToolCallStreamController, type ToolCallTiming, type ToolDeclaration, ToolExecutionStream, type ToolJSONSchema, type ToolModelContentPart, type ToolModelOutputFunction, ToolResponse, type ToolResponseLike, type ToolResultStreamOptions, type UIMessageStreamChunk, type UIMessageStreamDataChunk, UIMessageStreamDecoder, type UIMessageStreamDecoderOptions, createAssistantStream, createAssistantStreamController, createAssistantStreamResponse, createObjectStream, fromObjectStreamResponse, toGenericMessages, toJSONSchema, toPartialJSONSchema, toToolsJSONSchema, createInitialMessage as unstable_createInitialMessage, unstable_runPendingTools, toolResultStream as unstable_toolResultStream };
+//#region src/index.d.ts
+/** @deprecated Use the assistant-transport surface instead. */
+declare const createObjectStream: ({ execute, defaultValue }: {
+  execute: (controller: {
+    readonly abortSignal: AbortSignal;
+    enqueue(operations: readonly GorpStreamOperation[]): void;
+  }) => void | PromiseLike<void>;
+  defaultValue?: ReadonlyJSONValue;
+}) => ReadableStream<GorpStreamChunk>;
+/** @deprecated Use the assistant-transport surface instead. */
+declare const ObjectStreamResponse: typeof GorpStreamResponse;
+/** @deprecated Use the assistant-transport surface instead. */
+type ObjectStreamResponse = GorpStreamResponse;
+/** @deprecated Use the assistant-transport surface instead. */
+declare const fromObjectStreamResponse: (response: Response) => ReadableStream<GorpStreamChunk>;
+/** @deprecated Use the assistant-transport surface instead. */
+type ObjectStreamChunk = GorpStreamChunk;
+//#endregion
+export { type AssistantMessage, AssistantMessageAccumulator, AssistantMessageStream, type AssistantMessageTiming, AssistantStream, type AssistantStreamChunk, type AssistantStreamController, AssistantTransportDecoder, GorpStreamDeltaTracker as AssistantTransportDeltaTracker, AssistantTransportEncoder, type AssistantTransportStateOperation, type DataPart, DataStreamDecoder, DataStreamEncoder, type GenericAssistantMessage, type GenericFilePart, type GenericMessage, type GenericSystemMessage, type GenericTextPart, type GenericToolCallPart, type GenericToolMessage, type GenericToolResultPart, type GenericUserMessage, type McpServerConfig, ObjectStreamChunk, ObjectStreamResponse, PlainTextDecoder, PlainTextEncoder, type ProviderOptions, type TextStreamController, type ToToolsJSONSchemaOptions, type Tool, type ToolCallReader, type ToolCallStreamController, type ToolCallTiming, type ToolDeclaration, ToolExecutionStream, type ToolJSONSchema, type ToolModelContentPart, type ToolModelOutputFunction, ToolResponse, type ToolResponseLike, type ToolResultStreamOptions, type UIMessageStreamChunk, type UIMessageStreamDataChunk, UIMessageStreamDecoder, type UIMessageStreamDecoderOptions, createAssistantStream, createAssistantStreamController, createAssistantStreamResponse, createObjectStream, fromObjectStreamResponse, toGenericMessages, toJSONSchema, toPartialJSONSchema, toToolsJSONSchema, createInitialMessage as unstable_createInitialMessage, unstable_runPendingTools, toolResultStream as unstable_toolResultStream };
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","names":[],"sources":["../src/index.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;cA4Ea,uBAAkB,SAAA;;;;;;MAAmB,eAAA;;cAErC,6BAAoB;;KAErB,uBAAuB;;cAEtB,2BAAwB,UAAA,aAAA,eAAA;;KAEzB,oBAAoB"}

package/dist/index.js

@@ -10,7 +10,18 @@ import { ToolResponse } from "./core/tool/ToolResponse.js";
 import { ToolExecutionStream } from "./core/tool/ToolExecutionStream.js";
 import { toolResultStream, unstable_runPendingTools } from "./core/tool/toolResultStream.js";
 import { toJSONSchema, toPartialJSONSchema, toToolsJSONSchema } from "./core/tool/schema-utils.js";
-import { createObjectStream } from "./core/object/createObjectStream.js";
-import { ObjectStreamResponse, fromObjectStreamResponse } from "./core/object/ObjectStreamResponse.js";
+import { createGorpStream } from "./core/gorp/createGorpStream.js";
+import { GorpStreamResponse, fromGorpStreamResponse } from "./core/gorp/GorpStreamResponse.js";
+import { GorpStreamDeltaTracker } from "./core/gorp/GorpStreamDeltaTracker.js";
 import { toGenericMessages } from "./core/converters/toGenericMessages.js";
-export { AssistantMessageAccumulator, AssistantMessageStream, AssistantStream, AssistantTransportDecoder, AssistantTransportEncoder, DataStreamDecoder, DataStreamEncoder, ObjectStreamResponse, PlainTextDecoder, PlainTextEncoder, ToolExecutionStream, ToolResponse, UIMessageStreamDecoder, createAssistantStream, createAssistantStreamController, createAssistantStreamResponse, createObjectStream, fromObjectStreamResponse, toGenericMessages, toJSONSchema, toPartialJSONSchema, toToolsJSONSchema, createInitialMessage as unstable_createInitialMessage, unstable_runPendingTools, toolResultStream as unstable_toolResultStream };
+//#region src/index.ts
+/** @deprecated Use the assistant-transport surface instead. */
+const createObjectStream = createGorpStream;
+/** @deprecated Use the assistant-transport surface instead. */
+const ObjectStreamResponse = GorpStreamResponse;
+/** @deprecated Use the assistant-transport surface instead. */
+const fromObjectStreamResponse = fromGorpStreamResponse;
+//#endregion
+export { AssistantMessageAccumulator, AssistantMessageStream, AssistantStream, AssistantTransportDecoder, GorpStreamDeltaTracker as AssistantTransportDeltaTracker, AssistantTransportEncoder, DataStreamDecoder, DataStreamEncoder, ObjectStreamResponse, PlainTextDecoder, PlainTextEncoder, ToolExecutionStream, ToolResponse, UIMessageStreamDecoder, createAssistantStream, createAssistantStreamController, createAssistantStreamResponse, createObjectStream, fromObjectStreamResponse, toGenericMessages, toJSONSchema, toPartialJSONSchema, toToolsJSONSchema, createInitialMessage as unstable_createInitialMessage, unstable_runPendingTools, toolResultStream as unstable_toolResultStream };
+
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["export {\n  createAssistantStream,\n  createAssistantStreamResponse,\n  createAssistantStreamController,\n} from \"./core/modules/assistant-stream\";\nexport {\n  AssistantMessageAccumulator,\n  createInitialMessage as unstable_createInitialMessage,\n} from \"./core/accumulators/assistant-message-accumulator\";\nexport { AssistantStream } from \"./core/AssistantStream\";\nexport type { AssistantStreamController } from \"./core/modules/assistant-stream\";\nexport type { AssistantStreamChunk } from \"./core/AssistantStreamChunk\";\nexport {\n  DataStreamDecoder,\n  DataStreamEncoder,\n} from \"./core/serialization/data-stream/DataStream\";\nexport {\n  PlainTextDecoder,\n  PlainTextEncoder,\n} from \"./core/serialization/PlainText\";\nexport {\n  AssistantTransportDecoder,\n  AssistantTransportEncoder,\n} from \"./core/serialization/assistant-transport/AssistantTransport\";\nexport {\n  UIMessageStreamDecoder,\n  type UIMessageStreamChunk,\n  type UIMessageStreamDataChunk,\n  type UIMessageStreamDecoderOptions,\n} from \"./core/serialization/ui-message-stream/UIMessageStream\";\nexport { AssistantMessageStream } from \"./core/accumulators/AssistantMessageStream\";\nexport type {\n  AssistantMessage,\n  AssistantMessageTiming,\n  DataPart,\n  ToolCallTiming,\n} from \"./core/utils/types\";\n\nexport type {\n  Tool,\n  ToolDeclaration,\n  McpServerConfig,\n  ToolModelContentPart,\n  ToolModelOutputFunction,\n} from \"./core/tool/tool-types\";\nexport { ToolResponse, type ToolResponseLike } from \"./core/tool/ToolResponse\";\nexport { ToolExecutionStream } from \"./core/tool/ToolExecutionStream\";\nexport type { ProviderOptions, ToolCallReader } from \"./core/tool/tool-types\";\nexport {\n  toolResultStream as unstable_toolResultStream,\n  unstable_runPendingTools,\n  type ToolResultStreamOptions,\n} from \"./core/tool/toolResultStream\";\nexport {\n  toJSONSchema,\n  toPartialJSONSchema,\n  toToolsJSONSchema,\n  type ToolJSONSchema,\n  type ToToolsJSONSchemaOptions,\n} from \"./core/tool/schema-utils\";\n\nexport type { TextStreamController } from \"./core/modules/text\";\nexport type { ToolCallStreamController } from \"./core/modules/tool-call\";\n\nimport { createGorpStream } from \"./core/gorp/createGorpStream\";\nimport {\n  GorpStreamResponse,\n  fromGorpStreamResponse,\n} from \"./core/gorp/GorpStreamResponse\";\nimport type { GorpStreamChunk } from \"./core/gorp/types\";\nimport { GorpStreamDeltaTracker } from \"./core/gorp/GorpStreamDeltaTracker\";\n\nexport { GorpStreamDeltaTracker as AssistantTransportDeltaTracker };\nexport type { AssistantTransportStateOperation } from \"./core/gorp/types\";\n\n/** @deprecated Use the assistant-transport surface instead. */\nexport const createObjectStream = createGorpStream;\n/** @deprecated Use the assistant-transport surface instead. */\nexport const ObjectStreamResponse = GorpStreamResponse;\n/** @deprecated Use the assistant-transport surface instead. */\nexport type ObjectStreamResponse = GorpStreamResponse;\n/** @deprecated Use the assistant-transport surface instead. */\nexport const fromObjectStreamResponse = fromGorpStreamResponse;\n/** @deprecated Use the assistant-transport surface instead. */\nexport type ObjectStreamChunk = GorpStreamChunk;\n\nexport {\n  toGenericMessages,\n  type GenericMessage,\n  type GenericSystemMessage,\n  type GenericUserMessage,\n  type GenericAssistantMessage,\n  type GenericToolMessage,\n  type GenericTextPart,\n  type GenericFilePart,\n  type GenericToolCallPart,\n  type GenericToolResultPart,\n} from \"./core/converters/toGenericMessages\";\n"],"mappings":";;;;;;;;;;;;;;;;;;AA4EA,MAAa,qBAAqB;;AAElC,MAAa,uBAAuB;;AAIpC,MAAa,2BAA2B"}

package/dist/resumable/ResumableStreamContext.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ResumableStreamContext.d.ts","names":[],"sources":["../../src/resumable/ResumableStreamContext.ts"],"mappings":";;KAOY;WACD,OAAO;WACP;;;;;;WAMA,aAAa,SAAS;WACtB,aAAa,kBAAkB,MAAM;WACrC,YAAY,kBAAkB;WAC9B,cACP,kBACA,0BACA;WAEO,WAAW,kBAAkB;;UAGvB;;EAEf,IACE,kBACA,kBAAkB,eAAe,cAChC,QAAQ,eAAe;;EAG1B,OAAO,mBAAmB,QAAQ,eAAe;;EAGjD,cAAc,mBAAmB,QAAQ,eAAe;EAExD,OAAO,mBAAmB,QAAQ;EAElC,OAAO,mBAAmB;;iBAGZ,6BACd,SAAS,gCACR"}
+{"version":3,"file":"ResumableStreamContext.d.ts","names":[],"sources":["../../src/resumable/ResumableStreamContext.ts"],"mappings":";;KAQY;WACD,OAAO;WACP;;;;;;WAMA,aAAa,SAAS;WACtB,aAAa,kBAAkB,MAAM;WACrC,YAAY,kBAAkB;WAC9B,cACP,kBACA,0BACA;WAEO,WAAW,kBAAkB;;UAGvB;;EAEf,IACE,kBACA,kBAAkB,eAAe,cAChC,QAAQ,eAAe;;EAG1B,OAAO,mBAAmB,QAAQ,eAAe;;EAGjD,cAAc,mBAAmB,QAAQ,eAAe;EAExD,OAAO,mBAAmB,QAAQ;EAElC,OAAO,mBAAmB;;iBAkBZ,6BACd,SAAS,gCACR"}

package/dist/resumable/ResumableStreamContext.js

@@ -1,12 +1,19 @@
+import { withPromiseOrValue } from "../core/utils/withPromiseOrValue.js";
 import { ResumableStreamError } from "./errors.js";
 //#region src/resumable/ResumableStreamContext.ts
+const invokeObservabilityHook = (name, hook, ...args) => {
+	if (!hook) return;
+	withPromiseOrValue(() => hook(...args), () => {}, (error) => {
+		console.error(`resumable stream ${name} hook failed:`, error);
+	});
+};
 function createResumableStreamContext(options) {
 	const { store, waitUntil, onAcquire, onAppend, onFinalize, onError } = options;
 	const acquireOptions = options.ttlMs !== void 0 ? { ttlMs: options.ttlMs } : void 0;
 	return {
 		async run(streamId, makeStream) {
 			const role = await store.acquire(streamId, acquireOptions);
-			onAcquire?.(streamId, role);
+			invokeObservabilityHook("onAcquire", onAcquire, streamId, role);
 			if (role === "producer") startProducerTask(store, streamId, makeStream, {
 				waitUntil,
 				onAppend,
@@ -42,13 +49,13 @@ function startProducerTask(store, streamId, makeStream, hooks) {
 				const { done, value } = await reader.read();
 				if (done) break;
 				await store.append(streamId, value);
-				onAppend?.(streamId, value.byteLength);
+				invokeObservabilityHook("onAppend", onAppend, streamId, value.byteLength);
 			}
 			await store.finalize(streamId, "done");
-			onFinalize?.(streamId, "done");
+			invokeObservabilityHook("onFinalize", onFinalize, streamId, "done");
 		} catch (err) {
 			cancelled = true;
-			onError?.(streamId, err);
+			invokeObservabilityHook("onError", onError, streamId, err);
 			const message = err instanceof Error ? err.message : String(err);
 			try {
 				await reader?.cancel(err);
@@ -57,7 +64,7 @@ function startProducerTask(store, streamId, makeStream, hooks) {
 			}
 			try {
 				await store.finalize(streamId, "error", message);
-				onFinalize?.(streamId, "error", message);
+				invokeObservabilityHook("onFinalize", onFinalize, streamId, "error", message);
 			} catch (finalizeErr) {
 				console.error("resumable stream finalize failed:", finalizeErr);
 			}

package/dist/resumable/ResumableStreamContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"ResumableStreamContext.js","names":[],"sources":["../../src/resumable/ResumableStreamContext.ts"],"sourcesContent":["import { ResumableStreamError } from \"./errors\";\nimport type {\n  ResumableStreamRole,\n  ResumableStreamStatus,\n  ResumableStreamStore,\n} from \"./types\";\n\nexport type ResumableStreamContextOptions = {\n  readonly store: ResumableStreamStore;\n  readonly ttlMs?: number;\n  /**\n   * Required on serverless runtimes that terminate background work when the\n   * request handler returns (Vercel, Cloudflare). Pass `after` from\n   * `next/server` so the producer task outlives the response.\n   */\n  readonly waitUntil?: (promise: Promise<unknown>) => void;\n  readonly onAcquire?: (streamId: string, role: ResumableStreamRole) => void;\n  readonly onAppend?: (streamId: string, byteLength: number) => void;\n  readonly onFinalize?: (\n    streamId: string,\n    status: \"done\" | \"error\",\n    error?: string,\n  ) => void;\n  readonly onError?: (streamId: string, error: unknown) => void;\n};\n\nexport interface ResumableStreamContext {\n  /** Producer or consumer entrypoint. Atomically elects the role. */\n  run(\n    streamId: string,\n    makeStream: () => ReadableStream<Uint8Array>,\n  ): Promise<ReadableStream<Uint8Array>>;\n\n  /** Returns `null` when the stream does not exist. */\n  resume(streamId: string): Promise<ReadableStream<Uint8Array> | null>;\n\n  /** Throws `ResumableStreamError(\"missing\")` when the stream does not exist. */\n  requireResume(streamId: string): Promise<ReadableStream<Uint8Array>>;\n\n  status(streamId: string): Promise<ResumableStreamStatus>;\n\n  delete(streamId: string): Promise<void>;\n}\n\nexport function createResumableStreamContext(\n  options: ResumableStreamContextOptions,\n): ResumableStreamContext {\n  const { store, waitUntil, onAcquire, onAppend, onFinalize, onError } =\n    options;\n  const acquireOptions =\n    options.ttlMs !== undefined ? { ttlMs: options.ttlMs } : undefined;\n\n  return {\n    async run(streamId, makeStream) {\n      const role = await store.acquire(streamId, acquireOptions);\n      onAcquire?.(streamId, role);\n      if (role === \"producer\") {\n        startProducerTask(store, streamId, makeStream, {\n          waitUntil,\n          onAppend,\n          onFinalize,\n          onError,\n        });\n      }\n      return readFromStore(store, streamId);\n    },\n\n    async resume(streamId) {\n      const status = await store.status(streamId);\n      if (status === \"missing\") return null;\n      return readFromStore(store, streamId);\n    },\n\n    async requireResume(streamId) {\n      const status = await store.status(streamId);\n      if (status === \"missing\") {\n        throw new ResumableStreamError(\n          \"missing\",\n          `resumable stream not found: ${streamId}`,\n        );\n      }\n      return readFromStore(store, streamId);\n    },\n\n    async status(streamId) {\n      return store.status(streamId);\n    },\n\n    async delete(streamId) {\n      await store.delete(streamId);\n    },\n  };\n}\n\ntype ProducerHooks = {\n  readonly waitUntil: ((promise: Promise<unknown>) => void) | undefined;\n  readonly onAppend:\n    | ((streamId: string, byteLength: number) => void)\n    | undefined;\n  readonly onFinalize:\n    | ((streamId: string, status: \"done\" | \"error\", error?: string) => void)\n    | undefined;\n  readonly onError: ((streamId: string, error: unknown) => void) | undefined;\n};\n\nfunction startProducerTask(\n  store: ResumableStreamStore,\n  streamId: string,\n  makeStream: () => ReadableStream<Uint8Array>,\n  hooks: ProducerHooks,\n): void {\n  const { waitUntil, onAppend, onFinalize, onError } = hooks;\n  const task = (async () => {\n    let reader: ReadableStreamDefaultReader<Uint8Array> | undefined;\n    let cancelled = false;\n    try {\n      reader = makeStream().getReader();\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        await store.append(streamId, value);\n        onAppend?.(streamId, value.byteLength);\n      }\n      await store.finalize(streamId, \"done\");\n      onFinalize?.(streamId, \"done\");\n    } catch (err) {\n      cancelled = true;\n      onError?.(streamId, err);\n      const message = err instanceof Error ? err.message : String(err);\n      try {\n        await reader?.cancel(err);\n      } catch (cancelErr) {\n        console.error(\"resumable stream reader cancel failed:\", cancelErr);\n      }\n      try {\n        await store.finalize(streamId, \"error\", message);\n        onFinalize?.(streamId, \"error\", message);\n      } catch (finalizeErr) {\n        console.error(\"resumable stream finalize failed:\", finalizeErr);\n      }\n    } finally {\n      if (!cancelled) reader?.releaseLock();\n    }\n  })();\n\n  if (waitUntil) waitUntil(task);\n  task.catch((err) => {\n    console.error(\"resumable producer task failed:\", err);\n  });\n}\n\nfunction readFromStore(\n  store: ResumableStreamStore,\n  streamId: string,\n): ReadableStream<Uint8Array> {\n  const ac = new AbortController();\n  let iterator: AsyncIterator<{ chunk: Uint8Array }> | undefined;\n\n  return new ReadableStream<Uint8Array>({\n    start() {\n      iterator = store.read(streamId, \"\", ac.signal)[Symbol.asyncIterator]();\n    },\n    async pull(controller) {\n      try {\n        if (!iterator) return;\n        const { done, value } = await iterator.next();\n        if (done) {\n          controller.close();\n          return;\n        }\n        controller.enqueue(value.chunk);\n      } catch (err) {\n        // the platform never calls cancel() on errored streams, so unwind\n        // the store iterator and abort the signal explicitly.\n        ac.abort();\n        try {\n          await iterator?.return?.();\n        } catch {}\n        controller.error(err);\n      }\n    },\n    cancel() {\n      ac.abort();\n      iterator?.return?.().catch(() => {});\n    },\n  });\n}\n"],"mappings":";;AA4CA,SAAgB,6BACd,SACwB;CACxB,MAAM,EAAE,OAAO,WAAW,WAAW,UAAU,YAAY,YACzD;CACF,MAAM,iBACJ,QAAQ,UAAU,KAAA,IAAY,EAAE,OAAO,QAAQ,MAAM,IAAI,KAAA;CAE3D,OAAO;EACL,MAAM,IAAI,UAAU,YAAY;GAC9B,MAAM,OAAO,MAAM,MAAM,QAAQ,UAAU,cAAc;GACzD,YAAY,UAAU,IAAI;GAC1B,IAAI,SAAS,YACX,kBAAkB,OAAO,UAAU,YAAY;IAC7C;IACA;IACA;IACA;GACF,CAAC;GAEH,OAAO,cAAc,OAAO,QAAQ;EACtC;EAEA,MAAM,OAAO,UAAU;GAErB,IAAI,MADiB,MAAM,OAAO,QAAQ,MAC3B,WAAW,OAAO;GACjC,OAAO,cAAc,OAAO,QAAQ;EACtC;EAEA,MAAM,cAAc,UAAU;GAE5B,IAAI,MADiB,MAAM,OAAO,QAAQ,MAC3B,WACb,MAAM,IAAI,qBACR,WACA,+BAA+B,UACjC;GAEF,OAAO,cAAc,OAAO,QAAQ;EACtC;EAEA,MAAM,OAAO,UAAU;GACrB,OAAO,MAAM,OAAO,QAAQ;EAC9B;EAEA,MAAM,OAAO,UAAU;GACrB,MAAM,MAAM,OAAO,QAAQ;EAC7B;CACF;AACF;AAaA,SAAS,kBACP,OACA,UACA,YACA,OACM;CACN,MAAM,EAAE,WAAW,UAAU,YAAY,YAAY;CACrD,MAAM,QAAQ,YAAY;EACxB,IAAI;EACJ,IAAI,YAAY;EAChB,IAAI;GACF,SAAS,WAAW,CAAC,CAAC,UAAU;GAChC,OAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;IAC1C,IAAI,MAAM;IACV,MAAM,MAAM,OAAO,UAAU,KAAK;IAClC,WAAW,UAAU,MAAM,UAAU;GACvC;GACA,MAAM,MAAM,SAAS,UAAU,MAAM;GACrC,aAAa,UAAU,MAAM;EAC/B,SAAS,KAAK;GACZ,YAAY;GACZ,UAAU,UAAU,GAAG;GACvB,MAAM,UAAU,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;GAC/D,IAAI;IACF,MAAM,QAAQ,OAAO,GAAG;GAC1B,SAAS,WAAW;IAClB,QAAQ,MAAM,0CAA0C,SAAS;GACnE;GACA,IAAI;IACF,MAAM,MAAM,SAAS,UAAU,SAAS,OAAO;IAC/C,aAAa,UAAU,SAAS,OAAO;GACzC,SAAS,aAAa;IACpB,QAAQ,MAAM,qCAAqC,WAAW;GAChE;EACF,UAAU;GACR,IAAI,CAAC,WAAW,QAAQ,YAAY;EACtC;CACF,EAAA,CAAG;CAEH,IAAI,WAAW,UAAU,IAAI;CAC7B,KAAK,OAAO,QAAQ;EAClB,QAAQ,MAAM,mCAAmC,GAAG;CACtD,CAAC;AACH;AAEA,SAAS,cACP,OACA,UAC4B;CAC5B,MAAM,KAAK,IAAI,gBAAgB;CAC/B,IAAI;CAEJ,OAAO,IAAI,eAA2B;EACpC,QAAQ;GACN,WAAW,MAAM,KAAK,UAAU,IAAI,GAAG,MAAM,CAAC,CAAC,OAAO,cAAc,CAAC;EACvE;EACA,MAAM,KAAK,YAAY;GACrB,IAAI;IACF,IAAI,CAAC,UAAU;IACf,MAAM,EAAE,MAAM,UAAU,MAAM,SAAS,KAAK;IAC5C,IAAI,MAAM;KACR,WAAW,MAAM;KACjB;IACF;IACA,WAAW,QAAQ,MAAM,KAAK;GAChC,SAAS,KAAK;IAGZ,GAAG,MAAM;IACT,IAAI;KACF,MAAM,UAAU,SAAS;IAC3B,QAAQ,CAAC;IACT,WAAW,MAAM,GAAG;GACtB;EACF;EACA,SAAS;GACP,GAAG,MAAM;GACT,UAAU,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC;EACrC;CACF,CAAC;AACH"}
+{"version":3,"file":"ResumableStreamContext.js","names":[],"sources":["../../src/resumable/ResumableStreamContext.ts"],"sourcesContent":["import { ResumableStreamError } from \"./errors\";\nimport { withPromiseOrValue } from \"../core/utils/withPromiseOrValue\";\nimport type {\n  ResumableStreamRole,\n  ResumableStreamStatus,\n  ResumableStreamStore,\n} from \"./types\";\n\nexport type ResumableStreamContextOptions = {\n  readonly store: ResumableStreamStore;\n  readonly ttlMs?: number;\n  /**\n   * Required on serverless runtimes that terminate background work when the\n   * request handler returns (Vercel, Cloudflare). Pass `after` from\n   * `next/server` so the producer task outlives the response.\n   */\n  readonly waitUntil?: (promise: Promise<unknown>) => void;\n  readonly onAcquire?: (streamId: string, role: ResumableStreamRole) => void;\n  readonly onAppend?: (streamId: string, byteLength: number) => void;\n  readonly onFinalize?: (\n    streamId: string,\n    status: \"done\" | \"error\",\n    error?: string,\n  ) => void;\n  readonly onError?: (streamId: string, error: unknown) => void;\n};\n\nexport interface ResumableStreamContext {\n  /** Producer or consumer entrypoint. Atomically elects the role. */\n  run(\n    streamId: string,\n    makeStream: () => ReadableStream<Uint8Array>,\n  ): Promise<ReadableStream<Uint8Array>>;\n\n  /** Returns `null` when the stream does not exist. */\n  resume(streamId: string): Promise<ReadableStream<Uint8Array> | null>;\n\n  /** Throws `ResumableStreamError(\"missing\")` when the stream does not exist. */\n  requireResume(streamId: string): Promise<ReadableStream<Uint8Array>>;\n\n  status(streamId: string): Promise<ResumableStreamStatus>;\n\n  delete(streamId: string): Promise<void>;\n}\n\nconst invokeObservabilityHook = <TArgs extends unknown[]>(\n  name: string,\n  hook: ((...args: TArgs) => void) | undefined,\n  ...args: TArgs\n): void => {\n  if (!hook) return;\n  void withPromiseOrValue(\n    () => hook(...args),\n    () => {},\n    (error) => {\n      console.error(`resumable stream ${name} hook failed:`, error);\n    },\n  );\n};\n\nexport function createResumableStreamContext(\n  options: ResumableStreamContextOptions,\n): ResumableStreamContext {\n  const { store, waitUntil, onAcquire, onAppend, onFinalize, onError } =\n    options;\n  const acquireOptions =\n    options.ttlMs !== undefined ? { ttlMs: options.ttlMs } : undefined;\n\n  return {\n    async run(streamId, makeStream) {\n      const role = await store.acquire(streamId, acquireOptions);\n      invokeObservabilityHook(\"onAcquire\", onAcquire, streamId, role);\n      if (role === \"producer\") {\n        startProducerTask(store, streamId, makeStream, {\n          waitUntil,\n          onAppend,\n          onFinalize,\n          onError,\n        });\n      }\n      return readFromStore(store, streamId);\n    },\n\n    async resume(streamId) {\n      const status = await store.status(streamId);\n      if (status === \"missing\") return null;\n      return readFromStore(store, streamId);\n    },\n\n    async requireResume(streamId) {\n      const status = await store.status(streamId);\n      if (status === \"missing\") {\n        throw new ResumableStreamError(\n          \"missing\",\n          `resumable stream not found: ${streamId}`,\n        );\n      }\n      return readFromStore(store, streamId);\n    },\n\n    async status(streamId) {\n      return store.status(streamId);\n    },\n\n    async delete(streamId) {\n      await store.delete(streamId);\n    },\n  };\n}\n\ntype ProducerHooks = {\n  readonly waitUntil: ((promise: Promise<unknown>) => void) | undefined;\n  readonly onAppend:\n    | ((streamId: string, byteLength: number) => void)\n    | undefined;\n  readonly onFinalize:\n    | ((streamId: string, status: \"done\" | \"error\", error?: string) => void)\n    | undefined;\n  readonly onError: ((streamId: string, error: unknown) => void) | undefined;\n};\n\nfunction startProducerTask(\n  store: ResumableStreamStore,\n  streamId: string,\n  makeStream: () => ReadableStream<Uint8Array>,\n  hooks: ProducerHooks,\n): void {\n  const { waitUntil, onAppend, onFinalize, onError } = hooks;\n  const task = (async () => {\n    let reader: ReadableStreamDefaultReader<Uint8Array> | undefined;\n    let cancelled = false;\n    try {\n      reader = makeStream().getReader();\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        await store.append(streamId, value);\n        invokeObservabilityHook(\n          \"onAppend\",\n          onAppend,\n          streamId,\n          value.byteLength,\n        );\n      }\n      await store.finalize(streamId, \"done\");\n      invokeObservabilityHook(\"onFinalize\", onFinalize, streamId, \"done\");\n    } catch (err) {\n      cancelled = true;\n      invokeObservabilityHook(\"onError\", onError, streamId, err);\n      const message = err instanceof Error ? err.message : String(err);\n      try {\n        await reader?.cancel(err);\n      } catch (cancelErr) {\n        console.error(\"resumable stream reader cancel failed:\", cancelErr);\n      }\n      try {\n        await store.finalize(streamId, \"error\", message);\n        invokeObservabilityHook(\n          \"onFinalize\",\n          onFinalize,\n          streamId,\n          \"error\",\n          message,\n        );\n      } catch (finalizeErr) {\n        console.error(\"resumable stream finalize failed:\", finalizeErr);\n      }\n    } finally {\n      if (!cancelled) reader?.releaseLock();\n    }\n  })();\n\n  if (waitUntil) waitUntil(task);\n  task.catch((err) => {\n    console.error(\"resumable producer task failed:\", err);\n  });\n}\n\nfunction readFromStore(\n  store: ResumableStreamStore,\n  streamId: string,\n): ReadableStream<Uint8Array> {\n  const ac = new AbortController();\n  let iterator: AsyncIterator<{ chunk: Uint8Array }> | undefined;\n\n  return new ReadableStream<Uint8Array>({\n    start() {\n      iterator = store.read(streamId, \"\", ac.signal)[Symbol.asyncIterator]();\n    },\n    async pull(controller) {\n      try {\n        if (!iterator) return;\n        const { done, value } = await iterator.next();\n        if (done) {\n          controller.close();\n          return;\n        }\n        controller.enqueue(value.chunk);\n      } catch (err) {\n        // the platform never calls cancel() on errored streams, so unwind\n        // the store iterator and abort the signal explicitly.\n        ac.abort();\n        try {\n          await iterator?.return?.();\n        } catch {}\n        controller.error(err);\n      }\n    },\n    cancel() {\n      ac.abort();\n      iterator?.return?.().catch(() => {});\n    },\n  });\n}\n"],"mappings":";;;AA6CA,MAAM,2BACJ,MACA,MACA,GAAG,SACM;CACT,IAAI,CAAC,MAAM;CACX,yBACQ,KAAK,GAAG,IAAI,SACZ,CAAC,IACN,UAAU;EACT,QAAQ,MAAM,oBAAoB,KAAK,gBAAgB,KAAK;CAC9D,CACF;AACF;AAEA,SAAgB,6BACd,SACwB;CACxB,MAAM,EAAE,OAAO,WAAW,WAAW,UAAU,YAAY,YACzD;CACF,MAAM,iBACJ,QAAQ,UAAU,KAAA,IAAY,EAAE,OAAO,QAAQ,MAAM,IAAI,KAAA;CAE3D,OAAO;EACL,MAAM,IAAI,UAAU,YAAY;GAC9B,MAAM,OAAO,MAAM,MAAM,QAAQ,UAAU,cAAc;GACzD,wBAAwB,aAAa,WAAW,UAAU,IAAI;GAC9D,IAAI,SAAS,YACX,kBAAkB,OAAO,UAAU,YAAY;IAC7C;IACA;IACA;IACA;GACF,CAAC;GAEH,OAAO,cAAc,OAAO,QAAQ;EACtC;EAEA,MAAM,OAAO,UAAU;GAErB,IAAI,MADiB,MAAM,OAAO,QAAQ,MAC3B,WAAW,OAAO;GACjC,OAAO,cAAc,OAAO,QAAQ;EACtC;EAEA,MAAM,cAAc,UAAU;GAE5B,IAAI,MADiB,MAAM,OAAO,QAAQ,MAC3B,WACb,MAAM,IAAI,qBACR,WACA,+BAA+B,UACjC;GAEF,OAAO,cAAc,OAAO,QAAQ;EACtC;EAEA,MAAM,OAAO,UAAU;GACrB,OAAO,MAAM,OAAO,QAAQ;EAC9B;EAEA,MAAM,OAAO,UAAU;GACrB,MAAM,MAAM,OAAO,QAAQ;EAC7B;CACF;AACF;AAaA,SAAS,kBACP,OACA,UACA,YACA,OACM;CACN,MAAM,EAAE,WAAW,UAAU,YAAY,YAAY;CACrD,MAAM,QAAQ,YAAY;EACxB,IAAI;EACJ,IAAI,YAAY;EAChB,IAAI;GACF,SAAS,WAAW,CAAC,CAAC,UAAU;GAChC,OAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;IAC1C,IAAI,MAAM;IACV,MAAM,MAAM,OAAO,UAAU,KAAK;IAClC,wBACE,YACA,UACA,UACA,MAAM,UACR;GACF;GACA,MAAM,MAAM,SAAS,UAAU,MAAM;GACrC,wBAAwB,cAAc,YAAY,UAAU,MAAM;EACpE,SAAS,KAAK;GACZ,YAAY;GACZ,wBAAwB,WAAW,SAAS,UAAU,GAAG;GACzD,MAAM,UAAU,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;GAC/D,IAAI;IACF,MAAM,QAAQ,OAAO,GAAG;GAC1B,SAAS,WAAW;IAClB,QAAQ,MAAM,0CAA0C,SAAS;GACnE;GACA,IAAI;IACF,MAAM,MAAM,SAAS,UAAU,SAAS,OAAO;IAC/C,wBACE,cACA,YACA,UACA,SACA,OACF;GACF,SAAS,aAAa;IACpB,QAAQ,MAAM,qCAAqC,WAAW;GAChE;EACF,UAAU;GACR,IAAI,CAAC,WAAW,QAAQ,YAAY;EACtC;CACF,EAAA,CAAG;CAEH,IAAI,WAAW,UAAU,IAAI;CAC7B,KAAK,OAAO,QAAQ;EAClB,QAAQ,MAAM,mCAAmC,GAAG;CACtD,CAAC;AACH;AAEA,SAAS,cACP,OACA,UAC4B;CAC5B,MAAM,KAAK,IAAI,gBAAgB;CAC/B,IAAI;CAEJ,OAAO,IAAI,eAA2B;EACpC,QAAQ;GACN,WAAW,MAAM,KAAK,UAAU,IAAI,GAAG,MAAM,CAAC,CAAC,OAAO,cAAc,CAAC;EACvE;EACA,MAAM,KAAK,YAAY;GACrB,IAAI;IACF,IAAI,CAAC,UAAU;IACf,MAAM,EAAE,MAAM,UAAU,MAAM,SAAS,KAAK;IAC5C,IAAI,MAAM;KACR,WAAW,MAAM;KACjB;IACF;IACA,WAAW,QAAQ,MAAM,KAAK;GAChC,SAAS,KAAK;IAGZ,GAAG,MAAM;IACT,IAAI;KACF,MAAM,UAAU,SAAS;IAC3B,QAAQ,CAAC;IACT,WAAW,MAAM,GAAG;GACtB;EACF;EACA,SAAS;GACP,GAAG,MAAM;GACT,UAAU,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC;EACrC;CACF,CAAC;AACH"}

package/dist/resumable/errors.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"errors.d.ts","names":[],"sources":["../../src/resumable/errors.ts"],"mappings":";KAAY;cAMC,6BAA6B;WAC/B,MAAM;cAEH,MAAM,0BAA0B;;iBAS9B,iBAAiB"}
+{"version":3,"file":"errors.d.ts","names":[],"sources":["../../src/resumable/errors.ts"],"mappings":";KAAY;cAMC,6BAA6B;WAC/B,MAAM;EAEH,YAAA,MAAM,0BAA0B;;iBAS9B,iBAAiB"}

package/dist/resumable/stores/redis-impl.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"redis-impl.d.ts","names":[],"sources":["../../../src/resumable/stores/redis-impl.ts"],"mappings":";;KAsBY;WAEG;WACA;WACA,QAAQ,wBAAwB;;WAEhC;WAAyB;WAAsB;;WAE/C;WACA;WACA;WACA;;;;;;;UAQE;EACf,MAAM,aAAa,eAAe,iBAAiB;EACnD,IAAI,aAAa,eAAe,iBAAiB;EACjD,IAAI,cAAc;EAClB,OAAO,aAAa,iBAAiB;EACrC,OAAO,cAAc;EACrB,IAAI,iBAAiB;EACrB,KACE,aACA,QAAQ,wBAAwB,cAC/B;EACH,OACE,aACA,eACA,cACC,QACD;IAAQ;IAAY,QAAQ,wBAAwB;;;EAGtD,SAAS,mBAAmB,oBAAoB;;KAGtC;WACD;WACA;;WAEA;WACA;;cAGE,qCAAqC;mBAC/B;mBACA;mBACA;mBACA;mBACA;cAGf,QAAQ,iBACR,UAAS;EASL,QACJ,kBACA,UAAU,gCACT,QAAQ;EAiBL,OAAO,kBAAkB,OAAO,aAAa;EA8B7C,SACJ,kBACA,0BACA,iBACC;EA6BI,KACL,kBACA,gBACA,QAAQ,cACP,cAAc;EA2CX,OAAO,mBAAmB,QAAQ;EAWlC,OAAO,mBAAmB;UAKlB;UAQN;UAIA"}
+{"version":3,"file":"redis-impl.d.ts","names":[],"sources":["../../../src/resumable/stores/redis-impl.ts"],"mappings":";;KAsBY;WAEG;WACA;WACA,QAAQ,wBAAwB;;WAEhC;WAAyB;WAAsB;;WAE/C;WACA;WACA;WACA;;;;;;;UAQE;EACf,MAAM,aAAa,eAAe,iBAAiB;EACnD,IAAI,aAAa,eAAe,iBAAiB;EACjD,IAAI,cAAc;EAClB,OAAO,aAAa,iBAAiB;EACrC,OAAO,cAAc;EACrB,IAAI,iBAAiB;EACrB,KACE,aACA,QAAQ,wBAAwB,cAC/B;EACH,OACE,aACA,eACA,cACC,QACD;IAAQ;IAAY,QAAQ,wBAAwB;;;EAGtD,SAAS,mBAAmB,oBAAoB;;KAGtC;WACD;WACA;;WAEA;WACA;;cAGE,qCAAqC;mBAC/B;mBACA;mBACA;mBACA;mBACA;EAGf,YAAA,QAAQ,iBACR,UAAS;EASL,QACJ,kBACA,UAAU,gCACT,QAAQ;EAiBL,OAAO,kBAAkB,OAAO,aAAa;EA8B7C,SACJ,kBACA,0BACA,iBACC;EA6BI,KACL,kBACA,gBACA,QAAQ,cACP,cAAc;EA2CX,OAAO,mBAAmB,QAAQ;EAWlC,OAAO,mBAAmB;UAKlB;UAQN;UAIA"}

package/dist/utils/AsyncIterableStream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AsyncIterableStream.d.ts","names":[],"sources":["../../src/utils/AsyncIterableStream.ts"],"mappings":";KAAY,oBAAoB,KAAK,cAAc,KAAK,eAAe;iBAiBvD,sBAAsB,GACpC,QAAQ,eAAe,KACtB,oBAAoB"}
+{"version":3,"file":"AsyncIterableStream.d.ts","names":[],"sources":["../../src/utils/AsyncIterableStream.ts"],"mappings":";KAAY,oBAAoB,KAAK,cAAc,KAAK,eAAe;iBAgCvD,sBAAsB,GACpC,QAAQ,eAAe,KACtB,oBAAoB"}

package/dist/utils/AsyncIterableStream.js

@@ -1,14 +1,29 @@
 //#region src/utils/AsyncIterableStream.ts
 async function* streamGeneratorPolyfill() {
 	const reader = this.getReader();
+	let shouldCancel = true;
 	try {
 		while (true) {
-			const { done, value } = await reader.read();
-			if (done) break;
+			let result;
+			try {
+				result = await reader.read();
+			} catch (error) {
+				shouldCancel = false;
+				throw error;
+			}
+			if (result.done) {
+				shouldCancel = false;
+				break;
+			}
+			const { value } = result;
 			yield value;
 		}
 	} finally {
-		reader.releaseLock();
+		try {
+			if (shouldCancel) await reader.cancel();
+		} finally {
+			reader.releaseLock();
+		}
 	}
 }
 function asAsyncIterableStream(source) {

package/dist/utils/AsyncIterableStream.js.map

@@ -1 +1 @@
-{"version":3,"file":"AsyncIterableStream.js","names":[],"sources":["../../src/utils/AsyncIterableStream.ts"],"sourcesContent":["export type AsyncIterableStream<T> = AsyncIterable<T> & ReadableStream<T>;\n\nasync function* streamGeneratorPolyfill<T>(\n  this: ReadableStream<T>,\n): AsyncGenerator<T, undefined, unknown> {\n  const reader = this.getReader();\n  try {\n    while (true) {\n      const { done, value } = await reader.read();\n      if (done) break;\n      yield value;\n    }\n  } finally {\n    reader.releaseLock();\n  }\n}\n\nexport function asAsyncIterableStream<T>(\n  source: ReadableStream<T>,\n): AsyncIterableStream<T> {\n  (source as AsyncIterableStream<T>)[Symbol.asyncIterator] ??=\n    streamGeneratorPolyfill;\n  return source as AsyncIterableStream<T>;\n}\n"],"mappings":";AAEA,gBAAgB,0BAEyB;CACvC,MAAM,SAAS,KAAK,UAAU;CAC9B,IAAI;EACF,OAAO,MAAM;GACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;GAC1C,IAAI,MAAM;GACV,MAAM;EACR;CACF,UAAU;EACR,OAAO,YAAY;CACrB;AACF;AAEA,SAAgB,sBACd,QACwB;CACxB,OAAmC,OAAO,mBACxC;CACF,OAAO;AACT"}
+{"version":3,"file":"AsyncIterableStream.js","names":[],"sources":["../../src/utils/AsyncIterableStream.ts"],"sourcesContent":["export type AsyncIterableStream<T> = AsyncIterable<T> & ReadableStream<T>;\n\nasync function* streamGeneratorPolyfill<T>(\n  this: ReadableStream<T>,\n): AsyncGenerator<T, undefined, unknown> {\n  const reader = this.getReader();\n  let shouldCancel = true;\n  try {\n    while (true) {\n      let result: ReadableStreamReadResult<T>;\n      try {\n        result = await reader.read();\n      } catch (error) {\n        shouldCancel = false;\n        throw error;\n      }\n      if (result.done) {\n        shouldCancel = false;\n        break;\n      }\n      const { value } = result;\n      yield value;\n    }\n  } finally {\n    try {\n      if (shouldCancel) await reader.cancel();\n    } finally {\n      reader.releaseLock();\n    }\n  }\n}\n\nexport function asAsyncIterableStream<T>(\n  source: ReadableStream<T>,\n): AsyncIterableStream<T> {\n  (source as AsyncIterableStream<T>)[Symbol.asyncIterator] ??=\n    streamGeneratorPolyfill;\n  return source as AsyncIterableStream<T>;\n}\n"],"mappings":";AAEA,gBAAgB,0BAEyB;CACvC,MAAM,SAAS,KAAK,UAAU;CAC9B,IAAI,eAAe;CACnB,IAAI;EACF,OAAO,MAAM;GACX,IAAI;GACJ,IAAI;IACF,SAAS,MAAM,OAAO,KAAK;GAC7B,SAAS,OAAO;IACd,eAAe;IACf,MAAM;GACR;GACA,IAAI,OAAO,MAAM;IACf,eAAe;IACf;GACF;GACA,MAAM,EAAE,UAAU;GAClB,MAAM;EACR;CACF,UAAU;EACR,IAAI;GACF,IAAI,cAAc,MAAM,OAAO,OAAO;EACxC,UAAU;GACR,OAAO,YAAY;EACrB;CACF;AACF;AAEA,SAAgB,sBACd,QACwB;CACxB,OAAmC,OAAO,mBACxC;CACF,OAAO;AACT"}

package/dist/utils/promiseWithResolvers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"promiseWithResolvers.d.ts","names":[],"sources":["../../src/utils/promiseWithResolvers.ts"],"mappings":";cAAa,uBAAwB;;mBACb,IAAI,YAAY"}
+{"version":3,"file":"promiseWithResolvers.d.ts","names":[],"sources":["../../src/utils/promiseWithResolvers.ts"],"mappings":";cAAa,uBAAwB;;EACb,UAAA,OAAA,IAAI,YAAY;EACf,SAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-stream",
-  "version": "0.3.26",
+  "version": "0.3.28",
   "description": "Streaming utilities for AI assistants",
   "keywords": [
     "ai",
@@ -67,7 +67,7 @@
     "ioredis": "^5.11.1",
     "redis": "^6.1.0",
     "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.18"
+    "@assistant-ui/x-buildutils": "0.0.20"
   },
   "publishConfig": {
     "access": "public",

package/src/core/AssistantStreamChunk.ts

@@ -1,5 +1,5 @@
 import type { ReadonlyJSONValue } from "../utils/json/json-value";
-import type { ObjectStreamOperation } from "./object/types";
+import type { AssistantTransportStateOperation } from "./gorp/types";
 import type { ToolModelContentPart } from "./tool/tool-types";
 
 /**
@@ -136,8 +136,8 @@ export type AssistantStreamChunk = { readonly path: readonly number[] } & (
       readonly severity?: "critical" | "warning" | "info";
     }
   | {
-      /** Applies object-stream operations to state carried by this stream. */
+      /** Applies gorp-stream operations to state carried by this stream. */
       readonly type: "update-state";
-      readonly operations: ObjectStreamOperation[];
+      readonly operations: AssistantTransportStateOperation[];
     }
 );

package/src/core/accumulators/AssistantMessageStream.ts

@@ -3,7 +3,9 @@ import type { AssistantMessage } from "../utils/types";
 import { AssistantMessageAccumulator } from "./assistant-message-accumulator";
 
 export class AssistantMessageStream {
-  constructor(public readonly readable: ReadableStream<AssistantMessage>) {
+  public readonly readable: ReadableStream<AssistantMessage>;
+
+  constructor(readable: ReadableStream<AssistantMessage>) {
     this.readable = readable;
   }