BedrockChat
Amazon Bedrock 是一项完全托管的服务,通过单个 API 提供来自领先 AI 公司(如 AI21 Labs、Anthropic、Cohere、Meta、Stability AI 和 Amazon)的选择性高性能基础模型 (FM),以及构建安全、私密和负责任的 AI 应用程序所需的一系列广泛功能。
这将帮助您开始使用 Amazon Bedrock 聊天模型。有关所有 BedrockChat
功能和配置的详细文档,请访问API 参考。
新的ChatBedrockConverse
聊天模型现在可通过专用 @langchain/aws
集成包使用。使用此包与更多模型一起使用工具调用。
概述
集成详情
类 | 包 | 本地 | 可序列化 | PY 支持 | 包下载 | 包最新 |
---|---|---|---|---|---|---|
BedrockChat | @langchain/community | ❌ | ✅ | ✅ |
模型特性
有关如何使用特定功能的指南,请查看下方表头中的链接。
工具调用 | 结构化输出 | JSON 模式 | 图像输入 | 音频输入 | 视频输入 | 令牌级流 | 令牌使用 | Logprobs |
---|---|---|---|---|---|---|---|---|
✅ | ✅ | ❌ | ✅ | ❌ | ❌ | ✅ | ✅ | ❌ |
设置
要访问 Bedrock 模型,您需要创建一个 AWS 帐户,设置 Bedrock API 服务,获取访问密钥 ID 和密钥,并安装 @langchain/community
集成包。
凭据
访问AWS 文档,注册 AWS 并设置您的凭据。您还需要为您的帐户启用模型访问权限,您可以通过按照以下说明进行操作。
如果您想自动跟踪您的模型调用,还可以通过取消下面的注释设置您的LangSmith API 密钥。
# export LANGCHAIN_TRACING_V2="true"
# export LANGCHAIN_API_KEY="your-api-key"
安装
LangChain BedrockChat
集成位于 @langchain/community
包中。您还需要安装几个官方 AWS 包作为对等依赖项。
有关安装集成包的一般说明,请参阅此部分。
- npm
- yarn
- pnpm
npm i @langchain/community @aws-crypto/sha256-js @aws-sdk/credential-provider-node @smithy/protocol-http @smithy/signature-v4 @smithy/eventstream-codec @smithy/util-utf8 @aws-sdk/types
yarn add @langchain/community @aws-crypto/sha256-js @aws-sdk/credential-provider-node @smithy/protocol-http @smithy/signature-v4 @smithy/eventstream-codec @smithy/util-utf8 @aws-sdk/types
pnpm add @langchain/community @aws-crypto/sha256-js @aws-sdk/credential-provider-node @smithy/protocol-http @smithy/signature-v4 @smithy/eventstream-codec @smithy/util-utf8 @aws-sdk/types
您还可以通过省略 @aws-sdk/credential-provider-node 依赖项并使用网络入口点,在 Edge 函数或 Cloudflare Workers 等网络环境中使用 BedrockChat。
有关安装集成包的一般说明,请参阅此部分。
- npm
- yarn
- pnpm
npm i @langchain/community @aws-crypto/sha256-js @smithy/protocol-http @smithy/signature-v4 @smithy/eventstream-codec @smithy/util-utf8 @aws-sdk/types
yarn add @langchain/community @aws-crypto/sha256-js @smithy/protocol-http @smithy/signature-v4 @smithy/eventstream-codec @smithy/util-utf8 @aws-sdk/types
pnpm add @langchain/community @aws-crypto/sha256-js @smithy/protocol-http @smithy/signature-v4 @smithy/eventstream-codec @smithy/util-utf8 @aws-sdk/types
实例化
目前,只有 Anthropic、Cohere 和 Mistral 模型支持聊天模型集成。有关来自 AI21 或 Amazon 的基础模型,请参阅文本生成 Bedrock 变体。
有几种不同的方法可以使用 AWS 进行身份验证 - 以下示例依赖于在环境变量中设置的访问密钥、秘密访问密钥和区域。
import { BedrockChat } from "@langchain/community/chat_models/bedrock";
const llm = new BedrockChat({
model: "anthropic.claude-3-5-sonnet-20240620-v1:0",
region: process.env.BEDROCK_AWS_REGION,
credentials: {
accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID!,
secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY!,
},
// endpointUrl: "custom.amazonaws.com",
// modelKwargs: {
// anthropic_version: "bedrock-2023-05-31",
// },
});
调用
const aiMsg = await llm.invoke([
[
"system",
"You are a helpful assistant that translates English to French. Translate the user sentence.",
],
["human", "I love programming."],
]);
aiMsg;
AIMessage {
"content": "J'adore la programmation.",
"additional_kwargs": {
"id": "msg_bdrk_01RwhfuWkLLcp7ks1X3u8bwd"
},
"response_metadata": {
"type": "message",
"role": "assistant",
"model": "claude-3-5-sonnet-20240620",
"stop_reason": "end_turn",
"stop_sequence": null,
"usage": {
"input_tokens": 29,
"output_tokens": 11
}
},
"tool_calls": [],
"invalid_tool_calls": []
}
console.log(aiMsg.content);
J'adore la programmation.
链式调用
我们可以像这样使用链式调用我们的模型和提示模板。
import { ChatPromptTemplate } from "@langchain/core/prompts";
const prompt = ChatPromptTemplate.fromMessages([
[
"system",
"You are a helpful assistant that translates {input_language} to {output_language}.",
],
["human", "{input}"],
]);
const chain = prompt.pipe(llm);
await chain.invoke({
input_language: "English",
output_language: "German",
input: "I love programming.",
});
AIMessage {
"content": "Here's the German translation:\n\nIch liebe Programmieren.",
"additional_kwargs": {
"id": "msg_bdrk_01RtUH3qrYJPUdutYoxphFkv"
},
"response_metadata": {
"type": "message",
"role": "assistant",
"model": "claude-3-5-sonnet-20240620",
"stop_reason": "end_turn",
"stop_sequence": null,
"usage": {
"input_tokens": 23,
"output_tokens": 18
}
},
"tool_calls": [],
"invalid_tool_calls": []
}
工具调用
使用 Bedrock 模型进行工具调用与其他模型的工具调用方式类似,但请注意,并非所有 Bedrock 模型都支持工具调用。有关更多信息,请参阅AWS 模型文档。
API 参考
有关所有 BedrockChat
功能和配置的详细文档,请访问 API 参考:https://api.js.langchain.com/classes/langchain_community_chat_models_bedrock.BedrockChat.html