ai-stream-utils 0.0.2 → 0.0.3

package/README.md

@@ -271,16 +271,16 @@ function mapUIMessageStream<UI_MESSAGE extends UIMessage>(
 ): AsyncIterableStream<InferUIMessageChunk<UI_MESSAGE>>
 
 type MapUIMessageStreamFn<UI_MESSAGE extends UIMessage> = (
-  input: ChunkMapInput<UI_MESSAGE>,
-  context: ChunkMapContext<UI_MESSAGE>,
+  input: MapInput<UI_MESSAGE>,
+  context: MapContext<UI_MESSAGE>,
 ) => InferUIMessageChunk<UI_MESSAGE> | null;
 
-type ChunkMapInput<UI_MESSAGE extends UIMessage> = {
+type MapInput<UI_MESSAGE extends UIMessage> = {
   chunk: InferUIMessageChunk<UI_MESSAGE>;
-  part: PartialPart<UI_MESSAGE>;
+  part: InferPartialUIMessagePart<UI_MESSAGE>;
 };
 
-type ChunkMapContext<UI_MESSAGE extends UIMessage> = {
+type MapContext<UI_MESSAGE extends UIMessage> = {
   index: number;
   chunks: InferUIMessageChunk<UI_MESSAGE>[];
 };
@@ -303,17 +303,17 @@ function flatMapUIMessageStream<UI_MESSAGE extends UIMessage, PART extends Infer
 ): AsyncIterableStream<InferUIMessageChunk<UI_MESSAGE>>
 
 type FlatMapUIMessageStreamFn<UI_MESSAGE extends UIMessage, PART = InferUIMessagePart<UI_MESSAGE>> = (
-  input: PartFlatMapInput<UI_MESSAGE, PART>,
-  context: PartFlatMapContext<UI_MESSAGE>,
+  input: FlatMapInput<UI_MESSAGE, PART>,
+  context: FlatMapContext<UI_MESSAGE>,
 ) => PART | null;
 
-type PartFlatMapInput<UI_MESSAGE extends UIMessage, PART = InferUIMessagePart<UI_MESSAGE>> = {
+type FlatMapInput<UI_MESSAGE extends UIMessage, PART = InferUIMessagePart<UI_MESSAGE>> = {
   part: PART;
 };
 
-type PartFlatMapContext<UI_MESSAGE extends UIMessage> = {
+type FlatMapContext<UI_MESSAGE extends UIMessage> = {
   index: number;
-  parts: PartFlatMapInput<UI_MESSAGE>[];
+  parts: InferUIMessagePart<UI_MESSAGE>[];
 };
 ```
 
@@ -325,7 +325,7 @@ function partTypeIs<UI_MESSAGE extends UIMessage, T extends InferUIMessagePartTy
 ): FlatMapUIMessageStreamPredicate<UI_MESSAGE, Extract<InferUIMessagePart<UI_MESSAGE>, { type: T }>>
 
 type FlatMapUIMessageStreamPredicate<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE>> = 
-  (part: PartialPart<UI_MESSAGE>) => boolean;
+  (part: InferPartialUIMessagePart<UI_MESSAGE>) => boolean;
 ```
 
 ### `filterUIMessageStream`
@@ -337,8 +337,8 @@ function filterUIMessageStream<UI_MESSAGE extends UIMessage>(
 ): AsyncIterableStream<InferUIMessageChunk<UI_MESSAGE>>
 
 type FilterUIMessageStreamPredicate<UI_MESSAGE extends UIMessage> = (
-  input: ChunkMapInput<UI_MESSAGE>,
-  context: ChunkMapContext<UI_MESSAGE>,
+  input: MapInput<UI_MESSAGE>,
+  context: MapContext<UI_MESSAGE>,
 ) => boolean;
 ```
 

package/dist/index.d.mts

@@ -10,7 +10,7 @@ type InferUIMessagePartType<UI_MESSAGE extends UIMessage> = InferUIMessagePart<U
  * A partial part reconstructed from the current chunk.
  * Contains the part type and any available data from the chunk.
  */
-type PartialPart<UI_MESSAGE extends UIMessage> = {
+type InferPartialUIMessagePart<UI_MESSAGE extends UIMessage> = {
   /** The part type (e.g., 'text', 'reasoning', 'tool-weather', 'file') */
   type: InferUIMessagePartType<UI_MESSAGE>;
 } & Partial<InferUIMessagePart<UI_MESSAGE>>;
@@ -19,19 +19,19 @@ type PartialPart<UI_MESSAGE extends UIMessage> = {
 /**
  * Input object provided to the chunk map function.
  */
-type ChunkMapInput<UI_MESSAGE extends UIMessage> = {
+type MapInput<UI_MESSAGE extends UIMessage> = {
   /** The current chunk */
   chunk: InferUIMessageChunk<UI_MESSAGE>;
   /**
    * A partial representation of the part this chunk belongs to.
    * Use `part.type` to determine the part type.
    */
-  part: PartialPart<UI_MESSAGE>;
+  part: InferPartialUIMessagePart<UI_MESSAGE>;
 };
 /**
  * Context provided to the chunk map function (similar to Array.map callback).
  */
-type ChunkMapContext<UI_MESSAGE extends UIMessage> = {
+type MapContext<UI_MESSAGE extends UIMessage> = {
   /** The index of the current chunk in the stream (0-based) */
   index: number;
   /** All chunks seen so far (including the current one) */
@@ -42,7 +42,7 @@ type ChunkMapContext<UI_MESSAGE extends UIMessage> = {
  * Similar to Array.map, receives the input object and context.
  * Return the chunk (possibly transformed) to include it, or null to filter it out.
  */
-type MapUIMessageStreamFn<UI_MESSAGE extends UIMessage> = (input: ChunkMapInput<UI_MESSAGE>, context: ChunkMapContext<UI_MESSAGE>) => InferUIMessageChunk<UI_MESSAGE> | null;
+type MapUIMessageStreamFn<UI_MESSAGE extends UIMessage> = (input: MapInput<UI_MESSAGE>, context: MapContext<UI_MESSAGE>) => InferUIMessageChunk<UI_MESSAGE> | null;
 /**
  * Maps/filters a UIMessageStream at the chunk level.
  *
@@ -91,7 +91,7 @@ declare function mapUIMessageStream<UI_MESSAGE extends UIMessage>(stream: Readab
  * Filter function that receives the same input/context as mapUIMessageStream.
  * Return true to include the chunk, false to filter it out.
  */
-type FilterUIMessageStreamPredicate<UI_MESSAGE extends UIMessage> = (input: ChunkMapInput<UI_MESSAGE>, context: ChunkMapContext<UI_MESSAGE>) => boolean;
+type FilterUIMessageStreamPredicate<UI_MESSAGE extends UIMessage> = (input: MapInput<UI_MESSAGE>, context: MapContext<UI_MESSAGE>) => boolean;
 /**
  * Creates a filter predicate that includes only the specified part types.
  *
@@ -157,18 +157,18 @@ declare function filterUIMessageStream<UI_MESSAGE extends UIMessage>(stream: Rea
 /**
  * Input object provided to the part flatMap function.
  */
-type PartFlatMapInput<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE> = InferUIMessagePart<UI_MESSAGE>> = {
+type FlatMapInput<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE> = InferUIMessagePart<UI_MESSAGE>> = {
   /** The reconstructed part */
   part: PART;
 };
 /**
  * Context provided to the part flatMap function.
  */
-type PartFlatMapContext<UI_MESSAGE extends UIMessage> = {
+type FlatMapContext<UI_MESSAGE extends UIMessage> = {
   /** The index of the current part in the stream (0-based) */
   index: number;
   /** All parts seen so far (including the current one) */
-  parts: PartFlatMapInput<UI_MESSAGE>[];
+  parts: InferUIMessagePart<UI_MESSAGE>[];
 };
 /**
  * FlatMap function for part-level transformation.
@@ -176,13 +176,13 @@ type PartFlatMapContext<UI_MESSAGE extends UIMessage> = {
  * - The part (possibly transformed) to include it
  * - null to filter out the part
  */
-type FlatMapUIMessageStreamFn<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE> = InferUIMessagePart<UI_MESSAGE>> = (input: PartFlatMapInput<UI_MESSAGE, PART>, context: PartFlatMapContext<UI_MESSAGE>) => PART | null;
+type FlatMapUIMessageStreamFn<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE> = InferUIMessagePart<UI_MESSAGE>> = (input: FlatMapInput<UI_MESSAGE, PART>, context: FlatMapContext<UI_MESSAGE>) => PART | null;
 /**
  * Predicate function to determine which parts should be buffered.
  * Receives a PartialPart built from the current chunk.
  * Returns true to buffer the part for transformation, false to pass through immediately.
  */
-type FlatMapUIMessageStreamPredicate<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE> = InferUIMessagePart<UI_MESSAGE>> = (part: PartialPart<UI_MESSAGE>) => boolean;
+type FlatMapUIMessageStreamPredicate<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE> = InferUIMessagePart<UI_MESSAGE>> = (part: InferPartialUIMessagePart<UI_MESSAGE>) => boolean;
 /**
  * Creates a predicate that matches parts by their type.
  * Supports both single type and array of types with full type narrowing.
@@ -245,4 +245,4 @@ declare function partTypeIs<UI_MESSAGE extends UIMessage, PART_TYPE extends Infe
 declare function flatMapUIMessageStream<UI_MESSAGE extends UIMessage, PART extends InferUIMessagePart<UI_MESSAGE>>(stream: ReadableStream<InferUIMessageChunk<UI_MESSAGE>>, predicate: FlatMapUIMessageStreamPredicate<UI_MESSAGE, PART>, flatMapFn: FlatMapUIMessageStreamFn<UI_MESSAGE, PART>): AsyncIterableStream<InferUIMessageChunk<UI_MESSAGE>>;
 declare function flatMapUIMessageStream<UI_MESSAGE extends UIMessage>(stream: ReadableStream<InferUIMessageChunk<UI_MESSAGE>>, flatMapFn: FlatMapUIMessageStreamFn<UI_MESSAGE>): AsyncIterableStream<InferUIMessageChunk<UI_MESSAGE>>;
 //#endregion
-export { type ChunkMapContext, type ChunkMapInput, type FilterUIMessageStreamPredicate, type FlatMapUIMessageStreamFn, type FlatMapUIMessageStreamPredicate, type InferUIMessageData, type InferUIMessageMetadata, type InferUIMessagePart, type InferUIMessagePartType, type InferUIMessageTools, type MapUIMessageStreamFn, type PartFlatMapContext, type PartFlatMapInput, type PartialPart, excludeParts, filterUIMessageStream, flatMapUIMessageStream, includeParts, mapUIMessageStream, partTypeIs };
+export { type FilterUIMessageStreamPredicate, type FlatMapContext, type FlatMapInput, type FlatMapUIMessageStreamFn, type FlatMapUIMessageStreamPredicate, type InferPartialUIMessagePart, type InferUIMessageData, type InferUIMessageMetadata, type InferUIMessagePart, type InferUIMessagePartType, type InferUIMessageTools, type MapContext, type MapInput, type MapUIMessageStreamFn, excludeParts, filterUIMessageStream, flatMapUIMessageStream, includeParts, mapUIMessageStream, partTypeIs };

package/dist/index.mjs

@@ -552,7 +552,7 @@ function flatMapUIMessageStream(...args) {
 		isBufferingCurrentPart = false;
 		const part = reconstructPartFromChunks(chunks);
 		const input = { part };
-		allParts.push(input);
+		allParts.push(part);
 		const result = flatMapFn(input, {
 			index: currentIndex++,
 			parts: allParts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-stream-utils",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "AI SDK: Filter and transform UI messages while streaming to the client",
   "main": "./dist/index.mjs",
   "module": "./dist/index.mjs",