assistant-stream 0.0.0-rc.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Simon Farshid
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/AssistantStream-CYbWmRWJ.d.mts

@@ -0,0 +1,27 @@
+type AssistantStreamChunk = {
+    type: "text-delta";
+    textDelta: string;
+} | {
+    type: "tool-call-begin";
+    toolCallId: string;
+    toolName: string;
+} | {
+    type: "tool-call-delta";
+    toolCallId: string;
+    argsTextDelta: string;
+} | {
+    type: "tool-result";
+    toolCallId: string;
+    result: any;
+};
+declare class AssistantStream {
+    readonly readable: ReadableStream<AssistantStreamChunk>;
+    constructor(readable: ReadableStream<AssistantStreamChunk>);
+    toResponse(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): Response;
+    static fromResponse(response: Response, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
+    toByteStream(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): ReadableStream<Uint8Array>;
+    static fromByteStream(readable: ReadableStream<Uint8Array>, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
+    tee(): [AssistantStream, AssistantStream];
+}
+
+export { AssistantStream as A, type AssistantStreamChunk as a };

package/dist/AssistantStream-CYbWmRWJ.d.ts

@@ -0,0 +1,27 @@
+type AssistantStreamChunk = {
+    type: "text-delta";
+    textDelta: string;
+} | {
+    type: "tool-call-begin";
+    toolCallId: string;
+    toolName: string;
+} | {
+    type: "tool-call-delta";
+    toolCallId: string;
+    argsTextDelta: string;
+} | {
+    type: "tool-result";
+    toolCallId: string;
+    result: any;
+};
+declare class AssistantStream {
+    readonly readable: ReadableStream<AssistantStreamChunk>;
+    constructor(readable: ReadableStream<AssistantStreamChunk>);
+    toResponse(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): Response;
+    static fromResponse(response: Response, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
+    toByteStream(transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>): ReadableStream<Uint8Array>;
+    static fromByteStream(readable: ReadableStream<Uint8Array>, transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>): AssistantStream;
+    tee(): [AssistantStream, AssistantStream];
+}
+
+export { AssistantStream as A, type AssistantStreamChunk as a };

package/dist/ai-sdk.d.mts

@@ -0,0 +1,7 @@
+import { TextStreamPart, CoreTool, ObjectStreamPart } from 'ai';
+import { A as AssistantStream } from './AssistantStream-CYbWmRWJ.mjs';
+
+declare const fromAISDKStreamText: (stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>) => AssistantStream;
+declare const fromAISDKStreamObject: (stream: ReadableStream<ObjectStreamPart<unknown>>, toolName: string) => AssistantStream;
+
+export { fromAISDKStreamObject, fromAISDKStreamText };

package/dist/ai-sdk.d.ts

@@ -0,0 +1,7 @@
+import { TextStreamPart, CoreTool, ObjectStreamPart } from 'ai';
+import { A as AssistantStream } from './AssistantStream-CYbWmRWJ.js';
+
+declare const fromAISDKStreamText: (stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>) => AssistantStream;
+declare const fromAISDKStreamObject: (stream: ReadableStream<ObjectStreamPart<unknown>>, toolName: string) => AssistantStream;
+
+export { fromAISDKStreamObject, fromAISDKStreamText };

package/dist/ai-sdk.js

@@ -0,0 +1,187 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/ai-sdk.ts
+var ai_sdk_exports = {};
+__export(ai_sdk_exports, {
+  fromAISDKStreamObject: () => fromAISDKStreamObject,
+  fromAISDKStreamText: () => fromAISDKStreamText
+});
+module.exports = __toCommonJS(ai_sdk_exports);
+
+// ../core/src/AssistantStream.ts
+var AssistantStream = class _AssistantStream {
+  constructor(readable) {
+    this.readable = readable;
+    this.readable = readable;
+  }
+  toResponse(transformer) {
+    return new Response(this.toByteStream(transformer));
+  }
+  static fromResponse(response, transformer) {
+    return _AssistantStream.fromByteStream(response.body, transformer);
+  }
+  toByteStream(transformer) {
+    return this.readable.pipeThrough(transformer);
+  }
+  static fromByteStream(readable, transformer) {
+    return new _AssistantStream(readable.pipeThrough(transformer));
+  }
+  tee() {
+    const [readable1, readable2] = this.readable.tee();
+    return [new _AssistantStream(readable1), new _AssistantStream(readable2)];
+  }
+};
+
+// ../../../node_modules/.pnpm/nanoid@5.0.7/node_modules/nanoid/non-secure/index.js
+var customAlphabet = (alphabet, defaultSize = 21) => {
+  return (size = defaultSize) => {
+    let id = "";
+    let i = size;
+    while (i--) {
+      id += alphabet[Math.random() * alphabet.length | 0];
+    }
+    return id;
+  };
+};
+
+// ../core/src/utils/generateId.tsx
+var generateId = customAlphabet(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
+);
+
+// ../ai-sdk/src/index.ts
+var fromAISDKStreamText = (stream) => {
+  const transformer = new TransformStream({
+    transform(chunk, controller) {
+      const { type } = chunk;
+      switch (type) {
+        case "text-delta": {
+          const { textDelta } = chunk;
+          controller.enqueue({
+            type: "text-delta",
+            textDelta
+          });
+          break;
+        }
+        case "tool-call-streaming-start": {
+          const { toolCallId, toolName } = chunk;
+          controller.enqueue({
+            type: "tool-call-begin",
+            toolCallId,
+            toolName
+          });
+          break;
+        }
+        case "tool-call-delta": {
+          const { toolCallId, argsTextDelta } = chunk;
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId,
+            argsTextDelta
+          });
+          break;
+        }
+        case "tool-result": {
+          const { toolCallId, result } = chunk;
+          controller.enqueue({
+            type: "tool-result",
+            toolCallId,
+            result
+          });
+          break;
+        }
+        case "tool-call": {
+          const { toolCallId, toolName, args } = chunk;
+          controller.enqueue({
+            type: "tool-call-begin",
+            toolCallId,
+            toolName
+          });
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId,
+            argsTextDelta: JSON.stringify(args)
+          });
+          break;
+        }
+        case "step-finish":
+        case "error":
+        case "finish": {
+          break;
+        }
+        default: {
+          const unhandledType = type;
+          throw new Error(`Unhandled chunk type: ${unhandledType}`);
+        }
+      }
+    }
+  });
+  return new AssistantStream(stream.pipeThrough(transformer));
+};
+var fromAISDKStreamObject = (stream, toolName) => {
+  const toolCallId = generateId();
+  const transformer = new TransformStream({
+    start(controller) {
+      controller.enqueue({
+        type: "tool-call-begin",
+        toolName,
+        toolCallId
+      });
+    },
+    transform(chunk, controller) {
+      const { type } = chunk;
+      switch (type) {
+        case "text-delta": {
+          const { textDelta } = chunk;
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId,
+            argsTextDelta: textDelta
+          });
+          break;
+        }
+        case "finish": {
+          controller.enqueue({
+            type: "tool-result",
+            toolCallId,
+            result: ""
+          });
+          break;
+        }
+        case "object":
+        case "error": {
+          break;
+        }
+        default: {
+          const unhandledType = type;
+          throw new Error(`Unhandled chunk type: ${unhandledType}`);
+        }
+      }
+    }
+  });
+  return new AssistantStream(stream.pipeThrough(transformer));
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  fromAISDKStreamObject,
+  fromAISDKStreamText
+});
+//# sourceMappingURL=ai-sdk.js.map

package/dist/ai-sdk.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ai-sdk.ts","../../core/src/AssistantStream.ts","../../../../node_modules/.pnpm/nanoid@5.0.7/node_modules/nanoid/non-secure/index.js","../../core/src/utils/generateId.tsx","../../ai-sdk/src/index.ts"],"sourcesContent":["export * from \"@assistant-stream/ai-sdk\";\n ","export type AssistantStream2 = {\n  // kind: AssistantStreamKind;\n  readable: ReadableStream<AssistantStreamChunk>;\n  toResponse(\n    format: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n  ): Response;\n  toByteStream(\n    format: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n  ): ReadableStream<Uint8Array>;\n  tee(): [AssistantStream, AssistantStream];\n};\n\nexport type AssistantStreamChunk =\n  | {\n      type: \"text-delta\";\n      textDelta: string;\n    }\n  | {\n      type: \"tool-call-begin\";\n      toolCallId: string;\n      toolName: string;\n    }\n  | {\n      type: \"tool-call-delta\";\n      toolCallId: string;\n      argsTextDelta: string;\n    }\n  | {\n      type: \"tool-result\";\n      toolCallId: string;\n      result: any;\n    };\n\nexport class AssistantStream {\n  constructor(public readonly readable: ReadableStream<AssistantStreamChunk>) {\n    this.readable = readable;\n  }\n\n  toResponse(\n    transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n  ) {\n    return new Response(this.toByteStream(transformer));\n  }\n\n  static fromResponse(\n    response: Response,\n    transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>,\n  ) {\n    return AssistantStream.fromByteStream(response.body!, transformer);\n  }\n\n  toByteStream(\n    transformer: ReadableWritablePair<Uint8Array, AssistantStreamChunk>,\n  ) {\n    return this.readable.pipeThrough(transformer);\n  }\n\n  static fromByteStream(\n    readable: ReadableStream<Uint8Array>,\n    transformer: ReadableWritablePair<AssistantStreamChunk, Uint8Array>,\n  ) {\n    return new AssistantStream(readable.pipeThrough(transformer));\n  }\n\n  tee(): [AssistantStream, AssistantStream] {\n    const [readable1, readable2] = this.readable.tee();\n    return [new AssistantStream(readable1), new AssistantStream(readable2)];\n  }\n}\n","let urlAlphabet =\n  'useandom-26T198340PX75pxJACKVERYMINDBUSHWOLF_GQZbfghjklqvwyzrict'\nexport let customAlphabet = (alphabet, defaultSize = 21) => {\n  return (size = defaultSize) => {\n    let id = ''\n    let i = size\n    while (i--) {\n      id += alphabet[(Math.random() * alphabet.length) | 0]\n    }\n    return id\n  }\n}\nexport let nanoid = (size = 21) => {\n  let id = ''\n  let i = size\n  while (i--) {\n    id += urlAlphabet[(Math.random() * 64) | 0]\n  }\n  return id\n}\n","import { customAlphabet } from \"nanoid/non-secure\";\n\nexport const generateId = customAlphabet(\n  \"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz\",\n  7,\n);\n","import type { TextStreamPart, CoreTool, ObjectStreamPart } from \"ai\";\nimport { AssistantStream, AssistantStreamChunk } from \"@assistant-stream/core\";\nimport { generateId } from \"../../core/src/utils/generateId\";\n\nexport const fromAISDKStreamText = (\n  stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>,\n): AssistantStream => {\n  const transformer = new TransformStream<\n    TextStreamPart<Record<string, CoreTool>>,\n    AssistantStreamChunk\n  >({\n    transform(chunk, controller) {\n      const { type } = chunk;\n      switch (type) {\n        case \"text-delta\": {\n          const { textDelta } = chunk;\n          controller.enqueue({\n            type: \"text-delta\",\n            textDelta,\n          });\n          break;\n        }\n        case \"tool-call-streaming-start\": {\n          const { toolCallId, toolName } = chunk;\n          controller.enqueue({\n            type: \"tool-call-begin\",\n            toolCallId,\n            toolName,\n          });\n          break;\n        }\n        case \"tool-call-delta\": {\n          const { toolCallId, argsTextDelta } = chunk;\n          controller.enqueue({\n            type: \"tool-call-delta\",\n            toolCallId,\n            argsTextDelta,\n          });\n          break;\n        }\n        case \"tool-result\" as string: {\n          const { toolCallId, result } = chunk as unknown as {\n            toolCallId: string;\n            result: unknown;\n          };\n          controller.enqueue({\n            type: \"tool-result\",\n            toolCallId,\n            result,\n          });\n          break;\n        }\n        case \"tool-call\": {\n          const { toolCallId, toolName, args } = chunk;\n          controller.enqueue({\n            type: \"tool-call-begin\",\n            toolCallId,\n            toolName,\n          });\n          controller.enqueue({\n            type: \"tool-call-delta\",\n            toolCallId,\n            argsTextDelta: JSON.stringify(args),\n          });\n          break;\n        }\n        case \"step-finish\":\n        case \"error\":\n        case \"finish\": {\n          break;\n        }\n\n        default: {\n          const unhandledType: never = type;\n          throw new Error(`Unhandled chunk type: ${unhandledType}`);\n        }\n      }\n    },\n  });\n\n  return new AssistantStream(stream.pipeThrough(transformer));\n};\n\nexport const fromAISDKStreamObject = (\n  stream: ReadableStream<ObjectStreamPart<unknown>>,\n  toolName: string,\n): AssistantStream => {\n  const toolCallId = generateId();\n  const transformer = new TransformStream<\n    ObjectStreamPart<unknown>,\n    AssistantStreamChunk\n  >({\n    start(controller) {\n      controller.enqueue({\n        type: \"tool-call-begin\",\n        toolName,\n        toolCallId,\n      });\n    },\n    transform(chunk, controller) {\n      const { type } = chunk;\n      switch (type) {\n        case \"text-delta\": {\n          const { textDelta } = chunk;\n          controller.enqueue({\n            type: \"tool-call-delta\",\n            toolCallId,\n            argsTextDelta: textDelta,\n          });\n          break;\n        }\n        case \"finish\": {\n          controller.enqueue({\n            type: \"tool-result\",\n            toolCallId,\n            result: \"\",\n          });\n          break;\n        }\n\n        case \"object\":\n        case \"error\": {\n          break;\n        }\n\n        default: {\n          const unhandledType: never = type;\n          throw new Error(`Unhandled chunk type: ${unhandledType}`);\n        }\n      }\n    },\n  });\n\n  return new AssistantStream(stream.pipeThrough(transformer));\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACiCO,IAAM,kBAAN,MAAM,iBAAgB;AAAA,EAC3B,YAA4B,UAAgD;AAAhD;AAC1B,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,WACE,aACA;AACA,WAAO,IAAI,SAAS,KAAK,aAAa,WAAW,CAAC;AAAA,EACpD;AAAA,EAEA,OAAO,aACL,UACA,aACA;AACA,WAAO,iBAAgB,eAAe,SAAS,MAAO,WAAW;AAAA,EACnE;AAAA,EAEA,aACE,aACA;AACA,WAAO,KAAK,SAAS,YAAY,WAAW;AAAA,EAC9C;AAAA,EAEA,OAAO,eACL,UACA,aACA;AACA,WAAO,IAAI,iBAAgB,SAAS,YAAY,WAAW,CAAC;AAAA,EAC9D;AAAA,EAEA,MAA0C;AACxC,UAAM,CAAC,WAAW,SAAS,IAAI,KAAK,SAAS,IAAI;AACjD,WAAO,CAAC,IAAI,iBAAgB,SAAS,GAAG,IAAI,iBAAgB,SAAS,CAAC;AAAA,EACxE;AACF;;;AClEO,IAAI,iBAAiB,CAAC,UAAU,cAAc,OAAO;AAC1D,SAAO,CAAC,OAAO,gBAAgB;AAC7B,QAAI,KAAK;AACT,QAAI,IAAI;AACR,WAAO,KAAK;AACV,YAAM,SAAU,KAAK,OAAO,IAAI,SAAS,SAAU,CAAC;AAAA,IACtD;AACA,WAAO;AAAA,EACT;AACF;;;ACTO,IAAM,aAAa;AAAA,EACxB;AAAA,EACA;AACF;;;ACDO,IAAM,sBAAsB,CACjC,WACoB;AACpB,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,6BAA6B;AAChC,gBAAM,EAAE,YAAY,SAAS,IAAI;AACjC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,mBAAmB;AACtB,gBAAM,EAAE,YAAY,cAAc,IAAI;AACtC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,eAAyB;AAC5B,gBAAM,EAAE,YAAY,OAAO,IAAI;AAI/B,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,aAAa;AAChB,gBAAM,EAAE,YAAY,UAAU,KAAK,IAAI;AACvC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe,KAAK,UAAU,IAAI;AAAA,UACpC,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK,UAAU;AACb;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;AAEO,IAAM,wBAAwB,CACnC,QACA,aACoB;AACpB,QAAM,aAAa,WAAW;AAC9B,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,MAAM,YAAY;AAChB,iBAAW,QAAQ;AAAA,QACjB,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe;AAAA,UACjB,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,UAAU;AACb,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,QAAQ;AAAA,UACV,CAAC;AACD;AAAA,QACF;AAAA,QAEA,KAAK;AAAA,QACL,KAAK,SAAS;AACZ;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;","names":[]}

package/dist/ai-sdk.mjs

@@ -0,0 +1,122 @@
+import {
+  AssistantStream,
+  generateId
+} from "./chunk-JI2TPVLD.mjs";
+
+// ../ai-sdk/src/index.ts
+var fromAISDKStreamText = (stream) => {
+  const transformer = new TransformStream({
+    transform(chunk, controller) {
+      const { type } = chunk;
+      switch (type) {
+        case "text-delta": {
+          const { textDelta } = chunk;
+          controller.enqueue({
+            type: "text-delta",
+            textDelta
+          });
+          break;
+        }
+        case "tool-call-streaming-start": {
+          const { toolCallId, toolName } = chunk;
+          controller.enqueue({
+            type: "tool-call-begin",
+            toolCallId,
+            toolName
+          });
+          break;
+        }
+        case "tool-call-delta": {
+          const { toolCallId, argsTextDelta } = chunk;
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId,
+            argsTextDelta
+          });
+          break;
+        }
+        case "tool-result": {
+          const { toolCallId, result } = chunk;
+          controller.enqueue({
+            type: "tool-result",
+            toolCallId,
+            result
+          });
+          break;
+        }
+        case "tool-call": {
+          const { toolCallId, toolName, args } = chunk;
+          controller.enqueue({
+            type: "tool-call-begin",
+            toolCallId,
+            toolName
+          });
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId,
+            argsTextDelta: JSON.stringify(args)
+          });
+          break;
+        }
+        case "step-finish":
+        case "error":
+        case "finish": {
+          break;
+        }
+        default: {
+          const unhandledType = type;
+          throw new Error(`Unhandled chunk type: ${unhandledType}`);
+        }
+      }
+    }
+  });
+  return new AssistantStream(stream.pipeThrough(transformer));
+};
+var fromAISDKStreamObject = (stream, toolName) => {
+  const toolCallId = generateId();
+  const transformer = new TransformStream({
+    start(controller) {
+      controller.enqueue({
+        type: "tool-call-begin",
+        toolName,
+        toolCallId
+      });
+    },
+    transform(chunk, controller) {
+      const { type } = chunk;
+      switch (type) {
+        case "text-delta": {
+          const { textDelta } = chunk;
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId,
+            argsTextDelta: textDelta
+          });
+          break;
+        }
+        case "finish": {
+          controller.enqueue({
+            type: "tool-result",
+            toolCallId,
+            result: ""
+          });
+          break;
+        }
+        case "object":
+        case "error": {
+          break;
+        }
+        default: {
+          const unhandledType = type;
+          throw new Error(`Unhandled chunk type: ${unhandledType}`);
+        }
+      }
+    }
+  });
+  return new AssistantStream(stream.pipeThrough(transformer));
+};
+export {
+  fromAISDKStreamObject,
+  fromAISDKStreamText
+};
+//# sourceMappingURL=ai-sdk.mjs.map

package/dist/ai-sdk.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../ai-sdk/src/index.ts"],"sourcesContent":["import type { TextStreamPart, CoreTool, ObjectStreamPart } from \"ai\";\nimport { AssistantStream, AssistantStreamChunk } from \"@assistant-stream/core\";\nimport { generateId } from \"../../core/src/utils/generateId\";\n\nexport const fromAISDKStreamText = (\n  stream: ReadableStream<TextStreamPart<Record<string, CoreTool>>>,\n): AssistantStream => {\n  const transformer = new TransformStream<\n    TextStreamPart<Record<string, CoreTool>>,\n    AssistantStreamChunk\n  >({\n    transform(chunk, controller) {\n      const { type } = chunk;\n      switch (type) {\n        case \"text-delta\": {\n          const { textDelta } = chunk;\n          controller.enqueue({\n            type: \"text-delta\",\n            textDelta,\n          });\n          break;\n        }\n        case \"tool-call-streaming-start\": {\n          const { toolCallId, toolName } = chunk;\n          controller.enqueue({\n            type: \"tool-call-begin\",\n            toolCallId,\n            toolName,\n          });\n          break;\n        }\n        case \"tool-call-delta\": {\n          const { toolCallId, argsTextDelta } = chunk;\n          controller.enqueue({\n            type: \"tool-call-delta\",\n            toolCallId,\n            argsTextDelta,\n          });\n          break;\n        }\n        case \"tool-result\" as string: {\n          const { toolCallId, result } = chunk as unknown as {\n            toolCallId: string;\n            result: unknown;\n          };\n          controller.enqueue({\n            type: \"tool-result\",\n            toolCallId,\n            result,\n          });\n          break;\n        }\n        case \"tool-call\": {\n          const { toolCallId, toolName, args } = chunk;\n          controller.enqueue({\n            type: \"tool-call-begin\",\n            toolCallId,\n            toolName,\n          });\n          controller.enqueue({\n            type: \"tool-call-delta\",\n            toolCallId,\n            argsTextDelta: JSON.stringify(args),\n          });\n          break;\n        }\n        case \"step-finish\":\n        case \"error\":\n        case \"finish\": {\n          break;\n        }\n\n        default: {\n          const unhandledType: never = type;\n          throw new Error(`Unhandled chunk type: ${unhandledType}`);\n        }\n      }\n    },\n  });\n\n  return new AssistantStream(stream.pipeThrough(transformer));\n};\n\nexport const fromAISDKStreamObject = (\n  stream: ReadableStream<ObjectStreamPart<unknown>>,\n  toolName: string,\n): AssistantStream => {\n  const toolCallId = generateId();\n  const transformer = new TransformStream<\n    ObjectStreamPart<unknown>,\n    AssistantStreamChunk\n  >({\n    start(controller) {\n      controller.enqueue({\n        type: \"tool-call-begin\",\n        toolName,\n        toolCallId,\n      });\n    },\n    transform(chunk, controller) {\n      const { type } = chunk;\n      switch (type) {\n        case \"text-delta\": {\n          const { textDelta } = chunk;\n          controller.enqueue({\n            type: \"tool-call-delta\",\n            toolCallId,\n            argsTextDelta: textDelta,\n          });\n          break;\n        }\n        case \"finish\": {\n          controller.enqueue({\n            type: \"tool-result\",\n            toolCallId,\n            result: \"\",\n          });\n          break;\n        }\n\n        case \"object\":\n        case \"error\": {\n          break;\n        }\n\n        default: {\n          const unhandledType: never = type;\n          throw new Error(`Unhandled chunk type: ${unhandledType}`);\n        }\n      }\n    },\n  });\n\n  return new AssistantStream(stream.pipeThrough(transformer));\n};\n"],"mappings":";;;;;;AAIO,IAAM,sBAAsB,CACjC,WACoB;AACpB,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,6BAA6B;AAChC,gBAAM,EAAE,YAAY,SAAS,IAAI;AACjC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,mBAAmB;AACtB,gBAAM,EAAE,YAAY,cAAc,IAAI;AACtC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,eAAyB;AAC5B,gBAAM,EAAE,YAAY,OAAO,IAAI;AAI/B,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,aAAa;AAChB,gBAAM,EAAE,YAAY,UAAU,KAAK,IAAI;AACvC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF,CAAC;AACD,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe,KAAK,UAAU,IAAI;AAAA,UACpC,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK,UAAU;AACb;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;AAEO,IAAM,wBAAwB,CACnC,QACA,aACoB;AACpB,QAAM,aAAa,WAAW;AAC9B,QAAM,cAAc,IAAI,gBAGtB;AAAA,IACA,MAAM,YAAY;AAChB,iBAAW,QAAQ;AAAA,QACjB,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,UAAU,OAAO,YAAY;AAC3B,YAAM,EAAE,KAAK,IAAI;AACjB,cAAQ,MAAM;AAAA,QACZ,KAAK,cAAc;AACjB,gBAAM,EAAE,UAAU,IAAI;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,eAAe;AAAA,UACjB,CAAC;AACD;AAAA,QACF;AAAA,QACA,KAAK,UAAU;AACb,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN;AAAA,YACA,QAAQ;AAAA,UACV,CAAC;AACD;AAAA,QACF;AAAA,QAEA,KAAK;AAAA,QACL,KAAK,SAAS;AACZ;AAAA,QACF;AAAA,QAEA,SAAS;AACP,gBAAM,gBAAuB;AAC7B,gBAAM,IAAI,MAAM,yBAAyB,aAAa,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,IAAI,gBAAgB,OAAO,YAAY,WAAW,CAAC;AAC5D;","names":[]}