assistant-stream 0.3.26 → 0.3.27

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.27",
   "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.19"
   },
   "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/assistant-message-accumulator.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect } from "vitest";
+import { describe, it, expect, vi } from "vitest";
 import { AssistantMessageAccumulator } from "./assistant-message-accumulator";
 import type { AssistantStreamChunk } from "../AssistantStreamChunk";
 import type { AssistantMessage } from "../utils/types";
@@ -224,6 +224,88 @@ describe("AssistantMessageAccumulator timing", () => {
     expect(last.metadata.timing).toBeDefined();
     expect(last.metadata.timing!.firstTokenTime).toBeTypeOf("number");
   });
+
+  it("does not record firstTokenTime when a text-delta is dropped for an out-of-range path", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [5], textDelta: "x" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.metadata.timing).toBeDefined();
+    expect(last.metadata.timing!.firstTokenTime).toBeUndefined();
+    warn.mockRestore();
+  });
+
+  it("does not record firstTokenTime when a text-delta is dropped for a nested path", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [0, 1], textDelta: "x" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.metadata.timing).toBeDefined();
+    expect(last.metadata.timing!.firstTokenTime).toBeUndefined();
+    warn.mockRestore();
+  });
+
+  it("does not record firstTokenTime when a text-delta is dropped for a wrong part type", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      {
+        type: "part-start",
+        path: [0],
+        part: {
+          type: "source",
+          sourceType: "url",
+          id: "s1",
+          url: "https://example.com",
+        },
+      },
+      { type: "text-delta", path: [0], textDelta: "x" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.metadata.timing).toBeDefined();
+    expect(last.metadata.timing!.firstTokenTime).toBeUndefined();
+    warn.mockRestore();
+  });
+
+  it("records firstTokenTime on the first applied text-delta, not a preceding dropped one", async () => {
+    vi.useFakeTimers();
+    vi.setSystemTime(1000);
+
+    const accumulator = new AssistantMessageAccumulator();
+    const writer = accumulator.writable.getWriter();
+    const messages: AssistantMessage[] = [];
+    const reading = (async () => {
+      const reader = accumulator.readable.getReader();
+      for (;;) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        messages.push(value);
+      }
+    })();
+
+    await writer.write({
+      type: "part-start",
+      path: [0],
+      part: { type: "text" },
+    });
+    vi.setSystemTime(2000);
+    await writer.write({ type: "text-delta", path: [5], textDelta: "x" });
+    vi.setSystemTime(3000);
+    await writer.write({ type: "text-delta", path: [0], textDelta: "hi" });
+    await writer.close();
+    await reading;
+    vi.useRealTimers();
+
+    const last = messages.at(-1)!;
+    expect(last.metadata.timing).toBeDefined();
+    expect(last.metadata.timing!.firstTokenTime).toBe(2000);
+  });
 });
 
 describe("AssistantMessageAccumulator error chunks", () => {
@@ -240,6 +322,19 @@ describe("AssistantMessageAccumulator error chunks", () => {
     });
   });
 
+  it("defaults a missing error message", async () => {
+    const messages = await collectStream([
+      { type: "error", path: [] } as unknown as AssistantStreamChunk,
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.status).toEqual({
+      type: "incomplete",
+      reason: "error",
+      error: { code: "unknown", message: "unknown error" },
+    });
+  });
+
   it("carries code and severity through unchanged", async () => {
     const messages = await collectStream([
       {
@@ -288,3 +383,281 @@ describe("AssistantMessageAccumulator error chunks", () => {
     });
   });
 });
+
+describe("AssistantMessageAccumulator part path bounds", () => {
+  it("drops a text-delta whose path is out of range with a warning", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [5], textDelta: "x" },
+      { type: "text-delta", path: [0], textDelta: "hi" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(1);
+    expect(last.parts[0]).toMatchObject({ type: "text", text: "hi" });
+    expect(warn).toHaveBeenCalledWith(
+      "Dropped text-delta chunk: no part at path [5]",
+    );
+    warn.mockRestore();
+  });
+
+  it("drops an out-of-range part-finish instead of fabricating a part", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "part-finish", path: [3] },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(1);
+    expect(last.parts[0]!.type).toBe("text");
+    warn.mockRestore();
+  });
+
+  it("drops part chunks when no parts exist", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "text-delta", path: [0], textDelta: "x" },
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [0], textDelta: "ok" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(1);
+    expect(last.parts[0]).toMatchObject({ type: "text", text: "ok" });
+    warn.mockRestore();
+  });
+
+  it("drops chunks with nested paths", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [0, 1], textDelta: "x" },
+      { type: "text-delta", path: [0], textDelta: "ok" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(1);
+    expect(last.parts[0]).toMatchObject({ type: "text", text: "ok" });
+    warn.mockRestore();
+  });
+
+  it("drops an out-of-range result without touching existing tool calls", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      {
+        type: "part-start",
+        path: [0],
+        part: { type: "tool-call", toolCallId: "t1", toolName: "f" },
+      },
+      { type: "text-delta", path: [0], textDelta: "{}" },
+      { type: "result", path: [7], result: { ok: true }, isError: false },
+      { type: "result", path: [0], result: { ok: true }, isError: false },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(1);
+    expect(last.parts[0]).toMatchObject({
+      type: "tool-call",
+      state: "result",
+      result: { ok: true },
+    });
+    warn.mockRestore();
+  });
+});
+
+describe("AssistantMessageAccumulator wrong part type chunks", () => {
+  const sourcePart = {
+    type: "source",
+    sourceType: "url",
+    id: "s1",
+    url: "https://example.com",
+  } as const;
+
+  it("drops a text-delta addressed to a source part", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: sourcePart },
+      { type: "text-delta", path: [0], textDelta: "x" },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts[0]).toMatchObject({ type: "source", id: "s1" });
+    expect(warn).toHaveBeenCalledWith(
+      "Dropped text-delta chunk: part is neither text nor tool-call",
+    );
+    warn.mockRestore();
+  });
+
+  it("drops a result addressed to a text part", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [0], textDelta: "hi" },
+      { type: "result", path: [0], result: { ok: true }, isError: false },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts[0]).toMatchObject({ type: "text", text: "hi" });
+    expect(warn).toHaveBeenCalledWith(
+      "Dropped result chunk: part is not a tool-call",
+    );
+    warn.mockRestore();
+  });
+
+  it("drops an args-text-finish addressed to a text part", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "tool-call-args-text-finish", path: [0] },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts[0]).toMatchObject({ type: "text" });
+    expect(warn).toHaveBeenCalledWith(
+      "Dropped tool-call-args-text-finish chunk: part is not a tool-call",
+    );
+    warn.mockRestore();
+  });
+
+  it("inserts a placeholder for an unsupported part-start so later part indices stay aligned", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      {
+        type: "part-start",
+        path: [0],
+        part: { type: "bogus" },
+      } as unknown as AssistantStreamChunk,
+      { type: "part-start", path: [1], part: { type: "text" } },
+      { type: "text-delta", path: [1], textDelta: "ok" },
+      { type: "part-finish", path: [0] },
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(2);
+    expect(last.parts[0]).toMatchObject({
+      type: "reasoning",
+      text: "",
+      status: { type: "complete", reason: "unknown" },
+    });
+    expect(last.parts[1]).toMatchObject({ type: "text", text: "ok" });
+    expect(warn).toHaveBeenCalledWith(
+      "Unsupported part-start type bogus: inserting an empty reasoning part to preserve part indices",
+    );
+    warn.mockRestore();
+  });
+});
+
+describe("AssistantMessageAccumulator warn dedup", () => {
+  it("warns once per drop class per instance", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    await collectStream([
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [5], textDelta: "a" },
+      { type: "text-delta", path: [5], textDelta: "b" },
+      { type: "text-delta", path: [6], textDelta: "c" },
+      { type: "part-finish", path: [9] },
+    ]);
+
+    expect(warn).toHaveBeenCalledTimes(2);
+    expect(warn).toHaveBeenCalledWith(
+      "Dropped text-delta chunk: no part at path [5]",
+    );
+    expect(warn).toHaveBeenCalledWith(
+      "Dropped part-finish chunk: no part at path [9]",
+    );
+    warn.mockRestore();
+  });
+
+  it("resets dedup per accumulator instance", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const stream = [
+      { type: "part-start", path: [0], part: { type: "text" } },
+      { type: "text-delta", path: [5], textDelta: "x" },
+    ] satisfies AssistantStreamChunk[];
+    await collectStream(stream);
+    await collectStream(stream);
+
+    expect(warn).toHaveBeenCalledTimes(2);
+    warn.mockRestore();
+  });
+
+  it("dedupes repeated wrong-part-type drops", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    await collectStream([
+      {
+        type: "part-start",
+        path: [0],
+        part: { type: "source", sourceType: "url", id: "s", url: "https://x" },
+      },
+      { type: "text-delta", path: [0], textDelta: "a" },
+      { type: "text-delta", path: [0], textDelta: "b" },
+    ]);
+
+    expect(warn).toHaveBeenCalledTimes(1);
+    warn.mockRestore();
+  });
+});
+
+describe("AssistantMessageAccumulator warn dedup key independence", () => {
+  it("keeps drop classes as independent keys within one stream", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    await collectStream([
+      {
+        type: "part-start",
+        path: [0],
+        part: { type: "source", sourceType: "url", id: "s", url: "https://x" },
+      },
+      { type: "text-delta", path: [0], textDelta: "wrong-part" },
+      { type: "text-delta", path: [7], textDelta: "no-part" },
+    ]);
+
+    expect(warn).toHaveBeenCalledTimes(2);
+    warn.mockRestore();
+  });
+
+  it("keys unsupported part-starts per type", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    const messages = await collectStream([
+      {
+        type: "part-start",
+        path: [0],
+        part: { type: "bogus" },
+      } as unknown as AssistantStreamChunk,
+      {
+        type: "part-start",
+        path: [1],
+        part: { type: "bogus" },
+      } as unknown as AssistantStreamChunk,
+      {
+        type: "part-start",
+        path: [2],
+        part: { type: "video" },
+      } as unknown as AssistantStreamChunk,
+    ]);
+    const last = messages.at(-1)!;
+
+    expect(last.parts).toHaveLength(3);
+    expect(warn).toHaveBeenCalledTimes(2);
+    warn.mockRestore();
+  });
+
+  it("caps the number of warned keys per instance", async () => {
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    await collectStream(
+      Array.from(
+        { length: 20 },
+        (_, i) =>
+          ({
+            type: "part-start",
+            path: [i],
+            part: { type: `bogus-${i}` },
+          }) as unknown as AssistantStreamChunk,
+      ),
+    );
+
+    expect(warn).toHaveBeenCalledTimes(16);
+    warn.mockRestore();
+  });
+});