Skip to content

Commit ae0711e

Browse files
authored
Merge branch 'main' into build/database-docker
2 parents 1e1380e + 56ffa22 commit ae0711e

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

src/config/modelProviders/bedrock.ts

+4
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ const Bedrock: ModelProviderCard = {
3737
'Claude 3 Opus 是 Anthropic 最强大的人工智能模型,在处理高度复杂的任务方面具备顶尖性能。该模型能够以非凡的流畅性和类似人类的理解能力引导开放式的提示和未可见的场景。Claude 3 Opus 向我们展示生成式人工智能的美好前景。 Claude 3 Opus 可以处理图像和返回文本输出,并且提供 200K 上下文窗口。',
3838
displayName: 'Claude 3 Opus',
3939
enabled: true,
40+
functionCall: true,
4041
id: 'anthropic.claude-3-opus-20240229-v1:0',
4142
tokens: 200_000,
4243
vision: true,
@@ -46,6 +47,7 @@ const Bedrock: ModelProviderCard = {
4647
'Anthropic 推出的 Claude 3 Sonnet 模型在智能和速度之间取得理想的平衡,尤其是在处理企业工作负载方面。该模型提供最大的效用,同时价格低于竞争产品,并且其经过精心设计,是大规模部署人工智能的可信赖、高耐久性骨干模型。 Claude 3 Sonnet 可以处理图像和返回文本输出,并且提供 200K 上下文窗口。',
4748
displayName: 'Claude 3 Sonnet',
4849
enabled: true,
50+
functionCall: true,
4951
id: 'anthropic.claude-3-sonnet-20240229-v1:0',
5052
tokens: 200_000,
5153
vision: true,
@@ -55,6 +57,7 @@ const Bedrock: ModelProviderCard = {
5557
'Claude 3.5 Sonnet 提高了行业的智能标准, 在广泛的基准测试中超越了竞争对手模型以及 Claude 3 Opus, 以中端模型的速度和成本,展现出卓越性能。 Claude 3.5 Sonnet 可以处理图像和返回文本输出,并且提供 200K 上下文窗口。',
5658
displayName: 'Claude 3.5 Sonnet',
5759
enabled: true,
60+
functionCall: true,
5861
id: 'anthropic.claude-3-5-sonnet-20240620-v1:0',
5962
tokens: 200_000,
6063
vision: true,
@@ -64,6 +67,7 @@ const Bedrock: ModelProviderCard = {
6467
'Claude 3 Haiku 是 Anthropic 最快速、最紧凑的模型,具有近乎即时的响应能力。该模型可以快速回答简单的查询和请求。客户将能够构建模仿人类交互的无缝人工智能体验。 Claude 3 Haiku 可以处理图像和返回文本输出,并且提供 200K 上下文窗口。',
6568
displayName: 'Claude 3 Haiku',
6669
enabled: true,
70+
functionCall: true,
6771
id: 'anthropic.claude-3-haiku-20240307-v1:0',
6872
tokens: 200_000,
6973
vision: true,

src/libs/agent-runtime/bedrock/index.ts

+3-2
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import { experimental_buildLlama2Prompt } from 'ai/prompts';
77
import { LobeRuntimeAI } from '../BaseAI';
88
import { AgentRuntimeErrorType } from '../error';
99
import { ChatCompetitionOptions, ChatStreamPayload, ModelProvider } from '../types';
10-
import { buildAnthropicMessages } from '../utils/anthropicHelpers';
10+
import { buildAnthropicMessages, buildAnthropicTools } from '../utils/anthropicHelpers';
1111
import { AgentRuntimeError } from '../utils/createError';
1212
import { debugStream } from '../utils/debugStream';
1313
import { StreamingResponse } from '../utils/response';
@@ -53,7 +53,7 @@ export class LobeBedrockAI implements LobeRuntimeAI {
5353
payload: ChatStreamPayload,
5454
options?: ChatCompetitionOptions,
5555
): Promise<Response> => {
56-
const { max_tokens, messages, model, temperature, top_p } = payload;
56+
const { max_tokens, messages, model, temperature, top_p, tools } = payload;
5757
const system_message = messages.find((m) => m.role === 'system');
5858
const user_messages = messages.filter((m) => m.role !== 'system');
5959

@@ -65,6 +65,7 @@ export class LobeBedrockAI implements LobeRuntimeAI {
6565
messages: buildAnthropicMessages(user_messages),
6666
system: system_message?.content as string,
6767
temperature: temperature,
68+
tools: buildAnthropicTools(tools),
6869
top_p: top_p,
6970
}),
7071
contentType: 'application/json',

0 commit comments

Comments
 (0)