1package anthropic
 2
 3import (
 4	"cmp"
 5	"os"
 6	"strings"
 7
 8	"github.com/aws/aws-sdk-go-v2/aws"
 9	"github.com/aws/smithy-go/auth/bearer"
10)
11
12func bedrockBasicAuthConfig(apiKey string) aws.Config {
13	return aws.Config{
14		Region:                  cmp.Or(os.Getenv("AWS_REGION"), "us-east-1"),
15		BearerAuthTokenProvider: bearer.StaticTokenProvider{Token: bearer.Token{Value: apiKey}},
16	}
17}
18
19func bedrockPrefixModelWithRegion(modelID string) string {
20	region := os.Getenv("AWS_REGION")
21	if len(region) < 2 {
22		region = "us-east-1"
23	}
24	prefix := region[:2] + "."
25	if strings.HasPrefix(modelID, prefix) {
26		return modelID
27	}
28	return prefix + modelID
29}