@agentica/core 0.23.0 → 0.24.0

package/lib/utils/__map_take.spec.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/utils/__map_take.spec.js

@@ -0,0 +1,108 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const __map_take_1 = require("./__map_take");
+describe("__map_take", () => {
+    describe("basic functionality", () => {
+        it("should generate value for new key", () => {
+            const map = new Map();
+            const generator = () => 42;
+            const result = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result).toBe(42);
+            expect(map.get("test")).toBe(42);
+        });
+        it("should return existing value for existing key", () => {
+            const map = new Map();
+            map.set("test", 100);
+            const generator = () => 42;
+            const result = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result).toBe(100);
+            expect(map.get("test")).toBe(100);
+        });
+    });
+    describe("various type tests", () => {
+        it("should handle object type", () => {
+            const map = new Map();
+            const generator = () => ({ value: 42 });
+            const result = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result).toEqual({ value: 42 });
+            expect(map.get("test")).toEqual({ value: 42 });
+        });
+        it("should handle array type", () => {
+            const map = new Map();
+            const generator = () => [1, 2, 3];
+            const result = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result).toEqual([1, 2, 3]);
+            expect(map.get("test")).toEqual([1, 2, 3]);
+        });
+        it("should handle function type", () => {
+            var _a;
+            const map = new Map();
+            const generator = () => () => 42;
+            const result = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result()).toBe(42);
+            expect((_a = map.get("test")) === null || _a === void 0 ? void 0 : _a()).toBe(42);
+        });
+    });
+    describe("edge cases", () => {
+        it("should handle null key", () => {
+            const map = new Map();
+            const generator = () => "test";
+            const result = (0, __map_take_1.__map_take)(map, null, generator);
+            expect(result).toBe("test");
+            expect(map.get(null)).toBe("test");
+        });
+        it("should handle undefined key", () => {
+            const map = new Map();
+            const generator = () => "test";
+            const result = (0, __map_take_1.__map_take)(map, undefined, generator);
+            expect(result).toBe("test");
+            expect(map.get(undefined)).toBe("test");
+        });
+        it("should handle empty string key", () => {
+            const map = new Map();
+            const generator = () => "test";
+            const result = (0, __map_take_1.__map_take)(map, "", generator);
+            expect(result).toBe("test");
+            expect(map.get("")).toBe("test");
+        });
+    });
+    describe("generator function tests", () => {
+        it("should not call generator multiple times", () => {
+            const map = new Map();
+            let callCount = 0;
+            const generator = () => {
+                callCount++;
+                return 42;
+            };
+            (0, __map_take_1.__map_take)(map, "test", generator);
+            (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(callCount).toBe(1);
+        });
+        it("should handle generator throwing error", () => {
+            const map = new Map();
+            const generator = () => {
+                throw new Error("Generator error");
+            };
+            expect(() => (0, __map_take_1.__map_take)(map, "test", generator)).toThrow("Generator error");
+        });
+        it("should handle generator returning undefined", () => {
+            const map = new Map();
+            const generator = () => undefined;
+            const result = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result).toBeUndefined();
+            expect(map.get("test")).toBeUndefined();
+        });
+    });
+    describe("concurrency tests", () => {
+        it("should handle concurrent access to same key", () => {
+            const map = new Map();
+            const generator = () => 42;
+            const result1 = (0, __map_take_1.__map_take)(map, "test", generator);
+            const result2 = (0, __map_take_1.__map_take)(map, "test", generator);
+            expect(result1).toBe(42);
+            expect(result2).toBe(42);
+            expect(map.get("test")).toBe(42);
+        });
+    });
+});
+//# sourceMappingURL=__map_take.spec.js.map

package/lib/utils/__map_take.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"__map_take.spec.js","sourceRoot":"","sources":["../../src/utils/__map_take.spec.ts"],"names":[],"mappings":";;AAAA,6CAA0C;AAE1C,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YACtC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC;YAE3B,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YACtC,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAErB,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,GAAG,GAAG,IAAI,GAAG,EAA6B,CAAC;YACjD,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAExC,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACtC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAoB,CAAC;YACxC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAElC,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;;YACrC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAwB,CAAC;YAC5C,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAEjC,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,MAAA,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,2CAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAgB,CAAC;YACpC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;YAE/B,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAqB,CAAC;YACzC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;YAE/B,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YACtC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC;YAE/B,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;YAE9C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YACtC,IAAI,SAAS,GAAG,CAAC,CAAC;YAElB,MAAM,SAAS,GAAG,GAAG,EAAE;gBACrB,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC;YAEF,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YACnC,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAEnC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YACtC,MAAM,SAAS,GAAG,GAAG,EAAE;gBACrB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACrC,CAAC,CAAC;YAEF,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAqB,CAAC;YACzC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC;YAElC,MAAM,MAAM,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YAC/B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;YACtC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC;YAE3B,MAAM,OAAO,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YACnD,MAAM,OAAO,GAAG,IAAA,uBAAU,EAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzB,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentica/core",
-  "version": "0.23.0",
+  "version": "0.24.0",
   "description": "Agentic AI Library specialized in LLM Function Calling",
   "author": "Wrtn Technologies",
   "license": "MIT",

package/src/Agentica.ts

@@ -124,9 +124,17 @@ export class Agentica<Model extends ILlmSchema.Model> {
    * function calling information like {@link AgenticaExecuteHistory}.
    *
    * @param content The content to talk
+   * @param options Options
+   * @param options.abortSignal Abort signal
+   * @throws AbortError
    * @returns List of newly created chat prompts
    */
-  public async conversate(content: string | AgenticaUserInputHistory.Contents | Array<AgenticaUserInputHistory.Contents>): Promise<AgenticaHistory<Model>[]> {
+  public async conversate(
+    content: string | AgenticaUserInputHistory.Contents | Array<AgenticaUserInputHistory.Contents>,
+    options: {
+      abortSignal?: AbortSignal;
+    } = {},
+  ): Promise<AgenticaHistory<Model>[]> {
     const prompt: AgenticaUserInputHistory = createUserInputHistory({
       contents: Array.isArray(content)
         ? content
@@ -147,6 +155,7 @@ export class Agentica<Model extends ILlmSchema.Model> {
     const newbie: AgenticaHistory<Model>[] = await this.executor_(
       this.getContext({
         prompt,
+        abortSignal: options.abortSignal,
         usage: this.token_usage_,
       }),
     );
@@ -219,6 +228,7 @@ export class Agentica<Model extends ILlmSchema.Model> {
   public getContext(props: {
     prompt: AgenticaUserInputHistory;
     usage: AgenticaTokenUsage;
+    abortSignal?: AbortSignal;
   }): AgenticaContext<Model> {
     const dispatch = async (event: AgenticaEvent<Model>) => this.dispatch(event);
     return {
@@ -231,6 +241,7 @@ export class Agentica<Model extends ILlmSchema.Model> {
       stack: this.stack_,
       ready: () => this.ready_,
       prompt: props.prompt,
+      abortSignal: props.abortSignal,
 
       // HANDLERS
       dispatch: async event => this.dispatch(event),
@@ -246,7 +257,10 @@ export class Agentica<Model extends ILlmSchema.Model> {
               include_usage: true,
             },
           },
-          options: this.props.vendor.options,
+          options: {
+            ...this.props.vendor.options,
+            signal: props.abortSignal,
+          },
         });
         await dispatch(event);
 

package/src/context/AgenticaContext.ts

@@ -87,6 +87,11 @@ export interface AgenticaContext<Model extends ILlmSchema.Model> {
    */
   prompt: AgenticaUserInputHistory;
 
+  /**
+   * Abort signal.
+   */
+  abortSignal?: AbortSignal;
+
   /**
    * Whether the agent is ready.
    *

package/src/histories/AgenticaUserInputHistory.ts

@@ -1,3 +1,5 @@
+import type typia from "typia";
+
 import type { IAgenticaHistoryJson } from "../json/IAgenticaHistoryJson";
 
 import type { AgenticaHistoryBase } from "./AgenticaHistoryBase";
@@ -35,7 +37,7 @@ export namespace AgenticaUserInputHistory {
         /**
          * Either a URL of the image or the base64 encoded image data.
          */
-        url: string;
+        url: string & typia.tags.Format<"url">;
         /**
          * Specifies the detail level of the image. Learn more in the
          * [Vision guide](https://platform.openai.com/docs/guides/vision#low-or-high-fidelity-image-understanding).
@@ -46,6 +48,8 @@ export namespace AgenticaUserInputHistory {
 
     /**
      * Learn about [audio inputs](https://platform.openai.com/docs/guides/audio).
+     *
+     * Note: we not recommend it because audio input data only support base64 so it's too big data.
      */
     export interface InputAudio extends ContentsBase<"input_audio"> {
       input_audio: {
@@ -64,24 +68,26 @@ export namespace AgenticaUserInputHistory {
     /**
      * Learn about [file inputs](https://platform.openai.com/docs/guides/text) for text
      * generation.
+     *
+     * Note: we recommend use `file_id` instead of `file_data` because it's too big data.
      */
     export interface File extends ContentsBase<"file"> {
       file: {
         /**
-         * The base64 encoded file data, used when passing the file to the model as a
-         * string.
+         * The ID of an uploaded file to use as input.
          */
-        file_data?: string;
-
+        file_id: string;
+      } | {
         /**
-         * The ID of an uploaded file to use as input.
+         * The base64 encoded file data, used when passing the file to the model as a
+         * string.
          */
-        file_id?: string;
+        file_data: string;
 
         /**
          * The name of the file, used when passing the file to the model as a string.
          */
-        filename?: string;
+        filename: string;
       };
     }
   }

package/src/utils/ChatGptCompletionMessageUtil.spec.ts

@@ -0,0 +1,320 @@
+import type {
+  ChatCompletion,
+  ChatCompletionChunk,
+  ChatCompletionMessageToolCall,
+} from "openai/resources";
+
+import { ChatGptCompletionMessageUtil } from "./ChatGptCompletionMessageUtil";
+
+describe("chatGptCompletionMessageUtil", () => {
+  describe("transformCompletionChunk", () => {
+    it("should transform string chunk to ChatCompletionChunk", () => {
+      const chunk = {
+        id: "test-id",
+        choices: [{
+          index: 0,
+          delta: { content: "Hello" },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion.chunk",
+      };
+
+      const result = ChatGptCompletionMessageUtil.transformCompletionChunk(JSON.stringify(chunk));
+      expect(result).toEqual(chunk);
+    });
+
+    it("should transform Uint8Array chunk to ChatCompletionChunk", () => {
+      const chunk = {
+        id: "test-id",
+        choices: [{
+          index: 0,
+          delta: { content: "Hello" },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion.chunk",
+      };
+
+      const uint8Array = new TextEncoder().encode(JSON.stringify(chunk));
+      const result = ChatGptCompletionMessageUtil.transformCompletionChunk(uint8Array);
+      expect(result).toEqual(chunk);
+    });
+
+    it("should handle invalid JSON", () => {
+      expect(() => {
+        ChatGptCompletionMessageUtil.transformCompletionChunk("invalid json");
+      }).toThrow();
+    });
+  });
+
+  describe("accumulate", () => {
+    it("should accumulate content from chunks", () => {
+      const origin: ChatCompletion = {
+        id: "test-id",
+        choices: [{
+          index: 0,
+          // @ts-expect-error - refusal is not required
+          message: { role: "assistant", content: "Hello" },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion",
+      };
+
+      const chunk: ChatCompletionChunk = {
+        id: "test-id",
+        // @ts-expect-error - finish_reason is not required
+        choices: [{
+          index: 0,
+          delta: { content: " World" },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion.chunk",
+      };
+
+      const result = ChatGptCompletionMessageUtil.accumulate(origin, chunk);
+      expect(result.choices[0]?.message.content).toBe("Hello World");
+    });
+
+    it("should accumulate tool calls", () => {
+      const origin: ChatCompletion = {
+        id: "test-id",
+        choices: [{
+          index: 0,
+          // @ts-expect-error - finish_reason is not required
+          message: {
+            role: "assistant",
+            content: null,
+            tool_calls: [{
+              id: "call_1",
+              type: "function",
+              function: {
+                name: "test",
+                arguments: "{\"arg\": \"value\"}",
+              },
+            }],
+          },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion",
+      };
+
+      const chunk: ChatCompletionChunk = {
+        id: "test-id",
+        // @ts-expect-error - finish_reason is not required
+        choices: [{
+          index: 0,
+          delta: {
+            tool_calls: [{
+              index: 0,
+              id: "call_1",
+              function: {
+                name: "_function",
+                arguments: "{\"arg2\": \"value2\"}",
+              },
+            }],
+          },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion.chunk",
+      };
+
+      const result = ChatGptCompletionMessageUtil.accumulate(origin, chunk);
+      expect(result.choices[0]?.message.tool_calls?.[0]?.function.name).toBe("test_function");
+      expect(result.choices[0]?.message.tool_calls?.[0]?.function.arguments).toBe("{\"arg\": \"value\"}{\"arg2\": \"value2\"}");
+    });
+
+    it("should handle usage aggregation", () => {
+      const origin: ChatCompletion = {
+        id: "test-id",
+        choices: [{
+          index: 0,
+          // @ts-expect-error - finish_reason is not required
+          message: { role: "assistant", content: "Hello" },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion",
+        usage: {
+          prompt_tokens: 10,
+          completion_tokens: 5,
+          total_tokens: 15,
+        },
+      };
+
+      const chunk: ChatCompletionChunk = {
+        id: "test-id",
+        // @ts-expect-error - finish_reason is not required
+        choices: [{
+          index: 0,
+          delta: { content: " World" },
+        }],
+        created: 1234567890,
+        model: "gpt-4",
+        object: "chat.completion.chunk",
+        usage: {
+          prompt_tokens: 0,
+          completion_tokens: 6,
+          total_tokens: 6,
+        },
+      };
+
+      const result = ChatGptCompletionMessageUtil.accumulate(origin, chunk);
+      expect(result.usage).toEqual({
+        prompt_tokens: 10,
+        completion_tokens: 11,
+        total_tokens: 21,
+        completion_tokens_details: {
+          accepted_prediction_tokens: 0,
+          reasoning_tokens: 0,
+          rejected_prediction_tokens: 0,
+        },
+        prompt_tokens_details: {
+          audio_tokens: 0,
+          cached_tokens: 0,
+        },
+      });
+    });
+  });
+
+  describe("merge", () => {
+    it("should merge multiple chunks into completion", () => {
+      const chunks: ChatCompletionChunk[] = [
+        {
+          id: "test-id",
+          // @ts-expect-error - finish_reason is not required
+          choices: [{
+            index: 0,
+            delta: { content: "Hello" },
+          }],
+          created: 1234567890,
+          model: "gpt-4",
+          object: "chat.completion.chunk",
+        },
+        {
+          id: "test-id",
+          // @ts-expect-error - finish_reason is not required
+          choices: [{
+            index: 0,
+            delta: { content: " World" },
+          }],
+          created: 1234567890,
+          model: "gpt-4",
+          object: "chat.completion.chunk",
+        },
+      ];
+
+      const result = ChatGptCompletionMessageUtil.merge(chunks);
+      expect(result.choices[0]?.message.content).toBe("Hello World");
+    });
+
+    it("should throw error for empty chunks array", () => {
+      expect(() => {
+        ChatGptCompletionMessageUtil.merge([]);
+      }).toThrow("No chunks received");
+    });
+  });
+
+  describe("mergeChoice", () => {
+    it("should merge finish reason", () => {
+      const acc: ChatCompletion.Choice = {
+        index: 0,
+        // @ts-expect-error - finish_reason is not required
+        message: { role: "assistant", content: "Hello" },
+      };
+
+      const cur: ChatCompletionChunk.Choice = {
+        index: 0,
+        delta: {},
+        finish_reason: "stop",
+      };
+
+      const result = ChatGptCompletionMessageUtil.mergeChoice(acc, cur);
+      expect(result.finish_reason).toBe("stop");
+    });
+
+    it("should merge content", () => {
+      const acc: ChatCompletion.Choice = {
+        index: 0,
+        // @ts-expect-error - refusal is not required
+        message: { role: "assistant", content: "Hello" },
+      };
+
+      // @ts-expect-error - finish_reason is not required
+      const cur: ChatCompletionChunk.Choice = {
+        index: 0,
+        delta: { content: " World" },
+      };
+
+      const result = ChatGptCompletionMessageUtil.mergeChoice(acc, cur);
+      expect(result.message.content).toBe("Hello World");
+    });
+
+    it("should merge refusal", () => {
+      // @ts-expect-error - finish_reason is not required
+      const acc: ChatCompletion.Choice = {
+        index: 0,
+        message: { role: "assistant", content: null, refusal: "I cannot" },
+      };
+
+      // @ts-expect-error - finish_reason is not required
+      const cur: ChatCompletionChunk.Choice = {
+        index: 0,
+        delta: { refusal: " do that" },
+      };
+
+      const result = ChatGptCompletionMessageUtil.mergeChoice(acc, cur);
+      expect(result.message.refusal).toBe("I cannot do that");
+    });
+  });
+
+  describe("mergeToolCalls", () => {
+    it("should merge tool call function arguments", () => {
+      const acc: ChatCompletionMessageToolCall = {
+        id: "call_1",
+        type: "function",
+        function: {
+          name: "test",
+          arguments: "{\"arg\": \"value\"}",
+        },
+      };
+
+      const cur: ChatCompletionChunk.Choice.Delta.ToolCall = {
+        index: 0,
+        id: "call_1",
+        function: {
+          arguments: "{\"arg2\": \"value2\"}",
+        },
+      };
+
+      const result = ChatGptCompletionMessageUtil.mergeToolCalls(acc, cur);
+      expect(result.function.arguments).toBe("{\"arg\": \"value\"}{\"arg2\": \"value2\"}");
+    });
+
+    it("should merge tool call function name", () => {
+      const acc: ChatCompletionMessageToolCall = {
+        id: "call_1",
+        type: "function",
+        function: {
+          name: "test",
+          arguments: "",
+        },
+      };
+
+      const cur: ChatCompletionChunk.Choice.Delta.ToolCall = {
+        index: 0,
+        id: "call_1",
+        function: {
+          name: "_function",
+        },
+      };
+
+      const result = ChatGptCompletionMessageUtil.mergeToolCalls(acc, cur);
+      expect(result.function.name).toBe("test_function");
+    });
+  });
+});