@aigne/core 1.15.0 → 1.16.0

package/lib/cjs/{models → agents}/chat-model.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatModel = void 0;
 const zod_1 = require("zod");
-const agent_js_1 = require("../agents/agent.js");
+const agent_js_1 = require("./agent.js");
 /**
  * ChatModel is an abstract base class for interacting with Large Language Models (LLMs).
  *
@@ -12,19 +12,19 @@ const agent_js_1 = require("../agents/agent.js");
  *
  * @example
  * Here's how to implement a custom ChatModel:
- * {@includeCode ../../test/models/chat-model.test.ts#example-chat-model}
+ * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model}
  *
  * @example
  * Here's an example showing streaming response with readable stream:
- * {@includeCode ../../test/models/chat-model.test.ts#example-chat-model-streaming}
+ * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-streaming}
  *
  * @example
  * Here's an example showing streaming response with async generator:
- * {@includeCode ../../test/models/chat-model.test.ts#example-chat-model-streaming-async-generator}
+ * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-streaming-async-generator}
  *
  * @example
  * Here's an example with tool calls:
- * {@includeCode ../../test/models/chat-model.test.ts#example-chat-model-tools}
+ * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
 class ChatModel extends agent_js_1.Agent {
     constructor() {
@@ -59,6 +59,19 @@ class ChatModel extends agent_js_1.Agent {
             }
         }
     }
+    /**
+     * Normalizes tool names to ensure compatibility with language models
+     *
+     * This method converts tool names to a format that complies with model requirements
+     * by replacing hyphens and whitespace characters with underscores. The normalized
+     * names are used for tool calls while preserving the original names for reference.
+     *
+     * @param name - The original tool name to normalize
+     * @returns A promise that resolves to the normalized tool name
+     */
+    async normalizeToolName(name) {
+        return name.replaceAll(/[-\s]/g, "_");
+    }
     /**
      * Performs preprocessing operations before handling input
      *
@@ -68,14 +81,29 @@ class ChatModel extends agent_js_1.Agent {
      * @param context Execution context
      * @throws Error if token usage exceeds maximum limit
      */
-    preprocess(input, context) {
+    async preprocess(input, context) {
         super.preprocess(input, context);
         const { limits, usage } = context;
         const usedTokens = usage.outputTokens + usage.inputTokens;
         if (limits?.maxTokens && usedTokens >= limits.maxTokens) {
             throw new Error(`Exceeded max tokens ${usedTokens}/${limits.maxTokens}`);
         }
-        this.validateToolNames(input.tools);
+        // Automatically convert tool names to a valid format
+        if (input.tools?.length) {
+            const toolsMap = {};
+            const tools = [];
+            for (const originalTool of input.tools) {
+                const name = await this.normalizeToolName(originalTool.function.name);
+                const tool = {
+                    ...originalTool,
+                    function: { ...originalTool.function, name },
+                };
+                tools.push(tool);
+                toolsMap[name] = originalTool;
+            }
+            this.validateToolNames(tools);
+            Object.assign(input, { _toolsMap: toolsMap, tools });
+        }
     }
     /**
      * Performs postprocessing operations after handling output
@@ -87,6 +115,19 @@ class ChatModel extends agent_js_1.Agent {
      * @param context Execution context
      */
     postprocess(input, output, context) {
+        // Restore original tool names in the output
+        if (output.toolCalls?.length) {
+            const toolsMap = input._toolsMap;
+            if (toolsMap) {
+                for (const toolCall of output.toolCalls) {
+                    const originalTool = toolsMap[toolCall.function.name];
+                    if (!originalTool) {
+                        throw new Error(`Tool "${toolCall.function.name}" not found in tools map`);
+                    }
+                    toolCall.function.name = originalTool.function.name;
+                }
+            }
+        }
         super.postprocess(input, output, context);
         const { usage } = output;
         if (usage) {

package/lib/cjs/agents/guide-rail-agent.d.ts

@@ -0,0 +1,62 @@
+import type { Agent, AgentOptions, Message } from "./agent.js";
+/**
+ * Input interface for GuideRail agents
+ *
+ * GuideRail agents receive both input and expected output, allowing them to
+ * validate, transform, or control the flow of messages between other agents.
+ */
+export interface GuideRailAgentInput extends Message {
+    /**
+     * The input data to be processed
+     *
+     * This is the original message that would be sent to the target agent
+     */
+    input?: unknown;
+    /**
+     * The expected output data
+     *
+     * This is what the target agent is expected to produce, allowing
+     * the GuideRail agent to validate or transform the data flow
+     */
+    output?: unknown;
+}
+/**
+ * Output interface for GuideRail agents
+ *
+ * GuideRail agents can either allow the process to continue or abort it with a reason.
+ * This provides a mechanism for enforcing rules, validating data, and controlling
+ * the execution flow of the agent system.
+ */
+export interface GuideRailAgentOutput extends Message {
+    /**
+     * Whether to abort the current process
+     *
+     * When true, the agent system should stop the current execution flow
+     * and prevent further processing based on this input/output pair
+     *
+     * @default false
+     */
+    abort?: boolean;
+    /**
+     * Reason for aborting the process
+     *
+     * When abort is true, this provides a human-readable explanation
+     * for why the process was stopped
+     */
+    reason?: string;
+}
+/**
+ * GuideRail agent type definition
+ *
+ * GuideRail agents act as validators, transformers, or controllers for the message
+ * flow between agents. They can enforce rules, perform safety checks, ensure data
+ * quality, or implement business logic validations.
+ *
+ * Use GuideRail agents when you need to:
+ * - Validate inputs or outputs against specific criteria
+ * - Enforce security or safety policies
+ * - Implement business rules that control agent interactions
+ * - Monitor and audit agent behavior
+ */
+export type GuideRailAgent = Agent<GuideRailAgentInput, GuideRailAgentOutput>;
+export declare const guideRailAgentOptions: AgentOptions<GuideRailAgentInput, GuideRailAgentOutput>;

package/lib/cjs/agents/guide-rail-agent.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.guideRailAgentOptions = void 0;
+const zod_1 = require("zod");
+exports.guideRailAgentOptions = {
+    inputSchema: zod_1.z.object({
+        input: zod_1.z.unknown(),
+        output: zod_1.z.unknown(),
+    }),
+    outputSchema: zod_1.z.object({
+        abort: zod_1.z.boolean().optional().describe("Whether to abort the current process"),
+        reason: zod_1.z.string().optional().describe("Reason for aborting the process"),
+    }),
+};

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

@@ -84,25 +84,25 @@ class MCPAgent extends agent_js_1.Agent {
             version: MCP_AGENT_CLIENT_VERSION,
         }, undefined, isSSEServerParameters(options) ? { transportCreator, ...options } : undefined);
         const transport = transportCreator();
-        logger_js_1.logger.mcp(`Connecting to MCP server: ${getMCPServerString(options)}`);
+        logger_js_1.logger.debug(`Connecting to MCP server: ${getMCPServerString(options)}`);
         await client.connect(transport);
         const mcpServer = getMCPServerName(client);
         const { tools: isToolsAvailable, prompts: isPromptsAvailable, resources: isResourcesAvailable, } = client.getServerCapabilities() ?? {};
-        logger_js_1.logger.mcp(`Listing tools from ${mcpServer}`);
+        logger_js_1.logger.debug(`Listing tools from ${mcpServer}`);
         const skills = isToolsAvailable
             ? await client.listTools().then(({ tools }) => {
-                logger_js_1.logger.mcp(`Listing tools from ${mcpServer} completed %O`, tools?.map((i) => i.name));
+                logger_js_1.logger.debug(`Listing tools from ${mcpServer} completed %O`, tools?.map((i) => i.name));
                 return tools.map((tool) => (0, mcp_utils_js_1.toolFromMCPTool)(tool, { client }));
             })
             : undefined;
-        logger_js_1.logger.mcp(`Listing prompts from ${mcpServer}`);
+        logger_js_1.logger.debug(`Listing prompts from ${mcpServer}`);
         const prompts = isPromptsAvailable
             ? await client.listPrompts().then(({ prompts }) => {
-                logger_js_1.logger.mcp(`Listing prompts from ${mcpServer} completed %O`, prompts?.map((i) => i.name));
+                logger_js_1.logger.debug(`Listing prompts from ${mcpServer} completed %O`, prompts?.map((i) => i.name));
                 return prompts.map((prompt) => (0, mcp_utils_js_1.promptFromMCPPrompt)(prompt, { client }));
             })
             : undefined;
-        logger_js_1.logger.mcp(`Listing resources from ${mcpServer}`);
+        logger_js_1.logger.debug(`Listing resources from ${mcpServer}`);
         // TODO: should conditionally call listResourceTemplates based on the server capabilities
         // but the capability is not correct in the current SDK version
         const resources = isResourcesAvailable
@@ -111,7 +111,7 @@ class MCPAgent extends agent_js_1.Agent {
                 client.listResourceTemplates().catch(() => ({ resourceTemplates: [] })),
             ]).then(([{ resources }, { resourceTemplates }]) => {
                 const result = [...resources, ...resourceTemplates].map((resource) => (0, mcp_utils_js_1.resourceFromMCPResource)(resource, { client }));
-                logger_js_1.logger.mcp(`Listing resources from ${mcpServer} completed %O`, result.map((i) => i.name));
+                logger_js_1.logger.debug(`Listing resources from ${mcpServer} completed %O`, result.map((i) => i.name));
                 return result;
             })
             : undefined;
@@ -235,7 +235,7 @@ class ClientWithReconnect extends index_js_1.Client {
         }, {
             retries: this.reconnectOptions?.maxReconnects ?? DEFAULT_MAX_RECONNECTS,
             shouldRetry: this.shouldReconnect,
-            onFailedAttempt: (error) => logger_js_1.logger.mcp("Reconnect attempt failed: %O", error),
+            onFailedAttempt: (error) => logger_js_1.logger.error("Reconnect attempt failed: %O", error),
         });
     }
     async request(request, resultSchema, options) {
@@ -248,7 +248,7 @@ class ClientWithReconnect extends index_js_1.Client {
         }
         catch (error) {
             if (this.shouldReconnect(error)) {
-                logger_js_1.logger.mcp("Error occurred, reconnecting to MCP server: %O", error);
+                logger_js_1.logger.error("Error occurred, reconnecting to MCP server: %O", error);
                 await this.reconnect();
                 return await super.request(request, resultSchema, mergedOptions);
             }

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

@@ -1,6 +1,7 @@
 import { Agent, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { ChatModel } from "../agents/chat-model.js";
 import type { UserAgent } from "../agents/user-agent.js";
-import { ChatModel } from "../models/chat-model.js";
+import { type LoadOptions } from "../loader/index.js";
 import { AIGNEContext, type InvokeOptions } from "./context.js";
 import { type MessagePayload, MessageQueue, type MessageQueueListener, type Unsubscribe } from "./message-queue.js";
 import type { ContextLimits } from "./usage.js";
@@ -54,7 +55,7 @@ export declare class AIGNE {
      * @param options - Options to override the loaded configuration.
      * @returns A fully initialized AIGNE instance with configured agents and skills.
      */
-    static load(path: string, options?: AIGNEOptions): Promise<AIGNE>;
+    static load(path: string, options: AIGNEOptions & Pick<LoadOptions, "models">): Promise<AIGNE>;
     /**
      * Creates a new AIGNE instance with the specified options.
      *

package/lib/cjs/aigne/aigne.js

@@ -3,8 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIGNE = void 0;
 const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
+const chat_model_js_1 = require("../agents/chat-model.js");
 const index_js_1 = require("../loader/index.js");
-const chat_model_js_1 = require("../models/chat-model.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const context_js_1 = require("./context.js");
 const message_queue_js_1 = require("./message-queue.js");
@@ -30,7 +30,7 @@ class AIGNE {
      * @returns A fully initialized AIGNE instance with configured agents and skills.
      */
     static async load(path, options) {
-        const { model, agents, skills, ...aigne } = await (0, index_js_1.load)({ path });
+        const { model, agents, skills, ...aigne } = await (0, index_js_1.load)({ models: options.models, path });
         return new AIGNE({
             ...options,
             model: options?.model || model,

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

@@ -1,7 +1,7 @@
 import EventEmitter from "node:events";
 import { Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import type { ChatModel } from "../agents/chat-model.js";
 import { UserAgent } from "../agents/user-agent.js";
-import type { ChatModel } from "../models/chat-model.js";
 import { type OmitPropertiesFromArrayFirstElement } from "../utils/type-utils.js";
 import type { Args, Listener, TypedEventEmitter } from "../utils/typed-event-emtter.js";
 import { type MessagePayload, MessageQueue, type MessageQueueListener, type Unsubscribe } from "./message-queue.js";
@@ -41,6 +41,7 @@ export type ContextEmitEventMap = {
 export interface InvokeOptions extends AgentInvokeOptions {
     returnActiveAgent?: boolean;
     disableTransfer?: boolean;
+    sourceAgent?: Agent;
 }
 /**
  * @hidden

package/lib/cjs/aigne/context.js

@@ -174,6 +174,13 @@ class AIGNEContextInternal {
         let output;
         for (;;) {
             const result = {};
+            if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
+                options.sourceAgent.hooks.onHandoff?.({
+                    source: options.sourceAgent,
+                    target: activeAgent,
+                    input,
+                });
+            }
             const stream = await activeAgent.invoke(input, context, { streaming: true });
             for await (const value of stream) {
                 if (value.delta.text) {

package/lib/cjs/index.d.ts

@@ -5,6 +5,6 @@ export * from "./agents/team-agent.js";
 export * from "./agents/types.js";
 export * from "./agents/user-agent.js";
 export * from "./aigne/index.js";
-export * from "./models/chat-model.js";
+export * from "./agents/chat-model.js";
 export * from "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";

package/lib/cjs/index.js

@@ -21,6 +21,6 @@ __exportStar(require("./agents/team-agent.js"), exports);
 __exportStar(require("./agents/types.js"), exports);
 __exportStar(require("./agents/user-agent.js"), exports);
 __exportStar(require("./aigne/index.js"), exports);
-__exportStar(require("./models/chat-model.js"), exports);
+__exportStar(require("./agents/chat-model.js"), exports);
 __exportStar(require("./prompt/prompt-builder.js"), exports);
 __exportStar(require("./prompt/template.js"), exports);

package/lib/cjs/loader/agent-yaml.d.ts

@@ -14,13 +14,13 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
     }, {
         [x: string]: any;
     }> | undefined;
-    toolChoice?: AIAgentToolChoice | undefined;
     outputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {
         [x: string]: any;
     }> | undefined;
     outputKey?: string | undefined;
+    toolChoice?: AIAgentToolChoice | undefined;
 } | {
     type: "mcp";
     url?: string | undefined;

package/lib/cjs/loader/index.d.ts

@@ -1,7 +1,14 @@
+import type { Camelize } from "camelize-ts";
 import { z } from "zod";
 import { type Agent } from "../agents/agent.js";
-import type { ChatModel, ChatModelOptions } from "../models/chat-model.js";
+import type { ChatModel, ChatModelOptions } from "../agents/chat-model.js";
 export interface LoadOptions {
+    models: {
+        new (parameters: {
+            model?: string;
+            modelOptions?: ChatModelOptions;
+        }): ChatModel;
+    }[];
     path: string;
 }
 export declare function load(options: LoadOptions): Promise<{
@@ -15,12 +22,12 @@ export declare function load(options: LoadOptions): Promise<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
 }>;
 export declare function loadAgent(path: string): Promise<Agent>;
-export declare function loadModel(model?: z.infer<typeof aigneFileSchema>["chat_model"], modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
+export declare function loadModel(models: LoadOptions["models"], model?: Camelize<z.infer<typeof aigneFileSchema>["chat_model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
 declare const aigneFileSchema: z.ZodObject<{
     name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     description: z.ZodOptional<z.ZodNullable<z.ZodString>>;
@@ -29,35 +36,35 @@ declare const aigneFileSchema: z.ZodObject<{
         name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         temperature: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
         top_p: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        frequent_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+        frequency_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
         presence_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
     }, "strip", z.ZodTypeAny, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     }, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     }>]>>>, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined, string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined>;
     agents: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
@@ -71,7 +78,7 @@ declare const aigneFileSchema: z.ZodObject<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
     agents?: string[] | null | undefined;
@@ -84,7 +91,7 @@ declare const aigneFileSchema: z.ZodObject<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
     agents?: string[] | null | undefined;
@@ -98,7 +105,7 @@ export declare function loadAIGNEFile(path: string): Promise<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
     agents?: string[] | null | undefined;

package/lib/cjs/loader/index.js

@@ -11,14 +11,6 @@ const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
 const ai_agent_js_1 = require("../agents/ai-agent.js");
 const mcp_agent_js_1 = require("../agents/mcp-agent.js");
-const bedrock_chat_model_js_1 = require("../models/bedrock-chat-model.js");
-const claude_chat_model_js_1 = require("../models/claude-chat-model.js");
-const deepseek_chat_model_js_1 = require("../models/deepseek-chat-model.js");
-const gemini_chat_model_js_1 = require("../models/gemini-chat-model.js");
-const ollama_chat_model_js_1 = require("../models/ollama-chat-model.js");
-const open_router_chat_model_js_1 = require("../models/open-router-chat-model.js");
-const openai_chat_model_js_1 = require("../models/openai-chat-model.js");
-const xai_chat_model_js_1 = require("../models/xai-chat-model.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_js_1 = require("./agent-js.js");
 const agent_yaml_js_1 = require("./agent-yaml.js");
@@ -32,7 +24,7 @@ async function load(options) {
     const skills = await Promise.all((aigne.skills ?? []).map((filename) => loadAgent((0, node_path_1.join)(rootDir, filename))));
     return {
         ...aigne,
-        model: await loadModel(aigne.chat_model),
+        model: await loadModel(options.models, aigne.chat_model),
         agents,
         skills,
     };
@@ -70,26 +62,15 @@ async function loadAgent(path) {
 }
 const { MODEL_PROVIDER, MODEL_NAME } = process.env;
 const DEFAULT_MODEL_PROVIDER = "openai";
-const DEFAULT_MODEL_NAME = "gpt-4o-mini";
-async function loadModel(model, modelOptions) {
+async function loadModel(models, model, modelOptions) {
     const params = {
-        model: MODEL_NAME ?? model?.name ?? DEFAULT_MODEL_NAME,
+        model: MODEL_NAME ?? model?.name ?? undefined,
         temperature: model?.temperature ?? undefined,
-        topP: model?.top_p ?? undefined,
-        frequencyPenalty: model?.frequent_penalty ?? undefined,
-        presencePenalty: model?.presence_penalty ?? undefined,
+        topP: model?.topP ?? undefined,
+        frequencyPenalty: model?.frequencyPenalty ?? undefined,
+        presencePenalty: model?.presencePenalty ?? undefined,
     };
-    const availableModels = [
-        openai_chat_model_js_1.OpenAIChatModel,
-        claude_chat_model_js_1.ClaudeChatModel,
-        xai_chat_model_js_1.XAIChatModel,
-        gemini_chat_model_js_1.GeminiChatModel,
-        deepseek_chat_model_js_1.DeepSeekChatModel,
-        open_router_chat_model_js_1.OpenRouterChatModel,
-        ollama_chat_model_js_1.OllamaChatModel,
-        bedrock_chat_model_js_1.BedrockChatModel,
-    ];
-    const M = availableModels.find((m) => m.name
+    const M = models.find((m) => m.name
         .toLowerCase()
         .includes((MODEL_PROVIDER ?? model?.provider ?? DEFAULT_MODEL_PROVIDER).toLowerCase()));
     if (!M)
@@ -107,7 +88,7 @@ const aigneFileSchema = zod_1.z.object({
             name: zod_1.z.string().nullish(),
             temperature: zod_1.z.number().min(0).max(2).nullish(),
             top_p: zod_1.z.number().min(0).nullish(),
-            frequent_penalty: zod_1.z.number().min(-2).max(2).nullish(),
+            frequency_penalty: zod_1.z.number().min(-2).max(2).nullish(),
             presence_penalty: zod_1.z.number().min(-2).max(2).nullish(),
         }),
     ])

package/lib/cjs/memory/retriever.d.ts

@@ -39,11 +39,11 @@ export declare const memoryRetrieverInputSchema: z.ZodObject<{
     limit: z.ZodOptional<z.ZodNumber>;
     search: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
-    limit?: number | undefined;
     search?: string | undefined;
-}, {
     limit?: number | undefined;
+}, {
     search?: string | undefined;
+    limit?: number | undefined;
 }>;
 /**
  * @hidden

package/lib/cjs/prompt/prompt-builder.d.ts

@@ -1,9 +1,9 @@
 import type { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
 import { Agent, type Message } from "../agents/agent.js";
 import type { AIAgent } from "../agents/ai-agent.js";
+import type { ChatModel, ChatModelInput } from "../agents/chat-model.js";
 import type { Context } from "../aigne/context.js";
 import type { MemoryAgent } from "../memory/memory.js";
-import type { ChatModel, ChatModelInput } from "../models/chat-model.js";
 import { ChatMessagesTemplate } from "./template.js";
 export declare const MESSAGE_KEY = "$message";
 export declare function createMessage<I extends Message>(message: string | I): I;
@@ -11,7 +11,7 @@ export declare function getMessage(input: Message): string | undefined;
 export interface PromptBuilderOptions {
     instructions?: string | ChatMessagesTemplate;
 }
-export interface PromptBuilderBuildOptions {
+export interface PromptBuildOptions {
     memory?: MemoryAgent | MemoryAgent[];
     context: Context;
     agent?: AIAgent;
@@ -32,7 +32,7 @@ export declare class PromptBuilder {
     private static fromMCPPromptResult;
     constructor(options?: PromptBuilderOptions);
     instructions?: string | ChatMessagesTemplate;
-    build(options: PromptBuilderBuildOptions): Promise<ChatModelInput & {
+    build(options: PromptBuildOptions): Promise<ChatModelInput & {
         toolAgents?: Agent[];
     }>;
     private buildMessages;

package/lib/cjs/prompt/template.d.ts

@@ -1,4 +1,4 @@
-import type { ChatModelInputMessage, ChatModelInputMessageContent, ChatModelOutputToolCall } from "../models/chat-model.js";
+import type { ChatModelInputMessage, ChatModelInputMessageContent, ChatModelOutputToolCall } from "../agents/chat-model.js";
 export declare class PromptTemplate {
     template: string;
     static from(template: string): PromptTemplate;
@@ -24,7 +24,7 @@ export declare class AgentMessageTemplate extends ChatMessageTemplate {
     constructor(content?: ChatModelInputMessage["content"], toolCalls?: ChatModelOutputToolCall[] | undefined, name?: string);
     format(variables?: Record<string, unknown>): {
         toolCalls: ChatModelOutputToolCall[] | undefined;
-        role: import("../models/chat-model.js").Role;
+        role: import("../agents/chat-model.js").Role;
         content?: ChatModelInputMessageContent;
         toolCallId?: string;
         name?: string;
@@ -36,7 +36,7 @@ export declare class ToolMessageTemplate extends ChatMessageTemplate {
     constructor(content: object | string, toolCallId: string, name?: string);
     format(variables?: Record<string, unknown>): {
         toolCallId: string;
-        role: import("../models/chat-model.js").Role;
+        role: import("../agents/chat-model.js").Role;
         content?: ChatModelInputMessageContent;
         toolCalls?: {
             id: string;

package/lib/cjs/utils/json-schema.js

@@ -28,7 +28,7 @@ function parseJSON(json) {
         return JSON.parse(json);
     }
     catch (error) {
-        logger_js_1.logger.core("Failed to parse JSON", { json, error });
+        logger_js_1.logger.error("Failed to parse JSON", { json, error });
         throw error;
     }
 }

package/lib/cjs/utils/logger.d.ts

@@ -1,8 +1,33 @@
-import debug from "debug";
-export declare const logger: debug.Debug & {
-    debug: debug.Debug;
-    default: debug.Debug;
-} & {
-    core: debug.Debugger;
-    mcp: debug.Debugger;
-};
+export declare enum LogLevel {
+    ERROR = "error",
+    WARN = "warn",
+    INFO = "info",
+    DEBUG = "debug"
+}
+export declare class Logger {
+    constructor(options: {
+        level: LogLevel;
+        ns: string;
+    });
+    level: LogLevel;
+    private debugLogger;
+    private infoLogger;
+    private warnLogger;
+    private errorLogger;
+    logMessage: {
+        (...data: any[]): void;
+        (...data: any[]): void;
+        (message?: any, ...optionalParams: any[]): void;
+    };
+    logError: {
+        (...data: any[]): void;
+        (...data: any[]): void;
+        (message?: any, ...optionalParams: any[]): void;
+    };
+    debug(message: string, ...args: unknown[]): void;
+    info(message: string, ...args: unknown[]): void;
+    warn(message: string, ...args: unknown[]): void;
+    error(message: string, ...args: unknown[]): void;
+    enabled(level: LogLevel): boolean;
+}
+export declare const logger: Logger;