@aigne/test-utils 0.5.55 → 0.5.56-beta

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## [0.5.56-beta](https://github.com/AIGNE-io/aigne-framework/compare/test-utils-v0.5.55...test-utils-v0.5.56-beta) (2025-10-21)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/core bumped to 1.64.0-beta
+
 ## [0.5.55](https://github.com/AIGNE-io/aigne-framework/compare/test-utils-v0.5.55-beta.12...test-utils-v0.5.55) (2025-10-19)
 
 

package/lib/run-example-test.d.ts

@@ -0,0 +1,7 @@
+export interface TestConfig {
+    initialCall?: string;
+    scriptPath?: string;
+}
+export declare function runExampleTest(config?: TestConfig): Promise<{
+    status: number | null;
+}>;

package/lib/run-example-test.js

@@ -0,0 +1,12 @@
+import { spawnSync } from "node:child_process";
+import { join } from "node:path";
+export async function runExampleTest(config) {
+    const scriptPath = config?.scriptPath ?? join(process.cwd(), "index.ts");
+    return spawnSync("bun", [scriptPath], {
+        stdio: ["inherit", "inherit", "inherit"],
+        env: {
+            ...process.env,
+            INITIAL_CALL: config?.initialCall ?? process.env.INITIAL_CALL,
+        },
+    });
+}

package/lib/utils/agent-response.d.ts

@@ -0,0 +1,7 @@
+import { type AgentResponseChunk, type AgentResponseStream, type Message } from "@aigne/core";
+export declare function agentResponseStreamToArraySnapshot<T extends Message>(stream: AgentResponseStream<T>, options?: {
+    catchError?: false;
+}): Promise<AgentResponseChunk<T>[]>;
+export declare function agentResponseStreamToArraySnapshot<T extends Message>(stream: AgentResponseStream<T>, options?: {
+    catchError?: boolean;
+}): Promise<(AgentResponseChunk<T> | Error)[]>;

package/lib/utils/agent-response.js

@@ -0,0 +1,14 @@
+import { isAgentResponseProgress, } from "@aigne/core";
+import { readableStreamToArray } from "@aigne/core/utils/stream-utils.js";
+export async function agentResponseStreamToArraySnapshot(stream, options) {
+    return (await readableStreamToArray(stream, options)).map((i) => !(i instanceof Error) && isAgentResponseProgress(i)
+        ? {
+            progress: {
+                ...i.progress,
+                contextId: "TEST_CONTEXT_ID",
+                parentContextId: "TEST_PARENT_CONTEXT_ID",
+                timestamp: 123456,
+            },
+        }
+        : i);
+}

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: () => Promise<import("@aigne/core").ChatModelInputMessage[]>;
+export declare const createWeatherToolExpected: () => any;
+export declare const createWeatherToolCallMessages: () => Promise<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/lib/utils/with-env.d.ts

@@ -0,0 +1,3 @@
+export declare function withEnv(env: Record<string, string>): {
+    [Symbol.dispose]: () => void;
+};

package/lib/utils/with-env.js

@@ -0,0 +1,9 @@
+export function withEnv(env) {
+    const originalEnv = { ...process.env };
+    process.env = { ...process.env, ...env };
+    return {
+        [Symbol.dispose]: () => {
+            process.env = originalEnv;
+        },
+    };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/test-utils",
-  "version": "0.5.55",
+  "version": "0.5.56-beta",
   "description": "Test utils for AIGNE framework",
   "publishConfig": {
     "access": "public"
@@ -27,7 +27,7 @@
     }
   },
   "dependencies": {
-    "@aigne/core": "^1.63.0"
+    "@aigne/core": "^1.64.0-beta"
   },
   "devDependencies": {
     "@types/bun": "^1.2.22",