🔖 chore: update bedrock model maping (#169)

This commit is contained in:
Martial BE 2024-04-25 13:28:30 +08:00
parent 2e41f70e66
commit 58a0b6f9bc
No known key found for this signature in database
GPG Key ID: D06C32DF0EDB9084

View File

@ -34,6 +34,7 @@ func GetCategory(modelName string) (*Category, error) {
func GetModelName(modelName string) string {
bedrockMap := map[string]string{
"claude-3-opus-20240229": "anthropic.claude-3-opus-20240229-v1:0",
"claude-3-sonnet-20240229": "anthropic.claude-3-sonnet-20240229-v1:0",
"claude-3-haiku-20240307": "anthropic.claude-3-haiku-20240307-v1:0",
"claude-2.1": "anthropic.claude-v2:1",