@langchain/community/dist/chat_models/tests/chatbedrock.standard.test.js

Third-party integrations for LangChain.js

/* eslint-disable no-process-env */
import { test, expect } from "@jest/globals";
import { ChatModelUnitTests } from "@langchain/standard-tests";
import { BedrockChat } from "../bedrock/index.js";
class BedrockChatStandardUnitTests extends ChatModelUnitTests {
    constructor() {
        super({
            Cls: BedrockChat,
            chatModelHasToolCalling: true,
            chatModelHasStructuredOutput: true,
            constructorArgs: {
                model: "anthropic.claude-3-sonnet-20240229-v1:0",
            },
        });
        process.env.BEDROCK_AWS_SECRET_ACCESS_KEY = "test";
        process.env.BEDROCK_AWS_ACCESS_KEY_ID = "test";
        process.env.BEDROCK_AWS_SESSION_TOKEN = "test";
        process.env.AWS_DEFAULT_REGION = "us-east-1";
    }
    testChatModelInitApiKey() {
        this.skipTestMessage("testChatModelInitApiKey", "BedrockChat", this.multipleApiKeysRequiredMessage);
    }
}
const testClass = new BedrockChatStandardUnitTests();
test("BedrockChatStandardUnitTests", () => {
    const testResults = testClass.runTests();
    expect(testResults).toBe(true);
});