@ellyco/agentic 0.1.0

package/dist/models/BedrockModel.js

@@ -0,0 +1,315 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BedrockModel = void 0;
+const BaseModel_1 = require("./BaseModel");
+const client_bedrock_runtime_1 = require("@aws-sdk/client-bedrock-runtime");
+const messages_1 = require("../messages");
+const z = require("zod");
+/**
+ * Model implementation using AWS Bedrock as the backend.
+ * Handles communication with Bedrock API for inference, tools, and structured output.
+ *
+ * @class BedrockModel
+ * @extends {BaseModel}
+ *
+ * @example
+ * ```typescript
+ * const model = new BedrockModel({
+ *   modelId: "anthropic.claude-3-sonnet-20240229-v1:0",
+ *   temperature: 0.7,
+ *   maxTokens: 2048
+ * });
+ *
+ * model.withSystemMessage("You are a helpful assistant")
+ *   .withTools([searchTool]);
+ *
+ * const response = await model.invoke([userMessage]);
+ * ```
+ */
+class BedrockModel extends BaseModel_1.BaseModel {
+    modelId;
+    client;
+    /**
+     * Creates a new Bedrock model instance.
+     *
+     * @param {BedrockModelConfig} config - Configuration including model ID and AWS settings
+     */
+    constructor(config) {
+        super(config);
+        this.modelId = config.modelId;
+        this.client = new client_bedrock_runtime_1.BedrockRuntimeClient(config.aws ?? {});
+    }
+    /**
+     * Converts internal message format to Bedrock API message format.
+     * Handles BaseMessages (user, agent, system) and ToolUse messages (requests, responses, errors).
+     *
+     * @private
+     * @param {ModelMessages[]} messages - Messages to convert
+     * @returns {Message[]} Messages in Bedrock format
+     */
+    convertMessagesToBedrockMessages(messages) {
+        const bedrockMessages = [];
+        for (const message of messages) {
+            if (message instanceof messages_1.BaseMessage) {
+                const role = message.role;
+                if (role === messages_1.MessageRole.SYSTEM && !this.systemMessage) {
+                    this.systemMessage = message;
+                }
+                else {
+                    bedrockMessages.push({
+                        role: role === messages_1.MessageRole.USER
+                            ? client_bedrock_runtime_1.ConversationRole.USER
+                            : client_bedrock_runtime_1.ConversationRole.ASSISTANT,
+                        content: [
+                            {
+                                ...(message.hasText()
+                                    ? { text: message.text }
+                                    : {}),
+                            },
+                        ],
+                    });
+                }
+            }
+            if (message instanceof messages_1.ToolUse) {
+                if (message instanceof messages_1.ToolRequest) {
+                    bedrockMessages.push({
+                        role: client_bedrock_runtime_1.ConversationRole.ASSISTANT,
+                        content: [
+                            {
+                                toolUse: {
+                                    toolUseId: message.toolUseId,
+                                    name: message.toolName,
+                                    input: message.input,
+                                },
+                            },
+                        ],
+                    });
+                }
+                if (message instanceof messages_1.ToolResponse) {
+                    bedrockMessages.push({
+                        role: client_bedrock_runtime_1.ConversationRole.USER,
+                        content: [
+                            {
+                                toolResult: {
+                                    toolUseId: message.toolUseId,
+                                    content: [
+                                        {
+                                            json: message.output,
+                                        },
+                                    ],
+                                },
+                            },
+                        ],
+                    });
+                }
+                if (message instanceof messages_1.ToolError) {
+                    bedrockMessages.push({
+                        role: client_bedrock_runtime_1.ConversationRole.USER,
+                        content: [
+                            {
+                                toolResult: {
+                                    toolUseId: message.toolUseId,
+                                    content: [
+                                        {
+                                            text: message.error,
+                                        },
+                                    ],
+                                    status: client_bedrock_runtime_1.ToolResultStatus.ERROR,
+                                },
+                            },
+                        ],
+                    });
+                }
+            }
+        }
+        return bedrockMessages;
+    }
+    /**
+     * Converts a tool definition to Bedrock tool specification format.
+     *
+     * @private
+     * @param {ToolDefinition} tool - The tool definition to convert
+     * @returns {Tool} Bedrock formatted tool specification
+     */
+    convertToolDefinitionToBedrockTool(tool) {
+        return {
+            toolSpec: {
+                name: tool.name,
+                description: tool.description,
+                inputSchema: {
+                    json: z.toJSONSchema(tool.schema),
+                },
+            },
+        };
+    }
+    /**
+     * Builds the tool configuration for Bedrock.
+     * For structured output, creates a single "output" tool that captures the schema.
+     * For regular tools, converts all tool definitions.
+     *
+     * @private
+     * @returns {ToolConfiguration} Tool configuration for Bedrock
+     */
+    getBedrockToolConfig() {
+        if (this.structuredOutput) {
+            return {
+                tools: [
+                    {
+                        toolSpec: {
+                            name: "output",
+                            description: "Expected output from the model",
+                            inputSchema: {
+                                json: z.toJSONSchema(this.structuredOutput),
+                            },
+                        },
+                    },
+                ],
+                toolChoice: {
+                    tool: { name: "output" },
+                },
+            };
+        }
+        return {
+            tools: this.tools.map((tool) => this.convertToolDefinitionToBedrockTool(tool)),
+        };
+    }
+    /**
+     * Maps Bedrock stop reason to internal stop reason enum.
+     *
+     * @private
+     * @param {ConverseCommandOutput} response - The Bedrock API response
+     * @returns {InvokeResponseStopReason} Mapped stop reason
+     */
+    mapStopReason(response) {
+        switch (response.stopReason) {
+            case client_bedrock_runtime_1.StopReason.STOP_SEQUENCE:
+                return BaseModel_1.InvokeResponseStopReason.STOP_SEQUENCE;
+            case client_bedrock_runtime_1.StopReason.MAX_TOKENS:
+                return BaseModel_1.InvokeResponseStopReason.MAX_TOKENS;
+            case client_bedrock_runtime_1.StopReason.END_TURN:
+                return BaseModel_1.InvokeResponseStopReason.END_TURN;
+            case client_bedrock_runtime_1.StopReason.TOOL_USE:
+                return BaseModel_1.InvokeResponseStopReason.TOOL_USE;
+            default:
+                return BaseModel_1.InvokeResponseStopReason.UNKNOWN;
+        }
+    }
+    /**
+     * Builds inference configuration chunk with temperature, topP, and maxTokens if set.
+     *
+     * @private
+     * @returns {{inferenceConfig?: InferenceConfiguration}} Configuration chunk or empty object
+     */
+    getInferenceConfigChunk() {
+        const inferenceConfig = {
+            ...(this.temperature ? { temperature: this.temperature } : {}),
+            ...(this.topP ? { topP: this.topP } : {}),
+            ...(this.maxTokens ? { maxTokens: this.maxTokens } : {}),
+        };
+        if (Object.keys(inferenceConfig).length > 0) {
+            return { inferenceConfig };
+        }
+        return {};
+    }
+    /**
+     * Builds system message chunk.
+     * For structured output, adds a directive to only call the output tool.
+     *
+     * @private
+     * @returns {{system?: SystemContentBlock[]}} System messages or empty object
+     */
+    getSystemMessageChunk() {
+        if (this.systemMessage) {
+            return {
+                system: [
+                    {
+                        text: this.systemMessage.toString(),
+                    },
+                    ...(this.structuredOutput
+                        ? [
+                            {
+                                text: "IMPORTANT: ONLY call the output tool. No other messages!",
+                            },
+                        ]
+                        : []),
+                ],
+            };
+        }
+        else if (this.structuredOutput) {
+            return {
+                system: [
+                    {
+                        text: "IMPORTANT: ONLY call the output tool. No other messages!",
+                    },
+                ],
+            };
+        }
+        return {};
+    }
+    /**
+     * Builds tool configuration chunk.
+     *
+     * @private
+     * @returns {{toolConfig?: ToolConfiguration}} Tool configuration or empty object
+     */
+    getToolConfigChunk() {
+        const toolConfig = this.getBedrockToolConfig();
+        if (toolConfig.tools && toolConfig.tools.length > 0) {
+            return {
+                toolConfig,
+            };
+        }
+        return {};
+    }
+    /**
+     * Executes the model inference with Bedrock API.
+     * Sends messages to Bedrock, processes the response, and converts it to internal format.
+     *
+     * @protected
+     * @param {ModelMessages[]} inputMessages - Messages to send to the model
+     * @returns {Promise<InvokeResponse>} The model's response with messages and usage stats
+     */
+    async runModel(inputMessages) {
+        const bedrockMessages = this.convertMessagesToBedrockMessages(inputMessages);
+        const command = new client_bedrock_runtime_1.ConverseCommand({
+            modelId: this.modelId,
+            ...this.getInferenceConfigChunk(),
+            messages: bedrockMessages,
+            ...this.getSystemMessageChunk(),
+            ...this.getToolConfigChunk(),
+        });
+        const response = await this.client.send(command);
+        const message = response.output?.message;
+        const usage = {
+            inputTokens: response.usage?.inputTokens ?? 0,
+            outputTokens: response.usage?.outputTokens ?? 0,
+        };
+        const stopReason = this.mapStopReason(response);
+        if (!message || !message.content) {
+            return {
+                messages: [],
+                usage,
+                stopReason,
+            };
+        }
+        const messages = [];
+        for (const block of message.content) {
+            if ("text" in block) {
+                messages.push(new messages_1.AgentMessage(block.text));
+            }
+            else if ("toolUse" in block) {
+                messages.push(new messages_1.ToolRequest(block.toolUse.toolUseId, block.toolUse.name, block.toolUse.input));
+            }
+            else {
+                console.warn("Unknown block type", block);
+            }
+        }
+        return {
+            messages,
+            usage,
+            stopReason,
+        };
+    }
+}
+exports.BedrockModel = BedrockModel;
+//# sourceMappingURL=BedrockModel.js.map

package/dist/models/BedrockModel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BedrockModel.js","sourceRoot":"","sources":["../../src/models/BedrockModel.ts"],"names":[],"mappings":";;;AAAA,2CAMqB;AACrB,4EAcyC;AACzC,0CASqB;AACrB,yBAAyB;AAgBzB;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAa,YAAa,SAAQ,qBAAS;IAC/B,OAAO,CAAS;IAChB,MAAM,CAAuB;IAErC;;;;OAIG;IACH,YAAY,MAA0B;QAClC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,6CAAoB,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED;;;;;;;OAOG;IACK,gCAAgC,CACpC,QAAyB;QAEzB,MAAM,eAAe,GAAc,EAAE,CAAC;QACtC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,IAAI,OAAO,YAAY,sBAAW,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC1B,IAAI,IAAI,KAAK,sBAAW,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;oBACrD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;gBACjC,CAAC;qBAAM,CAAC;oBACJ,eAAe,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,IAAI,KAAK,sBAAW,CAAC,IAAI;4BAC3B,CAAC,CAAC,yCAAgB,CAAC,IAAI;4BACvB,CAAC,CAAC,yCAAgB,CAAC,SAAS;wBAChC,OAAO,EAAE;4BACL;gCACI,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE;oCACjB,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE;oCACxB,CAAC,CAAC,EAAE,CAAiB;6BAC5B;yBACJ;qBACJ,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YACD,IAAI,OAAO,YAAY,kBAAO,EAAE,CAAC;gBAC7B,IAAI,OAAO,YAAY,sBAAW,EAAE,CAAC;oBACjC,eAAe,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,yCAAgB,CAAC,SAAS;wBAChC,OAAO,EAAE;4BACL;gCACI,OAAO,EAAE;oCACL,SAAS,EAAE,OAAO,CAAC,SAAS;oCAC5B,IAAI,EAAE,OAAO,CAAC,QAAQ;oCACtB,KAAK,EAAE,OAAO,CAAC,KAAK;iCACvB;6BACJ;yBACJ;qBACJ,CAAC,CAAC;gBACP,CAAC;gBACD,IAAI,OAAO,YAAY,uBAAY,EAAE,CAAC;oBAClC,eAAe,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,yCAAgB,CAAC,IAAI;wBAC3B,OAAO,EAAE;4BACL;gCACI,UAAU,EAAE;oCACR,SAAS,EAAE,OAAO,CAAC,SAAS;oCAC5B,OAAO,EAAE;wCACL;4CACI,IAAI,EAAE,OAAO,CAAC,MAAM;yCACvB;qCACJ;iCACJ;6BACJ;yBACJ;qBACJ,CAAC,CAAC;gBACP,CAAC;gBACD,IAAI,OAAO,YAAY,oBAAS,EAAE,CAAC;oBAC/B,eAAe,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,yCAAgB,CAAC,IAAI;wBAC3B,OAAO,EAAE;4BACL;gCACI,UAAU,EAAE;oCACR,SAAS,EAAE,OAAO,CAAC,SAAS;oCAC5B,OAAO,EAAE;wCACL;4CACI,IAAI,EAAE,OAAO,CAAC,KAAK;yCACtB;qCACJ;oCACD,MAAM,EAAE,yCAAgB,CAAC,KAAK;iCACjC;6BACJ;yBACJ;qBACJ,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QACD,OAAO,eAAe,CAAC;IAC3B,CAAC;IAED;;;;;;OAMG;IACK,kCAAkC,CAAC,IAAoB;QAC3D,OAAO;YACH,QAAQ,EAAE;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE;oBACT,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAQ;iBAC3C;aACJ;SACJ,CAAC;IACN,CAAC;IAED;;;;;;;OAOG;IACK,oBAAoB;QACxB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACxB,OAAO;gBACH,KAAK,EAAE;oBACH;wBACI,QAAQ,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,gCAAgC;4BAC7C,WAAW,EAAE;gCACT,IAAI,EAAE,CAAC,CAAC,YAAY,CAChB,IAAI,CAAC,gBAAgB,CACjB;6BACX;yBACJ;qBACJ;iBACJ;gBACD,UAAU,EAAE;oBACR,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAC3B;aACJ,CAAC;QACN,CAAC;QACD,OAAO;YACH,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC3B,IAAI,CAAC,kCAAkC,CAAC,IAAI,CAAC,CAChD;SACJ,CAAC;IACN,CAAC;IAED;;;;;;OAMG;IACK,aAAa,CACjB,QAA+B;QAE/B,QAAQ,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC1B,KAAK,mCAAU,CAAC,aAAa;gBACzB,OAAO,oCAAwB,CAAC,aAAa,CAAC;YAClD,KAAK,mCAAU,CAAC,UAAU;gBACtB,OAAO,oCAAwB,CAAC,UAAU,CAAC;YAC/C,KAAK,mCAAU,CAAC,QAAQ;gBACpB,OAAO,oCAAwB,CAAC,QAAQ,CAAC;YAC7C,KAAK,mCAAU,CAAC,QAAQ;gBACpB,OAAO,oCAAwB,CAAC,QAAQ,CAAC;YAC7C;gBACI,OAAO,oCAAwB,CAAC,OAAO,CAAC;QAChD,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACK,uBAAuB;QAG3B,MAAM,eAAe,GAA2B;YAC5C,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC3D,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,OAAO,EAAE,eAAe,EAAE,CAAC;QAC/B,CAAC;QACD,OAAO,EAAE,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACK,qBAAqB;QACzB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,OAAO;gBACH,MAAM,EAAE;oBACJ;wBACI,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE;qBACtC;oBACD,GAAG,CAAC,IAAI,CAAC,gBAAgB;wBACrB,CAAC,CAAC;4BACE;gCACI,IAAI,EACA,0DAA0D;6BACjE;yBACJ;wBACD,CAAC,CAAC,EAAE,CAAC;iBACZ;aACJ,CAAC;QACN,CAAC;aAAM,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC/B,OAAO;gBACH,MAAM,EAAE;oBACJ;wBACI,IAAI,EACA,0DAA0D;qBACjE;iBACJ;aACJ,CAAC;QACN,CAAC;QACD,OAAO,EAAE,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACK,kBAAkB;QACtB,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC/C,IAAI,UAAU,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClD,OAAO;gBACH,UAAU;aACb,CAAC;QACN,CAAC;QACD,OAAO,EAAE,CAAC;IACd,CAAC;IAED;;;;;;;OAOG;IACO,KAAK,CAAC,QAAQ,CACpB,aAA8B;QAE9B,MAAM,eAAe,GAAG,IAAI,CAAC,gCAAgC,CACzD,aAAa,CAChB,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,wCAAe,CAAC;YAChC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,GAAG,IAAI,CAAC,uBAAuB,EAAE;YACjC,QAAQ,EAAE,eAAe;YACzB,GAAG,IAAI,CAAC,qBAAqB,EAAE;YAC/B,GAAG,IAAI,CAAC,kBAAkB,EAAE;SAC/B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;QACzC,MAAM,KAAK,GAAwB;YAC/B,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC;YAC7C,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;SAClD,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC/B,OAAO;gBACH,QAAQ,EAAE,EAAE;gBACZ,KAAK;gBACL,UAAU;aACb,CAAC;QACN,CAAC;QAED,MAAM,QAAQ,GACuB,EAAE,CAAC;QACxC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,MAAM,IAAI,KAAK,EAAE,CAAC;gBAClB,QAAQ,CAAC,IAAI,CACT,IAAI,uBAAY,CAAC,KAAK,CAAC,IAAK,CAAC,CAChC,CAAC;YACN,CAAC;iBAAM,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;gBAC5B,QAAQ,CAAC,IAAI,CACT,IAAI,sBAAW,CACX,KAAK,CAAC,OAAQ,CAAC,SAAU,EACzB,KAAK,CAAC,OAAQ,CAAC,IAAK,EACpB,KAAK,CAAC,OAAQ,CAAC,KAAY,CAC9B,CACJ,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,IAAI,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;YAC9C,CAAC;QACL,CAAC;QAED,OAAO;YACH,QAAQ;YACR,KAAK;YACL,UAAU;SACb,CAAC;IACN,CAAC;CACJ;AA/TD,oCA+TC"}

package/dist/models/TestModel.d.ts

@@ -0,0 +1,135 @@
+import { ToolDefinition } from "../tools";
+import { BaseModel, BaseModelConfig, InvokeResponse } from "./BaseModel";
+import * as z from "zod";
+import { AgentMessage, ModelMessages, ToolRequest } from "../messages";
+/**
+ * Configuration for test model responses.
+ * Allows specifying what the model should respond with for specific inputs.
+ *
+ * @class TestResponseConfig
+ *
+ * @example
+ * ```typescript
+ * const config = new TestResponseConfig()
+ *   .userSends([userMessage])
+ *   .respondWith([agentMessage]);
+ * ```
+ */
+export declare class TestResponseConfig {
+    /**
+     * Tools that should be available when matching this config
+     */
+    toolsIncluded: ToolDefinition[];
+    /**
+     * Messages the model should respond with
+     */
+    responseMessages: (AgentMessage | ToolRequest)[];
+    /**
+     * Input messages that trigger this config
+     */
+    inputMessages: ModelMessages[];
+    /**
+     * Structured output schema for this config (if using structured output)
+     */
+    structuredOutput?: z.ZodSchema<any>;
+    /**
+     * Specifies what messages the model should respond with.
+     *
+     * @param {(AgentMessage | ToolRequest)[]} messages - Response messages
+     * @returns {this} The config instance for method chaining
+     */
+    respondWith(messages: (AgentMessage | ToolRequest)[]): this;
+    /**
+     * Specifies which tools should be included when matching this config.
+     *
+     * @param {ToolDefinition[]} tools - Tools to include
+     * @returns {this} The config instance for method chaining
+     */
+    includedTools(tools: ToolDefinition[]): this;
+    /**
+     * Specifies the structured output schema for this config.
+     *
+     * @param {z.ZodSchema<any>} schema - The output schema
+     * @returns {this} The config instance for method chaining
+     */
+    providedStructuredOutput(schema: z.ZodSchema<any>): this;
+    /**
+     * Specifies the input messages that should trigger this config.
+     *
+     * @param {ModelMessages[]} messages - Input messages to match
+     * @returns {this} The config instance for method chaining
+     */
+    userSends(messages: ModelMessages[]): this;
+}
+/**
+ * A mock model implementation for testing purposes.
+ * Allows you to specify predefined responses for specific message inputs.
+ * Useful for testing agent behavior without hitting real model APIs.
+ *
+ * @class TestModel
+ * @extends {BaseModel}
+ *
+ * @example
+ * ```typescript
+ * const testModel = new TestModel({ temperature: 0.7 });
+ *
+ * // Configure a response
+ * const config = new TestResponseConfig()
+ *   .userSends([new UserMessage("Hello")])
+ *   .respondWith([new AgentMessage("Hi there!")]);
+ *
+ * testModel.addTestConfig(config);
+ *
+ * // When invoked with matching messages, returns the configured response
+ * const response = await testModel.invoke([new UserMessage("Hello")]);
+ * // response.messages[0].text === "Hi there!"
+ * ```
+ */
+export declare class TestModel extends BaseModel {
+    /**
+     * Array of configured test responses
+     */
+    testConfigs: TestResponseConfig[];
+    /**
+     * Creates a new test model instance.
+     *
+     * @param {BaseModelConfig} config - Model configuration
+     */
+    constructor(config: BaseModelConfig);
+    /**
+     * Adds a test configuration for a specific input-output pair.
+     *
+     * @param {TestResponseConfig} config - The test configuration
+     * @returns {this} The model instance for method chaining
+     *
+     * @example
+     * ```typescript
+     * model.addTestConfig(
+     *   new TestResponseConfig()
+     *     .userSends([userMsg])
+     *     .respondWith([agentMsg])
+     * );
+     * ```
+     */
+    addTestConfig(config: TestResponseConfig): this;
+    /**
+     * Finds a matching test configuration for the given messages.
+     * Matching is based on message content, tools, and structured output schema.
+     *
+     * @param {ModelMessages[]} messages - Messages to match against
+     * @returns {TestResponseConfig | undefined} The matching config, or undefined if no match
+     *
+     * @private
+     */
+    findMatchingConfig(messages: ModelMessages[]): TestResponseConfig | undefined;
+    /**
+     * Executes the test model by finding and returning a matching configuration.
+     *
+     * @protected
+     * @param {ModelMessages[]} messages - Messages to respond to
+     * @returns {Promise<InvokeResponse>} The configured response
+     * @throws {Error} If no matching test configuration is found
+     */
+    protected runModel(messages: ModelMessages[]): Promise<InvokeResponse>;
+}
+//# sourceMappingURL=TestModel.d.ts.map

package/dist/models/TestModel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestModel.d.ts","sourceRoot":"","sources":["../../src/models/TestModel.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EACH,SAAS,EACT,eAAe,EACf,cAAc,EAEjB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AACzB,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAEvE;;;;;;;;;;;;GAYG;AACH,qBAAa,kBAAkB;IAC3B;;OAEG;IACH,aAAa,EAAE,cAAc,EAAE,CAAM;IAErC;;OAEG;IACH,gBAAgB,EACZ,CAAC,YAAY,GAAG,WAAW,CAAC,EAAE,CAAM;IAExC;;OAEG;IACH,aAAa,EAAE,aAAa,EAAE,CAAM;IAEpC;;OAEG;IACH,gBAAgB,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAEpC;;;;;OAKG;IACH,WAAW,CACP,QAAQ,EAAE,CAAC,YAAY,GAAG,WAAW,CAAC,EAAE,GACzC,IAAI;IAKP;;;;;OAKG;IACH,aAAa,CAAC,KAAK,EAAE,cAAc,EAAE,GAAG,IAAI;IAK5C;;;;;OAKG;IACH,wBAAwB,CAAC,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI;IAKxD;;;;;OAKG;IACH,SAAS,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAI;CAI7C;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,qBAAa,SAAU,SAAQ,SAAS;IACpC;;OAEG;IACH,WAAW,EAAE,kBAAkB,EAAE,CAAM;IAEvC;;;;OAIG;gBACS,MAAM,EAAE,eAAe;IAInC;;;;;;;;;;;;;;OAcG;IACH,aAAa,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI;IAK/C;;;;;;;;OAQG;IACH,kBAAkB,CACd,QAAQ,EAAE,aAAa,EAAE,GAC1B,kBAAkB,GAAG,SAAS;IAkCjC;;;;;;;OAOG;IACH,SAAS,CAAC,QAAQ,CACd,QAAQ,EAAE,aAAa,EAAE,GAC1B,OAAO,CAAC,cAAc,CAAC;CAc7B"}

package/dist/models/TestModel.js

@@ -0,0 +1,191 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestModel = exports.TestResponseConfig = void 0;
+const BaseModel_1 = require("./BaseModel");
+const z = require("zod");
+/**
+ * Configuration for test model responses.
+ * Allows specifying what the model should respond with for specific inputs.
+ *
+ * @class TestResponseConfig
+ *
+ * @example
+ * ```typescript
+ * const config = new TestResponseConfig()
+ *   .userSends([userMessage])
+ *   .respondWith([agentMessage]);
+ * ```
+ */
+class TestResponseConfig {
+    /**
+     * Tools that should be available when matching this config
+     */
+    toolsIncluded = [];
+    /**
+     * Messages the model should respond with
+     */
+    responseMessages = [];
+    /**
+     * Input messages that trigger this config
+     */
+    inputMessages = [];
+    /**
+     * Structured output schema for this config (if using structured output)
+     */
+    structuredOutput;
+    /**
+     * Specifies what messages the model should respond with.
+     *
+     * @param {(AgentMessage | ToolRequest)[]} messages - Response messages
+     * @returns {this} The config instance for method chaining
+     */
+    respondWith(messages) {
+        this.responseMessages = messages;
+        return this;
+    }
+    /**
+     * Specifies which tools should be included when matching this config.
+     *
+     * @param {ToolDefinition[]} tools - Tools to include
+     * @returns {this} The config instance for method chaining
+     */
+    includedTools(tools) {
+        this.toolsIncluded = tools;
+        return this;
+    }
+    /**
+     * Specifies the structured output schema for this config.
+     *
+     * @param {z.ZodSchema<any>} schema - The output schema
+     * @returns {this} The config instance for method chaining
+     */
+    providedStructuredOutput(schema) {
+        this.structuredOutput = schema;
+        return this;
+    }
+    /**
+     * Specifies the input messages that should trigger this config.
+     *
+     * @param {ModelMessages[]} messages - Input messages to match
+     * @returns {this} The config instance for method chaining
+     */
+    userSends(messages) {
+        this.inputMessages = messages;
+        return this;
+    }
+}
+exports.TestResponseConfig = TestResponseConfig;
+/**
+ * A mock model implementation for testing purposes.
+ * Allows you to specify predefined responses for specific message inputs.
+ * Useful for testing agent behavior without hitting real model APIs.
+ *
+ * @class TestModel
+ * @extends {BaseModel}
+ *
+ * @example
+ * ```typescript
+ * const testModel = new TestModel({ temperature: 0.7 });
+ *
+ * // Configure a response
+ * const config = new TestResponseConfig()
+ *   .userSends([new UserMessage("Hello")])
+ *   .respondWith([new AgentMessage("Hi there!")]);
+ *
+ * testModel.addTestConfig(config);
+ *
+ * // When invoked with matching messages, returns the configured response
+ * const response = await testModel.invoke([new UserMessage("Hello")]);
+ * // response.messages[0].text === "Hi there!"
+ * ```
+ */
+class TestModel extends BaseModel_1.BaseModel {
+    /**
+     * Array of configured test responses
+     */
+    testConfigs = [];
+    /**
+     * Creates a new test model instance.
+     *
+     * @param {BaseModelConfig} config - Model configuration
+     */
+    constructor(config) {
+        super(config);
+    }
+    /**
+     * Adds a test configuration for a specific input-output pair.
+     *
+     * @param {TestResponseConfig} config - The test configuration
+     * @returns {this} The model instance for method chaining
+     *
+     * @example
+     * ```typescript
+     * model.addTestConfig(
+     *   new TestResponseConfig()
+     *     .userSends([userMsg])
+     *     .respondWith([agentMsg])
+     * );
+     * ```
+     */
+    addTestConfig(config) {
+        this.testConfigs.push(config);
+        return this;
+    }
+    /**
+     * Finds a matching test configuration for the given messages.
+     * Matching is based on message content, tools, and structured output schema.
+     *
+     * @param {ModelMessages[]} messages - Messages to match against
+     * @returns {TestResponseConfig | undefined} The matching config, or undefined if no match
+     *
+     * @private
+     */
+    findMatchingConfig(messages) {
+        const messagesSig = messages.map((message) => message.toJSON()).join("\n");
+        for (const config of this.testConfigs) {
+            const configMsgSig = config.inputMessages.map((message) => message.toJSON()).join("\n");
+            if (messagesSig !== configMsgSig)
+                continue;
+            if (config.structuredOutput) {
+                if (!this.structuredOutput)
+                    continue;
+                if (JSON.stringify(z.toJSONSchema(config.structuredOutput)) ===
+                    JSON.stringify(z.toJSONSchema(this.structuredOutput))) {
+                    return config;
+                }
+            }
+            else {
+                const configToolNames = config.toolsIncluded.map((tool) => tool.name);
+                const modelToolNames = this.tools.map((tool) => tool.name);
+                if (configToolNames.every((name) => modelToolNames.includes(name))) {
+                    return config;
+                }
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Executes the test model by finding and returning a matching configuration.
+     *
+     * @protected
+     * @param {ModelMessages[]} messages - Messages to respond to
+     * @returns {Promise<InvokeResponse>} The configured response
+     * @throws {Error} If no matching test configuration is found
+     */
+    runModel(messages) {
+        const matchingConfig = this.findMatchingConfig(messages);
+        if (!matchingConfig) {
+            throw new Error("No matching test config found");
+        }
+        return Promise.resolve({
+            messages: matchingConfig.responseMessages,
+            usage: {
+                inputTokens: 0,
+                outputTokens: 0,
+            },
+            stopReason: BaseModel_1.InvokeResponseStopReason.UNKNOWN,
+        });
+    }
+}
+exports.TestModel = TestModel;
+//# sourceMappingURL=TestModel.js.map

package/dist/models/TestModel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestModel.js","sourceRoot":"","sources":["../../src/models/TestModel.ts"],"names":[],"mappings":";;;AACA,2CAKqB;AACrB,yBAAyB;AAGzB;;;;;;;;;;;;GAYG;AACH,MAAa,kBAAkB;IAC3B;;OAEG;IACH,aAAa,GAAqB,EAAE,CAAC;IAErC;;OAEG;IACH,gBAAgB,GACqB,EAAE,CAAC;IAExC;;OAEG;IACH,aAAa,GAAoB,EAAE,CAAC;IAEpC;;OAEG;IACH,gBAAgB,CAAoB;IAEpC;;;;;OAKG;IACH,WAAW,CACP,QAAwC;QAExC,IAAI,CAAC,gBAAgB,GAAG,QAAQ,CAAC;QACjC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,aAAa,CAAC,KAAuB;QACjC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,wBAAwB,CAAC,MAAwB;QAC7C,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAC/B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,SAAS,CAAC,QAAyB;QAC/B,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC;QAC9B,OAAO,IAAI,CAAC;IAChB,CAAC;CACJ;AAnED,gDAmEC;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAa,SAAU,SAAQ,qBAAS;IACpC;;OAEG;IACH,WAAW,GAAyB,EAAE,CAAC;IAEvC;;;;OAIG;IACH,YAAY,MAAuB;QAC/B,KAAK,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,aAAa,CAAC,MAA0B;QACpC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;;;;OAQG;IACH,kBAAkB,CACd,QAAyB;QAEzB,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAChE,IAAI,CACP,CAAC;QACF,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CACtD,OAAO,CAAC,MAAM,EAAE,CACnB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,IAAI,WAAW,KAAK,YAAY;gBAAE,SAAS;YAC3C,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;gBAC1B,IAAI,CAAC,IAAI,CAAC,gBAAgB;oBAAE,SAAS;gBACrC,IACI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;oBACvD,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,EACvD,CAAC;oBACC,OAAO,MAAM,CAAC;gBAClB,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,MAAM,eAAe,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACtD,IAAI,CAAC,IAAI,CACZ,CAAC;gBACF,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC3D,IACI,eAAe,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAC3B,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAChC,EACH,CAAC;oBACC,OAAO,MAAM,CAAC;gBAClB,CAAC;YACL,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;;;;;;OAOG;IACO,QAAQ,CACd,QAAyB;QAEzB,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,cAAc,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC;YACnB,QAAQ,EAAE,cAAc,CAAC,gBAAgB;YACzC,KAAK,EAAE;gBACH,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;aAClB;YACD,UAAU,EAAE,oCAAwB,CAAC,OAAO;SAC/C,CAAC,CAAC;IACP,CAAC;CACJ;AAxGD,8BAwGC"}