Просмотр исходного кода

fix: remove legacy Claude 2 series models from Bedrock provider (#10501)

Co-authored-by: Roo Code <[email protected]>
roomote[bot] 2 недель назад
Родитель
Сommit
e287a82144

+ 0 - 33
packages/types/src/providers/bedrock.ts

@@ -264,39 +264,6 @@ export const bedrockModels = {
 		inputPrice: 0.25,
 		outputPrice: 1.25,
 	},
-	"anthropic.claude-2-1-v1:0": {
-		maxTokens: 4096,
-		contextWindow: 100_000,
-		supportsImages: false,
-		supportsPromptCache: false,
-		supportsNativeTools: true,
-		defaultToolProtocol: "native",
-		inputPrice: 8.0,
-		outputPrice: 24.0,
-		description: "Claude 2.1",
-	},
-	"anthropic.claude-2-0-v1:0": {
-		maxTokens: 4096,
-		contextWindow: 100_000,
-		supportsImages: false,
-		supportsPromptCache: false,
-		supportsNativeTools: true,
-		defaultToolProtocol: "native",
-		inputPrice: 8.0,
-		outputPrice: 24.0,
-		description: "Claude 2.0",
-	},
-	"anthropic.claude-instant-v1:0": {
-		maxTokens: 4096,
-		contextWindow: 100_000,
-		supportsImages: false,
-		supportsPromptCache: false,
-		supportsNativeTools: true,
-		defaultToolProtocol: "native",
-		inputPrice: 0.8,
-		outputPrice: 2.4,
-		description: "Claude Instant",
-	},
 	"deepseek.r1-v1:0": {
 		maxTokens: 32_768,
 		contextWindow: 128_000,

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

@@ -122,7 +122,7 @@ describe("AwsBedrockHandler with invokedModelId", () => {
 						trace: {
 							promptRouter: {
 								invokedModelId:
-									"arn:aws:bedrock:us-west-2:699475926481:inference-profile/us.anthropic.claude-2-1-v1:0",
+									"arn:aws:bedrock:us-west-2:699475926481:inference-profile/us.anthropic.claude-3-opus-20240229-v1:0",
 								usage: {
 									inputTokens: 150,
 									outputTokens: 250,
@@ -162,12 +162,12 @@ describe("AwsBedrockHandler with invokedModelId", () => {
 		}
 
 		// Verify that getModelById was called with the id, not the full arn
-		expect(getModelByIdSpy).toHaveBeenCalledWith("anthropic.claude-2-1-v1:0", "inference-profile")
+		expect(getModelByIdSpy).toHaveBeenCalledWith("anthropic.claude-3-opus-20240229-v1:0", "inference-profile")
 
 		// Verify that getModel returns the updated model info
 		const costModel = handler.getModel()
 		//expect(costModel.id).toBe("anthropic.claude-3-5-sonnet-20240620-v1:0")
-		expect(costModel.info.inputPrice).toBe(8)
+		expect(costModel.info.inputPrice).toBe(15)
 
 		// Verify that a usage event was emitted after updating the costModelConfig
 		const usageEvents = events.filter((event) => event.type === "usage")