Class: RubyAmazonBedrock::PayloadFactory

Inherits:

Object

Object
RubyAmazonBedrock::PayloadFactory

show all

Defined in:: lib/bedrock_runtime/payload_factory.rb

Overview

The PayloadFactory class is responsible for creating instances of payload builders based on the provided model identifier.

Attributes:

model_id [String]: The identifier of the model.
input [String]: The input string for what needs to be generated.
options [Hash]: Additional options for payload creation.

Instance Method Summary collapse

#create ⇒ Hash

Creates a payload using the appropriate builder based on the model identifier.
#initialize(model_id, input, options = {}) ⇒ PayloadFactory constructor

A new instance of PayloadFactory.
#models_to_builders ⇒ Hash

Defines a mapping from model identifiers to their respective builder classes.

Constructor Details

#initialize(model_id, input, options = {}) ⇒ `PayloadFactory`

Returns a new instance of PayloadFactory.

# File 'lib/bedrock_runtime/payload_factory.rb', line 32

def initialize(model_id, input, options = {})
  @model_id = model_id
  @prompt = input
  @options = options
end

Instance Method Details

#create ⇒ `Hash`

Creates a payload using the appropriate builder based on the model identifier.

Returns:

(Hash) —

The built payload.

Raises:

(UnknownModelError) —

if the model identifier is not recognized.

# File 'lib/bedrock_runtime/payload_factory.rb', line 42

def create
  builder_class = models_to_builders[@model_id]

  raise UnknownModelError, "Unknown modelId: #{@model_id}" unless builder_class

  builder_class.new(@prompt, @options)
end

#models_to_builders ⇒ `Hash`

Defines a mapping from model identifiers to their respective builder classes.