assistant-stream 0.2.0 → 0.2.1

package/dist/core/utils/Counter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Counter.d.ts","sourceRoot":"","sources":["../../../src/core/utils/Counter.ts"],"names":[],"mappings":"AAAA,qBAAa,OAAO;IACX,KAAK,SAAM;IAElB,EAAE;CAGH"}

package/dist/core/utils/generateId.d.ts

@@ -0,0 +1,2 @@
+export declare const generateId: (size?: number) => string;
+//# sourceMappingURL=generateId.d.ts.map

package/dist/core/utils/generateId.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generateId.d.ts","sourceRoot":"","sources":["../../../src/core/utils/generateId.tsx"],"names":[],"mappings":"AAEA,eAAO,MAAM,UAAU,2BAGtB,CAAC"}

package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts

@@ -0,0 +1,20 @@
+import { AssistantStreamChunk, PartInit } from "../../AssistantStreamChunk";
+/**
+ * For chunk types that are associated with a part,
+ * we require a non‑nullable meta field.
+ */
+export type AssistantMetaStreamChunk = (AssistantStreamChunk & {
+    type: "text-delta" | "part-finish";
+    meta: PartInit;
+}) | (AssistantStreamChunk & {
+    type: "result" | "tool-call-args-text-finish";
+    meta: PartInit & {
+        type: "tool-call";
+    };
+}) | (AssistantStreamChunk & {
+    type: Exclude<AssistantStreamChunk["type"], "text-delta" | "result" | "tool-call-args-text-finish" | "part-finish">;
+});
+export declare class AssistantMetaTransformStream extends TransformStream<AssistantStreamChunk, AssistantMetaStreamChunk> {
+    constructor();
+}
+//# sourceMappingURL=AssistantMetaTransformStream.d.ts.map

package/dist/core/utils/stream/AssistantMetaTransformStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AssistantMetaTransformStream.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/AssistantMetaTransformStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAE5E;;;GAGG;AACH,MAAM,MAAM,wBAAwB,GAChC,CAAC,oBAAoB,GAAG;IACtB,IAAI,EAAE,YAAY,GAAG,aAAa,CAAC;IACnC,IAAI,EAAE,QAAQ,CAAC;CAChB,CAAC,GACF,CAAC,oBAAoB,GAAG;IACtB,IAAI,EAAE,QAAQ,GAAG,4BAA4B,CAAC;IAC9C,IAAI,EAAE,QAAQ,GAAG;QAAE,IAAI,EAAE,WAAW,CAAA;KAAE,CAAC;CACxC,CAAC,GACF,CAAC,oBAAoB,GAAG;IACtB,IAAI,EAAE,OAAO,CACX,oBAAoB,CAAC,MAAM,CAAC,EAC5B,YAAY,GAAG,QAAQ,GAAG,4BAA4B,GAAG,aAAa,CACvE,CAAC;CACH,CAAC,CAAC;AACP,qBAAa,4BAA6B,SAAQ,eAAe,CAC/D,oBAAoB,EACpB,wBAAwB,CACzB;;CAiDA"}

package/dist/core/utils/stream/AssistantTransformStream.d.ts

@@ -0,0 +1,15 @@
+import { AssistantStreamChunk } from "../../AssistantStreamChunk";
+import { AssistantStreamController } from "../../modules/assistant-stream";
+type AssistantTransformerFlushCallback = (controller: AssistantStreamController) => void | PromiseLike<void>;
+type AssistantTransformerStartCallback = (controller: AssistantStreamController) => void | PromiseLike<void>;
+type AssistantTransformerTransformCallback<I> = (chunk: I, controller: AssistantStreamController) => void | PromiseLike<void>;
+type AssistantTransformer<I> = {
+    flush?: AssistantTransformerFlushCallback;
+    start?: AssistantTransformerStartCallback;
+    transform?: AssistantTransformerTransformCallback<I>;
+};
+export declare class AssistantTransformStream<I> extends TransformStream<I, AssistantStreamChunk> {
+    constructor(transformer: AssistantTransformer<I>, writableStrategy?: QueuingStrategy<I>, readableStrategy?: QueuingStrategy<AssistantStreamChunk>);
+}
+export {};
+//# sourceMappingURL=AssistantTransformStream.d.ts.map

package/dist/core/utils/stream/AssistantTransformStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AssistantTransformStream.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/AssistantTransformStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,EACL,yBAAyB,EAE1B,MAAM,gCAAgC,CAAC;AAExC,KAAK,iCAAiC,GAAG,CACvC,UAAU,EAAE,yBAAyB,KAClC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;AAE9B,KAAK,iCAAiC,GAAG,CACvC,UAAU,EAAE,yBAAyB,KAClC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;AAE9B,KAAK,qCAAqC,CAAC,CAAC,IAAI,CAC9C,KAAK,EAAE,CAAC,EACR,UAAU,EAAE,yBAAyB,KAClC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;AAE9B,KAAK,oBAAoB,CAAC,CAAC,IAAI;IAC7B,KAAK,CAAC,EAAE,iCAAiC,CAAC;IAC1C,KAAK,CAAC,EAAE,iCAAiC,CAAC;IAC1C,SAAS,CAAC,EAAE,qCAAqC,CAAC,CAAC,CAAC,CAAC;CACtD,CAAC;AAEF,qBAAa,wBAAwB,CAAC,CAAC,CAAE,SAAQ,eAAe,CAC9D,CAAC,EACD,oBAAoB,CACrB;gBAEG,WAAW,EAAE,oBAAoB,CAAC,CAAC,CAAC,EACpC,gBAAgB,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,EACrC,gBAAgB,CAAC,EAAE,eAAe,CAAC,oBAAoB,CAAC;CAyC3D"}

package/dist/core/utils/stream/LineDecoderStream.d.ts

@@ -0,0 +1,5 @@
+export declare class LineDecoderStream extends TransformStream<string, string> {
+    private buffer;
+    constructor();
+}
+//# sourceMappingURL=LineDecoderStream.d.ts.map

package/dist/core/utils/stream/LineDecoderStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LineDecoderStream.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/LineDecoderStream.ts"],"names":[],"mappings":"AAAA,qBAAa,iBAAkB,SAAQ,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC;IACpE,OAAO,CAAC,MAAM,CAAM;;CA2BrB"}

package/dist/core/utils/stream/PipeableTransformStream.d.ts

@@ -0,0 +1,4 @@
+export declare class PipeableTransformStream<I, O> extends TransformStream<I, O> {
+    constructor(transform: (readable: ReadableStream<I>) => ReadableStream<O>);
+}
+//# sourceMappingURL=PipeableTransformStream.d.ts.map

package/dist/core/utils/stream/PipeableTransformStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PipeableTransformStream.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/PipeableTransformStream.ts"],"names":[],"mappings":"AAAA,qBAAa,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAE,SAAQ,eAAe,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC1D,SAAS,EAAE,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,KAAK,cAAc,CAAC,CAAC,CAAC;CAQ1E"}

package/dist/core/utils/stream/UnderlyingReadable.d.ts

@@ -0,0 +1,6 @@
+export type UnderlyingReadable<TController> = {
+    start?: (controller: TController) => void;
+    pull?: (controller: TController) => void | PromiseLike<void>;
+    cancel?: UnderlyingSourceCancelCallback;
+};
+//# sourceMappingURL=UnderlyingReadable.d.ts.map

package/dist/core/utils/stream/UnderlyingReadable.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"UnderlyingReadable.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/UnderlyingReadable.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,kBAAkB,CAAC,WAAW,IAAI;IAC5C,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,WAAW,KAAK,IAAI,CAAC;IAC1C,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,WAAW,KAAK,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC7D,MAAM,CAAC,EAAE,8BAA8B,CAAC;CACzC,CAAC"}

package/dist/core/utils/stream/merge.d.ts

@@ -0,0 +1,9 @@
+import { AssistantStreamChunk } from "../../AssistantStreamChunk";
+export declare const createMergeStream: () => {
+    readable: ReadableStream<AssistantStreamChunk>;
+    isSealed(): boolean;
+    seal(): void;
+    addStream(stream: ReadableStream<AssistantStreamChunk>): void;
+    enqueue(chunk: AssistantStreamChunk): void;
+};
+//# sourceMappingURL=merge.d.ts.map

package/dist/core/utils/stream/merge.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"merge.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/merge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAQlE,eAAO,MAAM,iBAAiB;;;;sBA0ER,cAAc,CAAC,oBAAoB,CAAC;mBAUvC,oBAAoB;CAWtC,CAAC"}

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

@@ -0,0 +1,12 @@
+import { AssistantStreamChunk } from "../../AssistantStreamChunk";
+import { Counter } from "../Counter";
+export declare class PathAppendEncoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
+    constructor(idx: number);
+}
+export declare class PathAppendDecoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
+    constructor(idx: number);
+}
+export declare class PathMergeEncoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
+    constructor(counter: Counter);
+}
+//# sourceMappingURL=path-utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"path-utils.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/path-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,qBAAa,iBAAkB,SAAQ,eAAe,CACpD,oBAAoB,EACpB,oBAAoB,CACrB;gBACa,GAAG,EAAE,MAAM;CAUxB;AAED,qBAAa,iBAAkB,SAAQ,eAAe,CACpD,oBAAoB,EACpB,oBAAoB,CACrB;gBACa,GAAG,EAAE,MAAM;CAiBxB;AAED,qBAAa,gBAAiB,SAAQ,eAAe,CACnD,oBAAoB,EACpB,oBAAoB,CACrB;gBACa,OAAO,EAAE,OAAO;CAwB7B"}

package/dist/core/utils/types.d.ts

@@ -0,0 +1,102 @@
+import { ReadonlyJSONObject, ReadonlyJSONValue } from "../../utils/json/json-value";
+type TextStatus = {
+    type: "running";
+} | {
+    type: "complete";
+    reason: "stop" | "unknown";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "length" | "content-filter" | "other";
+};
+export type TextPart = {
+    type: "text";
+    text: string;
+    status: TextStatus;
+};
+export type ReasoningPart = {
+    type: "reasoning";
+    text: string;
+    status: TextStatus;
+};
+type ToolCallStatus = {
+    type: "running";
+    isArgsComplete: boolean;
+} | {
+    type: "requires-action";
+    reason: "tool-call-result";
+} | {
+    type: "complete";
+    reason: "stop" | "unknown";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "length" | "content-filter" | "other";
+};
+export type ToolCallPart = {
+    type: "tool-call";
+    state: "partial-call" | "call" | "result";
+    status: ToolCallStatus;
+    toolCallId: string;
+    toolName: string;
+    argsText: string;
+    args: ReadonlyJSONObject;
+    artifact?: unknown;
+    result?: ReadonlyJSONValue;
+    isError?: boolean;
+};
+export type SourcePart = {
+    type: "source";
+    sourceType: "url";
+    id: string;
+    url: string;
+    title?: string;
+};
+export type FilePart = {
+    type: "file";
+    data: string;
+    mimeType: string;
+};
+export type AssistantMessagePart = TextPart | ReasoningPart | ToolCallPart | SourcePart | FilePart;
+type AssistantMessageStepUsage = {
+    promptTokens: number;
+    completionTokens: number;
+};
+type AssistantMessageStepMetadata = {
+    state: "started";
+    messageId: string;
+} | {
+    state: "finished";
+    messageId: string;
+    finishReason: "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown";
+    usage?: AssistantMessageStepUsage;
+    isContinued: boolean;
+};
+export type AssistantMessageStatus = {
+    type: "running";
+} | {
+    type: "requires-action";
+    reason: "tool-calls";
+} | {
+    type: "complete";
+    reason: "stop" | "unknown";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "tool-calls" | "length" | "content-filter" | "other" | "error";
+    error?: ReadonlyJSONValue;
+};
+export type AssistantMessage = {
+    role: "assistant";
+    status: AssistantMessageStatus;
+    parts: AssistantMessagePart[];
+    /**
+     * @deprecated Use `parts` instead.
+     */
+    content: AssistantMessagePart[];
+    metadata: {
+        unstable_data: ReadonlyJSONValue[];
+        unstable_annotations: ReadonlyJSONValue[];
+        steps: AssistantMessageStepMetadata[];
+        custom: Record<string, unknown>;
+    };
+};
+export {};
+//# sourceMappingURL=types.d.ts.map

package/dist/core/utils/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/core/utils/types.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,6BAA6B,CAAC;AAErC,KAAK,UAAU,GACX;IACE,IAAI,EAAE,SAAS,CAAC;CACjB,GACD;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;CAC5B,GACD;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,gBAAgB,GAAG,OAAO,CAAC;CAC7D,CAAC;AAMN,MAAM,MAAM,QAAQ,GAAG;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,UAAU,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,IAAI,EAAE,WAAW,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,UAAU,CAAC;CACpB,CAAC;AAEF,KAAK,cAAc,GACf;IACE,IAAI,EAAE,SAAS,CAAC;IAChB,cAAc,EAAE,OAAO,CAAC;CACzB,GACD;IACE,IAAI,EAAE,iBAAiB,CAAC;IACxB,MAAM,EAAE,kBAAkB,CAAC;CAC5B,GACD;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;CAC5B,GACD;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,gBAAgB,GAAG,OAAO,CAAC;CAC7D,CAAC;AAEN,MAAM,MAAM,YAAY,GAAG;IACzB,IAAI,EAAE,WAAW,CAAC;IAClB,KAAK,EAAE,cAAc,GAAG,MAAM,GAAG,QAAQ,CAAC;IAC1C,MAAM,EAAE,cAAc,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,kBAAkB,CAAC;IACzB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,QAAQ,CAAC;IACf,UAAU,EAAE,KAAK,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,QAAQ,GAAG;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAC5B,QAAQ,GACR,aAAa,GACb,YAAY,GACZ,UAAU,GACV,QAAQ,CAAC;AAEb,KAAK,yBAAyB,GAAG;IAC/B,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;CAC1B,CAAC;AAEF,KAAK,4BAA4B,GAC7B;IACE,KAAK,EAAE,SAAS,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,KAAK,EAAE,UAAU,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EACR,MAAM,GACN,QAAQ,GACR,gBAAgB,GAChB,YAAY,GACZ,OAAO,GACP,OAAO,GACP,SAAS,CAAC;IACd,KAAK,CAAC,EAAE,yBAAyB,CAAC;IAClC,WAAW,EAAE,OAAO,CAAC;CACtB,CAAC;AAEN,MAAM,MAAM,sBAAsB,GAC9B;IACE,IAAI,EAAE,SAAS,CAAC;CACjB,GACD;IACE,IAAI,EAAE,iBAAiB,CAAC;IACxB,MAAM,EAAE,YAAY,CAAC;CACtB,GACD;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;CAC5B,GACD;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,MAAM,EACF,WAAW,GACX,YAAY,GACZ,QAAQ,GACR,gBAAgB,GAChB,OAAO,GACP,OAAO,CAAC;IACZ,KAAK,CAAC,EAAE,iBAAiB,CAAC;CAC3B,CAAC;AAEN,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,EAAE,WAAW,CAAC;IAClB,MAAM,EAAE,sBAAsB,CAAC;IAC/B,KAAK,EAAE,oBAAoB,EAAE,CAAC;IAC9B;;OAEG;IACH,OAAO,EAAE,oBAAoB,EAAE,CAAC;IAChC,QAAQ,EAAE;QACR,aAAa,EAAE,iBAAiB,EAAE,CAAC;QACnC,oBAAoB,EAAE,iBAAiB,EAAE,CAAC;QAC1C,KAAK,EAAE,4BAA4B,EAAE,CAAC;QACtC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACjC,CAAC;CACH,CAAC"}

package/dist/core/utils/withPromiseOrValue.d.ts

@@ -0,0 +1,2 @@
+export declare function withPromiseOrValue<T>(callback: () => T | PromiseLike<T>, thenHandler: (value: T) => PromiseLike<void> | void, catchHandler: (error: unknown) => PromiseLike<void> | void): PromiseLike<void> | void;
+//# sourceMappingURL=withPromiseOrValue.d.ts.map

package/dist/core/utils/withPromiseOrValue.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"withPromiseOrValue.d.ts","sourceRoot":"","sources":["../../../src/core/utils/withPromiseOrValue.ts"],"names":[],"mappings":"AAAA,wBAAgB,kBAAkB,CAAC,CAAC,EAClC,QAAQ,EAAE,MAAM,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,EAClC,WAAW,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,EACnD,YAAY,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,GACzD,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,CAe1B"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./core";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC"}

package/dist/utils/AsyncIterableStream.d.ts

@@ -0,0 +1,3 @@
+export type AsyncIterableStream<T> = AsyncIterable<T> & ReadableStream<T>;
+export declare function asAsyncIterableStream<T>(source: ReadableStream<T>): AsyncIterableStream<T>;
+//# sourceMappingURL=AsyncIterableStream.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AsyncIterableStream.d.ts","sourceRoot":"","sources":["../../src/utils/AsyncIterableStream.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,mBAAmB,CAAC,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;AAiB1E,wBAAgB,qBAAqB,CAAC,CAAC,EACrC,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,GACxB,mBAAmB,CAAC,CAAC,CAAC,CAIxB"}

package/dist/utils/json/fix-json.d.ts

@@ -0,0 +1,2 @@
+export declare function fixJson(input: string): [string, string[]];
+//# sourceMappingURL=fix-json.d.ts.map

package/dist/utils/json/fix-json.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fix-json.d.ts","sourceRoot":"","sources":["../../../src/utils/json/fix-json.ts"],"names":[],"mappings":"AA+DA,wBAAgB,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAwazD"}

package/dist/utils/json/is-json.d.ts

@@ -0,0 +1,5 @@
+import { ReadonlyJSONArray, ReadonlyJSONObject, ReadonlyJSONValue } from "./json-value";
+export declare function isJSONValue(value: unknown): value is ReadonlyJSONValue;
+export declare function isJSONArray(value: unknown): value is ReadonlyJSONArray;
+export declare function isJSONObject(value: unknown): value is ReadonlyJSONObject;
+//# sourceMappingURL=is-json.d.ts.map

package/dist/utils/json/is-json.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"is-json.d.ts","sourceRoot":"","sources":["../../../src/utils/json/is-json.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,cAAc,CAAC;AAEtB,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAqBtE;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAEtE;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,kBAAkB,CASxE"}

package/dist/utils/json/json-value.d.ts

@@ -0,0 +1,6 @@
+export type ReadonlyJSONValue = null | string | number | boolean | ReadonlyJSONObject | ReadonlyJSONArray;
+export type ReadonlyJSONObject = {
+    readonly [key: string]: ReadonlyJSONValue;
+};
+export type ReadonlyJSONArray = readonly ReadonlyJSONValue[];
+//# sourceMappingURL=json-value.d.ts.map

package/dist/utils/json/json-value.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-value.d.ts","sourceRoot":"","sources":["../../../src/utils/json/json-value.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,iBAAiB,GACzB,IAAI,GACJ,MAAM,GACN,MAAM,GACN,OAAO,GACP,kBAAkB,GAClB,iBAAiB,CAAC;AAEtB,MAAM,MAAM,kBAAkB,GAAG;IAC/B,QAAQ,EAAE,GAAG,EAAE,MAAM,GAAG,iBAAiB,CAAC;CAC3C,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,SAAS,iBAAiB,EAAE,CAAC"}

package/dist/utils/json/parse-partial-json-object.d.ts

@@ -0,0 +1,14 @@
+import { ReadonlyJSONObject } from "./json-value";
+declare const PARTIAL_JSON_OBJECT_META_SYMBOL: unique symbol;
+type FieldState = "complete" | "partial";
+type PartialJsonObjectMeta = {
+    state: "complete" | "partial";
+    partialPath: string[];
+};
+export declare const getPartialJsonObjectMeta: (obj: Record<symbol, unknown>) => PartialJsonObjectMeta | undefined;
+export declare const parsePartialJsonObject: (json: string) => (ReadonlyJSONObject & {
+    [PARTIAL_JSON_OBJECT_META_SYMBOL]: PartialJsonObjectMeta;
+}) | undefined;
+export declare const getPartialJsonObjectFieldState: (obj: Record<string, unknown>, fieldPath: (string | number)[]) => FieldState;
+export {};
+//# sourceMappingURL=parse-partial-json-object.d.ts.map

package/dist/utils/json/parse-partial-json-object.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-partial-json-object.d.ts","sourceRoot":"","sources":["../../../src/utils/json/parse-partial-json-object.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,QAAA,MAAM,+BAA+B,eAEpC,CAAC;AAEF,KAAK,UAAU,GAAG,UAAU,GAAG,SAAS,CAAC;AAEzC,KAAK,qBAAqB,GAAG;IAC3B,KAAK,EAAE,UAAU,GAAG,SAAS,CAAC;IAC9B,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB,CAAC;AAEF,eAAO,MAAM,wBAAwB,GACnC,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC3B,qBAAqB,GAAG,SAE1B,CAAC;AAEF,eAAO,MAAM,sBAAsB,GACjC,MAAM,MAAM,KAEV,CAAC,kBAAkB,GAAG;IACpB,CAAC,+BAA+B,CAAC,EAAE,qBAAqB,CAAC;CAC1D,CAAC,GACF,SAgCH,CAAC;AAmCF,eAAO,MAAM,8BAA8B,GACzC,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC5B,WAAW,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,KAC7B,UAKF,CAAC"}

package/dist/utils/json/parse-partial-json-object.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=parse-partial-json-object.test.d.ts.map

package/dist/utils/json/parse-partial-json-object.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-partial-json-object.test.d.ts","sourceRoot":"","sources":["../../../src/utils/json/parse-partial-json-object.test.ts"],"names":[],"mappings":""}

package/dist/utils/promiseWithResolvers.d.ts

@@ -0,0 +1,6 @@
+export declare const promiseWithResolvers: <T>() => {
+    promise: Promise<T>;
+    resolve: (value: T | PromiseLike<T>) => void;
+    reject: (reason?: unknown) => void;
+};
+//# sourceMappingURL=promiseWithResolvers.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"promiseWithResolvers.d.ts","sourceRoot":"","sources":["../../src/utils/promiseWithResolvers.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,oBAAoB,GAAa,CAAC;;qBACvB,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,KAAK,IAAI;sBAC1B,OAAO,KAAK,IAAI;CAOxC,CAAC"}

package/dist/utils.d.ts

@@ -0,0 +1,4 @@
+export { parsePartialJsonObject, getPartialJsonObjectFieldState, } from "./utils/json/parse-partial-json-object";
+export { type AsyncIterableStream, asAsyncIterableStream, } from "./utils/AsyncIterableStream";
+export type { ReadonlyJSONValue, ReadonlyJSONArray, ReadonlyJSONObject, } from "./utils/json/json-value";
+//# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EACtB,8BAA8B,GAC/B,MAAM,wCAAwC,CAAC;AAChD,OAAO,EACL,KAAK,mBAAmB,EACxB,qBAAqB,GACtB,MAAM,6BAA6B,CAAC;AACrC,YAAY,EACV,iBAAiB,EACjB,iBAAiB,EACjB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-stream",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -22,6 +22,7 @@
   "types": "./dist/index.d.ts",
   "files": [
     "dist",
+    "src",
     "ai-sdk",
     "utils",
     "README.md"
@@ -44,7 +45,7 @@
   "homepage": "https://www.assistant-ui.com/",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/assistant-ui/assistant-ui.git"
+    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/assistant-stream"
   },
   "bugs": {
     "url": "https://github.com/assistant-ui/assistant-ui/issues"

package/src/ai-sdk/index.ts

@@ -0,0 +1,204 @@
+import type { TextStreamPart, ObjectStreamPart, Tool } from "ai";
+import { AssistantStream } from "../core/AssistantStream";
+import { AssistantTransformStream } from "../core/utils/stream/AssistantTransformStream";
+import { ToolCallStreamController } from "../core/modules/tool-call";
+import { ReadonlyJSONValue } from "../utils/json/json-value";
+
+export const fromStreamText = (
+  stream: ReadableStream<TextStreamPart<Record<string, Tool>>>,
+): AssistantStream => {
+  const toolControllers = new Map<string, ToolCallStreamController>();
+  let currentToolCallArgsText: ToolCallStreamController | undefined;
+
+  const endCurrentToolCallArgsText = () => {
+    if (!currentToolCallArgsText) return;
+    currentToolCallArgsText.argsText.close();
+    currentToolCallArgsText = undefined;
+  };
+
+  const transformer = new AssistantTransformStream<
+    TextStreamPart<Record<string, Tool>>
+  >({
+    transform(chunk, controller) {
+      const { type } = chunk;
+
+      if (
+        type !== "tool-call-delta" &&
+        type !== "tool-call" &&
+        type !== "error" &&
+        (type as string) !== "tool-result"
+      ) {
+        endCurrentToolCallArgsText();
+      }
+
+      switch (type) {
+        case "text-delta": {
+          const { textDelta } = chunk;
+          controller.appendText(textDelta);
+          break;
+        }
+        case "reasoning": {
+          const { textDelta } = chunk;
+          controller.appendReasoning(textDelta);
+          break;
+        }
+        case "tool-call-streaming-start": {
+          const { toolCallId, toolName } = chunk;
+          currentToolCallArgsText = controller.addToolCallPart({
+            toolCallId,
+            toolName,
+          });
+          toolControllers.set(toolCallId, currentToolCallArgsText);
+          break;
+        }
+        case "tool-call-delta": {
+          const { toolCallId, argsTextDelta } = chunk;
+          const toolController = toolControllers.get(toolCallId);
+          if (!toolController) throw new Error("Tool call not found");
+          toolController.argsText.append(argsTextDelta);
+          break;
+        }
+        case "tool-result" as string: {
+          const { toolCallId, result } = chunk as unknown as {
+            toolCallId: string;
+            result: ReadonlyJSONValue;
+          };
+          const toolController = toolControllers.get(toolCallId);
+          if (!toolController) throw new Error("Tool call not found");
+          toolController.setResponse({
+            result,
+          });
+          toolController.close();
+          toolControllers.delete(toolCallId);
+          break;
+        }
+        case "tool-call": {
+          const { toolCallId, toolName, args } = chunk;
+          const toolController = controller.addToolCallPart({
+            toolCallId,
+            toolName,
+          });
+          toolController.argsText.append(JSON.stringify(args));
+          toolController.argsText.close();
+          toolControllers.set(toolCallId, toolController);
+          break;
+        }
+
+        case "step-start":
+          controller.enqueue({
+            type: "step-start",
+            path: [],
+            messageId: chunk.messageId,
+          });
+          break;
+        case "step-finish":
+          controller.enqueue({
+            type: "step-finish",
+            path: [],
+            finishReason: chunk.finishReason,
+            usage: chunk.usage,
+            isContinued: chunk.isContinued,
+          });
+          break;
+        case "error":
+          controller.enqueue({
+            type: "error",
+            path: [],
+            error: String(chunk.error),
+          });
+          break;
+
+        case "finish": {
+          controller.enqueue({
+            type: "message-finish",
+            path: [],
+            finishReason: chunk.finishReason,
+            usage: chunk.usage,
+          });
+          break;
+        }
+
+        case "source":
+          controller.appendSource({
+            type: "source",
+            ...chunk.source,
+          });
+          break;
+
+        case "file":
+          controller.appendFile({
+            type: "file",
+            mimeType: chunk.mimeType,
+            data: chunk.base64,
+          });
+          break;
+
+        case "reasoning-signature":
+        case "redacted-reasoning":
+          // ignore these for now
+          break;
+
+        default: {
+          const unhandledType: never = type;
+          throw new Error(`Unhandled chunk type: ${unhandledType}`);
+        }
+      }
+    },
+    flush() {
+      for (const toolController of toolControllers.values()) {
+        toolController.close();
+      }
+      toolControllers.clear();
+    },
+  });
+
+  return stream.pipeThrough(transformer);
+};
+
+export const fromStreamObject = (
+  stream: ReadableStream<ObjectStreamPart<unknown>>,
+  toolName: string,
+): AssistantStream => {
+  let toolCall!: ToolCallStreamController;
+  const transformer = new AssistantTransformStream<ObjectStreamPart<unknown>>({
+    start(controller) {
+      toolCall = controller.addToolCallPart(toolName);
+    },
+    transform(chunk, controller) {
+      const { type } = chunk;
+      switch (type) {
+        case "text-delta": {
+          const { textDelta } = chunk;
+          toolCall.argsText.append(textDelta);
+          break;
+        }
+        case "finish": {
+          toolCall.argsText.close();
+          toolCall.setResponse({
+            result: "{}",
+          });
+          break;
+        }
+
+        case "object":
+          break;
+
+        case "error": {
+          controller.enqueue({
+            type: "error",
+            path: [],
+            error: String(chunk.error),
+          });
+          break;
+        }
+
+        default: {
+          const unhandledType: never = type;
+          throw new Error(`Unhandled chunk type: ${unhandledType}`);
+        }
+      }
+    },
+  });
+
+  return stream.pipeThrough(transformer);
+};