ruby-amazon-bedrock 0.1.1

data/lib/bedrock_runtime/payload_builders/cohere/embed_english_v3.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_relative 'embed_base'
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module Cohere
+      # EmbedEnglishV3 is a subclass of Base. It provides functionalities specific to the Cohere
+      # Embed English model.
+      #
+      # @see https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/providers?model=cohere.embed-english-v3
+      # for more information about the Cohere model.
+      class EmbedEnglishV3 < EmbedBase
+        # Returns the model ID for the Cohere Embed English model.
+        #
+        # @return [String] 'cohere.embed-english-v3'
+        def model_id
+          'cohere.embed-english-v3'
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/cohere/embed_multilingual_v3.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module Cohere
+      # EmbedMultilingualV3 is a subclass of Base. It provides functionalities specific to the Cohere
+      # Embed Multilingual model.
+      #
+      # @see https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/providers?model=cohere.embed-multilingual-v3
+      # for more information about the Cohere model.
+      class EmbedMultilingualV3 < EmbedBase
+        # Returns the model ID for the Cohere Embed Multilingual model.
+        #
+        # @return [String] 'cohere.embed-multilingual-v3'
+        def model_id
+          'cohere.embed-multilingual-v3'
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/meta/base.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module Meta
+      # Builds and returns a payload hash suitable for the Meta model processing.
+      # This method constructs a payload with specific parameters like `model_id`,
+      # `content_type`, `accept`, and a `body` that includes various AI-related settings.
+      #
+      # @return [Hash] The constructed payload containing AI model parameters and settings.
+      class Base < RubyAmazonBedrock::PayloadBuilders::Base
+        # Constructs and returns a payload formatted for text generation requests.
+        # This method assembles the necessary data structure for processing text input through
+        # an AI model, with various parameters to guide the generation process.
+        #
+        # @return [Hash] A structured payload containing:
+        #   - :model_id [String] Identifier for the AI model that will process the text generation request.
+        #   - :content_type [String] Specifies the content type of the payload, set to 'application/json'.
+        #   - :accept [String] Indicates the MIME type for the expected response.
+        #   - :body [String] A JSON string encapsulating the following details:
+        #     - :prompt [String] The input text for the model to generate content from.
+        #     - :max_gen_len [Integer] Maximum length for the generated content, measured in tokens.
+        #     - :temperature [Float] A parameter controlling the randomness in the generated content.
+        #     - :top_p [Float] Nucleus sampling parameter controlling the diversity of the generated text.
+        def build
+          {
+            model_id: model_id,
+            content_type: 'application/json',
+            accept: '*/*',
+            body: {
+              prompt: @input,
+              max_gen_len: 512,
+              temperature: 0.2,
+              top_p: 0.9
+            }.to_json
+          }
+        end
+
+        def model_id
+          # noop
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/meta/llama213b_chat_v1.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module Meta
+      # Llama213bChatV1 is a subclass of Base. It provides functionalities specific to the Meta
+      # Llama 2 Chat 13B model.
+      #
+      # @see https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/providers?model=meta.llama2-13b-chat-v1
+      # for more information about the Meta model.
+      class Llama213bChatV1 < Base
+        # Returns the model ID for the Meta Llama 2 Chat 13B model.
+        #
+        # @return [String] 'meta.llama2-13b-chat-v1'
+        def model_id
+          'meta.llama2-13b-chat-v1'
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/meta/llama270b_chat_v1.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module Meta
+      # Llama270bChatV1 is a subclass of Base. It provides functionalities specific to the Meta
+      # Llama 2 Chat 70B model.
+      #
+      # @see https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/providers?model=meta.llama2-70b-chat-v1
+      # for more information about the Meta model.
+      class Llama270bChatV1 < Base
+        # Returns the model ID for the Meta Llama 2 Chat 70B model.
+        #
+        # @return [String] 'meta.llama2-70b-chat-v1'
+        def model_id
+          'meta.llama2-70b-chat-v1'
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/stability_ai/base.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module StabilityAi
+      # Builds and returns a payload hash suitable for the Stability AI model processing.
+      # This method constructs a payload with specific parameters like `model_id`,
+      # `content_type`, `accept`, and a `body` that includes various AI-related settings.
+      #
+      # @return [Hash] The constructed payload containing AI model parameters and settings.
+      class Base < RubyAmazonBedrock::PayloadBuilders::Base
+        # Constructs and returns a structured payload for processing by an AI model.
+        # This method assembles data in a format suitable for tasks requiring configurable prompts,
+        # such as text or image generation, with control over creativity and randomness.
+        #
+        # @return [Hash] A hash containing the necessary details for the AI model to process:
+        #   - :model_id [String] Identifier for the AI model to which the request is directed.
+        #   - :content_type [String] Specifies the content type of the payload, typically set to 'application/json'.
+        #   - :accept [String] Indicates the MIME type that the response should conform to.
+        #   - :body [String] A JSON string including:
+        #     - :text_prompts [Array<Hash>] An array of hashes, each containing a 'text' key with a string value
+        #       representing a prompt.
+        #     - :cfg_scale [Integer] A parameter influencing the level of control versus freedom in content generation.
+        #     - :seed [Integer] Seed value for deterministic outputs, enabling reproducibility of results.
+        #     - :steps [Integer] Specifies the number of steps the model should take in generating the output.
+        def build
+          {
+            model_id: model_id,
+            content_type: 'application/json',
+            accept: '*/*',
+            body: {
+              text_prompts: [
+                { text: @input }
+              ],
+              cfg_scale: 10,
+              seed: 0,
+              steps: 50
+            }.to_json
+          }
+        end
+
+        def model_id
+          # noop
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module StabilityAi
+      # StableDifussionXlV0 is a subclass of Base. It provides functionalities specific to the Stability AI
+      # SDXL 0.8 model.
+      #
+      # @see https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/providers?model=stability.stable-diffusion-xl-v0
+      # for more information about the Anthropic model.
+      class StableDiffusionXlV0 < Base
+        # Returns the model ID for the SDXL 0.8 model.
+        #
+        # @return [String] 'stability.stable-diffusion-xl-v0'
+        def model_id
+          'stability.stable-diffusion-xl-v0'
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module RubyAmazonBedrock
+  module PayloadBuilders
+    module StabilityAi
+      # StableDifussionXlV1 is a subclass of Base. It provides functionalities specific to the Stability AI
+      # SDXL 1.0 model.
+      #
+      # @see https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/providers?model=stability.stable-diffusion-xl-v1
+      # for more information about the Stability AI model.
+      class StableDiffusionXlV1 < Base
+        # Returns the model ID for the SDXL 1.0 model.
+        #
+        # @return [String] 'stability.stable-diffusion-xl-v1'
+        def model_id
+          'stability.stable-diffusion-xl-v1'
+        end
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/payload_factory.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require_relative 'payload_builders/ai_21_labs/j2_mid_v1'
+require_relative 'payload_builders/ai_21_labs/j2_ultra_v1'
+require_relative 'payload_builders/amazon/titan_image_generator_v1'
+require_relative 'payload_builders/amazon/titan_text_lite_v1'
+require_relative 'payload_builders/amazon/titan_text_express_v1'
+require_relative 'payload_builders/anthropic/claude_v1'
+require_relative 'payload_builders/anthropic/claude_instant_v1'
+require_relative 'payload_builders/anthropic/claude_v2'
+require_relative 'payload_builders/cohere/command_light_text_v14'
+require_relative 'payload_builders/cohere/command_text_v14'
+require_relative 'payload_builders/cohere/embed_english_v3'
+require_relative 'payload_builders/cohere/embed_multilingual_v3'
+require_relative 'payload_builders/meta/llama213b_chat_v1'
+require_relative 'payload_builders/meta/llama270b_chat_v1'
+require_relative 'payload_builders/stability_ai/stable_diffusion_xl_v0'
+require_relative 'payload_builders/stability_ai/stable_diffusion_xl_v1'
+
+module RubyAmazonBedrock
+  # 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.
+  #
+  class PayloadFactory
+    def initialize(model_id, input, options = {})
+      @model_id = model_id
+      @input = input
+      @options = options
+    end
+
+    # Creates a payload using the appropriate builder based on the model identifier.
+    #
+    # @return [Hash] The built payload.
+    # @raise [UnknownModelError] if the model identifier is not recognized.
+    def create
+      builder_class = models_to_builders[@model_id]
+
+      raise UnknownModelError, "Unknown modelId: #{@model_id}" unless builder_class
+
+      builder_class.new(@input, @options)
+    end
+
+    # Defines a mapping from model identifiers to their respective builder classes.
+    #
+    # @return [Hash] The mapping of model identifiers to builder classes.
+    def models_to_builders
+      {
+        'ai21labs.j2-mid-v1' => PayloadBuilders::Ai21Labs::J2MidV1,
+        'ai21labs.j2-ultra-v1' => PayloadBuilders::Ai21Labs::J2UltraV1,
+        'amazon.titan-image-generator-v1' => PayloadBuilders::Amazon::TitanImageGeneratorV1,
+        'amazon.titan-text-lite-v1' => PayloadBuilders::Amazon::TitanTextLiteV1,
+        'amazon.titan-text-express-v1' => PayloadBuilders::Amazon::TitanTextExpressV1,
+        'anthropic.claude-v1' => PayloadBuilders::Anthropic::ClaudeV1,
+        'anthropic.claude-instant-v1' => PayloadBuilders::Anthropic::ClaudeInstantV1,
+        'anthropic.claude-v2' => PayloadBuilders::Anthropic::ClaudeV2,
+        'cohere.command-light-text-v14' => PayloadBuilders::Cohere::CommandLightTextV14,
+        'cohere.command-text-v14' => PayloadBuilders::Cohere::CommandTextV14,
+        'cohere.embed-english-v3' => PayloadBuilders::Cohere::EmbedEnglishV3,
+        'cohere.embed-multilingual-v3' => PayloadBuilders::Cohere::EmbedMultilingualV3,
+        'meta.llama2-13b-chat-v1' => PayloadBuilders::Meta::Llama213bChatV1,
+        'meta.llama2-70b-chat-v1' => PayloadBuilders::Meta::Llama270bChatV1,
+        'stability.stable-diffusion-xl-v0' => PayloadBuilders::StabilityAi::StableDiffusionXlV0,
+        'stability.stable-diffusion-xl-v1' => PayloadBuilders::StabilityAi::StableDiffusionXlV1
+      }
+    end
+  end
+end

data/ruby-amazon-bedrock.gemspec

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative "lib/bedrock/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "ruby-amazon-bedrock"
+  spec.version = Ruby::Amazon::Bedrock::VERSION
+  spec.authors = ["Alvaro Delgado"]
+  spec.email = ["hola@alvarodelgado.dev"]
+
+  spec.summary = "Seamless Integration with Amazon Bedrock for AI-Powered Text and Image Generation in Ruby."
+  spec.description = "The ruby-amazon-bedrock gem offers Ruby developers an efficient and user-friendly interface to \
+    Amazon Bedrock, a powerful library for AI-driven text and image generation. This gem simplifies the process of \
+    connecting to Amazon Bedrock's APIs, enabling developers to easily harness the capabilities of advanced machine \
+    learning models for generating high-quality text and images."
+  spec.homepage = "https://github.com/AAlvAAro/ruby-amazon-bedrock"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 2.6.0"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/AAlvAAro/ruby-amazon-bedrock"
+  spec.metadata["changelog_uri"] = "https://github.com/AAlvAAro/ruby-amazon-bedrock/blob/main/CHANGELOG.md"
+  spec.metadata["rubygems_mfa_required"] = "true"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (File.expand_path(f) == __FILE__) ||
+        f.start_with?(*%w[bin/ test/ spec/ features/ .git .github appveyor Gemfile])
+      f.end_with?(*%w[.gem])
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "aws-sdk-bedrockruntime", "~> 1.0"
+end

data/sig/ruby/amazon/bedrock.rbs

@@ -0,0 +1,8 @@
+module Ruby
+  module Amazon
+    module Bedrock
+      VERSION: String
+      # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+    end
+  end
+end

data/spec/bedrock_runtime/client_spec.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+require 'bedrock_runtime/client'
+require 'spec_helper'
+require 'webmock/rspec'
+
+RSpec.describe RubyAmazonBedrock::Client do
+  let(:client) do
+    described_class.new(
+      region: ENV.fetch('AWS_REGION', nil),
+      access_key_id: ENV.fetch('AWS_ACCESS_KEY_ID', nil),
+      secret_access_key: ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
+    )
+  end
+
+  describe '#invoke_model' do
+    models = [
+      # 'amazon.titan-image-generator-v1', This model is currently in preview release. Changes may be made to this model.
+      'amazon.titan-text-lite-v1',
+      'amazon.titan-text-express-v1',
+      'anthropic.claude-instant-v1',
+      'anthropic.claude-v1',
+      'anthropic.claude-v2',
+      'cohere.command-light-text-v14',
+      'cohere.command-text-v14',
+      'cohere.embed-english-v3',
+      'cohere.embed-multilingual-v3',
+      'meta.llama2-13b-chat-v1',
+      'meta.llama2-70b-chat-v1',
+      'stability.stable-diffusion-xl-v0',
+      'stability.stable-diffusion-xl-v1'
+    ]
+
+    models.each do |model|
+      it "invokes #{model} model with the given payload" do
+        VCR.use_cassette("models/#{model}") do
+          response = client.invoke_model(id: model, input: 'Give me a short list of steps for creating a Ruby gem', options: {})
+          expect(response).to be_a(Hash)
+        end
+      end
+    end
+
+    context 'when the model is not found' do
+      it 'raises an UnkownModelError' do
+        VCR.use_cassette('models/not_found') do
+          expect { client.invoke_model(id: 'unknown-model', input: 'Hello World', options: {}) }.to raise_error(RubyAmazonBedrock::UnknownModelError)
+        end
+      end
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/ai_21_labs/base_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/ai_21_labs/base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Ai21Labs::Base do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: input,
+      maxTokens: 200,
+      temperature: 0,
+      topP: 250,
+      stop_sequences: [],
+      countPenalty: { scale: 0 },
+      presencePenalty: { scale: 0 },
+      frequencyPenalty: { scale: 0 }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('*/*')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1_spec.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Ai21Labs::J2MidV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: input,
+      maxTokens: 200,
+      temperature: 0,
+      topP: 250,
+      stop_sequences: [],
+      countPenalty: { scale: 0 },
+      presencePenalty: { scale: 0 },
+      frequencyPenalty: { scale: 0 }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:model_id]).to eq('ai21labs.j2-mid-v1')
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('*/*')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1_spec.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Ai21Labs::J2UltraV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: input,
+      maxTokens: 200,
+      temperature: 0,
+      topP: 250,
+      stop_sequences: [],
+      countPenalty: { scale: 0 },
+      presencePenalty: { scale: 0 },
+      frequencyPenalty: { scale: 0 }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:model_id]).to eq('ai21labs.j2-ultra-v1')
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('*/*')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/amazon/base_spec.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/amazon/base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Amazon::Base do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      inputText: input,
+      textGenerationConfig: {
+        maxTokenCount: 4096,
+        stopSequences: [],
+        temperature: 0,
+        topP: 1
+      }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('*/*')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/amazon/titan_image_generator_v1_spec.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/amazon/titan_image_generator_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Amazon::TitanImageGeneratorV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      taskType: "TEXT_IMAGE",
+      textToImageParams: {
+        text: input,
+        negativeTtext: nil
+      },
+      imageGenerationConfig: {
+        numberOfImages: 1,
+        quality: 'standard',
+        height: 1024,
+        width: 1024,
+        cfgScale: 8.0,
+        seed: 0
+      }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:model_id]).to eq('amazon.titan-image-generator-v1')
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('application/json')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/amazon/titan_text_express_v1_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/amazon/titan_text_express_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Amazon::TitanTextExpressV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      inputText: input,
+      textGenerationConfig: {
+        maxTokenCount: 4096,
+        stopSequences: [],
+        temperature: 0,
+        topP: 1
+      }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:model_id]).to eq('amazon.titan-text-express-v1')
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('*/*')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/amazon/titan_text_lite_v1_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/amazon/titan_text_lite_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Amazon::TitanTextLiteV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      inputText: input,
+      textGenerationConfig: {
+        maxTokenCount: 4096,
+        stopSequences: [],
+        temperature: 0,
+        topP: 1
+      }
+    }.to_json
+  end
+
+  describe '#build' do
+    it 'returns a hash with the expected structure' do
+      payload_builder = described_class.new(input, options)
+      payload = payload_builder.build
+
+      expect(payload[:model_id]).to eq('amazon.titan-text-lite-v1')
+      expect(payload[:content_type]).to eq('application/json')
+      expect(payload[:accept]).to eq('*/*')
+      expect(payload[:body]).to eq(body)
+    end
+  end
+end