package llm_cli

import (
	"context"
	"os"

	"github.com/anthropics/anthropic-sdk-go"
	"github.com/anthropics/anthropic-sdk-go/bedrock"
	"github.com/aws/aws-sdk-go-v2/config"
)

func NewClaudeClient(bearerToken string) anthropic.Client {
	os.Setenv("AWS_BEARER_TOKEN_BEDROCK", bearerToken)
	client := anthropic.NewClient(
		bedrock.WithLoadDefaultConfig(context.Background(),
			config.WithRegion("us-west-2"),
		),
	)
	return client
}