assistant-stream 0.1.5 → 0.1.7

package/dist/core/tool/ToolCallReader.js

@@ -0,0 +1,339 @@
+"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/core/tool/ToolCallReader.ts
+var ToolCallReader_exports = {};
+__export(ToolCallReader_exports, {
+  ToolCallArgsReaderImpl: () => ToolCallArgsReaderImpl,
+  ToolCallReaderImpl: () => ToolCallReaderImpl,
+  ToolCallResponseReaderImpl: () => ToolCallResponseReaderImpl
+});
+module.exports = __toCommonJS(ToolCallReader_exports);
+var import_promiseWithResolvers = require("../../utils/promiseWithResolvers.js");
+var import_parse_partial_json_object = require("../../utils/json/parse-partial-json-object.js");
+function getField(obj, fieldPath) {
+  let current = obj;
+  for (const key of fieldPath) {
+    if (current === void 0 || current === null) {
+      return void 0;
+    }
+    current = current[key];
+  }
+  return current;
+}
+var GetHandle = class {
+  resolve;
+  reject;
+  disposed = false;
+  fieldPath;
+  constructor(resolve, reject, fieldPath) {
+    this.resolve = resolve;
+    this.reject = reject;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      if ((0, import_parse_partial_json_object.getPartialJsonObjectFieldState)(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        const value = getField(args, this.fieldPath);
+        if (value !== void 0) {
+          this.resolve(value);
+          this.dispose();
+        }
+      }
+    } catch (e) {
+      this.reject(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var StreamValuesHandle = class {
+  controller;
+  disposed = false;
+  fieldPath;
+  constructor(controller, fieldPath) {
+    this.controller = controller;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      const value = getField(args, this.fieldPath);
+      if (value !== void 0) {
+        this.controller.enqueue(value);
+      }
+      if ((0, import_parse_partial_json_object.getPartialJsonObjectFieldState)(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        this.controller.close();
+        this.dispose();
+      }
+    } catch (e) {
+      this.controller.error(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var StreamTextHandle = class {
+  controller;
+  disposed = false;
+  fieldPath;
+  lastValue = void 0;
+  constructor(controller, fieldPath) {
+    this.controller = controller;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      const value = getField(args, this.fieldPath);
+      if (value !== void 0 && typeof value === "string") {
+        const delta = value.substring(this.lastValue?.length || 0);
+        this.lastValue = value;
+        this.controller.enqueue(delta);
+      }
+      if ((0, import_parse_partial_json_object.getPartialJsonObjectFieldState)(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        this.controller.close();
+        this.dispose();
+      }
+    } catch (e) {
+      this.controller.error(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var ForEachHandle = class {
+  controller;
+  disposed = false;
+  fieldPath;
+  processedIndexes = /* @__PURE__ */ new Set();
+  constructor(controller, fieldPath) {
+    this.controller = controller;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      const array = getField(args, this.fieldPath);
+      if (!Array.isArray(array)) {
+        return;
+      }
+      for (let i = 0; i < array.length; i++) {
+        if (!this.processedIndexes.has(i)) {
+          const elementPath = [...this.fieldPath, i];
+          if ((0, import_parse_partial_json_object.getPartialJsonObjectFieldState)(
+            args,
+            elementPath
+          ) === "complete") {
+            this.controller.enqueue(array[i]);
+            this.processedIndexes.add(i);
+          }
+        }
+      }
+      if ((0, import_parse_partial_json_object.getPartialJsonObjectFieldState)(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        this.controller.close();
+        this.dispose();
+      }
+    } catch (e) {
+      this.controller.error(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var ToolCallArgsReaderImpl = class {
+  argTextDeltas;
+  handles = /* @__PURE__ */ new Set();
+  args = (0, import_parse_partial_json_object.parsePartialJsonObject)("");
+  constructor(argTextDeltas) {
+    this.argTextDeltas = argTextDeltas;
+    this.processStream();
+  }
+  async processStream() {
+    try {
+      let accumulatedText = "";
+      const reader = this.argTextDeltas.getReader();
+      while (true) {
+        const { value, done } = await reader.read();
+        if (done) break;
+        accumulatedText += value;
+        const parsedArgs = (0, import_parse_partial_json_object.parsePartialJsonObject)(accumulatedText);
+        if (parsedArgs !== void 0) {
+          this.args = parsedArgs;
+          for (const handle of this.handles) {
+            handle.update(parsedArgs);
+          }
+        }
+      }
+    } catch (error) {
+      console.error("Error processing argument stream:", error);
+      for (const handle of this.handles) {
+        handle.dispose();
+      }
+    }
+  }
+  get(...fieldPath) {
+    return new Promise((resolve, reject) => {
+      const handle = new GetHandle(resolve, reject, fieldPath);
+      if (this.args && (0, import_parse_partial_json_object.getPartialJsonObjectFieldState)(
+        this.args,
+        fieldPath
+      ) === "complete") {
+        const value = getField(this.args, fieldPath);
+        if (value !== void 0) {
+          resolve(value);
+          return;
+        }
+      }
+      this.handles.add(handle);
+      handle.update(this.args);
+    });
+  }
+  streamValues(...fieldPath) {
+    const simplePath = fieldPath;
+    const stream = new ReadableStream({
+      start: (controller) => {
+        const handle = new StreamValuesHandle(controller, simplePath);
+        this.handles.add(handle);
+        handle.update(this.args);
+      },
+      cancel: () => {
+        for (const handle of this.handles) {
+          if (handle instanceof StreamValuesHandle) {
+            handle.dispose();
+            this.handles.delete(handle);
+            break;
+          }
+        }
+      }
+    });
+    return stream;
+  }
+  streamText(...fieldPath) {
+    const simplePath = fieldPath;
+    const stream = new ReadableStream({
+      start: (controller) => {
+        const handle = new StreamTextHandle(controller, simplePath);
+        this.handles.add(handle);
+        handle.update(this.args);
+      },
+      cancel: () => {
+        for (const handle of this.handles) {
+          if (handle instanceof StreamTextHandle) {
+            handle.dispose();
+            this.handles.delete(handle);
+            break;
+          }
+        }
+      }
+    });
+    return stream;
+  }
+  forEach(...fieldPath) {
+    const simplePath = fieldPath;
+    const stream = new ReadableStream({
+      start: (controller) => {
+        const handle = new ForEachHandle(controller, simplePath);
+        this.handles.add(handle);
+        handle.update(this.args);
+      },
+      cancel: () => {
+        for (const handle of this.handles) {
+          if (handle instanceof ForEachHandle) {
+            handle.dispose();
+            this.handles.delete(handle);
+            break;
+          }
+        }
+      }
+    });
+    return stream;
+  }
+};
+var ToolCallResponseReaderImpl = class {
+  constructor(promise) {
+    this.promise = promise;
+  }
+  get() {
+    return this.promise;
+  }
+};
+var ToolCallReaderImpl = class {
+  args;
+  response;
+  writable;
+  resolve;
+  argsText = "";
+  constructor() {
+    const stream = new TransformStream();
+    this.writable = stream.writable;
+    this.args = new ToolCallArgsReaderImpl(stream.readable);
+    const { promise, resolve } = (0, import_promiseWithResolvers.promiseWithResolvers)();
+    this.resolve = resolve;
+    this.response = new ToolCallResponseReaderImpl(promise);
+  }
+  async appendArgsTextDelta(text) {
+    const writer = this.writable.getWriter();
+    try {
+      await writer.write(text);
+    } catch (err) {
+      console.warn(err);
+    } finally {
+      writer.releaseLock();
+    }
+    this.argsText += text;
+  }
+  setResponse(value) {
+    this.resolve(value);
+  }
+  result = {
+    get: async () => {
+      const response = await this.response.get();
+      return response.result;
+    }
+  };
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ToolCallArgsReaderImpl,
+  ToolCallReaderImpl,
+  ToolCallResponseReaderImpl
+});
+//# sourceMappingURL=ToolCallReader.js.map

package/dist/core/tool/ToolCallReader.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/tool/ToolCallReader.ts"],"sourcesContent":["import { promiseWithResolvers } from \"../../utils/promiseWithResolvers\";\nimport {\n  parsePartialJsonObject,\n  getPartialJsonObjectFieldState,\n} from \"../../utils/json/parse-partial-json-object\";\nimport {\n  ToolCallArgsReader,\n  ToolCallReader,\n  ToolCallResponseReader,\n} from \"./tool-types\";\nimport { TypeAtPath, TypePath } from \"./type-path-utils\";\nimport { ToolResponse } from \"./ToolResponse\";\n\n// TODO: remove dispose\n\nfunction getField<T>(obj: T, fieldPath: (string | number)[]): any {\n  let current: any = obj;\n  for (const key of fieldPath) {\n    if (current === undefined || current === null) {\n      return undefined;\n    }\n    current = current[key as string | number];\n  }\n  return current;\n}\n\ninterface Handle {\n  update(args: unknown): void;\n  dispose(): void;\n}\n\nclass GetHandle<T> implements Handle {\n  private resolve: (value: any) => void;\n  private reject: (reason: unknown) => void;\n  private disposed = false;\n  private fieldPath: (string | number)[];\n\n  constructor(\n    resolve: (value: any) => void,\n    reject: (reason: unknown) => void,\n    fieldPath: (string | number)[],\n  ) {\n    this.resolve = resolve;\n    this.reject = reject;\n    this.fieldPath = fieldPath;\n  }\n\n  update(args: unknown): void {\n    if (this.disposed) return;\n\n    try {\n      // Check if the field is complete\n      if (\n        getPartialJsonObjectFieldState(\n          args as Record<string, unknown>,\n          this.fieldPath,\n        ) === \"complete\"\n      ) {\n        const value = getField(args as T, this.fieldPath);\n        if (value !== undefined) {\n          this.resolve(value);\n          this.dispose();\n        }\n      }\n    } catch (e) {\n      this.reject(e);\n      this.dispose();\n    }\n  }\n\n  dispose(): void {\n    this.disposed = true;\n  }\n}\n\nclass StreamValuesHandle<T> implements Handle {\n  private controller: ReadableStreamDefaultController<any>;\n  private disposed = false;\n  private fieldPath: (string | number)[];\n\n  constructor(\n    controller: ReadableStreamDefaultController<any>,\n    fieldPath: (string | number)[],\n  ) {\n    this.controller = controller;\n    this.fieldPath = fieldPath;\n  }\n\n  update(args: unknown): void {\n    if (this.disposed) return;\n\n    try {\n      const value = getField(args as T, this.fieldPath);\n\n      if (value !== undefined) {\n        this.controller.enqueue(value);\n      }\n\n      // Check if the field is complete, if so close the stream\n      if (\n        getPartialJsonObjectFieldState(\n          args as Record<string, unknown>,\n          this.fieldPath,\n        ) === \"complete\"\n      ) {\n        this.controller.close();\n        this.dispose();\n      }\n    } catch (e) {\n      this.controller.error(e);\n      this.dispose();\n    }\n  }\n\n  dispose(): void {\n    this.disposed = true;\n  }\n}\n\nclass StreamTextHandle<T> implements Handle {\n  private controller: ReadableStreamDefaultController<any>;\n  private disposed = false;\n  private fieldPath: (string | number)[];\n  private lastValue: any = undefined;\n\n  constructor(\n    controller: ReadableStreamDefaultController<any>,\n    fieldPath: (string | number)[],\n  ) {\n    this.controller = controller;\n    this.fieldPath = fieldPath;\n  }\n\n  update(args: unknown): void {\n    if (this.disposed) return;\n\n    try {\n      const value = getField(args as T, this.fieldPath);\n\n      if (value !== undefined && typeof value === \"string\") {\n        const delta = value.substring(this.lastValue?.length || 0);\n        this.lastValue = value;\n        this.controller.enqueue(delta);\n      }\n\n      // Check if the field is complete, if so close the stream\n      if (\n        getPartialJsonObjectFieldState(\n          args as Record<string, unknown>,\n          this.fieldPath,\n        ) === \"complete\"\n      ) {\n        this.controller.close();\n        this.dispose();\n      }\n    } catch (e) {\n      this.controller.error(e);\n      this.dispose();\n    }\n  }\n\n  dispose(): void {\n    this.disposed = true;\n  }\n}\n\nclass ForEachHandle<T> implements Handle {\n  private controller: ReadableStreamDefaultController<any>;\n  private disposed = false;\n  private fieldPath: (string | number)[];\n  private processedIndexes = new Set<number>();\n\n  constructor(\n    controller: ReadableStreamDefaultController<any>,\n    fieldPath: (string | number)[],\n  ) {\n    this.controller = controller;\n    this.fieldPath = fieldPath;\n  }\n\n  update(args: unknown): void {\n    if (this.disposed) return;\n\n    try {\n      const array = getField(args as T, this.fieldPath) as unknown as any[];\n\n      if (!Array.isArray(array)) {\n        return;\n      }\n\n      // Check each array element and emit completed ones that haven't been processed\n      for (let i = 0; i < array.length; i++) {\n        if (!this.processedIndexes.has(i)) {\n          const elementPath = [...this.fieldPath, i];\n          if (\n            getPartialJsonObjectFieldState(\n              args as Record<string, unknown>,\n              elementPath,\n            ) === \"complete\"\n          ) {\n            this.controller.enqueue(array[i]);\n            this.processedIndexes.add(i);\n          }\n        }\n      }\n\n      // Check if the entire array is complete\n      if (\n        getPartialJsonObjectFieldState(\n          args as Record<string, unknown>,\n          this.fieldPath,\n        ) === \"complete\"\n      ) {\n        this.controller.close();\n        this.dispose();\n      }\n    } catch (e) {\n      this.controller.error(e);\n      this.dispose();\n    }\n  }\n\n  dispose(): void {\n    this.disposed = true;\n  }\n}\n\n// Implementation of ToolCallReader that uses stream of partial JSON\nexport class ToolCallArgsReaderImpl<T> implements ToolCallArgsReader<T> {\n  private argTextDeltas: ReadableStream<string>;\n  private handles: Set<Handle> = new Set();\n  private args: any = parsePartialJsonObject(\"\");\n\n  constructor(argTextDeltas: ReadableStream<string>) {\n    this.argTextDeltas = argTextDeltas;\n    this.processStream();\n  }\n\n  private async processStream(): Promise<void> {\n    try {\n      let accumulatedText = \"\";\n      const reader = this.argTextDeltas.getReader();\n\n      while (true) {\n        const { value, done } = await reader.read();\n        if (done) break;\n\n        accumulatedText += value;\n        const parsedArgs = parsePartialJsonObject(accumulatedText);\n\n        if (parsedArgs !== undefined) {\n          this.args = parsedArgs;\n          // Notify all handles of the updated args\n          for (const handle of this.handles) {\n            handle.update(parsedArgs);\n          }\n        }\n      }\n    } catch (error) {\n      console.error(\"Error processing argument stream:\", error);\n      // Notify handles of the error\n      for (const handle of this.handles) {\n        handle.dispose();\n      }\n    }\n  }\n\n  get<PathT extends TypePath<T>>(\n    ...fieldPath: PathT\n  ): Promise<TypeAtPath<T, PathT>> {\n    return new Promise<any>((resolve, reject) => {\n      const handle = new GetHandle<T>(resolve, reject, fieldPath);\n\n      // Check if the field is already complete in current args\n      if (\n        this.args &&\n        getPartialJsonObjectFieldState(\n          this.args as Record<string, unknown>,\n          fieldPath,\n        ) === \"complete\"\n      ) {\n        const value = getField(this.args as T, fieldPath);\n        if (value !== undefined) {\n          resolve(value);\n          return;\n        }\n      }\n\n      this.handles.add(handle);\n      handle.update(this.args);\n    });\n  }\n\n  streamValues<PathT extends TypePath<T>>(...fieldPath: PathT): any {\n    // Use a type assertion to convert the complex TypePath to a simple array\n    const simplePath = fieldPath as unknown as (string | number)[];\n\n    const stream = new ReadableStream<any>({\n      start: (controller) => {\n        const handle = new StreamValuesHandle<T>(controller, simplePath);\n        this.handles.add(handle);\n\n        // Check current args immediately\n        handle.update(this.args);\n      },\n      cancel: () => {\n        // Find and dispose the corresponding handle\n        for (const handle of this.handles) {\n          if (handle instanceof StreamValuesHandle) {\n            handle.dispose();\n            this.handles.delete(handle);\n            break;\n          }\n        }\n      },\n    });\n\n    // For type compatibility, cast the stream to the required type\n    return stream as any;\n  }\n\n  streamText<PathT extends TypePath<T>>(...fieldPath: PathT): any {\n    // Use a type assertion to convert the complex TypePath to a simple array\n    const simplePath = fieldPath as unknown as (string | number)[];\n\n    const stream = new ReadableStream<any>({\n      start: (controller) => {\n        const handle = new StreamTextHandle<T>(controller, simplePath);\n        this.handles.add(handle);\n\n        // Check current args immediately\n        handle.update(this.args);\n      },\n      cancel: () => {\n        // Find and dispose the corresponding handle\n        for (const handle of this.handles) {\n          if (handle instanceof StreamTextHandle) {\n            handle.dispose();\n            this.handles.delete(handle);\n            break;\n          }\n        }\n      },\n    });\n\n    // For type compatibility, cast the stream to the required type\n    return stream as any;\n  }\n\n  forEach<PathT extends TypePath<T>>(...fieldPath: PathT): any {\n    // Use a type assertion to convert the complex TypePath to a simple array\n    const simplePath = fieldPath as unknown as (string | number)[];\n\n    const stream = new ReadableStream<any>({\n      start: (controller) => {\n        const handle = new ForEachHandle<T>(controller, simplePath);\n        this.handles.add(handle);\n\n        // Check current args immediately\n        handle.update(this.args);\n      },\n      cancel: () => {\n        // Find and dispose the corresponding handle\n        for (const handle of this.handles) {\n          if (handle instanceof ForEachHandle) {\n            handle.dispose();\n            this.handles.delete(handle);\n            break;\n          }\n        }\n      },\n    });\n\n    // For type compatibility, cast the stream to the required type\n    return stream as any;\n  }\n}\n\nexport class ToolCallResponseReaderImpl<TResult>\n  implements ToolCallResponseReader<TResult>\n{\n  constructor(private readonly promise: Promise<ToolResponse<TResult>>) {}\n\n  public get() {\n    return this.promise;\n  }\n}\n\nexport class ToolCallReaderImpl<TArgs, TResult>\n  implements ToolCallReader<TArgs, TResult>\n{\n  public readonly args: ToolCallArgsReaderImpl<TArgs>;\n  public readonly response: ToolCallResponseReaderImpl<TResult>;\n  private readonly writable: WritableStream<string>;\n  private readonly resolve: (value: ToolResponse<TResult>) => void;\n\n  public argsText: string = \"\";\n\n  constructor() {\n    const stream = new TransformStream<string, string>();\n    this.writable = stream.writable;\n    this.args = new ToolCallArgsReaderImpl<TArgs>(stream.readable);\n\n    const { promise, resolve } = promiseWithResolvers<ToolResponse<TResult>>();\n    this.resolve = resolve;\n    this.response = new ToolCallResponseReaderImpl<TResult>(promise);\n  }\n\n  async appendArgsTextDelta(text: string): Promise<void> {\n    const writer = this.writable.getWriter();\n    try {\n      await writer.write(text);\n    } catch (err) {\n      console.warn(err);\n    } finally {\n      writer.releaseLock();\n    }\n\n    this.argsText += text;\n  }\n\n  setResponse(value: ToolResponse<TResult>): void {\n    this.resolve(value);\n  }\n\n  result = {\n    get: async () => {\n      const response = await this.response.get();\n      return response.result;\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kCAAqC;AACrC,uCAGO;AAWP,SAAS,SAAY,KAAQ,WAAqC;AAChE,MAAI,UAAe;AACnB,aAAW,OAAO,WAAW;AAC3B,QAAI,YAAY,UAAa,YAAY,MAAM;AAC7C,aAAO;AAAA,IACT;AACA,cAAU,QAAQ,GAAsB;AAAA,EAC1C;AACA,SAAO;AACT;AAOA,IAAM,YAAN,MAAqC;AAAA,EAC3B;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EAER,YACE,SACA,QACA,WACA;AACA,SAAK,UAAU;AACf,SAAK,SAAS;AACd,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,OAAO,MAAqB;AAC1B,QAAI,KAAK,SAAU;AAEnB,QAAI;AAEF,cACE;AAAA,QACE;AAAA,QACA,KAAK;AAAA,MACP,MAAM,YACN;AACA,cAAM,QAAQ,SAAS,MAAW,KAAK,SAAS;AAChD,YAAI,UAAU,QAAW;AACvB,eAAK,QAAQ,KAAK;AAClB,eAAK,QAAQ;AAAA,QACf;AAAA,MACF;AAAA,IACF,SAAS,GAAG;AACV,WAAK,OAAO,CAAC;AACb,WAAK,QAAQ;AAAA,IACf;AAAA,EACF;AAAA,EAEA,UAAgB;AACd,SAAK,WAAW;AAAA,EAClB;AACF;AAEA,IAAM,qBAAN,MAA8C;AAAA,EACpC;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EAER,YACE,YACA,WACA;AACA,SAAK,aAAa;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,OAAO,MAAqB;AAC1B,QAAI,KAAK,SAAU;AAEnB,QAAI;AACF,YAAM,QAAQ,SAAS,MAAW,KAAK,SAAS;AAEhD,UAAI,UAAU,QAAW;AACvB,aAAK,WAAW,QAAQ,KAAK;AAAA,MAC/B;AAGA,cACE;AAAA,QACE;AAAA,QACA,KAAK;AAAA,MACP,MAAM,YACN;AACA,aAAK,WAAW,MAAM;AACtB,aAAK,QAAQ;AAAA,MACf;AAAA,IACF,SAAS,GAAG;AACV,WAAK,WAAW,MAAM,CAAC;AACvB,WAAK,QAAQ;AAAA,IACf;AAAA,EACF;AAAA,EAEA,UAAgB;AACd,SAAK,WAAW;AAAA,EAClB;AACF;AAEA,IAAM,mBAAN,MAA4C;AAAA,EAClC;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA,YAAiB;AAAA,EAEzB,YACE,YACA,WACA;AACA,SAAK,aAAa;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,OAAO,MAAqB;AAC1B,QAAI,KAAK,SAAU;AAEnB,QAAI;AACF,YAAM,QAAQ,SAAS,MAAW,KAAK,SAAS;AAEhD,UAAI,UAAU,UAAa,OAAO,UAAU,UAAU;AACpD,cAAM,QAAQ,MAAM,UAAU,KAAK,WAAW,UAAU,CAAC;AACzD,aAAK,YAAY;AACjB,aAAK,WAAW,QAAQ,KAAK;AAAA,MAC/B;AAGA,cACE;AAAA,QACE;AAAA,QACA,KAAK;AAAA,MACP,MAAM,YACN;AACA,aAAK,WAAW,MAAM;AACtB,aAAK,QAAQ;AAAA,MACf;AAAA,IACF,SAAS,GAAG;AACV,WAAK,WAAW,MAAM,CAAC;AACvB,WAAK,QAAQ;AAAA,IACf;AAAA,EACF;AAAA,EAEA,UAAgB;AACd,SAAK,WAAW;AAAA,EAClB;AACF;AAEA,IAAM,gBAAN,MAAyC;AAAA,EAC/B;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA,mBAAmB,oBAAI,IAAY;AAAA,EAE3C,YACE,YACA,WACA;AACA,SAAK,aAAa;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,OAAO,MAAqB;AAC1B,QAAI,KAAK,SAAU;AAEnB,QAAI;AACF,YAAM,QAAQ,SAAS,MAAW,KAAK,SAAS;AAEhD,UAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB;AAAA,MACF;AAGA,eAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,YAAI,CAAC,KAAK,iBAAiB,IAAI,CAAC,GAAG;AACjC,gBAAM,cAAc,CAAC,GAAG,KAAK,WAAW,CAAC;AACzC,kBACE;AAAA,YACE;AAAA,YACA;AAAA,UACF,MAAM,YACN;AACA,iBAAK,WAAW,QAAQ,MAAM,CAAC,CAAC;AAChC,iBAAK,iBAAiB,IAAI,CAAC;AAAA,UAC7B;AAAA,QACF;AAAA,MACF;AAGA,cACE;AAAA,QACE;AAAA,QACA,KAAK;AAAA,MACP,MAAM,YACN;AACA,aAAK,WAAW,MAAM;AACtB,aAAK,QAAQ;AAAA,MACf;AAAA,IACF,SAAS,GAAG;AACV,WAAK,WAAW,MAAM,CAAC;AACvB,WAAK,QAAQ;AAAA,IACf;AAAA,EACF;AAAA,EAEA,UAAgB;AACd,SAAK,WAAW;AAAA,EAClB;AACF;AAGO,IAAM,yBAAN,MAAiE;AAAA,EAC9D;AAAA,EACA,UAAuB,oBAAI,IAAI;AAAA,EAC/B,WAAY,yDAAuB,EAAE;AAAA,EAE7C,YAAY,eAAuC;AACjD,SAAK,gBAAgB;AACrB,SAAK,cAAc;AAAA,EACrB;AAAA,EAEA,MAAc,gBAA+B;AAC3C,QAAI;AACF,UAAI,kBAAkB;AACtB,YAAM,SAAS,KAAK,cAAc,UAAU;AAE5C,aAAO,MAAM;AACX,cAAM,EAAE,OAAO,KAAK,IAAI,MAAM,OAAO,KAAK;AAC1C,YAAI,KAAM;AAEV,2BAAmB;AACnB,cAAM,iBAAa,yDAAuB,eAAe;AAEzD,YAAI,eAAe,QAAW;AAC5B,eAAK,OAAO;AAEZ,qBAAW,UAAU,KAAK,SAAS;AACjC,mBAAO,OAAO,UAAU;AAAA,UAC1B;AAAA,QACF;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,cAAQ,MAAM,qCAAqC,KAAK;AAExD,iBAAW,UAAU,KAAK,SAAS;AACjC,eAAO,QAAQ;AAAA,MACjB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OACK,WAC4B;AAC/B,WAAO,IAAI,QAAa,CAAC,SAAS,WAAW;AAC3C,YAAM,SAAS,IAAI,UAAa,SAAS,QAAQ,SAAS;AAG1D,UACE,KAAK,YACL;AAAA,QACE,KAAK;AAAA,QACL;AAAA,MACF,MAAM,YACN;AACA,cAAM,QAAQ,SAAS,KAAK,MAAW,SAAS;AAChD,YAAI,UAAU,QAAW;AACvB,kBAAQ,KAAK;AACb;AAAA,QACF;AAAA,MACF;AAEA,WAAK,QAAQ,IAAI,MAAM;AACvB,aAAO,OAAO,KAAK,IAAI;AAAA,IACzB,CAAC;AAAA,EACH;AAAA,EAEA,gBAA2C,WAAuB;AAEhE,UAAM,aAAa;AAEnB,UAAM,SAAS,IAAI,eAAoB;AAAA,MACrC,OAAO,CAAC,eAAe;AACrB,cAAM,SAAS,IAAI,mBAAsB,YAAY,UAAU;AAC/D,aAAK,QAAQ,IAAI,MAAM;AAGvB,eAAO,OAAO,KAAK,IAAI;AAAA,MACzB;AAAA,MACA,QAAQ,MAAM;AAEZ,mBAAW,UAAU,KAAK,SAAS;AACjC,cAAI,kBAAkB,oBAAoB;AACxC,mBAAO,QAAQ;AACf,iBAAK,QAAQ,OAAO,MAAM;AAC1B;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAGD,WAAO;AAAA,EACT;AAAA,EAEA,cAAyC,WAAuB;AAE9D,UAAM,aAAa;AAEnB,UAAM,SAAS,IAAI,eAAoB;AAAA,MACrC,OAAO,CAAC,eAAe;AACrB,cAAM,SAAS,IAAI,iBAAoB,YAAY,UAAU;AAC7D,aAAK,QAAQ,IAAI,MAAM;AAGvB,eAAO,OAAO,KAAK,IAAI;AAAA,MACzB;AAAA,MACA,QAAQ,MAAM;AAEZ,mBAAW,UAAU,KAAK,SAAS;AACjC,cAAI,kBAAkB,kBAAkB;AACtC,mBAAO,QAAQ;AACf,iBAAK,QAAQ,OAAO,MAAM;AAC1B;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAGD,WAAO;AAAA,EACT;AAAA,EAEA,WAAsC,WAAuB;AAE3D,UAAM,aAAa;AAEnB,UAAM,SAAS,IAAI,eAAoB;AAAA,MACrC,OAAO,CAAC,eAAe;AACrB,cAAM,SAAS,IAAI,cAAiB,YAAY,UAAU;AAC1D,aAAK,QAAQ,IAAI,MAAM;AAGvB,eAAO,OAAO,KAAK,IAAI;AAAA,MACzB;AAAA,MACA,QAAQ,MAAM;AAEZ,mBAAW,UAAU,KAAK,SAAS;AACjC,cAAI,kBAAkB,eAAe;AACnC,mBAAO,QAAQ;AACf,iBAAK,QAAQ,OAAO,MAAM;AAC1B;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAGD,WAAO;AAAA,EACT;AACF;AAEO,IAAM,6BAAN,MAEP;AAAA,EACE,YAA6B,SAAyC;AAAzC;AAAA,EAA0C;AAAA,EAEhE,MAAM;AACX,WAAO,KAAK;AAAA,EACd;AACF;AAEO,IAAM,qBAAN,MAEP;AAAA,EACkB;AAAA,EACA;AAAA,EACC;AAAA,EACA;AAAA,EAEV,WAAmB;AAAA,EAE1B,cAAc;AACZ,UAAM,SAAS,IAAI,gBAAgC;AACnD,SAAK,WAAW,OAAO;AACvB,SAAK,OAAO,IAAI,uBAA8B,OAAO,QAAQ;AAE7D,UAAM,EAAE,SAAS,QAAQ,QAAI,kDAA4C;AACzE,SAAK,UAAU;AACf,SAAK,WAAW,IAAI,2BAAoC,OAAO;AAAA,EACjE;AAAA,EAEA,MAAM,oBAAoB,MAA6B;AACrD,UAAM,SAAS,KAAK,SAAS,UAAU;AACvC,QAAI;AACF,YAAM,OAAO,MAAM,IAAI;AAAA,IACzB,SAAS,KAAK;AACZ,cAAQ,KAAK,GAAG;AAAA,IAClB,UAAE;AACA,aAAO,YAAY;AAAA,IACrB;AAEA,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,YAAY,OAAoC;AAC9C,SAAK,QAAQ,KAAK;AAAA,EACpB;AAAA,EAEA,SAAS;AAAA,IACP,KAAK,YAAY;AACf,YAAM,WAAW,MAAM,KAAK,SAAS,IAAI;AACzC,aAAO,SAAS;AAAA,IAClB;AAAA,EACF;AACF;","names":[]}

package/dist/core/tool/ToolCallReader.mjs

@@ -0,0 +1,315 @@
+// src/core/tool/ToolCallReader.ts
+import { promiseWithResolvers } from "../../utils/promiseWithResolvers.mjs";
+import {
+  parsePartialJsonObject,
+  getPartialJsonObjectFieldState
+} from "../../utils/json/parse-partial-json-object.mjs";
+function getField(obj, fieldPath) {
+  let current = obj;
+  for (const key of fieldPath) {
+    if (current === void 0 || current === null) {
+      return void 0;
+    }
+    current = current[key];
+  }
+  return current;
+}
+var GetHandle = class {
+  resolve;
+  reject;
+  disposed = false;
+  fieldPath;
+  constructor(resolve, reject, fieldPath) {
+    this.resolve = resolve;
+    this.reject = reject;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      if (getPartialJsonObjectFieldState(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        const value = getField(args, this.fieldPath);
+        if (value !== void 0) {
+          this.resolve(value);
+          this.dispose();
+        }
+      }
+    } catch (e) {
+      this.reject(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var StreamValuesHandle = class {
+  controller;
+  disposed = false;
+  fieldPath;
+  constructor(controller, fieldPath) {
+    this.controller = controller;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      const value = getField(args, this.fieldPath);
+      if (value !== void 0) {
+        this.controller.enqueue(value);
+      }
+      if (getPartialJsonObjectFieldState(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        this.controller.close();
+        this.dispose();
+      }
+    } catch (e) {
+      this.controller.error(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var StreamTextHandle = class {
+  controller;
+  disposed = false;
+  fieldPath;
+  lastValue = void 0;
+  constructor(controller, fieldPath) {
+    this.controller = controller;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      const value = getField(args, this.fieldPath);
+      if (value !== void 0 && typeof value === "string") {
+        const delta = value.substring(this.lastValue?.length || 0);
+        this.lastValue = value;
+        this.controller.enqueue(delta);
+      }
+      if (getPartialJsonObjectFieldState(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        this.controller.close();
+        this.dispose();
+      }
+    } catch (e) {
+      this.controller.error(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var ForEachHandle = class {
+  controller;
+  disposed = false;
+  fieldPath;
+  processedIndexes = /* @__PURE__ */ new Set();
+  constructor(controller, fieldPath) {
+    this.controller = controller;
+    this.fieldPath = fieldPath;
+  }
+  update(args) {
+    if (this.disposed) return;
+    try {
+      const array = getField(args, this.fieldPath);
+      if (!Array.isArray(array)) {
+        return;
+      }
+      for (let i = 0; i < array.length; i++) {
+        if (!this.processedIndexes.has(i)) {
+          const elementPath = [...this.fieldPath, i];
+          if (getPartialJsonObjectFieldState(
+            args,
+            elementPath
+          ) === "complete") {
+            this.controller.enqueue(array[i]);
+            this.processedIndexes.add(i);
+          }
+        }
+      }
+      if (getPartialJsonObjectFieldState(
+        args,
+        this.fieldPath
+      ) === "complete") {
+        this.controller.close();
+        this.dispose();
+      }
+    } catch (e) {
+      this.controller.error(e);
+      this.dispose();
+    }
+  }
+  dispose() {
+    this.disposed = true;
+  }
+};
+var ToolCallArgsReaderImpl = class {
+  argTextDeltas;
+  handles = /* @__PURE__ */ new Set();
+  args = parsePartialJsonObject("");
+  constructor(argTextDeltas) {
+    this.argTextDeltas = argTextDeltas;
+    this.processStream();
+  }
+  async processStream() {
+    try {
+      let accumulatedText = "";
+      const reader = this.argTextDeltas.getReader();
+      while (true) {
+        const { value, done } = await reader.read();
+        if (done) break;
+        accumulatedText += value;
+        const parsedArgs = parsePartialJsonObject(accumulatedText);
+        if (parsedArgs !== void 0) {
+          this.args = parsedArgs;
+          for (const handle of this.handles) {
+            handle.update(parsedArgs);
+          }
+        }
+      }
+    } catch (error) {
+      console.error("Error processing argument stream:", error);
+      for (const handle of this.handles) {
+        handle.dispose();
+      }
+    }
+  }
+  get(...fieldPath) {
+    return new Promise((resolve, reject) => {
+      const handle = new GetHandle(resolve, reject, fieldPath);
+      if (this.args && getPartialJsonObjectFieldState(
+        this.args,
+        fieldPath
+      ) === "complete") {
+        const value = getField(this.args, fieldPath);
+        if (value !== void 0) {
+          resolve(value);
+          return;
+        }
+      }
+      this.handles.add(handle);
+      handle.update(this.args);
+    });
+  }
+  streamValues(...fieldPath) {
+    const simplePath = fieldPath;
+    const stream = new ReadableStream({
+      start: (controller) => {
+        const handle = new StreamValuesHandle(controller, simplePath);
+        this.handles.add(handle);
+        handle.update(this.args);
+      },
+      cancel: () => {
+        for (const handle of this.handles) {
+          if (handle instanceof StreamValuesHandle) {
+            handle.dispose();
+            this.handles.delete(handle);
+            break;
+          }
+        }
+      }
+    });
+    return stream;
+  }
+  streamText(...fieldPath) {
+    const simplePath = fieldPath;
+    const stream = new ReadableStream({
+      start: (controller) => {
+        const handle = new StreamTextHandle(controller, simplePath);
+        this.handles.add(handle);
+        handle.update(this.args);
+      },
+      cancel: () => {
+        for (const handle of this.handles) {
+          if (handle instanceof StreamTextHandle) {
+            handle.dispose();
+            this.handles.delete(handle);
+            break;
+          }
+        }
+      }
+    });
+    return stream;
+  }
+  forEach(...fieldPath) {
+    const simplePath = fieldPath;
+    const stream = new ReadableStream({
+      start: (controller) => {
+        const handle = new ForEachHandle(controller, simplePath);
+        this.handles.add(handle);
+        handle.update(this.args);
+      },
+      cancel: () => {
+        for (const handle of this.handles) {
+          if (handle instanceof ForEachHandle) {
+            handle.dispose();
+            this.handles.delete(handle);
+            break;
+          }
+        }
+      }
+    });
+    return stream;
+  }
+};
+var ToolCallResponseReaderImpl = class {
+  constructor(promise) {
+    this.promise = promise;
+  }
+  get() {
+    return this.promise;
+  }
+};
+var ToolCallReaderImpl = class {
+  args;
+  response;
+  writable;
+  resolve;
+  argsText = "";
+  constructor() {
+    const stream = new TransformStream();
+    this.writable = stream.writable;
+    this.args = new ToolCallArgsReaderImpl(stream.readable);
+    const { promise, resolve } = promiseWithResolvers();
+    this.resolve = resolve;
+    this.response = new ToolCallResponseReaderImpl(promise);
+  }
+  async appendArgsTextDelta(text) {
+    const writer = this.writable.getWriter();
+    try {
+      await writer.write(text);
+    } catch (err) {
+      console.warn(err);
+    } finally {
+      writer.releaseLock();
+    }
+    this.argsText += text;
+  }
+  setResponse(value) {
+    this.resolve(value);
+  }
+  result = {
+    get: async () => {
+      const response = await this.response.get();
+      return response.result;
+    }
+  };
+};
+export {
+  ToolCallArgsReaderImpl,
+  ToolCallReaderImpl,
+  ToolCallResponseReaderImpl
+};
+//# sourceMappingURL=ToolCallReader.mjs.map