ai-stream-utils 2.0.0 → 2.0.1

package/dist/index.d.mts

@@ -529,9 +529,9 @@ declare class ChunkPipeline<UI_MESSAGE extends UIMessage, CHUNK extends InferUIM
    * Use with includeChunks(), includeParts(), excludeChunks(), or excludeParts().
    * The callback only receives content chunks because meta chunks pass through unchanged.
    */
-  filter<NARROWED_CHUNK extends InferUIMessageChunk<UI_MESSAGE>, NARROWED_PART extends {
-    type: string;
-  }>(guard: FilterGuard<UI_MESSAGE, NARROWED_CHUNK, NARROWED_PART>): ChunkPipeline<UI_MESSAGE, NARROWED_CHUNK, NARROWED_PART>;
+  filter<NARROWED_CHUNK extends CHUNK, NARROWED_PART extends {
+    type: PART[`type`];
+  }>(guard: FilterGuard<UI_MESSAGE, NARROWED_CHUNK, NARROWED_PART>): ChunkPipeline<UI_MESSAGE, CHUNK & NARROWED_CHUNK, PART & NARROWED_PART>;
   /**
    * Filters chunks using a generic predicate function.
    * The callback only receives content chunks because meta chunks pass through unchanged.
@@ -549,8 +549,8 @@ declare class ChunkPipeline<UI_MESSAGE extends UIMessage, CHUNK extends InferUIM
    * Content chunks include a part object with the type, while meta chunks have undefined part.
    * All chunks pass through regardless of whether the callback is invoked.
    */
-  on<NARROWED_CHUNK extends InferUIMessageChunk<UI_MESSAGE>, NARROWED_PART extends {
-    type: string;
+  on<NARROWED_CHUNK extends CHUNK, NARROWED_PART extends {
+    type: PART[`type`];
   } | undefined>(guard: ObserveGuard<UI_MESSAGE, NARROWED_CHUNK, NARROWED_PART>, callback: ChunkObserveFn<NARROWED_CHUNK, NARROWED_PART>): ChunkPipeline<UI_MESSAGE, CHUNK, PART>;
   /**
    * Observes chunks matching a predicate without filtering them.
@@ -638,7 +638,7 @@ declare function includeParts<UI_MESSAGE extends UIMessage, PART_TYPE extends In
  *   .map(({ chunk }) => chunk); // excludes text-start and text-end
  * ```
  */
-declare function excludeChunks<UI_MESSAGE extends UIMessage, CHUNK_TYPE extends ContentChunkType<UI_MESSAGE>>(types: CHUNK_TYPE | Array<CHUNK_TYPE>): FilterGuard<UI_MESSAGE, Exclude<ExtractChunk<UI_MESSAGE, ContentChunkType<UI_MESSAGE>>, ExtractChunk<UI_MESSAGE, CHUNK_TYPE>>, {
+declare function excludeChunks<UI_MESSAGE extends UIMessage, CHUNK_TYPE extends ContentChunkType<UI_MESSAGE>>(types: CHUNK_TYPE | Array<CHUNK_TYPE>): FilterGuard<UI_MESSAGE, ExtractChunk<UI_MESSAGE, Exclude<ContentChunkType<UI_MESSAGE>, CHUNK_TYPE>>, {
   type: ExcludePartForChunks<UI_MESSAGE, CHUNK_TYPE>;
 }>;
 /**
@@ -656,7 +656,7 @@ declare function excludeChunks<UI_MESSAGE extends UIMessage, CHUNK_TYPE extends
  *   .map(({ chunk }) => chunk); // excludes text and reasoning chunks
  * ```
  */
-declare function excludeParts<UI_MESSAGE extends UIMessage, PART_TYPE extends InferUIMessagePartType<UI_MESSAGE>>(types: PART_TYPE | Array<PART_TYPE>): FilterGuard<UI_MESSAGE, Exclude<ExtractChunk<UI_MESSAGE, ContentChunkType<UI_MESSAGE>>, ExtractChunkForPart<UI_MESSAGE, ExtractPart<UI_MESSAGE, PART_TYPE>>>, {
+declare function excludeParts<UI_MESSAGE extends UIMessage, PART_TYPE extends InferUIMessagePartType<UI_MESSAGE>>(types: PART_TYPE | Array<PART_TYPE>): FilterGuard<UI_MESSAGE, ExtractChunk<UI_MESSAGE, Exclude<ContentChunkType<UI_MESSAGE>, PartTypeToChunkTypes<UI_MESSAGE, PART_TYPE>>>, {
   type: Exclude<InferUIMessagePartType<UI_MESSAGE>, PART_TYPE>;
 }>;
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-stream-utils",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "AI SDK: Filter and transform UI messages while streaming to the client",
   "keywords": [
     "ai",