Method withAWSBedrockOptions

intercept/messages/base.go:267–312

(ctx context.Context, cfg *aibconfig.AWSBedrock)

}

func (*interceptionBase) withAWSBedrockOptions(ctx context.Context, cfg *aibconfig.AWSBedrock) ([]option.RequestOption, error) {
	if cfg == nil {
		return nil, xerrors.New("nil config given")
	}
	if cfg.Region == "" && cfg.BaseURL == "" {
		return nil, xerrors.New("region or base url required")
	}
	if cfg.AccessKey == "" {
		return nil, xerrors.New("access key required")
	}
	if cfg.AccessKeySecret == "" {
		return nil, xerrors.New("access key secret required")
	}
	if cfg.Model == "" {
		return nil, xerrors.New("model required")
	}
	if cfg.SmallFastModel == "" {
		return nil, xerrors.New("small fast model required")
	}

	opts := []func(*config.LoadOptions) error{
		config.WithRegion(cfg.Region),
		config.WithCredentialsProvider(
			credentials.NewStaticCredentialsProvider(
				cfg.AccessKey,
				cfg.AccessKeySecret,
				"",
			),
		),
	}

	awsCfg, err := config.LoadDefaultConfig(ctx, opts...)
	if err != nil {
		return nil, xerrors.Errorf("failed to load AWS Bedrock config: %w", err)
	}

	var out []option.RequestOption
	out = append(out, bedrock.WithConfig(awsCfg))

	// If a custom base URL is set, override the default endpoint constructed by the bedrock middleware.
	if cfg.BaseURL != "" {
		out = append(out, option.WithBaseURL(cfg.BaseURL))
	}

	return out, nil
}

// augmentRequestForBedrock will change the model used for the request since AWS Bedrock doesn't support
// Anthropics' model names. It also converts adaptive thinking to enabled with a budget for models that