chore: allows an environmental override of the default AWS region and… (

#1025) * chore: allows an environmental override of the default AWS region and using it for bedrock Signed-off-by: Alex Jones <alexsimonjones@gmail.com> * chore: missing provider region Signed-off-by: Alex Jones <alexsimonjones@gmail.com> --------- Signed-off-by: Alex Jones <alexsimonjones@gmail.com>
k8sgpt-ai · Mar 22, 2024 · 8f8f5c6 · 8f8f5c6
1 parent 3c1c055
commit 8f8f5c6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git 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
@@ -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 {