@ellyco/agentic 0.1.0

package/dist/models/BaseModel.d.ts

@@ -0,0 +1,232 @@
+import * as z from "zod";
+import { AgentMessage, MessageContent, ModelMessages, SystemMessage, ToolRequest } from "../messages";
+import { ToolDefinition } from "../tools";
+/**
+ * Configuration options for model initialization.
+ *
+ * @interface BaseModelConfig
+ * @property {number} [temperature] - Controls randomness/creativity (0-1). Lower = more deterministic
+ * @property {number} [topP] - Nucleus sampling parameter (0-1). Controls diversity of output
+ * @property {number} [maxTokens] - Maximum number of tokens in the response
+ */
+export interface BaseModelConfig {
+    temperature?: number;
+    topP?: number;
+    maxTokens?: number;
+}
+/**
+ * Token usage information for a model invocation.
+ *
+ * @interface InvokeResponseUsage
+ * @property {number} inputTokens - Number of tokens in the input/prompt
+ * @property {number} outputTokens - Number of tokens in the output/response
+ */
+export interface InvokeResponseUsage {
+    inputTokens: number;
+    outputTokens: number;
+}
+/**
+ * Enum representing the reason why model generation stopped.
+ *
+ * @enum {string}
+ * @property {string} UNKNOWN - Stop reason is not determined
+ * @property {string} STOP_SEQUENCE - Generation ended at a stop sequence
+ * @property {string} MAX_TOKENS - Generation reached maximum token limit
+ * @property {string} END_TURN - Model indicated end of turn
+ * @property {string} TOOL_USE - Generation stopped to request tool use
+ */
+export declare enum InvokeResponseStopReason {
+    UNKNOWN = "unknown",
+    STOP_SEQUENCE = "stop_sequence",
+    MAX_TOKENS = "max_tokens",
+    END_TURN = "end_turn",
+    TOOL_USE = "tool_use"
+}
+/**
+ * Response from invoking a model.
+ *
+ * @interface InvokeResponse
+ * @property {(AgentMessage | ToolRequest)[]} messages - Messages generated by the model
+ * @property {InvokeResponseUsage} usage - Token usage statistics
+ * @property {InvokeResponseStopReason} [stopReason] - Reason why generation stopped
+ */
+export interface InvokeResponse {
+    messages: (AgentMessage | ToolRequest)[];
+    usage: InvokeResponseUsage;
+    stopReason?: InvokeResponseStopReason;
+}
+/**
+ * Abstract base class for AI model implementations.
+ * Provides common functionality for configuring and invoking models.
+ * Subclasses must implement runModel() for specific model providers.
+ *
+ * @abstract
+ * @example
+ * ```typescript
+ * class MyCustomModel extends BaseModel {
+ *   protected async runModel(messages: ModelMessages[]): Promise<InvokeResponse> {
+ *     // Implementation specific to your model provider
+ *   }
+ * }
+ *
+ * const model = new MyCustomModel({ temperature: 0.7 })
+ *   .withSystemMessage("You are a helpful assistant")
+ *   .withTools([searchTool, calculatorTool]);
+ *
+ * const response = await model.invoke([userMessage]);
+ * ```
+ */
+export declare abstract class BaseModel {
+    protected temperature?: number;
+    protected topP?: number;
+    protected maxTokens?: number;
+    protected systemMessage?: SystemMessage;
+    protected tools: ToolDefinition[];
+    protected structuredOutput?: z.ZodSchema<any>;
+    /**
+     * Creates a new model instance with the provided configuration.
+     * Configuration values are clamped to valid ranges.
+     *
+     * @param {BaseModelConfig} config - Model configuration
+     */
+    constructor(config: BaseModelConfig);
+    /**
+     * Sets the system message for the model.
+     * The system message provides context and instructions for the model's behavior.
+     *
+     * @param {SystemMessage | MessageContent} message - The system message
+     * @returns {this} The model instance for method chaining
+     *
+     * @example
+     * ```typescript
+     * model.withSystemMessage("You are an expert Python developer");
+     * ```
+     */
+    withSystemMessage(message: SystemMessage | MessageContent): this;
+    /**
+     * Sets the tools available to the model.
+     * Cannot be combined with structured output.
+     *
+     * @param {ToolDefinition[]} tools - Array of tool definitions
+     * @returns {this} The model instance for method chaining
+     * @throws {Error} If structured output is already configured
+     *
+     * @example
+     * ```typescript
+     * model.withTools([webSearchTool, calculatorTool]);
+     * ```
+     */
+    withTools(tools: ToolDefinition[]): this;
+    /**
+     * Configures structured output using a Zod schema.
+     * The model will be forced to output data matching this schema.
+     * Cannot be combined with tools.
+     *
+     * @template T - The Zod schema type
+     * @param {T} schema - Zod schema defining the output structure
+     * @returns {StructuredOutputWrapper<T>} A wrapper for invoking with structured output
+     * @throws {Error} If tools are already configured
+     *
+     * @example
+     * ```typescript
+     * const schema = z.object({
+     *   sentiment: z.enum(["positive", "negative", "neutral"]),
+     *   confidence: z.number().min(0).max(1)
+     * });
+     * const wrapper = model.withStructuredOutput(schema);
+     * const result = await wrapper.invoke(messages);
+     * ```
+     */
+    withStructuredOutput<T extends z.ZodObject>(schema: T): StructuredOutputWrapper<T>;
+    /**
+     * Internal method to run the model implementation.
+     * Must be implemented by subclasses.
+     *
+     * @protected
+     * @abstract
+     * @param {ModelMessages[]} messages - Messages to send to the model
+     * @returns {Promise<InvokeResponse>} The model's response
+     */
+    protected abstract runModel(messages: ModelMessages[]): Promise<InvokeResponse>;
+    /**
+     * Invokes the model with messages and optional template variable substitution.
+     * Messages can contain template variables like {name} that will be replaced
+     * with values from the properties object.
+     *
+     * @param {ModelMessages[]} messages - Messages to send to the model
+     * @param {Record<string, any>} [properties] - Optional property values for template substitution
+     * @returns {Promise<InvokeResponse>} The model's response
+     * @throws {Error} If a template variable is not found in properties
+     *
+     * @example
+     * ```typescript
+     * const response = await model.invoke(
+     *   [new UserMessage("Hello, my name is {name}")],
+     *   { name: "Alice" }
+     * );
+     * ```
+     */
+    invoke(messages: ModelMessages[], properties?: Record<string, any>): Promise<InvokeResponse>;
+}
+/**
+ * Error thrown when response is not structured output as expected.
+ *
+ * @class ResponseNotStructuredOutputError
+ * @extends {Error}
+ */
+export declare class ResponseNotStructuredOutputError extends Error {
+    /**
+     * Creates a new error instance.
+     */
+    constructor();
+}
+/**
+ * Wrapper class for models configured with structured output.
+ * Ensures responses conform to the specified Zod schema.
+ *
+ * @template T - The Zod schema type
+ *
+ * @example
+ * ```typescript
+ * const schema = z.object({ message: z.string(), score: z.number() });
+ * const wrapper = model.withStructuredOutput(schema);
+ * const parsed = await wrapper.invoke(messages);
+ * // parsed is guaranteed to match the schema type
+ * ```
+ */
+export declare class StructuredOutputWrapper<T extends z.ZodObject> {
+    private readonly model;
+    private readonly schema;
+    /**
+     * Creates a wrapper for structured output.
+     *
+     * @param {BaseModel} model - The model instance
+     * @param {T} schema - The output schema
+     */
+    constructor(model: BaseModel, schema: T);
+    /**
+     * Sets the system message for this model instance.
+     *
+     * @param {SystemMessage | MessageContent} message - The system message
+     * @returns {this} The wrapper instance for method chaining
+     */
+    withSystemMessage(message: SystemMessage | MessageContent): this;
+    /**
+     * Invokes the model and parses the response using the configured schema.
+     * The model is forced to call an "output" tool containing the structured data.
+     *
+     * @param {ModelMessages[]} messages - Messages to send to the model
+     * @param {Record<string, any>} [properties] - Optional template variables
+     * @returns {Promise<z.infer<T>>} The parsed response matching the schema
+     * @throws {ResponseNotStructuredOutputError} If response doesn't match expected format
+     *
+     * @example
+     * ```typescript
+     * const result = await wrapper.invoke([userMessage]);
+     * console.log(result.message); // TypeScript knows this is a string
+     * console.log(result.score);   // TypeScript knows this is a number
+     * ```
+     */
+    invoke(messages: ModelMessages[], properties?: Record<string, any>): Promise<z.infer<T>>;
+}
+//# sourceMappingURL=BaseModel.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"BaseModel.d.ts","sourceRoot":"","sources":["../../src/models/BaseModel.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AACzB,OAAO,EACH,YAAY,EAEZ,cAAc,EACd,aAAa,EACb,aAAa,EACb,WAAW,EACd,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C;;;;;;;GAOG;AACH,MAAM,WAAW,eAAe;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;;;;GAMG;AACH,MAAM,WAAW,mBAAmB;IAChC,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;;;;GASG;AACH,oBAAY,wBAAwB;IAChC,OAAO,YAAY;IACnB,aAAa,kBAAkB;IAC/B,UAAU,eAAe;IACzB,QAAQ,aAAa;IACrB,QAAQ,aAAa;CACxB;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,cAAc;IAC3B,QAAQ,EAAE,CAAC,YAAY,GAAG,WAAW,CAAC,EAAE,CAAC;IACzC,KAAK,EAAE,mBAAmB,CAAC;IAC3B,UAAU,CAAC,EAAE,wBAAwB,CAAC;CACzC;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,8BAAsB,SAAS;IAC3B,SAAS,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC/B,SAAS,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAE7B,SAAS,CAAC,aAAa,CAAC,EAAE,aAAa,CAAC;IACxC,SAAS,CAAC,KAAK,EAAE,cAAc,EAAE,CAAM;IACvC,SAAS,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAE9C;;;;;OAKG;gBACS,MAAM,EAAE,eAAe;IAYnC;;;;;;;;;;;OAWG;IACH,iBAAiB,CAAC,OAAO,EAAE,aAAa,GAAG,cAAc,GAAG,IAAI;IAShE;;;;;;;;;;;;OAYG;IACH,SAAS,CAAC,KAAK,EAAE,cAAc,EAAE,GAAG,IAAI;IAQxC;;;;;;;;;;;;;;;;;;;OAmBG;IACH,oBAAoB,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS,EACtC,MAAM,EAAE,CAAC,GACV,uBAAuB,CAAC,CAAC,CAAC;IAM7B;;;;;;;;OAQG;IACH,SAAS,CAAC,QAAQ,CAAC,QAAQ,CACvB,QAAQ,EAAE,aAAa,EAAE,GAC1B,OAAO,CAAC,cAAc,CAAC;IAE1B;;;;;;;;;;;;;;;;;OAiBG;IACG,MAAM,CACR,QAAQ,EAAE,aAAa,EAAE,EACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GACjC,OAAO,CAAC,cAAc,CAAC;CAW7B;AAED;;;;;GAKG;AACH,qBAAa,gCAAiC,SAAQ,KAAK;IACvD;;OAEG;;CAKN;AAED;;;;;;;;;;;;;GAaG;AACH,qBAAa,uBAAuB,CAAC,CAAC,SAAS,CAAC,CAAC,SAAS;IAO1C,OAAO,CAAC,QAAQ,CAAC,KAAK;IAAa,OAAO,CAAC,QAAQ,CAAC,MAAM;IANtE;;;;;OAKG;gBAC0B,KAAK,EAAE,SAAS,EAAmB,MAAM,EAAE,CAAC;IAEzE;;;;;OAKG;IACH,iBAAiB,CAAC,OAAO,EAAE,aAAa,GAAG,cAAc,GAAG,IAAI;IAKhE;;;;;;;;;;;;;;;OAeG;IACG,MAAM,CACR,QAAQ,EAAE,aAAa,EAAE,EACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GACjC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;CAWzB"}

package/dist/models/BaseModel.js

@@ -0,0 +1,247 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StructuredOutputWrapper = exports.ResponseNotStructuredOutputError = exports.BaseModel = exports.InvokeResponseStopReason = void 0;
+const messages_1 = require("../messages");
+/**
+ * Enum representing the reason why model generation stopped.
+ *
+ * @enum {string}
+ * @property {string} UNKNOWN - Stop reason is not determined
+ * @property {string} STOP_SEQUENCE - Generation ended at a stop sequence
+ * @property {string} MAX_TOKENS - Generation reached maximum token limit
+ * @property {string} END_TURN - Model indicated end of turn
+ * @property {string} TOOL_USE - Generation stopped to request tool use
+ */
+var InvokeResponseStopReason;
+(function (InvokeResponseStopReason) {
+    InvokeResponseStopReason["UNKNOWN"] = "unknown";
+    InvokeResponseStopReason["STOP_SEQUENCE"] = "stop_sequence";
+    InvokeResponseStopReason["MAX_TOKENS"] = "max_tokens";
+    InvokeResponseStopReason["END_TURN"] = "end_turn";
+    InvokeResponseStopReason["TOOL_USE"] = "tool_use";
+})(InvokeResponseStopReason || (exports.InvokeResponseStopReason = InvokeResponseStopReason = {}));
+/**
+ * Abstract base class for AI model implementations.
+ * Provides common functionality for configuring and invoking models.
+ * Subclasses must implement runModel() for specific model providers.
+ *
+ * @abstract
+ * @example
+ * ```typescript
+ * class MyCustomModel extends BaseModel {
+ *   protected async runModel(messages: ModelMessages[]): Promise<InvokeResponse> {
+ *     // Implementation specific to your model provider
+ *   }
+ * }
+ *
+ * const model = new MyCustomModel({ temperature: 0.7 })
+ *   .withSystemMessage("You are a helpful assistant")
+ *   .withTools([searchTool, calculatorTool]);
+ *
+ * const response = await model.invoke([userMessage]);
+ * ```
+ */
+class BaseModel {
+    temperature;
+    topP;
+    maxTokens;
+    systemMessage;
+    tools = [];
+    structuredOutput;
+    /**
+     * Creates a new model instance with the provided configuration.
+     * Configuration values are clamped to valid ranges.
+     *
+     * @param {BaseModelConfig} config - Model configuration
+     */
+    constructor(config) {
+        if (config.temperature) {
+            this.temperature = Math.max(0, Math.min(1, config.temperature));
+        }
+        if (config.topP) {
+            this.topP = Math.max(0, Math.min(1, config.topP));
+        }
+        if (config.maxTokens) {
+            this.maxTokens = Math.max(0, config.maxTokens);
+        }
+    }
+    /**
+     * Sets the system message for the model.
+     * The system message provides context and instructions for the model's behavior.
+     *
+     * @param {SystemMessage | MessageContent} message - The system message
+     * @returns {this} The model instance for method chaining
+     *
+     * @example
+     * ```typescript
+     * model.withSystemMessage("You are an expert Python developer");
+     * ```
+     */
+    withSystemMessage(message) {
+        if (message instanceof messages_1.SystemMessage) {
+            this.systemMessage = message;
+        }
+        else {
+            this.systemMessage = new messages_1.SystemMessage(message);
+        }
+        return this;
+    }
+    /**
+     * Sets the tools available to the model.
+     * Cannot be combined with structured output.
+     *
+     * @param {ToolDefinition[]} tools - Array of tool definitions
+     * @returns {this} The model instance for method chaining
+     * @throws {Error} If structured output is already configured
+     *
+     * @example
+     * ```typescript
+     * model.withTools([webSearchTool, calculatorTool]);
+     * ```
+     */
+    withTools(tools) {
+        if (this.structuredOutput) {
+            throw new Error("Cannot set tools with structured output");
+        }
+        this.tools = tools;
+        return this;
+    }
+    /**
+     * Configures structured output using a Zod schema.
+     * The model will be forced to output data matching this schema.
+     * Cannot be combined with tools.
+     *
+     * @template T - The Zod schema type
+     * @param {T} schema - Zod schema defining the output structure
+     * @returns {StructuredOutputWrapper<T>} A wrapper for invoking with structured output
+     * @throws {Error} If tools are already configured
+     *
+     * @example
+     * ```typescript
+     * const schema = z.object({
+     *   sentiment: z.enum(["positive", "negative", "neutral"]),
+     *   confidence: z.number().min(0).max(1)
+     * });
+     * const wrapper = model.withStructuredOutput(schema);
+     * const result = await wrapper.invoke(messages);
+     * ```
+     */
+    withStructuredOutput(schema) {
+        this.tools = [];
+        this.structuredOutput = schema;
+        return new StructuredOutputWrapper(this, schema);
+    }
+    /**
+     * Invokes the model with messages and optional template variable substitution.
+     * Messages can contain template variables like {name} that will be replaced
+     * with values from the properties object.
+     *
+     * @param {ModelMessages[]} messages - Messages to send to the model
+     * @param {Record<string, any>} [properties] - Optional property values for template substitution
+     * @returns {Promise<InvokeResponse>} The model's response
+     * @throws {Error} If a template variable is not found in properties
+     *
+     * @example
+     * ```typescript
+     * const response = await model.invoke(
+     *   [new UserMessage("Hello, my name is {name}")],
+     *   { name: "Alice" }
+     * );
+     * ```
+     */
+    async invoke(messages, properties) {
+        if (!properties) {
+            return this.runModel(messages);
+        }
+        return this.runModel(messages.map((message) => {
+            if (message instanceof messages_1.BaseMessage) {
+                message.interpolate(properties);
+            }
+            return message;
+        }));
+    }
+}
+exports.BaseModel = BaseModel;
+/**
+ * Error thrown when response is not structured output as expected.
+ *
+ * @class ResponseNotStructuredOutputError
+ * @extends {Error}
+ */
+class ResponseNotStructuredOutputError extends Error {
+    /**
+     * Creates a new error instance.
+     */
+    constructor() {
+        super("Response is not structured output");
+        this.name = "ResponseNotStructuredOutputError";
+    }
+}
+exports.ResponseNotStructuredOutputError = ResponseNotStructuredOutputError;
+/**
+ * Wrapper class for models configured with structured output.
+ * Ensures responses conform to the specified Zod schema.
+ *
+ * @template T - The Zod schema type
+ *
+ * @example
+ * ```typescript
+ * const schema = z.object({ message: z.string(), score: z.number() });
+ * const wrapper = model.withStructuredOutput(schema);
+ * const parsed = await wrapper.invoke(messages);
+ * // parsed is guaranteed to match the schema type
+ * ```
+ */
+class StructuredOutputWrapper {
+    model;
+    schema;
+    /**
+     * Creates a wrapper for structured output.
+     *
+     * @param {BaseModel} model - The model instance
+     * @param {T} schema - The output schema
+     */
+    constructor(model, schema) {
+        this.model = model;
+        this.schema = schema;
+    }
+    /**
+     * Sets the system message for this model instance.
+     *
+     * @param {SystemMessage | MessageContent} message - The system message
+     * @returns {this} The wrapper instance for method chaining
+     */
+    withSystemMessage(message) {
+        this.model.withSystemMessage(message);
+        return this;
+    }
+    /**
+     * Invokes the model and parses the response using the configured schema.
+     * The model is forced to call an "output" tool containing the structured data.
+     *
+     * @param {ModelMessages[]} messages - Messages to send to the model
+     * @param {Record<string, any>} [properties] - Optional template variables
+     * @returns {Promise<z.infer<T>>} The parsed response matching the schema
+     * @throws {ResponseNotStructuredOutputError} If response doesn't match expected format
+     *
+     * @example
+     * ```typescript
+     * const result = await wrapper.invoke([userMessage]);
+     * console.log(result.message); // TypeScript knows this is a string
+     * console.log(result.score);   // TypeScript knows this is a number
+     * ```
+     */
+    async invoke(messages, properties) {
+        const response = await this.model.invoke(messages, properties);
+        if (!response.messages.length) {
+            throw new ResponseNotStructuredOutputError();
+        }
+        if (!(response.messages[0] instanceof messages_1.ToolRequest)) {
+            throw new ResponseNotStructuredOutputError();
+        }
+        const toolRequest = response.messages[0];
+        return this.schema.parse(toolRequest.input);
+    }
+}
+exports.StructuredOutputWrapper = StructuredOutputWrapper;
+//# sourceMappingURL=BaseModel.js.map

package/dist/models/BaseModel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseModel.js","sourceRoot":"","sources":["../../src/models/BaseModel.ts"],"names":[],"mappings":";;;AACA,0CAOqB;AA6BrB;;;;;;;;;GASG;AACH,IAAY,wBAMX;AAND,WAAY,wBAAwB;IAChC,+CAAmB,CAAA;IACnB,2DAA+B,CAAA;IAC/B,qDAAyB,CAAA;IACzB,iDAAqB,CAAA;IACrB,iDAAqB,CAAA;AACzB,CAAC,EANW,wBAAwB,wCAAxB,wBAAwB,QAMnC;AAgBD;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAsB,SAAS;IACjB,WAAW,CAAU;IACrB,IAAI,CAAU;IACd,SAAS,CAAU;IAEnB,aAAa,CAAiB;IAC9B,KAAK,GAAqB,EAAE,CAAC;IAC7B,gBAAgB,CAAoB;IAE9C;;;;;OAKG;IACH,YAAY,MAAuB;QAC/B,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACd,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QACnD,CAAC;IACL,CAAC;IAED;;;;;;;;;;;OAWG;IACH,iBAAiB,CAAC,OAAuC;QACrD,IAAI,OAAO,YAAY,wBAAa,EAAE,CAAC;YACnC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QACjC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,aAAa,GAAG,IAAI,wBAAa,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,SAAS,CAAC,KAAuB;QAC7B,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;;;;;;;;;;;;;;;OAmBG;IACH,oBAAoB,CAChB,MAAS;QAET,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAC/B,OAAO,IAAI,uBAAuB,CAAI,IAAI,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC;IAeD;;;;;;;;;;;;;;;;;OAiBG;IACH,KAAK,CAAC,MAAM,CACR,QAAyB,EACzB,UAAgC;QAEhC,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YAC1C,IAAI,OAAO,YAAY,sBAAW,EAAE,CAAC;gBACjC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACpC,CAAC;YACD,OAAO,OAAO,CAAC;QACnB,CAAC,CAAC,CAAC,CAAC;IACR,CAAC;CACJ;AA9ID,8BA8IC;AAED;;;;;GAKG;AACH,MAAa,gCAAiC,SAAQ,KAAK;IACvD;;OAEG;IACH;QACI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,GAAG,kCAAkC,CAAC;IACnD,CAAC;CACJ;AARD,4EAQC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAa,uBAAuB;IAOH;IAAmC;IANhE;;;;;OAKG;IACH,YAA6B,KAAgB,EAAmB,MAAS;QAA5C,UAAK,GAAL,KAAK,CAAW;QAAmB,WAAM,GAAN,MAAM,CAAG;IAAI,CAAC;IAE9E;;;;;OAKG;IACH,iBAAiB,CAAC,OAAuC;QACrD,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACtC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;;;;;;;;;;;OAeG;IACH,KAAK,CAAC,MAAM,CACR,QAAyB,EACzB,UAAgC;QAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC5B,MAAM,IAAI,gCAAgC,EAAE,CAAC;QACjD,CAAC;QACD,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,sBAAW,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,gCAAgC,EAAE,CAAC;QACjD,CAAC;QACD,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAgB,CAAC;QACxD,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;CACJ;AAlDD,0DAkDC"}

package/dist/models/BedrockModel.d.ts

@@ -0,0 +1,112 @@
+import { BaseModel, type BaseModelConfig, InvokeResponse } from "./BaseModel";
+import { BedrockRuntimeClientConfig } from "@aws-sdk/client-bedrock-runtime";
+import { ModelMessages } from "../messages";
+/**
+ * Configuration for BedrockModel combining BaseModel config with AWS-specific settings.
+ *
+ * @interface BedrockModelConfig
+ * @extends {BaseModelConfig}
+ * @property {string} modelId - The ID of the Bedrock model (e.g., "anthropic.claude-3-sonnet-20240229-v1:0")
+ * @property {BedrockRuntimeClientConfig} [aws] - Optional AWS SDK configuration
+ */
+export type BedrockModelConfig = BaseModelConfig & {
+    modelId: string;
+    aws?: BedrockRuntimeClientConfig;
+};
+/**
+ * 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]);
+ * ```
+ */
+export declare class BedrockModel extends BaseModel {
+    private modelId;
+    private client;
+    /**
+     * Creates a new Bedrock model instance.
+     *
+     * @param {BedrockModelConfig} config - Configuration including model ID and AWS settings
+     */
+    constructor(config: BedrockModelConfig);
+    /**
+     * 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
+     */
+    private convertMessagesToBedrockMessages;
+    /**
+     * Converts a tool definition to Bedrock tool specification format.
+     *
+     * @private
+     * @param {ToolDefinition} tool - The tool definition to convert
+     * @returns {Tool} Bedrock formatted tool specification
+     */
+    private convertToolDefinitionToBedrockTool;
+    /**
+     * 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
+     */
+    private getBedrockToolConfig;
+    /**
+     * Maps Bedrock stop reason to internal stop reason enum.
+     *
+     * @private
+     * @param {ConverseCommandOutput} response - The Bedrock API response
+     * @returns {InvokeResponseStopReason} Mapped stop reason
+     */
+    private mapStopReason;
+    /**
+     * Builds inference configuration chunk with temperature, topP, and maxTokens if set.
+     *
+     * @private
+     * @returns {{inferenceConfig?: InferenceConfiguration}} Configuration chunk or empty object
+     */
+    private getInferenceConfigChunk;
+    /**
+     * 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
+     */
+    private getSystemMessageChunk;
+    /**
+     * Builds tool configuration chunk.
+     *
+     * @private
+     * @returns {{toolConfig?: ToolConfiguration}} Tool configuration or empty object
+     */
+    private getToolConfigChunk;
+    /**
+     * 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
+     */
+    protected runModel(inputMessages: ModelMessages[]): Promise<InvokeResponse>;
+}
+//# sourceMappingURL=BedrockModel.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"BedrockModel.d.ts","sourceRoot":"","sources":["../../src/models/BedrockModel.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,SAAS,EACT,KAAK,eAAe,EACpB,cAAc,EAGjB,MAAM,aAAa,CAAC;AACrB,OAAO,EAEH,0BAA0B,EAY7B,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAIH,aAAa,EAKhB,MAAM,aAAa,CAAC;AAIrB;;;;;;;GAOG;AACH,MAAM,MAAM,kBAAkB,GAAG,eAAe,GAAG;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,0BAA0B,CAAC;CACpC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,qBAAa,YAAa,SAAQ,SAAS;IACvC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAAuB;IAErC;;;;OAIG;gBACS,MAAM,EAAE,kBAAkB;IAMtC;;;;;;;OAOG;IACH,OAAO,CAAC,gCAAgC;IA+ExC;;;;;;OAMG;IACH,OAAO,CAAC,kCAAkC;IAY1C;;;;;;;OAOG;IACH,OAAO,CAAC,oBAAoB;IA4B5B;;;;;;OAMG;IACH,OAAO,CAAC,aAAa;IAiBrB;;;;;OAKG;IACH,OAAO,CAAC,uBAAuB;IAc/B;;;;;;OAMG;IACH,OAAO,CAAC,qBAAqB;IA8B7B;;;;;OAKG;IACH,OAAO,CAAC,kBAAkB;IAU1B;;;;;;;OAOG;cACa,QAAQ,CACpB,aAAa,EAAE,aAAa,EAAE,GAC/B,OAAO,CAAC,cAAc,CAAC;CAuD7B"}