diff --git a/cmd/auth/add.go b/cmd/auth/add.go index ae9735b425..ed455d9add 100644 --- a/cmd/auth/add.go +++ b/cmd/auth/add.go @@ -45,6 +45,9 @@ var addCmd = &cobra.Command{ _ = cmd.MarkFlagRequired("endpointname") _ = cmd.MarkFlagRequired("providerRegion") } + if strings.ToLower(backend) == "amazonbedrock" { + _ = cmd.MarkFlagRequired("providerRegion") + } }, Run: func(cmd *cobra.Command, args []string) { diff --git a/pkg/ai/amazonbedrock.go b/pkg/ai/amazonbedrock.go index 8f3ecdbfaf..f9cd80fa61 100644 --- a/pkg/ai/amazonbedrock.go +++ b/pkg/ai/amazonbedrock.go @@ -4,6 +4,7 @@ import ( "context" "encoding/json" "fmt" + "os" "github.com/aws/aws-sdk-go/aws" "github.com/aws/aws-sdk-go/aws/session" @@ -76,6 +77,9 @@ func GetModelOrDefault(model string) string { // GetModelOrDefault check config region func GetRegionOrDefault(region string) string { + if os.Getenv("AWS_DEFAULT_REGION") != "" { + region = os.Getenv("AWS_DEFAULT_REGION") + } // Check if the provided model is in the list for _, m := range BEDROCKER_SUPPORTED_REGION { if m == region {