@langchain/community
Version:
Third-party integrations for LangChain.js
34 lines (33 loc) • 1.41 kB
JavaScript
/* eslint-disable no-process-env */
import { test, expect } from "@jest/globals";
import { ChatModelIntegrationTests } from "@langchain/standard-tests";
import { BedrockChat } from "../bedrock/index.js";
class BedrockChatStandardIntegrationTests extends ChatModelIntegrationTests {
constructor() {
const region = process.env.BEDROCK_AWS_REGION ?? "us-east-1";
super({
Cls: BedrockChat,
chatModelHasToolCalling: true,
chatModelHasStructuredOutput: true,
constructorArgs: {
region,
model: "anthropic.claude-3-sonnet-20240229-v1:0",
credentials: {
secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY,
accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID,
},
},
});
}
async testUsageMetadataStreaming() {
this.skipTestMessage("testUsageMetadataStreaming", "BedrockChat", "Streaming tokens is not currently supported.");
}
async testUsageMetadata() {
this.skipTestMessage("testUsageMetadata", "BedrockChat", "Usage metadata tokens is not currently supported.");
}
}
const testClass = new BedrockChatStandardIntegrationTests();
test("BedrockChatStandardIntegrationTests", async () => {
const testResults = await testClass.runTests();
expect(testResults).toBe(true);
});