@copilotkit/runtime/src/service-adapters/bedrock/bedrock-adapter.ts

<div align="center"> <a href="https://copilotkit.ai" target="_blank"> <img src="https://github.com/copilotkit/copilotkit/raw/main/assets/banner.png" alt="CopilotKit Logo"> </a>

/**
 * Copilot Runtime adapter for AWS Bedrock.
 *
 * ## Example
 *
 * ```ts
 * import { CopilotRuntime, BedrockAdapter } from "@copilotkit/runtime";
 *
 * const copilotKit = new CopilotRuntime();
 *
 * return new BedrockAdapter({
 *   model: "amazon.nova-lite-v1:0",
 *   region: "us-east-1",
 *   credentials: {
 *     accessKeyId: process.env.AWS_ACCESS_KEY_ID,
 *     secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY
 *   }
 * });
 * ```
 */

import { ChatBedrockConverse } from "@langchain/aws";
import { LangChainAdapter } from "../langchain/langchain-adapter";

export interface BedrockAdapterParams {
  /**
   * AWS Bedrock model ID to use.
   * @default "amazon.nova-lite-v1:0"
   */
  model?: string;

  /**
   * AWS region where Bedrock is available.
   * @default "us-east-1"
   */
  region?: string;

  /**
   * AWS credentials for Bedrock access.
   */
  credentials?: {
    accessKeyId?: string;
    secretAccessKey?: string;
  };
}

export class BedrockAdapter extends LangChainAdapter {
  constructor(options?: BedrockAdapterParams) {
    super({
      chainFn: async ({ messages, tools, threadId }) => {
        const model = new ChatBedrockConverse({
          model: options?.model ?? "amazon.nova-lite-v1:0",
          region: options?.region ?? "us-east-1",
          credentials: options?.credentials
            ? {
                accessKeyId: options.credentials.accessKeyId,
                secretAccessKey: options.credentials.secretAccessKey,
              }
            : undefined,
        }).bindTools(tools);
        return model.stream(messages);
      },
    });
  }
}