bedrock_support

package

v0.3.50 Latest Latest Go to latest Published: Feb 24, 2025 License: Apache-2.0 Imports: 3 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type AI21 struct {
	// contains filtered or unexported fields
}

func (a *AI21) GetCompletion(ctx context.Context, prompt string, modelConfig BedrockModelConfig) ([]byte, error)

type AI21Response struct {
	// contains filtered or unexported fields
}

func (a *AI21Response) ParseResponse(rawResponse []byte) (string, error)

type AmazonCompletion struct {
	// contains filtered or unexported fields
}

func (a *AmazonCompletion) GetCompletion(ctx context.Context, prompt string, modelConfig BedrockModelConfig) ([]byte, error)

type AmazonResponse struct {
	// contains filtered or unexported fields
}

func (a *AmazonResponse) ParseResponse(rawResponse []byte) (string, error)

type BedrockModel struct {
	Name       string
	Completion ICompletion
	Response   IResponse
	Config     BedrockModelConfig
}

type BedrockModelConfig struct {
	MaxTokens   int
	Temperature float32
	TopP        float32
}

type CohereCompletion struct {
	// contains filtered or unexported fields
}

func (a *CohereCompletion) GetCompletion(ctx context.Context, prompt string, modelConfig BedrockModelConfig) ([]byte, error)

type CohereResponse struct {
	// contains filtered or unexported fields
}

func (a *CohereResponse) ParseResponse(rawResponse []byte) (string, error)

type ICompletion interface {
	GetCompletion(ctx context.Context, prompt string, modelConfig BedrockModelConfig) ([]byte, error)
}

type IResponse interface {
	ParseResponse(rawResponse []byte) (string, error)
}