@ax-llm/ax-ai-aws-bedrock 14.0.32

package/index.cjs

@@ -0,0 +1,492 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// index.ts
+var index_exports = {};
+__export(index_exports, {
+  AxAIBedrock: () => AxAIBedrock,
+  AxAIBedrockEmbedModel: () => AxAIBedrockEmbedModel,
+  AxAIBedrockModel: () => AxAIBedrockModel,
+  axModelInfoBedrock: () => axModelInfoBedrock
+});
+module.exports = __toCommonJS(index_exports);
+
+// api.ts
+var import_client_bedrock_runtime = require("@aws-sdk/client-bedrock-runtime");
+var import_ax = require("@ax-llm/ax");
+
+// types.ts
+var AxAIBedrockModel = /* @__PURE__ */ ((AxAIBedrockModel2) => {
+  AxAIBedrockModel2["ClaudeSonnet4"] = "us.anthropic.claude-sonnet-4-20250514-v1:0";
+  AxAIBedrockModel2["Claude37Sonnet"] = "anthropic.claude-3-7-sonnet-20250219-v1:0";
+  AxAIBedrockModel2["Claude35Sonnet"] = "anthropic.claude-3-5-sonnet-20240620-v1:0";
+  AxAIBedrockModel2["GptOss120B"] = "openai.gpt-oss-120b-1:0";
+  AxAIBedrockModel2["GptOss20B"] = "openai.gpt-oss-20b-1:0";
+  return AxAIBedrockModel2;
+})(AxAIBedrockModel || {});
+var AxAIBedrockEmbedModel = /* @__PURE__ */ ((AxAIBedrockEmbedModel2) => {
+  AxAIBedrockEmbedModel2["TitanEmbedV2"] = "amazon.titan-embed-text-v2:0";
+  return AxAIBedrockEmbedModel2;
+})(AxAIBedrockEmbedModel || {});
+
+// info.ts
+var axModelInfoBedrock = [
+  // ========================================================================
+  // Claude Models
+  // ========================================================================
+  {
+    name: "us.anthropic.claude-sonnet-4-20250514-v1:0" /* ClaudeSonnet4 */,
+    currency: "usd",
+    promptTokenCostPer1M: 3,
+    completionTokenCostPer1M: 15,
+    maxTokens: 64e3,
+    contextWindow: 2e5,
+    supported: { thinkingBudget: true, showThoughts: true }
+  },
+  {
+    name: "anthropic.claude-3-7-sonnet-20250219-v1:0" /* Claude37Sonnet */,
+    currency: "usd",
+    promptTokenCostPer1M: 3,
+    completionTokenCostPer1M: 15,
+    maxTokens: 64e3,
+    contextWindow: 2e5
+  },
+  {
+    name: "anthropic.claude-3-5-sonnet-20240620-v1:0" /* Claude35Sonnet */,
+    currency: "usd",
+    promptTokenCostPer1M: 3,
+    completionTokenCostPer1M: 15,
+    maxTokens: 8192,
+    contextWindow: 2e5
+  },
+  // ========================================================================
+  // GPT OSS Models
+  // ========================================================================
+  {
+    name: "openai.gpt-oss-120b-1:0" /* GptOss120B */,
+    currency: "usd",
+    promptTokenCostPer1M: 0.5,
+    completionTokenCostPer1M: 1.5,
+    maxTokens: 16384,
+    contextWindow: 128e3
+  },
+  {
+    name: "openai.gpt-oss-20b-1:0" /* GptOss20B */,
+    currency: "usd",
+    promptTokenCostPer1M: 0.25,
+    completionTokenCostPer1M: 0.75,
+    maxTokens: 16384,
+    contextWindow: 128e3
+  },
+  // ========================================================================
+  // Embed Models
+  // ========================================================================
+  {
+    name: "amazon.titan-embed-text-v2:0" /* TitanEmbedV2 */,
+    currency: "usd",
+    promptTokenCostPer1M: 0.02,
+    completionTokenCostPer1M: 0,
+    maxTokens: 8192,
+    contextWindow: 8192
+  }
+];
+
+// api.ts
+var AxAIBedrockImpl = class {
+  constructor(config, primaryRegion, fallbackRegions, gptRegion, gptFallbackRegions) {
+    this.config = config;
+    this.primaryRegion = primaryRegion;
+    this.fallbackRegions = fallbackRegions;
+    this.gptRegion = gptRegion;
+    this.gptFallbackRegions = gptFallbackRegions;
+  }
+  clients = /* @__PURE__ */ new Map();
+  tokensUsed;
+  getTokenUsage() {
+    return this.tokensUsed;
+  }
+  getModelConfig() {
+    return {
+      maxTokens: this.config.maxTokens,
+      temperature: this.config.temperature,
+      topP: this.config.topP,
+      stopSequences: this.config.stopSequences
+    };
+  }
+  getClient(region) {
+    let client = this.clients.get(region);
+    if (!client) {
+      client = new import_client_bedrock_runtime.BedrockRuntimeClient({ region });
+      this.clients.set(region, client);
+    }
+    return client;
+  }
+  /**
+   * Detect model family from model ID
+   */
+  getModelFamily(modelId) {
+    if (modelId.includes("anthropic.claude") || modelId.includes("us.anthropic.claude")) {
+      return "claude";
+    }
+    if (modelId.includes("openai.gpt")) {
+      return "gpt";
+    }
+    if (modelId.includes("amazon.titan-embed")) {
+      return "titan";
+    }
+    throw new Error(`Unknown model family for: ${modelId}`);
+  }
+  /**
+   * Get appropriate regions for model
+   */
+  getRegionsForModel(modelId) {
+    const family = this.getModelFamily(modelId);
+    if (family === "gpt") {
+      return [this.gptRegion, ...this.gptFallbackRegions];
+    }
+    return [this.primaryRegion, ...this.fallbackRegions];
+  }
+  /**
+   * Regional failover logic - tries primary region, then fallbacks
+   */
+  async invokeWithFailover(modelId, handler) {
+    const regions = this.getRegionsForModel(modelId);
+    let lastError;
+    for (const region of regions) {
+      try {
+        const client = this.getClient(region);
+        return await handler(client);
+      } catch (error) {
+        lastError = error;
+        console.warn(
+          `[Bedrock] Region ${region} failed for ${modelId}:`,
+          error
+        );
+      }
+    }
+    throw lastError || new Error(`All Bedrock regions failed for ${modelId}`);
+  }
+  /**
+   * Transform AX chat request → Bedrock request (Claude or GPT)
+   */
+  createChatReq = async (req, _config) => {
+    const family = this.getModelFamily(req.model);
+    const maxTokens = req.modelConfig?.maxTokens ?? this.config.maxTokens ?? 4096;
+    const temperature = req.modelConfig?.temperature ?? this.config.temperature;
+    const topP = req.modelConfig?.topP ?? this.config.topP;
+    let bedrockRequest;
+    if (family === "claude") {
+      const systemMessages = req.chatPrompt.filter((msg) => msg.role === "system").map((msg) => msg.content).join("\n\n");
+      const messages = req.chatPrompt.filter((msg) => msg.role !== "system").map((msg) => {
+        if (msg.role === "user") {
+          return {
+            role: "user",
+            content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content)
+          };
+        }
+        if (msg.role === "assistant") {
+          return {
+            role: "assistant",
+            content: msg.content || ""
+          };
+        }
+        throw new Error(`Unsupported role: ${msg.role}`);
+      });
+      bedrockRequest = {
+        anthropic_version: "bedrock-2023-05-31",
+        max_tokens: maxTokens,
+        messages,
+        ...systemMessages ? { system: systemMessages } : {},
+        ...temperature !== void 0 ? { temperature } : {},
+        ...topP !== void 0 ? { top_p: topP } : {}
+      };
+    } else if (family === "gpt") {
+      const messages = req.chatPrompt.filter((msg) => msg.role !== "function").map((msg) => {
+        let content;
+        if ("content" in msg) {
+          content = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
+        } else {
+          content = "";
+        }
+        return {
+          role: msg.role,
+          content
+        };
+      });
+      bedrockRequest = {
+        messages,
+        max_tokens: maxTokens,
+        ...temperature !== void 0 ? { temperature } : {},
+        ...topP !== void 0 ? { top_p: topP } : {}
+      };
+    } else {
+      throw new Error(`Chat not supported for model family: ${family}`);
+    }
+    const apiConfig = {
+      name: `bedrock-${family}`,
+      localCall: async (data) => {
+        const reqBody = data;
+        const result = await this.invokeWithFailover(
+          req.model,
+          async (client) => {
+            const command = new import_client_bedrock_runtime.InvokeModelCommand({
+              modelId: req.model,
+              body: JSON.stringify(reqBody),
+              contentType: "application/json",
+              accept: "application/json"
+            });
+            const response = await client.send(command);
+            return JSON.parse(new TextDecoder().decode(response.body));
+          }
+        );
+        return result;
+      }
+    };
+    return [apiConfig, bedrockRequest];
+  };
+  /**
+   * Transform Bedrock response → AX chat response (Claude or GPT)
+   */
+  createChatResp(resp) {
+    if ("content" in resp && Array.isArray(resp.content)) {
+      return this.createClaudeChatResp(resp);
+    } else if ("choices" in resp && Array.isArray(resp.choices)) {
+      return this.createGptChatResp(resp);
+    }
+    throw new Error("Unknown response format");
+  }
+  /**
+   * Handle Claude-specific response format
+   */
+  createClaudeChatResp(resp) {
+    let content = "";
+    for (const block of resp.content) {
+      if (block.type === "text") {
+        content += block.text;
+      }
+    }
+    this.tokensUsed = {
+      promptTokens: resp.usage.input_tokens,
+      completionTokens: resp.usage.output_tokens,
+      totalTokens: resp.usage.input_tokens + resp.usage.output_tokens
+    };
+    let finishReason;
+    switch (resp.stop_reason) {
+      case "end_turn":
+      case "stop_sequence":
+        finishReason = "stop";
+        break;
+      case "max_tokens":
+        finishReason = "length";
+        break;
+      default:
+        finishReason = void 0;
+    }
+    return {
+      results: [
+        {
+          index: 0,
+          id: resp.id,
+          content,
+          finishReason
+        }
+      ],
+      remoteId: resp.id
+    };
+  }
+  /**
+   * Handle GPT-specific response format
+   */
+  createGptChatResp(resp) {
+    const choice = resp.choices[0];
+    if (!choice) {
+      throw new Error("No choices in GPT response");
+    }
+    let content = "";
+    if (typeof choice.message.content === "string") {
+      content = choice.message.content;
+    } else if (Array.isArray(choice.message.content)) {
+      content = choice.message.content.map((part) => {
+        if (typeof part === "string") return part;
+        return part.text || part.content || "";
+      }).join("");
+    }
+    if (resp.usage) {
+      this.tokensUsed = {
+        promptTokens: resp.usage.prompt_tokens,
+        completionTokens: resp.usage.completion_tokens,
+        totalTokens: resp.usage.total_tokens
+      };
+    }
+    let finishReason;
+    switch (choice.finish_reason) {
+      case "stop":
+        finishReason = "stop";
+        break;
+      case "length":
+        finishReason = "length";
+        break;
+      case "content_filter":
+        finishReason = "content_filter";
+        break;
+      default:
+        finishReason = void 0;
+    }
+    return {
+      results: [
+        {
+          index: choice.index,
+          id: resp.id,
+          content,
+          finishReason
+        }
+      ],
+      remoteId: resp.id
+    };
+  }
+  /**
+   * Create embed request for Titan
+   */
+  createEmbedReq = async (req) => {
+    if (!req.texts || req.texts.length === 0) {
+      throw new Error("No texts provided for embedding");
+    }
+    const embedRequest = {
+      inputText: req.texts[0],
+      // Take first text
+      dimensions: 512,
+      normalize: true
+    };
+    const apiConfig = {
+      name: "bedrock-titan-embed",
+      localCall: async (data) => {
+        const reqBody = data;
+        const result = await this.invokeWithFailover(
+          req.embedModel,
+          async (client) => {
+            const command = new import_client_bedrock_runtime.InvokeModelCommand({
+              modelId: req.embedModel,
+              body: JSON.stringify(reqBody),
+              contentType: "application/json",
+              accept: "application/json"
+            });
+            const response = await client.send(command);
+            return JSON.parse(new TextDecoder().decode(response.body));
+          }
+        );
+        return result;
+      }
+    };
+    return [apiConfig, embedRequest];
+  };
+  /**
+   * Create embed response from Titan
+   */
+  createEmbedResp(resp) {
+    return {
+      embeddings: [resp.embedding]
+    };
+  }
+};
+var AxAIBedrock = class extends import_ax.AxBaseAI {
+  constructor({
+    region = "us-east-2",
+    fallbackRegions = ["us-west-2", "us-east-1"],
+    gptRegion = "us-west-2",
+    gptFallbackRegions = ["us-east-1"],
+    config,
+    options
+  }) {
+    const fullConfig = {
+      ...(0, import_ax.axBaseAIDefaultConfig)(),
+      model: "us.anthropic.claude-sonnet-4-20250514-v1:0" /* ClaudeSonnet4 */,
+      region,
+      fallbackRegions,
+      gptRegion,
+      gptFallbackRegions,
+      ...config
+    };
+    const aiImpl = new AxAIBedrockImpl(
+      fullConfig,
+      region,
+      fallbackRegions,
+      gptRegion,
+      gptFallbackRegions
+    );
+    const supportFor = () => ({
+      functions: false,
+      // Not implemented yet - add when needed
+      streaming: false,
+      // Not implemented yet - add when needed
+      functionCot: false,
+      hasThinkingBudget: false,
+      hasShowThoughts: false,
+      media: {
+        images: {
+          supported: false,
+          // Add when needed
+          formats: []
+        },
+        audio: {
+          supported: false,
+          formats: []
+        },
+        files: {
+          supported: false,
+          formats: [],
+          uploadMethod: "none"
+        },
+        urls: {
+          supported: false,
+          webSearch: false,
+          contextFetching: false
+        }
+      },
+      caching: {
+        supported: false,
+        types: []
+      },
+      thinking: false,
+      multiTurn: true
+      // All models support multi-turn conversations
+    });
+    super(aiImpl, {
+      name: "Bedrock",
+      apiURL: "",
+      // Not used - we use SDK directly
+      headers: async () => ({}),
+      // AWS SDK handles auth
+      modelInfo: axModelInfoBedrock,
+      defaults: {
+        model: fullConfig.model,
+        embedModel: fullConfig.embedModel
+      },
+      options,
+      supportFor
+    });
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AxAIBedrock,
+  AxAIBedrockEmbedModel,
+  AxAIBedrockModel,
+  axModelInfoBedrock
+});
+//# sourceMappingURL=index.cjs.map

package/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../index.ts","../api.ts","../types.ts","../info.ts"],"sourcesContent":["/**\n * AWS Bedrock Provider for AX\n *\n * Production-ready AWS Bedrock integration supporting Claude, GPT OSS, and Titan Embed models.\n *\n * @example\n * ```typescript\n * import { AxAIBedrock, AxAIBedrockModel } from '@ax-llm/ax-ai-aws-bedrock';\n *\n * const ai = new AxAIBedrock({\n *   region: 'us-east-2',\n *   config: { model: AxAIBedrockModel.ClaudeSonnet4 }\n * });\n *\n * const response = await ai.chat({\n *   chatPrompt: [\n *     { role: 'system', content: 'You are a helpful assistant.' },\n *     { role: 'user', content: 'What is AWS Bedrock?' }\n *   ]\n * });\n * ```\n *\n * @packageDocumentation\n */\n\nexport { AxAIBedrock } from './api.js';\nexport { axModelInfoBedrock } from './info.js';\nexport { AxAIBedrockEmbedModel, AxAIBedrockModel } from './types.js';\n\nexport type {\n  AxAIBedrockConfig,\n  BedrockChatRequest,\n  BedrockChatResponse,\n  BedrockClaudeRequest,\n  BedrockClaudeResponse,\n  BedrockGptRequest,\n  BedrockGptResponse,\n  BedrockTitanEmbedRequest,\n  BedrockTitanEmbedResponse,\n} from './types.js';\n","/**\n * AWS Bedrock Provider for AX Library\n *\n * Supports Claude, GPT OSS, and Titan Embed models for use with AX's\n * compiler, signatures, flows, and optimization features.\n *\n * Usage:\n *   const ai = new AxAIBedrock({\n *     region: 'us-east-2',\n *     config: { model: AxAIBedrockModel.ClaudeSonnet4 }\n *   });\n *\n *   const sig = new AxSignature('input -> output');\n *   const gen = new AxGen(sig, { ai });\n *   const result = await gen.forward({ input: 'test' });\n */\n\nimport {\n  BedrockRuntimeClient,\n  InvokeModelCommand,\n} from '@aws-sdk/client-bedrock-runtime';\nimport type {\n  AxAIServiceImpl,\n  AxAIServiceOptions,\n  AxAPI,\n  AxChatResponse,\n  AxChatResponseResult,\n  AxEmbedResponse,\n  AxInternalChatRequest,\n  AxInternalEmbedRequest,\n  AxModelConfig,\n  AxTokenUsage,\n} from '@ax-llm/ax';\nimport { type AxAIFeatures, AxBaseAI, axBaseAIDefaultConfig } from '@ax-llm/ax';\nimport { axModelInfoBedrock } from './info.js';\nimport {\n  type AxAIBedrockConfig,\n  type AxAIBedrockEmbedModel,\n  AxAIBedrockModel,\n  type BedrockChatRequest,\n  type BedrockChatResponse,\n  type BedrockClaudeRequest,\n  type BedrockClaudeResponse,\n  type BedrockGptRequest,\n  type BedrockGptResponse,\n  type BedrockTitanEmbedRequest,\n  type BedrockTitanEmbedResponse,\n} from './types.js';\n\n// ============================================================================\n// IMPLEMENTATION - Converts between AX format and Bedrock format\n// ============================================================================\n\ntype ModelFamily = 'claude' | 'gpt' | 'titan';\n\nclass AxAIBedrockImpl\n  implements\n    AxAIServiceImpl<\n      AxAIBedrockModel,\n      AxAIBedrockEmbedModel,\n      BedrockChatRequest,\n      BedrockTitanEmbedRequest,\n      BedrockChatResponse,\n      never, // No streaming for now\n      BedrockTitanEmbedResponse\n    >\n{\n  private clients: Map<string, BedrockRuntimeClient> = new Map();\n  private tokensUsed?: AxTokenUsage;\n\n  constructor(\n    private config: AxAIBedrockConfig,\n    private primaryRegion: string,\n    private fallbackRegions: string[],\n    private gptRegion: string,\n    private gptFallbackRegions: string[]\n  ) {}\n\n  getTokenUsage(): AxTokenUsage | undefined {\n    return this.tokensUsed;\n  }\n\n  getModelConfig(): AxModelConfig {\n    return {\n      maxTokens: this.config.maxTokens,\n      temperature: this.config.temperature,\n      topP: this.config.topP,\n      stopSequences: this.config.stopSequences,\n    };\n  }\n\n  private getClient(region: string): BedrockRuntimeClient {\n    let client = this.clients.get(region);\n    if (!client) {\n      client = new BedrockRuntimeClient({ region });\n      this.clients.set(region, client);\n    }\n    return client;\n  }\n\n  /**\n   * Detect model family from model ID\n   */\n  private getModelFamily(modelId: string): ModelFamily {\n    if (\n      modelId.includes('anthropic.claude') ||\n      modelId.includes('us.anthropic.claude')\n    ) {\n      return 'claude';\n    }\n    if (modelId.includes('openai.gpt')) {\n      return 'gpt';\n    }\n    if (modelId.includes('amazon.titan-embed')) {\n      return 'titan';\n    }\n    throw new Error(`Unknown model family for: ${modelId}`);\n  }\n\n  /**\n   * Get appropriate regions for model\n   */\n  private getRegionsForModel(modelId: string): string[] {\n    const family = this.getModelFamily(modelId);\n    if (family === 'gpt') {\n      return [this.gptRegion, ...this.gptFallbackRegions];\n    }\n    return [this.primaryRegion, ...this.fallbackRegions];\n  }\n\n  /**\n   * Regional failover logic - tries primary region, then fallbacks\n   */\n  private async invokeWithFailover<T>(\n    modelId: string,\n    handler: (client: BedrockRuntimeClient) => Promise<T>\n  ): Promise<T> {\n    const regions = this.getRegionsForModel(modelId);\n    let lastError: Error | undefined;\n\n    for (const region of regions) {\n      try {\n        const client = this.getClient(region);\n        return await handler(client);\n      } catch (error) {\n        lastError = error as Error;\n        console.warn(\n          `[Bedrock] Region ${region} failed for ${modelId}:`,\n          error\n        );\n      }\n    }\n\n    throw lastError || new Error(`All Bedrock regions failed for ${modelId}`);\n  }\n\n  /**\n   * Transform AX chat request → Bedrock request (Claude or GPT)\n   */\n  createChatReq = async (\n    req: Readonly<AxInternalChatRequest<AxAIBedrockModel>>,\n    _config: Readonly<AxAIServiceOptions>\n  ): Promise<[AxAPI, BedrockChatRequest]> => {\n    const family = this.getModelFamily(req.model);\n    const maxTokens =\n      req.modelConfig?.maxTokens ?? this.config.maxTokens ?? 4096;\n    const temperature = req.modelConfig?.temperature ?? this.config.temperature;\n    const topP = req.modelConfig?.topP ?? this.config.topP;\n\n    let bedrockRequest: BedrockChatRequest;\n\n    if (family === 'claude') {\n      // Extract system messages for Claude\n      const systemMessages = req.chatPrompt\n        .filter((msg) => msg.role === 'system')\n        .map((msg) => msg.content)\n        .join('\\n\\n');\n\n      // Convert other messages to Claude format\n      const messages = req.chatPrompt\n        .filter((msg) => msg.role !== 'system')\n        .map((msg) => {\n          if (msg.role === 'user') {\n            return {\n              role: 'user' as const,\n              content:\n                typeof msg.content === 'string'\n                  ? msg.content\n                  : JSON.stringify(msg.content),\n            };\n          }\n          if (msg.role === 'assistant') {\n            return {\n              role: 'assistant' as const,\n              content: msg.content || '',\n            };\n          }\n          throw new Error(`Unsupported role: ${msg.role}`);\n        });\n\n      bedrockRequest = {\n        anthropic_version: 'bedrock-2023-05-31',\n        max_tokens: maxTokens,\n        messages,\n        ...(systemMessages ? { system: systemMessages } : {}),\n        ...(temperature !== undefined ? { temperature } : {}),\n        ...(topP !== undefined ? { top_p: topP } : {}),\n      } as BedrockClaudeRequest;\n    } else if (family === 'gpt') {\n      // GPT uses OpenAI-style format with system messages in array\n      const messages = req.chatPrompt\n        .filter((msg) => msg.role !== 'function') // Skip function messages\n        .map((msg) => {\n          // Get content based on role\n          let content: string;\n          if ('content' in msg) {\n            content =\n              typeof msg.content === 'string'\n                ? msg.content\n                : JSON.stringify(msg.content);\n          } else {\n            content = '';\n          }\n\n          return {\n            role: msg.role as 'system' | 'user' | 'assistant',\n            content,\n          };\n        });\n\n      bedrockRequest = {\n        messages,\n        max_tokens: maxTokens,\n        ...(temperature !== undefined ? { temperature } : {}),\n        ...(topP !== undefined ? { top_p: topP } : {}),\n      } as BedrockGptRequest;\n    } else {\n      throw new Error(`Chat not supported for model family: ${family}`);\n    }\n\n    // Create API config with local call (uses SDK instead of HTTP)\n    const apiConfig: AxAPI = {\n      name: `bedrock-${family}`,\n      localCall: async <TRequest, TResponse>(data: TRequest) => {\n        const reqBody = data as unknown as BedrockChatRequest;\n        const result = await this.invokeWithFailover(\n          req.model,\n          async (client) => {\n            const command = new InvokeModelCommand({\n              modelId: req.model,\n              body: JSON.stringify(reqBody),\n              contentType: 'application/json',\n              accept: 'application/json',\n            });\n            const response = await client.send(command);\n            return JSON.parse(new TextDecoder().decode(response.body));\n          }\n        );\n        return result as TResponse;\n      },\n    };\n\n    return [apiConfig, bedrockRequest];\n  };\n\n  /**\n   * Transform Bedrock response → AX chat response (Claude or GPT)\n   */\n  createChatResp(resp: Readonly<BedrockChatResponse>): AxChatResponse {\n    // Detect response type\n    if ('content' in resp && Array.isArray(resp.content)) {\n      // Claude response\n      return this.createClaudeChatResp(resp as BedrockClaudeResponse);\n    } else if ('choices' in resp && Array.isArray(resp.choices)) {\n      // GPT response\n      return this.createGptChatResp(resp as BedrockGptResponse);\n    }\n    throw new Error('Unknown response format');\n  }\n\n  /**\n   * Handle Claude-specific response format\n   */\n  private createClaudeChatResp(\n    resp: Readonly<BedrockClaudeResponse>\n  ): AxChatResponse {\n    // Extract text content from response\n    let content = '';\n    for (const block of resp.content) {\n      if (block.type === 'text') {\n        content += block.text;\n      }\n    }\n\n    // Track token usage for AX's optimizer\n    this.tokensUsed = {\n      promptTokens: resp.usage.input_tokens,\n      completionTokens: resp.usage.output_tokens,\n      totalTokens: resp.usage.input_tokens + resp.usage.output_tokens,\n    };\n\n    // Map finish reason\n    let finishReason: AxChatResponseResult['finishReason'];\n    switch (resp.stop_reason) {\n      case 'end_turn':\n      case 'stop_sequence':\n        finishReason = 'stop';\n        break;\n      case 'max_tokens':\n        finishReason = 'length';\n        break;\n      default:\n        finishReason = undefined;\n    }\n\n    return {\n      results: [\n        {\n          index: 0,\n          id: resp.id,\n          content,\n          finishReason,\n        },\n      ],\n      remoteId: resp.id,\n    };\n  }\n\n  /**\n   * Handle GPT-specific response format\n   */\n  private createGptChatResp(\n    resp: Readonly<BedrockGptResponse>\n  ): AxChatResponse {\n    const choice = resp.choices[0];\n    if (!choice) {\n      throw new Error('No choices in GPT response');\n    }\n\n    // Extract content (can be string or array)\n    let content = '';\n    if (typeof choice.message.content === 'string') {\n      content = choice.message.content;\n    } else if (Array.isArray(choice.message.content)) {\n      content = choice.message.content\n        .map((part) => {\n          if (typeof part === 'string') return part;\n          return part.text || part.content || '';\n        })\n        .join('');\n    }\n\n    // Track token usage if available\n    if (resp.usage) {\n      this.tokensUsed = {\n        promptTokens: resp.usage.prompt_tokens,\n        completionTokens: resp.usage.completion_tokens,\n        totalTokens: resp.usage.total_tokens,\n      };\n    }\n\n    // Map finish reason\n    let finishReason: AxChatResponseResult['finishReason'];\n    switch (choice.finish_reason) {\n      case 'stop':\n        finishReason = 'stop';\n        break;\n      case 'length':\n        finishReason = 'length';\n        break;\n      case 'content_filter':\n        finishReason = 'content_filter';\n        break;\n      default:\n        finishReason = undefined;\n    }\n\n    return {\n      results: [\n        {\n          index: choice.index,\n          id: resp.id,\n          content,\n          finishReason,\n        },\n      ],\n      remoteId: resp.id,\n    };\n  }\n\n  /**\n   * Create embed request for Titan\n   */\n  createEmbedReq = async (\n    req: Readonly<AxInternalEmbedRequest<AxAIBedrockEmbedModel>>\n  ): Promise<[AxAPI, BedrockTitanEmbedRequest]> => {\n    if (!req.texts || req.texts.length === 0) {\n      throw new Error('No texts provided for embedding');\n    }\n\n    const embedRequest: BedrockTitanEmbedRequest = {\n      inputText: req.texts[0], // Take first text\n      dimensions: 512,\n      normalize: true,\n    };\n\n    const apiConfig: AxAPI = {\n      name: 'bedrock-titan-embed',\n      localCall: async <TRequest, TResponse>(data: TRequest) => {\n        const reqBody = data as unknown as BedrockTitanEmbedRequest;\n        const result = await this.invokeWithFailover(\n          req.embedModel,\n          async (client) => {\n            const command = new InvokeModelCommand({\n              modelId: req.embedModel,\n              body: JSON.stringify(reqBody),\n              contentType: 'application/json',\n              accept: 'application/json',\n            });\n            const response = await client.send(command);\n            return JSON.parse(new TextDecoder().decode(response.body));\n          }\n        );\n        return result as TResponse;\n      },\n    };\n\n    return [apiConfig, embedRequest];\n  };\n\n  /**\n   * Create embed response from Titan\n   */\n  createEmbedResp(resp: Readonly<BedrockTitanEmbedResponse>): AxEmbedResponse {\n    return {\n      embeddings: [resp.embedding],\n    };\n  }\n}\n\n// ============================================================================\n// PROVIDER CLASS - Main entry point\n// ============================================================================\n\nexport class AxAIBedrock extends AxBaseAI<\n  AxAIBedrockModel,\n  AxAIBedrockEmbedModel,\n  BedrockChatRequest,\n  BedrockTitanEmbedRequest,\n  BedrockChatResponse,\n  never, // No streaming yet\n  BedrockTitanEmbedResponse,\n  string\n> {\n  constructor({\n    region = 'us-east-2',\n    fallbackRegions = ['us-west-2', 'us-east-1'],\n    gptRegion = 'us-west-2',\n    gptFallbackRegions = ['us-east-1'],\n    config,\n    options,\n  }: Readonly<{\n    region?: string;\n    fallbackRegions?: string[];\n    gptRegion?: string;\n    gptFallbackRegions?: string[];\n    config: Readonly<Partial<AxAIBedrockConfig>>;\n    options?: Readonly<AxAIServiceOptions>;\n  }>) {\n    // Merge user config with defaults\n    const fullConfig: AxAIBedrockConfig = {\n      ...axBaseAIDefaultConfig(),\n      model: AxAIBedrockModel.ClaudeSonnet4,\n      region,\n      fallbackRegions,\n      gptRegion,\n      gptFallbackRegions,\n      ...config,\n    };\n\n    // Create implementation\n    const aiImpl = new AxAIBedrockImpl(\n      fullConfig,\n      region,\n      fallbackRegions,\n      gptRegion,\n      gptFallbackRegions\n    );\n\n    // Define feature support\n    const supportFor = (): AxAIFeatures => ({\n      functions: false, // Not implemented yet - add when needed\n      streaming: false, // Not implemented yet - add when needed\n      functionCot: false,\n      hasThinkingBudget: false,\n      hasShowThoughts: false,\n      media: {\n        images: {\n          supported: false, // Add when needed\n          formats: [],\n        },\n        audio: {\n          supported: false,\n          formats: [],\n        },\n        files: {\n          supported: false,\n          formats: [],\n          uploadMethod: 'none',\n        },\n        urls: {\n          supported: false,\n          webSearch: false,\n          contextFetching: false,\n        },\n      },\n      caching: {\n        supported: false,\n        types: [],\n      },\n      thinking: false,\n      multiTurn: true, // All models support multi-turn conversations\n    });\n\n    // Initialize base class\n    super(aiImpl, {\n      name: 'Bedrock',\n      apiURL: '', // Not used - we use SDK directly\n      headers: async () => ({}), // AWS SDK handles auth\n      modelInfo: axModelInfoBedrock,\n      defaults: {\n        model: fullConfig.model,\n        embedModel: fullConfig.embedModel,\n      },\n      options,\n      supportFor,\n    });\n  }\n}\n\nexport type { AxAIBedrockConfig } from './types.js';\n// Re-export types for convenience\nexport { AxAIBedrockEmbedModel, AxAIBedrockModel } from './types.js';\n","/**\n * AWS Bedrock provider types for AX integration\n * Supports Claude, GPT OSS, and Titan models\n */\n\nimport type { AxModelConfig } from '@ax-llm/ax';\n\n// All Bedrock models\nexport enum AxAIBedrockModel {\n  // Claude models\n  ClaudeSonnet4 = 'us.anthropic.claude-sonnet-4-20250514-v1:0',\n  Claude37Sonnet = 'anthropic.claude-3-7-sonnet-20250219-v1:0',\n  Claude35Sonnet = 'anthropic.claude-3-5-sonnet-20240620-v1:0',\n\n  // GPT OSS models\n  GptOss120B = 'openai.gpt-oss-120b-1:0',\n  GptOss20B = 'openai.gpt-oss-20b-1:0',\n}\n\n// Embed models\nexport enum AxAIBedrockEmbedModel {\n  TitanEmbedV2 = 'amazon.titan-embed-text-v2:0',\n}\n\nexport interface AxAIBedrockConfig extends AxModelConfig {\n  model: AxAIBedrockModel;\n  embedModel?: AxAIBedrockEmbedModel;\n  region?: string;\n  fallbackRegions?: string[];\n  gptRegion?: string;\n  gptFallbackRegions?: string[];\n}\n\n// ============================================================================\n// Claude Request/Response Types\n// ============================================================================\n\nexport interface BedrockClaudeRequest {\n  anthropic_version: string;\n  max_tokens: number;\n  messages: Array<{\n    role: 'user' | 'assistant';\n    content: string | Array<{ type: 'text'; text: string }>;\n  }>;\n  system?: string;\n  temperature?: number;\n  top_p?: number;\n}\n\nexport interface BedrockClaudeResponse {\n  id: string;\n  type: 'message';\n  role: 'assistant';\n  content: Array<{ type: 'text'; text: string }>;\n  model: string;\n  stop_reason: 'end_turn' | 'max_tokens' | 'stop_sequence';\n  usage: {\n    input_tokens: number;\n    output_tokens: number;\n  };\n}\n\n// ============================================================================\n// GPT OSS Request/Response Types (OpenAI-compatible format)\n// ============================================================================\n\nexport interface BedrockGptRequest {\n  messages: Array<{\n    role: 'system' | 'user' | 'assistant';\n    content: string;\n  }>;\n  max_tokens: number;\n  temperature?: number;\n  top_p?: number;\n}\n\nexport interface BedrockGptResponse {\n  id?: string;\n  choices: Array<{\n    index: number;\n    message: {\n      role: 'assistant';\n      content:\n        | string\n        | Array<{ type?: string; text?: string; content?: string }>;\n    };\n    finish_reason: 'stop' | 'length' | 'content_filter' | null;\n  }>;\n  usage?: {\n    prompt_tokens: number;\n    completion_tokens: number;\n    total_tokens: number;\n  };\n}\n\n// ============================================================================\n// Titan Embed Request/Response Types\n// ============================================================================\n\nexport interface BedrockTitanEmbedRequest {\n  inputText: string;\n  dimensions?: number;\n  normalize?: boolean;\n}\n\nexport interface BedrockTitanEmbedResponse {\n  embedding: number[];\n  inputTextTokenCount: number;\n}\n\n// Union types for all models\nexport type BedrockChatRequest = BedrockClaudeRequest | BedrockGptRequest;\nexport type BedrockChatResponse = BedrockClaudeResponse | BedrockGptResponse;\n","/**\n * Bedrock model information (pricing, limits, features)\n */\n\nimport type { AxModelInfo } from '@ax-llm/ax';\nimport { AxAIBedrockEmbedModel, AxAIBedrockModel } from './types.js';\n\nexport const axModelInfoBedrock: AxModelInfo[] = [\n  // ========================================================================\n  // Claude Models\n  // ========================================================================\n  {\n    name: AxAIBedrockModel.ClaudeSonnet4,\n    currency: 'usd',\n    promptTokenCostPer1M: 3.0,\n    completionTokenCostPer1M: 15.0,\n    maxTokens: 64000,\n    contextWindow: 200000,\n    supported: { thinkingBudget: true, showThoughts: true },\n  },\n  {\n    name: AxAIBedrockModel.Claude37Sonnet,\n    currency: 'usd',\n    promptTokenCostPer1M: 3.0,\n    completionTokenCostPer1M: 15.0,\n    maxTokens: 64000,\n    contextWindow: 200000,\n  },\n  {\n    name: AxAIBedrockModel.Claude35Sonnet,\n    currency: 'usd',\n    promptTokenCostPer1M: 3.0,\n    completionTokenCostPer1M: 15.0,\n    maxTokens: 8192,\n    contextWindow: 200000,\n  },\n\n  // ========================================================================\n  // GPT OSS Models\n  // ========================================================================\n  {\n    name: AxAIBedrockModel.GptOss120B,\n    currency: 'usd',\n    promptTokenCostPer1M: 0.5,\n    completionTokenCostPer1M: 1.5,\n    maxTokens: 16384,\n    contextWindow: 128000,\n  },\n  {\n    name: AxAIBedrockModel.GptOss20B,\n    currency: 'usd',\n    promptTokenCostPer1M: 0.25,\n    completionTokenCostPer1M: 0.75,\n    maxTokens: 16384,\n    contextWindow: 128000,\n  },\n\n  // ========================================================================\n  // Embed Models\n  // ========================================================================\n  {\n    name: AxAIBedrockEmbedModel.TitanEmbedV2,\n    currency: 'usd',\n    promptTokenCostPer1M: 0.02,\n    completionTokenCostPer1M: 0,\n    maxTokens: 8192,\n    contextWindow: 8192,\n  },\n];\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACiBA,oCAGO;AAaP,gBAAmE;;;ACzB5D,IAAK,mBAAL,kBAAKA,sBAAL;AAEL,EAAAA,kBAAA,mBAAgB;AAChB,EAAAA,kBAAA,oBAAiB;AACjB,EAAAA,kBAAA,oBAAiB;AAGjB,EAAAA,kBAAA,gBAAa;AACb,EAAAA,kBAAA,eAAY;AARF,SAAAA;AAAA,GAAA;AAYL,IAAK,wBAAL,kBAAKC,2BAAL;AACL,EAAAA,uBAAA,kBAAe;AADL,SAAAA;AAAA,GAAA;;;ACbL,IAAM,qBAAoC;AAAA;AAAA;AAAA;AAAA,EAI/C;AAAA,IACE;AAAA,IACA,UAAU;AAAA,IACV,sBAAsB;AAAA,IACtB,0BAA0B;AAAA,IAC1B,WAAW;AAAA,IACX,eAAe;AAAA,IACf,WAAW,EAAE,gBAAgB,MAAM,cAAc,KAAK;AAAA,EACxD;AAAA,EACA;AAAA,IACE;AAAA,IACA,UAAU;AAAA,IACV,sBAAsB;AAAA,IACtB,0BAA0B;AAAA,IAC1B,WAAW;AAAA,IACX,eAAe;AAAA,EACjB;AAAA,EACA;AAAA,IACE;AAAA,IACA,UAAU;AAAA,IACV,sBAAsB;AAAA,IACtB,0BAA0B;AAAA,IAC1B,WAAW;AAAA,IACX,eAAe;AAAA,EACjB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,IACE;AAAA,IACA,UAAU;AAAA,IACV,sBAAsB;AAAA,IACtB,0BAA0B;AAAA,IAC1B,WAAW;AAAA,IACX,eAAe;AAAA,EACjB;AAAA,EACA;AAAA,IACE;AAAA,IACA,UAAU;AAAA,IACV,sBAAsB;AAAA,IACtB,0BAA0B;AAAA,IAC1B,WAAW;AAAA,IACX,eAAe;AAAA,EACjB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,IACE;AAAA,IACA,UAAU;AAAA,IACV,sBAAsB;AAAA,IACtB,0BAA0B;AAAA,IAC1B,WAAW;AAAA,IACX,eAAe;AAAA,EACjB;AACF;;;AFbA,IAAM,kBAAN,MAWA;AAAA,EAIE,YACU,QACA,eACA,iBACA,WACA,oBACR;AALQ;AACA;AACA;AACA;AACA;AAAA,EACP;AAAA,EATK,UAA6C,oBAAI,IAAI;AAAA,EACrD;AAAA,EAUR,gBAA0C;AACxC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,iBAAgC;AAC9B,WAAO;AAAA,MACL,WAAW,KAAK,OAAO;AAAA,MACvB,aAAa,KAAK,OAAO;AAAA,MACzB,MAAM,KAAK,OAAO;AAAA,MAClB,eAAe,KAAK,OAAO;AAAA,IAC7B;AAAA,EACF;AAAA,EAEQ,UAAU,QAAsC;AACtD,QAAI,SAAS,KAAK,QAAQ,IAAI,MAAM;AACpC,QAAI,CAAC,QAAQ;AACX,eAAS,IAAI,mDAAqB,EAAE,OAAO,CAAC;AAC5C,WAAK,QAAQ,IAAI,QAAQ,MAAM;AAAA,IACjC;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,eAAe,SAA8B;AACnD,QACE,QAAQ,SAAS,kBAAkB,KACnC,QAAQ,SAAS,qBAAqB,GACtC;AACA,aAAO;AAAA,IACT;AACA,QAAI,QAAQ,SAAS,YAAY,GAAG;AAClC,aAAO;AAAA,IACT;AACA,QAAI,QAAQ,SAAS,oBAAoB,GAAG;AAC1C,aAAO;AAAA,IACT;AACA,UAAM,IAAI,MAAM,6BAA6B,OAAO,EAAE;AAAA,EACxD;AAAA;AAAA;AAAA;AAAA,EAKQ,mBAAmB,SAA2B;AACpD,UAAM,SAAS,KAAK,eAAe,OAAO;AAC1C,QAAI,WAAW,OAAO;AACpB,aAAO,CAAC,KAAK,WAAW,GAAG,KAAK,kBAAkB;AAAA,IACpD;AACA,WAAO,CAAC,KAAK,eAAe,GAAG,KAAK,eAAe;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,mBACZ,SACA,SACY;AACZ,UAAM,UAAU,KAAK,mBAAmB,OAAO;AAC/C,QAAI;AAEJ,eAAW,UAAU,SAAS;AAC5B,UAAI;AACF,cAAM,SAAS,KAAK,UAAU,MAAM;AACpC,eAAO,MAAM,QAAQ,MAAM;AAAA,MAC7B,SAAS,OAAO;AACd,oBAAY;AACZ,gBAAQ;AAAA,UACN,oBAAoB,MAAM,eAAe,OAAO;AAAA,UAChD;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,UAAM,aAAa,IAAI,MAAM,kCAAkC,OAAO,EAAE;AAAA,EAC1E;AAAA;AAAA;AAAA;AAAA,EAKA,gBAAgB,OACd,KACA,YACyC;AACzC,UAAM,SAAS,KAAK,eAAe,IAAI,KAAK;AAC5C,UAAM,YACJ,IAAI,aAAa,aAAa,KAAK,OAAO,aAAa;AACzD,UAAM,cAAc,IAAI,aAAa,eAAe,KAAK,OAAO;AAChE,UAAM,OAAO,IAAI,aAAa,QAAQ,KAAK,OAAO;AAElD,QAAI;AAEJ,QAAI,WAAW,UAAU;AAEvB,YAAM,iBAAiB,IAAI,WACxB,OAAO,CAAC,QAAQ,IAAI,SAAS,QAAQ,EACrC,IAAI,CAAC,QAAQ,IAAI,OAAO,EACxB,KAAK,MAAM;AAGd,YAAM,WAAW,IAAI,WAClB,OAAO,CAAC,QAAQ,IAAI,SAAS,QAAQ,EACrC,IAAI,CAAC,QAAQ;AACZ,YAAI,IAAI,SAAS,QAAQ;AACvB,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SACE,OAAO,IAAI,YAAY,WACnB,IAAI,UACJ,KAAK,UAAU,IAAI,OAAO;AAAA,UAClC;AAAA,QACF;AACA,YAAI,IAAI,SAAS,aAAa;AAC5B,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS,IAAI,WAAW;AAAA,UAC1B;AAAA,QACF;AACA,cAAM,IAAI,MAAM,qBAAqB,IAAI,IAAI,EAAE;AAAA,MACjD,CAAC;AAEH,uBAAiB;AAAA,QACf,mBAAmB;AAAA,QACnB,YAAY;AAAA,QACZ;AAAA,QACA,GAAI,iBAAiB,EAAE,QAAQ,eAAe,IAAI,CAAC;AAAA,QACnD,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;AAAA,QACnD,GAAI,SAAS,SAAY,EAAE,OAAO,KAAK,IAAI,CAAC;AAAA,MAC9C;AAAA,IACF,WAAW,WAAW,OAAO;AAE3B,YAAM,WAAW,IAAI,WAClB,OAAO,CAAC,QAAQ,IAAI,SAAS,UAAU,EACvC,IAAI,CAAC,QAAQ;AAEZ,YAAI;AACJ,YAAI,aAAa,KAAK;AACpB,oBACE,OAAO,IAAI,YAAY,WACnB,IAAI,UACJ,KAAK,UAAU,IAAI,OAAO;AAAA,QAClC,OAAO;AACL,oBAAU;AAAA,QACZ;AAEA,eAAO;AAAA,UACL,MAAM,IAAI;AAAA,UACV;AAAA,QACF;AAAA,MACF,CAAC;AAEH,uBAAiB;AAAA,QACf;AAAA,QACA,YAAY;AAAA,QACZ,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;AAAA,QACnD,GAAI,SAAS,SAAY,EAAE,OAAO,KAAK,IAAI,CAAC;AAAA,MAC9C;AAAA,IACF,OAAO;AACL,YAAM,IAAI,MAAM,wCAAwC,MAAM,EAAE;AAAA,IAClE;AAGA,UAAM,YAAmB;AAAA,MACvB,MAAM,WAAW,MAAM;AAAA,MACvB,WAAW,OAA4B,SAAmB;AACxD,cAAM,UAAU;AAChB,cAAM,SAAS,MAAM,KAAK;AAAA,UACxB,IAAI;AAAA,UACJ,OAAO,WAAW;AAChB,kBAAM,UAAU,IAAI,iDAAmB;AAAA,cACrC,SAAS,IAAI;AAAA,cACb,MAAM,KAAK,UAAU,OAAO;AAAA,cAC5B,aAAa;AAAA,cACb,QAAQ;AAAA,YACV,CAAC;AACD,kBAAM,WAAW,MAAM,OAAO,KAAK,OAAO;AAC1C,mBAAO,KAAK,MAAM,IAAI,YAAY,EAAE,OAAO,SAAS,IAAI,CAAC;AAAA,UAC3D;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO,CAAC,WAAW,cAAc;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA,EAKA,eAAe,MAAqD;AAElE,QAAI,aAAa,QAAQ,MAAM,QAAQ,KAAK,OAAO,GAAG;AAEpD,aAAO,KAAK,qBAAqB,IAA6B;AAAA,IAChE,WAAW,aAAa,QAAQ,MAAM,QAAQ,KAAK,OAAO,GAAG;AAE3D,aAAO,KAAK,kBAAkB,IAA0B;AAAA,IAC1D;AACA,UAAM,IAAI,MAAM,yBAAyB;AAAA,EAC3C;AAAA;AAAA;AAAA;AAAA,EAKQ,qBACN,MACgB;AAEhB,QAAI,UAAU;AACd,eAAW,SAAS,KAAK,SAAS;AAChC,UAAI,MAAM,SAAS,QAAQ;AACzB,mBAAW,MAAM;AAAA,MACnB;AAAA,IACF;AAGA,SAAK,aAAa;AAAA,MAChB,cAAc,KAAK,MAAM;AAAA,MACzB,kBAAkB,KAAK,MAAM;AAAA,MAC7B,aAAa,KAAK,MAAM,eAAe,KAAK,MAAM;AAAA,IACpD;AAGA,QAAI;AACJ,YAAQ,KAAK,aAAa;AAAA,MACxB,KAAK;AAAA,MACL,KAAK;AACH,uBAAe;AACf;AAAA,MACF,KAAK;AACH,uBAAe;AACf;AAAA,MACF;AACE,uBAAe;AAAA,IACnB;AAEA,WAAO;AAAA,MACL,SAAS;AAAA,QACP;AAAA,UACE,OAAO;AAAA,UACP,IAAI,KAAK;AAAA,UACT;AAAA,UACA;AAAA,QACF;AAAA,MACF;AAAA,MACA,UAAU,KAAK;AAAA,IACjB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,kBACN,MACgB;AAChB,UAAM,SAAS,KAAK,QAAQ,CAAC;AAC7B,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI,MAAM,4BAA4B;AAAA,IAC9C;AAGA,QAAI,UAAU;AACd,QAAI,OAAO,OAAO,QAAQ,YAAY,UAAU;AAC9C,gBAAU,OAAO,QAAQ;AAAA,IAC3B,WAAW,MAAM,QAAQ,OAAO,QAAQ,OAAO,GAAG;AAChD,gBAAU,OAAO,QAAQ,QACtB,IAAI,CAAC,SAAS;AACb,YAAI,OAAO,SAAS,SAAU,QAAO;AACrC,eAAO,KAAK,QAAQ,KAAK,WAAW;AAAA,MACtC,CAAC,EACA,KAAK,EAAE;AAAA,IACZ;AAGA,QAAI,KAAK,OAAO;AACd,WAAK,aAAa;AAAA,QAChB,cAAc,KAAK,MAAM;AAAA,QACzB,kBAAkB,KAAK,MAAM;AAAA,QAC7B,aAAa,KAAK,MAAM;AAAA,MAC1B;AAAA,IACF;AAGA,QAAI;AACJ,YAAQ,OAAO,eAAe;AAAA,MAC5B,KAAK;AACH,uBAAe;AACf;AAAA,MACF,KAAK;AACH,uBAAe;AACf;AAAA,MACF,KAAK;AACH,uBAAe;AACf;AAAA,MACF;AACE,uBAAe;AAAA,IACnB;AAEA,WAAO;AAAA,MACL,SAAS;AAAA,QACP;AAAA,UACE,OAAO,OAAO;AAAA,UACd,IAAI,KAAK;AAAA,UACT;AAAA,UACA;AAAA,QACF;AAAA,MACF;AAAA,MACA,UAAU,KAAK;AAAA,IACjB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,iBAAiB,OACf,QAC+C;AAC/C,QAAI,CAAC,IAAI,SAAS,IAAI,MAAM,WAAW,GAAG;AACxC,YAAM,IAAI,MAAM,iCAAiC;AAAA,IACnD;AAEA,UAAM,eAAyC;AAAA,MAC7C,WAAW,IAAI,MAAM,CAAC;AAAA;AAAA,MACtB,YAAY;AAAA,MACZ,WAAW;AAAA,IACb;AAEA,UAAM,YAAmB;AAAA,MACvB,MAAM;AAAA,MACN,WAAW,OAA4B,SAAmB;AACxD,cAAM,UAAU;AAChB,cAAM,SAAS,MAAM,KAAK;AAAA,UACxB,IAAI;AAAA,UACJ,OAAO,WAAW;AAChB,kBAAM,UAAU,IAAI,iDAAmB;AAAA,cACrC,SAAS,IAAI;AAAA,cACb,MAAM,KAAK,UAAU,OAAO;AAAA,cAC5B,aAAa;AAAA,cACb,QAAQ;AAAA,YACV,CAAC;AACD,kBAAM,WAAW,MAAM,OAAO,KAAK,OAAO;AAC1C,mBAAO,KAAK,MAAM,IAAI,YAAY,EAAE,OAAO,SAAS,IAAI,CAAC;AAAA,UAC3D;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO,CAAC,WAAW,YAAY;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA,EAKA,gBAAgB,MAA4D;AAC1E,WAAO;AAAA,MACL,YAAY,CAAC,KAAK,SAAS;AAAA,IAC7B;AAAA,EACF;AACF;AAMO,IAAM,cAAN,cAA0B,mBAS/B;AAAA,EACA,YAAY;AAAA,IACV,SAAS;AAAA,IACT,kBAAkB,CAAC,aAAa,WAAW;AAAA,IAC3C,YAAY;AAAA,IACZ,qBAAqB,CAAC,WAAW;AAAA,IACjC;AAAA,IACA;AAAA,EACF,GAOI;AAEF,UAAM,aAAgC;AAAA,MACpC,OAAG,iCAAsB;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,GAAG;AAAA,IACL;AAGA,UAAM,SAAS,IAAI;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAGA,UAAM,aAAa,OAAqB;AAAA,MACtC,WAAW;AAAA;AAAA,MACX,WAAW;AAAA;AAAA,MACX,aAAa;AAAA,MACb,mBAAmB;AAAA,MACnB,iBAAiB;AAAA,MACjB,OAAO;AAAA,QACL,QAAQ;AAAA,UACN,WAAW;AAAA;AAAA,UACX,SAAS,CAAC;AAAA,QACZ;AAAA,QACA,OAAO;AAAA,UACL,WAAW;AAAA,UACX,SAAS,CAAC;AAAA,QACZ;AAAA,QACA,OAAO;AAAA,UACL,WAAW;AAAA,UACX,SAAS,CAAC;AAAA,UACV,cAAc;AAAA,QAChB;AAAA,QACA,MAAM;AAAA,UACJ,WAAW;AAAA,UACX,WAAW;AAAA,UACX,iBAAiB;AAAA,QACnB;AAAA,MACF;AAAA,MACA,SAAS;AAAA,QACP,WAAW;AAAA,QACX,OAAO,CAAC;AAAA,MACV;AAAA,MACA,UAAU;AAAA,MACV,WAAW;AAAA;AAAA,IACb;AAGA,UAAM,QAAQ;AAAA,MACZ,MAAM;AAAA,MACN,QAAQ;AAAA;AAAA,MACR,SAAS,aAAa,CAAC;AAAA;AAAA,MACvB,WAAW;AAAA,MACX,UAAU;AAAA,QACR,OAAO,WAAW;AAAA,QAClB,YAAY,WAAW;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACF;","names":["AxAIBedrockModel","AxAIBedrockEmbedModel"]}

package/index.d.cts

@@ -0,0 +1,131 @@
+import { AxModelConfig, AxBaseAI, AxAIServiceOptions, AxModelInfo } from '@ax-llm/ax';
+
+/**
+ * AWS Bedrock provider types for AX integration
+ * Supports Claude, GPT OSS, and Titan models
+ */
+
+declare enum AxAIBedrockModel {
+    ClaudeSonnet4 = "us.anthropic.claude-sonnet-4-20250514-v1:0",
+    Claude37Sonnet = "anthropic.claude-3-7-sonnet-20250219-v1:0",
+    Claude35Sonnet = "anthropic.claude-3-5-sonnet-20240620-v1:0",
+    GptOss120B = "openai.gpt-oss-120b-1:0",
+    GptOss20B = "openai.gpt-oss-20b-1:0"
+}
+declare enum AxAIBedrockEmbedModel {
+    TitanEmbedV2 = "amazon.titan-embed-text-v2:0"
+}
+interface AxAIBedrockConfig extends AxModelConfig {
+    model: AxAIBedrockModel;
+    embedModel?: AxAIBedrockEmbedModel;
+    region?: string;
+    fallbackRegions?: string[];
+    gptRegion?: string;
+    gptFallbackRegions?: string[];
+}
+interface BedrockClaudeRequest {
+    anthropic_version: string;
+    max_tokens: number;
+    messages: Array<{
+        role: 'user' | 'assistant';
+        content: string | Array<{
+            type: 'text';
+            text: string;
+        }>;
+    }>;
+    system?: string;
+    temperature?: number;
+    top_p?: number;
+}
+interface BedrockClaudeResponse {
+    id: string;
+    type: 'message';
+    role: 'assistant';
+    content: Array<{
+        type: 'text';
+        text: string;
+    }>;
+    model: string;
+    stop_reason: 'end_turn' | 'max_tokens' | 'stop_sequence';
+    usage: {
+        input_tokens: number;
+        output_tokens: number;
+    };
+}
+interface BedrockGptRequest {
+    messages: Array<{
+        role: 'system' | 'user' | 'assistant';
+        content: string;
+    }>;
+    max_tokens: number;
+    temperature?: number;
+    top_p?: number;
+}
+interface BedrockGptResponse {
+    id?: string;
+    choices: Array<{
+        index: number;
+        message: {
+            role: 'assistant';
+            content: string | Array<{
+                type?: string;
+                text?: string;
+                content?: string;
+            }>;
+        };
+        finish_reason: 'stop' | 'length' | 'content_filter' | null;
+    }>;
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+}
+interface BedrockTitanEmbedRequest {
+    inputText: string;
+    dimensions?: number;
+    normalize?: boolean;
+}
+interface BedrockTitanEmbedResponse {
+    embedding: number[];
+    inputTextTokenCount: number;
+}
+type BedrockChatRequest = BedrockClaudeRequest | BedrockGptRequest;
+type BedrockChatResponse = BedrockClaudeResponse | BedrockGptResponse;
+
+/**
+ * AWS Bedrock Provider for AX Library
+ *
+ * Supports Claude, GPT OSS, and Titan Embed models for use with AX's
+ * compiler, signatures, flows, and optimization features.
+ *
+ * Usage:
+ *   const ai = new AxAIBedrock({
+ *     region: 'us-east-2',
+ *     config: { model: AxAIBedrockModel.ClaudeSonnet4 }
+ *   });
+ *
+ *   const sig = new AxSignature('input -> output');
+ *   const gen = new AxGen(sig, { ai });
+ *   const result = await gen.forward({ input: 'test' });
+ */
+
+declare class AxAIBedrock extends AxBaseAI<AxAIBedrockModel, AxAIBedrockEmbedModel, BedrockChatRequest, BedrockTitanEmbedRequest, BedrockChatResponse, never, // No streaming yet
+BedrockTitanEmbedResponse, string> {
+    constructor({ region, fallbackRegions, gptRegion, gptFallbackRegions, config, options, }: Readonly<{
+        region?: string;
+        fallbackRegions?: string[];
+        gptRegion?: string;
+        gptFallbackRegions?: string[];
+        config: Readonly<Partial<AxAIBedrockConfig>>;
+        options?: Readonly<AxAIServiceOptions>;
+    }>);
+}
+
+/**
+ * Bedrock model information (pricing, limits, features)
+ */
+
+declare const axModelInfoBedrock: AxModelInfo[];
+
+export { AxAIBedrock, type AxAIBedrockConfig, AxAIBedrockEmbedModel, AxAIBedrockModel, type BedrockChatRequest, type BedrockChatResponse, type BedrockClaudeRequest, type BedrockClaudeResponse, type BedrockGptRequest, type BedrockGptResponse, type BedrockTitanEmbedRequest, type BedrockTitanEmbedResponse, axModelInfoBedrock };