@aigne/core 1.72.0-beta.2 → 1.72.0-beta.23

package/lib/cjs/agents/chat-model.js

@@ -33,13 +33,14 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.chatModelOutputUsageSchema = exports.unionContentSchema = exports.textContentSchema = exports.roleSchema = exports.ChatModel = exports.StructuredOutputError = void 0;
+exports.chatModelOutputUsageSchema = exports.DEFAULT_CACHE_CONFIG = exports.unionContentSchema = exports.textContentSchema = exports.roleSchema = exports.ChatModel = exports.StructuredOutputError = void 0;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
 const zod_1 = require("zod");
 const zod_from_json_schema_1 = require("zod-from-json-schema");
 const schema_js_1 = require("../loader/schema.js");
 const json_schema_js_1 = require("../utils/json-schema.js");
 const logger_js_1 = require("../utils/logger.js");
+const token_estimator_js_1 = require("../utils/token-estimator.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
 const model_js_1 = require("./model.js");
@@ -117,6 +118,19 @@ class ChatModel extends model_js_1.Model {
             supportsParallelToolCalls: this.supportsParallelToolCalls,
         };
     }
+    async getModelOptions(input, options) {
+        const modelOptions = (await super.getModelOptions(input, options));
+        return {
+            ...modelOptions,
+            cacheConfig: {
+                ...modelOptions.cacheConfig,
+                autoBreakpoints: {
+                    ...modelOptions.cacheConfig?.autoBreakpoints,
+                    lastMessage: modelOptions.cacheConfig?.autoBreakpoints?.lastMessage ?? true,
+                },
+            },
+        };
+    }
     validateToolNames(tools) {
         for (const tool of tools ?? []) {
             if (!/^[a-zA-Z0-9_]+$/.test(tool.function.name)) {
@@ -124,6 +138,9 @@ class ChatModel extends model_js_1.Model {
             }
         }
     }
+    async countTokens(input) {
+        return (0, token_estimator_js_1.estimateTokens)(JSON.stringify(input));
+    }
     /**
      * Normalizes tool names to ensure compatibility with language models
      *
@@ -225,6 +242,10 @@ class ChatModel extends model_js_1.Model {
             options.context.usage.inputTokens += usage.inputTokens;
             if (usage.aigneHubCredits)
                 options.context.usage.aigneHubCredits += usage.aigneHubCredits;
+            if (usage.cacheCreationInputTokens)
+                options.context.usage.cacheCreationInputTokens += usage.cacheCreationInputTokens;
+            if (usage.cacheReadInputTokens)
+                options.context.usage.cacheReadInputTokens += usage.cacheReadInputTokens;
             if (usage.creditPrefix)
                 options.context.usage.creditPrefix = usage.creditPrefix;
         }
@@ -234,7 +255,7 @@ class ChatModel extends model_js_1.Model {
             const files = zod_1.z.array(model_js_1.fileUnionContentSchema).parse(output.files);
             output = {
                 ...output,
-                files: await Promise.all(files.map((file) => this.transformFileType(input.outputFileType, file, options))),
+                files: await Promise.all(files.map((file) => this.transformFileType(input.outputFileType, file))),
             };
         }
         // Remove fields with `null` value for validation
@@ -282,6 +303,10 @@ exports.roleSchema = zod_1.z.union([
 exports.textContentSchema = zod_1.z.object({
     type: zod_1.z.literal("text"),
     text: zod_1.z.string(),
+    cacheControl: (0, schema_js_1.optionalize)(zod_1.z.object({
+        type: zod_1.z.literal("ephemeral"),
+        ttl: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("5m"), zod_1.z.literal("1h")])),
+    })),
 });
 exports.unionContentSchema = zod_1.z.discriminatedUnion("type", [
     exports.textContentSchema,
@@ -303,6 +328,10 @@ const chatModelInputMessageSchema = zod_1.z.object({
     }))),
     toolCallId: (0, schema_js_1.optionalize)(zod_1.z.string()),
     name: (0, schema_js_1.optionalize)(zod_1.z.string()),
+    cacheControl: (0, schema_js_1.optionalize)(zod_1.z.object({
+        type: zod_1.z.literal("ephemeral"),
+        ttl: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("5m"), zod_1.z.literal("1h")])),
+    })),
 });
 const chatModelInputResponseFormatSchema = zod_1.z.discriminatedUnion("type", [
     zod_1.z.object({ type: zod_1.z.literal("text") }),
@@ -324,6 +353,10 @@ const chatModelInputToolSchema = zod_1.z.object({
         parameters: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()),
     }),
     metadata: (0, schema_js_1.optionalize)(zod_1.z.record(zod_1.z.string(), zod_1.z.unknown())),
+    cacheControl: (0, schema_js_1.optionalize)(zod_1.z.object({
+        type: zod_1.z.literal("ephemeral"),
+        ttl: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("5m"), zod_1.z.literal("1h")])),
+    })),
 });
 const chatModelInputToolChoiceSchema = zod_1.z.union([
     zod_1.z.literal("auto"),
@@ -331,6 +364,22 @@ const chatModelInputToolChoiceSchema = zod_1.z.union([
     zod_1.z.literal("required"),
     chatModelInputToolSchema,
 ]);
+/**
+ * Default cache configuration
+ *
+ * Enables automatic caching for system messages and tool definitions,
+ * which typically provides the best cost/performance tradeoff.
+ */
+exports.DEFAULT_CACHE_CONFIG = {
+    enabled: true,
+    ttl: "5m",
+    strategy: "auto",
+    autoBreakpoints: {
+        tools: true,
+        system: true,
+        lastMessage: false,
+    },
+};
 const modelOptionsSchemaProperties = {
     model: zod_1.z.string(),
     temperature: zod_1.z.number(),
@@ -346,15 +395,29 @@ const modelOptionsSchemaProperties = {
         zod_1.z.literal("medium"),
         zod_1.z.literal("high"),
     ]),
+    cacheConfig: zod_1.z.object({
+        enabled: (0, schema_js_1.optionalize)(zod_1.z.boolean().default(true)),
+        ttl: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("5m"), zod_1.z.literal("1h"), zod_1.z.number()]).default("5m")),
+        strategy: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("auto"), zod_1.z.literal("manual")]).default("auto")),
+        autoBreakpoints: (0, schema_js_1.optionalize)(zod_1.z.object({
+            tools: (0, schema_js_1.optionalize)(zod_1.z.boolean().default(true)),
+            system: (0, schema_js_1.optionalize)(zod_1.z.boolean().default(true)),
+            lastMessage: (0, schema_js_1.optionalize)(zod_1.z.boolean().default(false)),
+        })),
+    }),
 };
-const modelOptionsSchema = zod_1.z.object(Object.fromEntries(Object.entries(modelOptionsSchemaProperties).map(([key, schema]) => [
+const modelOptionsSchema = zod_1.z
+    .object(Object.fromEntries(Object.entries(modelOptionsSchemaProperties).map(([key, schema]) => [
     key,
     (0, schema_js_1.optionalize)(schema),
-])));
-const modelOptionsWithGetterSchema = zod_1.z.object(Object.fromEntries(Object.entries(modelOptionsSchemaProperties).map(([key, schema]) => [
+])))
+    .passthrough();
+const modelOptionsWithGetterSchema = zod_1.z
+    .object(Object.fromEntries(Object.entries(modelOptionsSchemaProperties).map(([key, schema]) => [
     key,
     (0, schema_js_1.optionalize)((0, agent_js_1.getterSchema)(schema)),
-])));
+])))
+    .passthrough();
 const chatModelOptionsSchema = agent_js_1.agentOptionsSchema.extend({
     model: (0, schema_js_1.optionalize)(zod_1.z.string()),
     modelOptions: (0, schema_js_1.optionalize)(modelOptionsWithGetterSchema),
@@ -379,6 +442,8 @@ exports.chatModelOutputUsageSchema = zod_1.z.object({
     inputTokens: zod_1.z.number(),
     outputTokens: zod_1.z.number(),
     aigneHubCredits: (0, schema_js_1.optionalize)(zod_1.z.number()),
+    cacheCreationInputTokens: (0, schema_js_1.optionalize)(zod_1.z.number()),
+    cacheReadInputTokens: (0, schema_js_1.optionalize)(zod_1.z.number()),
     creditPrefix: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("$"), zod_1.z.literal("€"), zod_1.z.literal("¥")])),
 });
 const chatModelOutputSchema = zod_1.z.object({

package/lib/cjs/agents/image-agent.d.ts

@@ -1,4 +1,4 @@
-import { type ZodObject, type ZodType } from "zod";
+import z, { type ZodObject, type ZodType } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type Message } from "./agent.js";
 import { type ImageModelOutput } from "./image-model.js";
@@ -13,6 +13,22 @@ export declare const imageAgentOptionsSchema: ZodObject<{
 }>;
 export declare class ImageAgent<I extends Message = any, O extends ImageModelOutput = any> extends Agent<I, O> {
     tag: string;
+    static schema({ filepath }: {
+        filepath: string;
+    }): z.ZodObject<{
+        instructions: z.ZodType<import("../loader/schema.js").Instructions, z.ZodTypeDef, import("../loader/schema.js").Instructions>;
+        inputFileKey: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        instructions: import("../loader/schema.js").Instructions;
+        inputFileKey?: string | undefined;
+    }, {
+        instructions: import("../loader/schema.js").Instructions;
+        inputFileKey?: string | undefined;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+    }): Promise<Agent<I, O>>;
     static from<I extends Message = any, O extends ImageModelOutput = any>(options: ImageAgentOptions<I, O>): ImageAgent<I, O>;
     constructor(options: ImageAgentOptions<I, O>);
     instructions: PromptBuilder;

package/lib/cjs/agents/image-agent.js

@@ -5,6 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ImageAgent = exports.imageAgentOptionsSchema = void 0;
 const zod_1 = __importDefault(require("zod"));
+const schema_js_1 = require("../loader/schema.js");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
@@ -16,6 +17,21 @@ exports.imageAgentOptionsSchema = agent_js_1.agentOptionsSchema.extend({
 });
 class ImageAgent extends agent_js_1.Agent {
     tag = "ImageAgent";
+    static schema({ filepath }) {
+        const instructionsSchema = (0, schema_js_1.getInstructionsSchema)({ filepath });
+        return (0, schema_js_1.camelizeSchema)(zod_1.default.object({
+            instructions: instructionsSchema,
+            inputFileKey: (0, schema_js_1.optionalize)(zod_1.default.string()),
+        }));
+    }
+    static async load(options) {
+        const valid = await ImageAgent.schema({ filepath: options.filepath }).parseAsync(options.parsed);
+        return new ImageAgent({
+            ...options.parsed,
+            ...valid,
+            instructions: (0, schema_js_1.instructionsToPromptBuilder)(valid.instructions),
+        });
+    }
     static from(options) {
         return new ImageAgent(options);
     }

package/lib/cjs/agents/image-model.d.ts

@@ -107,8 +107,8 @@ export declare const imageModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[] | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     n?: number | undefined;
+    outputFileType?: "local" | "file" | "url" | undefined;
 }, {
     prompt: string;
     modelOptions?: Record<string, unknown> | undefined;
@@ -128,8 +128,8 @@ export declare const imageModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[] | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     n?: number | undefined;
+    outputFileType?: "local" | "file" | "url" | undefined;
 }>;
 export interface ImageModelOutput extends Message {
     images: FileUnionContent[];
@@ -196,16 +196,22 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         inputTokens: z.ZodNumber;
         outputTokens: z.ZodNumber;
         aigneHubCredits: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+        cacheCreationInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+        cacheReadInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
         creditPrefix: ZodType<"$" | "€" | "¥" | undefined, z.ZodTypeDef, "$" | "€" | "¥" | undefined>;
     }, "strip", z.ZodTypeAny, {
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     }, {
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     }>>;
     model: z.ZodOptional<z.ZodString>;
@@ -231,6 +237,8 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
 }, {
@@ -255,6 +263,8 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
 }>;

package/lib/cjs/agents/image-model.js

@@ -70,7 +70,7 @@ class ImageModel extends model_js_1.Model {
             const images = zod_1.z.array(model_js_1.fileUnionContentSchema).parse(output.images);
             output = {
                 ...output,
-                images: await Promise.all(images.map((image) => this.transformFileType(input.outputFileType, image, options))),
+                images: await Promise.all(images.map((image) => this.transformFileType(input.outputFileType, image))),
             };
         }
         return super.processAgentOutput(input, output, options);

package/lib/cjs/agents/mcp-agent.d.ts

@@ -58,6 +58,23 @@ export type SSEServerParameters = {
  */
 export declare class MCPAgent extends Agent {
     tag: string;
+    static schema(): z.ZodObject<{
+        url: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        command: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        args: ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        url?: string | undefined;
+        command?: string | undefined;
+        args?: string[] | undefined;
+    }, {
+        url?: string | undefined;
+        command?: string | undefined;
+        args?: string[] | undefined;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+    }): Promise<Agent<I, O>>;
     /**
      * Create an MCPAgent from a connection to an SSE server.
      *

package/lib/cjs/agents/mcp-agent.js

@@ -41,6 +41,7 @@ const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
 const uriTemplate_js_1 = require("@modelcontextprotocol/sdk/shared/uriTemplate.js");
 const zod_1 = require("zod");
+const schema_js_1 = require("../loader/schema.js");
 const logger_js_1 = require("../utils/logger.js");
 const mcp_utils_js_1 = require("../utils/mcp-utils.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
@@ -85,6 +86,23 @@ function getMCPServerString(options) {
  */
 class MCPAgent extends agent_js_1.Agent {
     tag = "MCPAgent";
+    static schema() {
+        return zod_1.z.object({
+            url: (0, schema_js_1.optionalize)(zod_1.z.string()),
+            command: (0, schema_js_1.optionalize)(zod_1.z.string()),
+            args: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
+        });
+    }
+    static async load(options) {
+        const valid = await MCPAgent.schema().parseAsync(options.parsed);
+        if (!valid.url && !valid.command) {
+            throw new Error(`Missing url or command in mcp agent: ${options.filepath}`);
+        }
+        return MCPAgent.from({
+            ...options.parsed,
+            ...valid,
+        });
+    }
     static from(options) {
         (0, type_utils_js_1.checkArguments)("MCPAgent.from", mcpAgentOptionsSchema, options);
         if (isSSEServerParameters(options)) {

package/lib/cjs/agents/model.d.ts

@@ -17,9 +17,9 @@ export declare abstract class Model<I extends Message = any, O extends Message =
      */
     getModelOptions(input: Message, options: AgentInvokeOptions): Promise<Record<string, unknown>>;
     protected preprocess(input: I, options: AgentInvokeOptions): Promise<void>;
-    transformFileType(fileType: "file", data: FileUnionContent, options: AgentInvokeOptions): Promise<FileContent>;
-    transformFileType(fileType: "local" | undefined, data: FileUnionContent, options: AgentInvokeOptions): Promise<LocalContent>;
-    transformFileType(fileType: FileType | undefined, data: FileUnionContent, options: AgentInvokeOptions): Promise<FileUnionContent>;
+    transformFileType(fileType: "file", data: FileUnionContent): Promise<FileContent>;
+    transformFileType(fileType: "local" | undefined, data: FileUnionContent): Promise<LocalContent>;
+    transformFileType(fileType: FileType | undefined, data: FileUnionContent): Promise<FileUnionContent>;
     static getFileExtension(type: string): Promise<string | undefined>;
     static getMimeType(filename: string): Promise<string | undefined>;
     downloadFile(url: string): Promise<Response>;

package/lib/cjs/agents/model.js

@@ -89,13 +89,13 @@ class Model extends agent_js_1.Agent {
         Object.assign(input, { modelOptions: await this.getModelOptions(input, options) });
         return super.preprocess(input, options);
     }
-    async transformFileType(fileType = "local", data, options) {
+    async transformFileType(fileType = "local", data) {
         if (fileType === data.type)
             return data;
         const common = (0, type_utils_js_1.pick)(data, "filename", "mimeType");
         switch (fileType) {
             case "local": {
-                const dir = index_js_1.nodejs.path.join(index_js_1.nodejs.os.tmpdir(), options.context.id);
+                const dir = index_js_1.nodejs.path.join(index_js_1.nodejs.os.tmpdir(), (0, uuid_1.v7)());
                 await index_js_1.nodejs.fs.mkdir(dir, { recursive: true });
                 const ext = await Model.getFileExtension(data.mimeType || data.filename || "");
                 const id = (0, uuid_1.v7)();

package/lib/cjs/agents/team-agent.d.ts

@@ -1,3 +1,5 @@
+import { z } from "zod";
+import type { AgentLoadOptions } from "../loader/index.js";
 import { type PromiseOrValue } from "../utils/type-utils.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type Message } from "./agent.js";
 /**
@@ -200,6 +202,59 @@ export interface TeamAgentOptions<I extends Message, O extends Message> extends
  */
 export declare class TeamAgent<I extends Message, O extends Message> extends Agent<I, O> {
     tag: string;
+    static schema({ filepath }: {
+        filepath: string;
+    }): z.ZodObject<{
+        mode: z.ZodType<ProcessMode | undefined, z.ZodTypeDef, ProcessMode | undefined>;
+        iterateOn: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        concurrency: z.ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+        iterateWithPreviousOutput: z.ZodType<boolean | undefined, z.ZodTypeDef, boolean | undefined>;
+        includeAllStepsOutput: z.ZodType<boolean | undefined, z.ZodTypeDef, boolean | undefined>;
+        reflection: z.ZodType<{
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined, z.ZodTypeDef, {
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        mode?: ProcessMode | undefined;
+        reflection?: {
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined;
+        iterateOn?: string | undefined;
+        concurrency?: number | undefined;
+        iterateWithPreviousOutput?: boolean | undefined;
+        includeAllStepsOutput?: boolean | undefined;
+    }, {
+        mode?: ProcessMode | undefined;
+        reflection?: {
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined;
+        iterateOn?: string | undefined;
+        concurrency?: number | undefined;
+        iterateWithPreviousOutput?: boolean | undefined;
+        includeAllStepsOutput?: boolean | undefined;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+        options: AgentLoadOptions;
+    }): Promise<Agent<I, O>>;
     /**
      * Create a TeamAgent from the provided options.
      *

package/lib/cjs/agents/team-agent.js

@@ -39,6 +39,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.TeamAgent = exports.DEFAULT_REFLECTION_MAX_ITERATIONS = exports.ProcessMode = void 0;
 const node_assert_1 = __importDefault(require("node:assert"));
 const immer_1 = require("immer");
+const zod_1 = require("zod");
+const agent_yaml_js_1 = require("../loader/agent-yaml.js");
+const schema_js_1 = require("../loader/schema.js");
 const fastq = __importStar(require("../utils/queue.js"));
 const stream_utils_js_1 = require("../utils/stream-utils.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
@@ -86,6 +89,34 @@ exports.DEFAULT_REFLECTION_MAX_ITERATIONS = 3;
  */
 class TeamAgent extends agent_js_1.Agent {
     tag = "TeamAgent";
+    static schema({ filepath }) {
+        const nestAgentSchema = (0, agent_yaml_js_1.getNestAgentSchema)({ filepath });
+        return zod_1.z.object({
+            mode: (0, schema_js_1.optionalize)(zod_1.z.nativeEnum(ProcessMode)),
+            iterateOn: (0, schema_js_1.optionalize)(zod_1.z.string()),
+            concurrency: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(1)),
+            iterateWithPreviousOutput: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
+            includeAllStepsOutput: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
+            reflection: (0, schema_js_1.camelizeSchema)((0, schema_js_1.optionalize)(zod_1.z.object({
+                reviewer: nestAgentSchema,
+                isApproved: zod_1.z.string(),
+                maxIterations: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(1)),
+                returnLastOnMaxIterations: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
+                customErrorMessage: (0, schema_js_1.optionalize)(zod_1.z.string()),
+            }))),
+        });
+    }
+    static async load(options) {
+        const valid = await TeamAgent.schema({ filepath: options.filepath }).parseAsync(options.parsed);
+        return TeamAgent.from({
+            ...options.parsed,
+            ...valid,
+            reflection: valid.reflection && {
+                ...valid.reflection,
+                reviewer: await options.options.loadNestAgent(options.filepath, valid.reflection.reviewer, options.options),
+            },
+        });
+    }
     /**
      * Create a TeamAgent from the provided options.
      *

package/lib/cjs/agents/transform-agent.d.ts

@@ -1,3 +1,4 @@
+import { z } from "zod";
 import { Agent, type AgentOptions, type Message } from "./agent.js";
 /**
  * Configuration options for TransformAgent
@@ -47,6 +48,17 @@ export interface TransformAgentOptions<I extends Message, O extends Message> ext
  */
 export declare class TransformAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
     static type: string;
+    static schema(): z.ZodObject<{
+        jsonata: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        jsonata: string;
+    }, {
+        jsonata: string;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+    }): Promise<Agent<I, O>>;
     /**
      * Factory method to create a new TransformAgent instance
      *

package/lib/cjs/agents/transform-agent.js

@@ -5,6 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TransformAgent = void 0;
 const jsonata_1 = __importDefault(require("jsonata"));
+const zod_1 = require("zod");
 const agent_js_1 = require("./agent.js");
 /**
  * TransformAgent - A specialized agent for data transformation using JSONata expressions
@@ -23,6 +24,18 @@ const agent_js_1 = require("./agent.js");
  */
 class TransformAgent extends agent_js_1.Agent {
     static type = "TransformAgent";
+    static schema() {
+        return zod_1.z.object({
+            jsonata: zod_1.z.string(),
+        });
+    }
+    static async load(options) {
+        const parsed = await TransformAgent.schema().parseAsync(options.parsed);
+        return TransformAgent.from({
+            ...options.parsed,
+            ...parsed,
+        });
+    }
     /**
      * Factory method to create a new TransformAgent instance
      *

package/lib/cjs/agents/video-model.d.ts

@@ -203,16 +203,22 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         inputTokens: z.ZodNumber;
         outputTokens: z.ZodNumber;
         aigneHubCredits: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+        cacheCreationInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+        cacheReadInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
         creditPrefix: ZodType<"$" | "€" | "¥" | undefined, z.ZodTypeDef, "$" | "€" | "¥" | undefined>;
     }, "strip", z.ZodTypeAny, {
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     }, {
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     }>>;
     model: z.ZodOptional<z.ZodString>;
@@ -239,6 +245,8 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
     seconds?: number | undefined;
@@ -264,6 +272,8 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         inputTokens: number;
         outputTokens: number;
         aigneHubCredits?: number | undefined;
+        cacheCreationInputTokens?: number | undefined;
+        cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
     seconds?: number | undefined;

package/lib/cjs/agents/video-model.js

@@ -43,7 +43,7 @@ class VideoModel extends model_js_1.Model {
             const videos = zod_1.z.array(model_js_1.fileUnionContentSchema).parse(output.videos);
             output = {
                 ...output,
-                videos: await Promise.all(videos.map((video) => this.transformFileType(input.outputFileType, video, options))),
+                videos: await Promise.all(videos.map((video) => this.transformFileType(input.outputFileType, video))),
             };
         }
         return super.processAgentOutput(input, output, options);

package/lib/cjs/aigne/context.js

@@ -347,11 +347,9 @@ class AIGNEContextShared {
                     }
                 }
                 const stream = await activeAgent.invoke(input, {
-                    hooks: options.hooks,
+                    ...(0, type_utils_js_1.pick)(options, "hooks", "model", "imageModel", "caller"),
                     context,
                     streaming: true,
-                    model: options.model,
-                    imageModel: options.imageModel,
                 });
                 for await (const value of stream) {
                     if ((0, agent_js_1.isAgentResponseDelta)(value)) {

package/lib/cjs/aigne/usage.d.ts

@@ -8,6 +8,10 @@ export interface ContextUsage {
     creditPrefix?: "$" | "€" | "¥";
     agentCalls: number;
     duration: number;
+    /** Number of tokens written to cache (first time caching) */
+    cacheCreationInputTokens: number;
+    /** Number of tokens read from cache (cache hit) */
+    cacheReadInputTokens: number;
 }
 /**
  * @hidden

package/lib/cjs/aigne/usage.js

@@ -12,6 +12,8 @@ function newEmptyContextUsage() {
         aigneHubCredits: 0,
         agentCalls: 0,
         duration: 0,
+        cacheCreationInputTokens: 0,
+        cacheReadInputTokens: 0,
     };
 }
 function mergeContextUsage(usage, additional) {
@@ -25,4 +27,8 @@ function mergeContextUsage(usage, additional) {
         usage.agentCalls += additional.agentCalls;
     if (additional.duration)
         usage.duration += additional.duration;
+    if (additional.cacheCreationInputTokens)
+        usage.cacheCreationInputTokens += additional.cacheCreationInputTokens;
+    if (additional.cacheReadInputTokens)
+        usage.cacheReadInputTokens += additional.cacheReadInputTokens;
 }

package/lib/cjs/index.d.ts

@@ -13,6 +13,7 @@ export * from "./agents/user-agent.js";
 export * from "./agents/video-model.js";
 export * from "./aigne/index.js";
 export * from "./memory/index.js";
+export * from "./prompt/agent-session.js";
 export * from "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";
 export * from "./utils/json-utils.js";

package/lib/cjs/index.js

@@ -29,6 +29,7 @@ __exportStar(require("./agents/user-agent.js"), exports);
 __exportStar(require("./agents/video-model.js"), exports);
 __exportStar(require("./aigne/index.js"), exports);
 __exportStar(require("./memory/index.js"), exports);
+__exportStar(require("./prompt/agent-session.js"), exports);
 __exportStar(require("./prompt/prompt-builder.js"), exports);
 __exportStar(require("./prompt/template.js"), exports);
 __exportStar(require("./utils/json-utils.js"), exports);