Skip to content

Commit 0ebbe91

Browse files
committed
fix: update test to use current model ID anthropic.claude-3-5-sonnet-20241022-v2:0
1 parent 89d4063 commit 0ebbe91

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/api/providers/__tests__/bedrock-invokedModelId.spec.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ describe("AwsBedrockHandler with invokedModelId", () => {
122122
trace: {
123123
promptRouter: {
124124
invokedModelId:
125-
"arn:aws:bedrock:us-west-2:699475926481:inference-profile/us.anthropic.claude-2-1-v1:0",
125+
"arn:aws:bedrock:us-west-2:699475926481:inference-profile/us.anthropic.claude-3-5-sonnet-20241022-v2:0",
126126
usage: {
127127
inputTokens: 150,
128128
outputTokens: 250,
@@ -162,7 +162,7 @@ describe("AwsBedrockHandler with invokedModelId", () => {
162162
}
163163

164164
// Verify that getModelById was called with the id, not the full arn
165-
expect(getModelByIdSpy).toHaveBeenCalledWith("anthropic.claude-2-1-v1:0", "inference-profile")
165+
expect(getModelByIdSpy).toHaveBeenCalledWith("anthropic.claude-3-5-sonnet-20241022-v2:0", "inference-profile")
166166

167167
// Verify that getModel returns the updated model info
168168
// 在这里模拟 invokedModelId 后模型的 inputPrice 被更新为 8

0 commit comments

Comments
 (0)