@aigne/test-utils 0.2.0 → 0.3.1

package/CHANGELOG.md

@@ -1,5 +1,28 @@
 # Changelog
 
+## [0.3.1](https://github.com/AIGNE-io/aigne-framework/compare/test-utils-v0.3.0...test-utils-v0.3.1) (2025-05-25)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/core bumped to 1.17.0
+
+## [0.3.0](https://github.com/AIGNE-io/aigne-framework/compare/test-utils-v0.2.0...test-utils-v0.3.0) (2025-05-23)
+
+
+### Features
+
+* **models:** publish model adapters as standalone packages ([#126](https://github.com/AIGNE-io/aigne-framework/issues/126)) ([588b8ae](https://github.com/AIGNE-io/aigne-framework/commit/588b8aea6abcee5fa87def1358bf51f84021c6ef))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/core bumped to 1.16.0
+
 ## [0.2.0](https://github.com/AIGNE-io/aigne-framework/compare/test-utils-v0.1.0...test-utils-v0.2.0) (2025-05-12)
 
 

package/lib/utils/event-stream.d.ts

@@ -0,0 +1,5 @@
+export declare function createMockEventStream<T>(data: {
+    path: string;
+} | {
+    raw: string;
+}): T;

package/lib/utils/event-stream.js

@@ -0,0 +1,13 @@
+import { readFile } from "node:fs/promises";
+export function createMockEventStream(data) {
+    return new ReadableStream({
+        async start(controller) {
+            const file = "path" in data ? await readFile(data.path) : data.raw;
+            for (const line of file.toString().split("\n")) {
+                if (line)
+                    controller.enqueue(JSON.parse(line.replace("data:", "")));
+            }
+            controller.close();
+        },
+    });
+}

package/lib/utils/openai-like-utils.d.ts

@@ -0,0 +1,7 @@
+import { type ChatModelInputResponseFormat, type ChatModelInputTool, type ChatModelOutputToolCall, type Message } from "@aigne/core";
+export declare const COMMON_TOOLS: ChatModelInputTool[];
+export declare const COMMON_RESPONSE_FORMAT: ChatModelInputResponseFormat;
+export declare const createWeatherToolMessages: () => import("@aigne/core").ChatModelInputMessage[];
+export declare const createWeatherToolExpected: () => any;
+export declare const createWeatherToolCallMessages: () => import("@aigne/core").ChatModelInputMessage[];
+export declare function createToolCallResponse(functionName: string, args: Message): ChatModelOutputToolCall;

package/lib/utils/openai-like-utils.js

@@ -0,0 +1,76 @@
+import { expect } from "bun:test";
+import { AgentMessageTemplate, ChatMessagesTemplate, SystemMessageTemplate, ToolMessageTemplate, UserMessageTemplate, } from "@aigne/core";
+export const COMMON_TOOLS = [
+    {
+        type: "function",
+        function: {
+            name: "get_weather",
+            parameters: {
+                type: "object",
+                properties: {
+                    city: {
+                        type: "string",
+                    },
+                },
+                required: ["city"],
+            },
+        },
+    },
+];
+export const COMMON_RESPONSE_FORMAT = {
+    type: "json_schema",
+    jsonSchema: {
+        name: "output",
+        schema: {
+            type: "object",
+            properties: {
+                text: {
+                    type: "string",
+                },
+            },
+            required: ["text"],
+            additionalProperties: false,
+        },
+        strict: true,
+    },
+};
+const createBaseMessages = () => [
+    SystemMessageTemplate.from("You are a chatbot"),
+    UserMessageTemplate.from([{ type: "text", text: "What is the weather in New York?" }]),
+];
+export const createWeatherToolMessages = () => ChatMessagesTemplate.from(createBaseMessages()).format();
+export const createWeatherToolExpected = () => expect.objectContaining({
+    toolCalls: [
+        {
+            function: {
+                arguments: {
+                    city: "New York",
+                },
+                name: "get_weather",
+            },
+            id: expect.any(String),
+            type: "function",
+        },
+    ],
+});
+export const createWeatherToolCallMessages = () => ChatMessagesTemplate.from([
+    ...createBaseMessages(),
+    AgentMessageTemplate.from(undefined, [
+        {
+            id: "get_weather",
+            type: "function",
+            function: { name: "get_weather", arguments: { city: "New York" } },
+        },
+    ]),
+    ToolMessageTemplate.from({ temperature: 20 }, "get_weather"),
+]).format();
+export function createToolCallResponse(functionName, args) {
+    return {
+        id: functionName,
+        type: "function",
+        function: {
+            name: functionName,
+            arguments: args,
+        },
+    };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/test-utils",
-  "version": "0.2.0",
+  "version": "0.3.1",
   "description": "Test utils for AIGNE framework",
   "publishConfig": {
     "access": "public"
@@ -23,7 +23,9 @@
       ]
     }
   },
-  "dependencies": {},
+  "dependencies": {
+    "@aigne/core": "^1.17.0"
+  },
   "devDependencies": {
     "@types/bun": "^1.2.12",
     "@types/node": "^22.15.15",

package/src/utils/event-stream.ts

@@ -0,0 +1,13 @@
+import { readFile } from "node:fs/promises";
+
+export function createMockEventStream<T>(data: { path: string } | { raw: string }): T {
+  return new ReadableStream({
+    async start(controller) {
+      const file = "path" in data ? await readFile(data.path) : data.raw;
+      for (const line of file.toString().split("\n")) {
+        if (line) controller.enqueue(JSON.parse(line.replace("data:", "")));
+      }
+      controller.close();
+    },
+  }) as unknown as T;
+}

package/src/utils/openai-like-utils.ts

@@ -0,0 +1,99 @@
+import { expect } from "bun:test";
+import {
+  AgentMessageTemplate,
+  ChatMessagesTemplate,
+  type ChatModelInputResponseFormat,
+  type ChatModelInputTool,
+  type ChatModelOutputToolCall,
+  type Message,
+  SystemMessageTemplate,
+  ToolMessageTemplate,
+  UserMessageTemplate,
+} from "@aigne/core";
+
+export const COMMON_TOOLS: ChatModelInputTool[] = [
+  {
+    type: "function",
+    function: {
+      name: "get_weather",
+      parameters: {
+        type: "object",
+        properties: {
+          city: {
+            type: "string",
+          },
+        },
+        required: ["city"],
+      },
+    },
+  },
+];
+
+export const COMMON_RESPONSE_FORMAT: ChatModelInputResponseFormat = {
+  type: "json_schema",
+  jsonSchema: {
+    name: "output",
+    schema: {
+      type: "object",
+      properties: {
+        text: {
+          type: "string",
+        },
+      },
+      required: ["text"],
+      additionalProperties: false,
+    },
+    strict: true,
+  },
+};
+
+const createBaseMessages = () => [
+  SystemMessageTemplate.from("You are a chatbot"),
+  UserMessageTemplate.from([{ type: "text", text: "What is the weather in New York?" }]),
+];
+
+export const createWeatherToolMessages = () =>
+  ChatMessagesTemplate.from(createBaseMessages()).format();
+
+export const createWeatherToolExpected = () =>
+  expect.objectContaining({
+    toolCalls: [
+      {
+        function: {
+          arguments: {
+            city: "New York",
+          },
+          name: "get_weather",
+        },
+        id: expect.any(String),
+        type: "function",
+      },
+    ],
+  });
+
+export const createWeatherToolCallMessages = () =>
+  ChatMessagesTemplate.from([
+    ...createBaseMessages(),
+    AgentMessageTemplate.from(undefined, [
+      {
+        id: "get_weather",
+        type: "function",
+        function: { name: "get_weather", arguments: { city: "New York" } },
+      },
+    ]),
+    ToolMessageTemplate.from({ temperature: 20 }, "get_weather"),
+  ]).format();
+
+export function createToolCallResponse(
+  functionName: string,
+  args: Message,
+): ChatModelOutputToolCall {
+  return {
+    id: functionName,
+    type: "function",
+    function: {
+      name: functionName,
+      arguments: args,
+    },
+  };
+}

package/tsconfig.build.tsbuildinfo

@@ -1 +1 @@
-{"root":["./src/run-example-test.ts"],"version":"5.8.3"}
+{"root":["./src/run-example-test.ts","./src/utils/event-stream.ts","./src/utils/openai-like-utils.ts"],"version":"5.8.3"}