ruby-amazon-bedrock 0.2.2 → 0.2.4

data/lib/bedrock_runtime/response_builders/cohere_embed.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module ResponseBuilders
+    # The CohereEmbed class is responsible for parsing and building a structured response from a raw text response.
+    # It converts the HTTP response for an Cohere Embed model into a structured format to make it easier to access
+    # the response data.
+    class CohereEmbed
+      # Initializes a new instance of the Text class.
+      #
+      # @param response [Object] The raw response object with is an HTTP response.
+      def initialize(response, _options = {})
+        @response = response
+      end
+
+      # Builds and returns a structured representation of the raw text response.
+      # This method parses the raw response body as JSON and symbolizes the names
+      # for easier access in Ruby.
+      #
+      # @return [Hash] A hash representing the structured data from the response body.
+      # The keys of the hash are symbolized for convenient access.
+      def build
+        response = JSON.parse(@response.body.read, symbolize_names: true)
+
+        {
+          text: response[:texts].first,
+          full_response: response
+        }
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/response_builders/{text.rb → meta.rb}

@@ -2,13 +2,14 @@
 
 module RubyAmazonBedrock
   module ResponseBuilders
-    # The Text class is responsible for parsing and building a structured response from a raw text response.
-    # It converts the HTTP response into a structured format to make it easier to access the data.
-    class Text
+    # The Meta class is responsible for parsing and building a structured response from a raw text response.
+    # It converts the HTTP response for an Meta model into a structured format to make it easier to access
+    # the response data.
+    class Meta
       # Initializes a new instance of the Text class.
       #
       # @param response [Object] The raw response object with is an HTTP response.
-      def initialize(response)
+      def initialize(response, _options = {})
         @response = response
       end
 
@@ -19,7 +20,12 @@ module RubyAmazonBedrock
       # @return [Hash] A hash representing the structured data from the response body.
       # The keys of the hash are symbolized for convenient access.
       def build
-        JSON.parse(@response.body.read, symbolize_names: true)
+        response = JSON.parse(@response.body.read, symbolize_names: true)
+
+        {
+          text: response[:generation],
+          full_response: response
+        }
       end
     end
   end

data/lib/bedrock_runtime/response_builders/stability_ai.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+require 'aws-sdk-s3'
+require 'base64'
+require 'digest'
+
+module RubyAmazonBedrock
+  module ResponseBuilders
+    # The StabilityAi class is responsible for handling and processing image data received in a response.
+    # It parses the response, extracts the image data, decodes it from Base64, and saves it as a file.
+    class StabilityAi
+      # Initializes a new instance of the StabilityAi class.
+      #
+      # @param response [Object] The raw response object which is an HTTP response.
+      # The response should contain the image data in Base64 format.
+      # @param options [Hash] Optional parameters, currently supporting :file_path for specifying
+      # the location and name of the file to save the image. Default is 'image.jpg'.
+      def initialize(response, options = {})
+        @response = response
+        @file_path = options[:file_path] || 'image.jpg'
+        @upload = options[:upload] || false
+      end
+
+      # Processes the response to extract and decode image data, then saves it as a file.
+      # This method parses the response body as JSON, extracts the first artifact which is
+      # expected to be an image in Base64 format, decodes it, and writes it to a file.
+      # If the operation is successful, a hash with a success status and file path is returned.
+      # If an error occurs, a hash with a failure status and error details is returned.
+      #
+      # @return [Hash] A hash indicating the result of the operation.
+      # If successful, the hash includes :result set to :success and :file_path.
+      # If upload option is provided it will also include :s3_url after saving the image in S3.
+      # If failure, the hash includes :result set to :failure and :error with exception details.
+      def build
+        response_object = JSON.parse(@response.body.read, symbolize_names: true)
+        image_data = Base64.decode64(response_object[:artifacts].first[:base64])
+        File.binwrite(@file_path, image_data)
+
+        upload_image if can_upload?
+
+        {
+          result: :success,
+          file_path: @file_path,
+          s3_url: s3_url
+        }.compact
+      rescue StandardError => e
+        {
+          result: :failure,
+          error: e
+        }
+      end
+
+      private
+
+      def bucket
+        @bucket ||= ENV.fetch('AWS_S3_BUCKET')
+      end
+
+      def key
+        @key ||= Digest::SHA1.hexdigest(Time.now.to_s)[0..11]
+      end
+
+      def can_upload?
+        @upload && bucket && File.exist?(@file_path)
+      end
+
+      def s3_url
+        s3_resource = Aws::S3::Resource.new
+        object = s3_resource.bucket(bucket).object(key)
+        object.presigned_url(:get)
+      end
+
+      def upload_image
+        s3_client = Aws::S3::Client.new
+        s3_client.put_object(
+          bucket: bucket,
+          key: key,
+          body: File.read(@file_path)
+        )
+      rescue StandardError => e
+        Rails.logger.error("Error uploading image to S3: #{e}")
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/response_factory.rb

@@ -1,19 +1,25 @@
 # frozen_string_literal: true
 
-require_relative 'response_builders/image'
-require_relative 'response_builders/text'
+require_relative 'response_builders/ai_21_labs'
+require_relative 'response_builders/amazon_text'
+require_relative 'response_builders/amazon_image'
+require_relative 'response_builders/anthropic'
+require_relative 'response_builders/cohere_command'
+require_relative 'response_builders/cohere_embed'
+require_relative 'response_builders/meta'
+require_relative 'response_builders/stability_ai'
 
 module RubyAmazonBedrock
   # The ResponseFactory class is a factory for creating different types of response builder objects.
-  # It is designed to instantiate and return either a Text or Image response builder based on the type specified.
+  # It is designed to instantiate and return an object of a specific response builder class
   class ResponseFactory
     # Initializes a new instance of the ResponseFactory class.
     #
-    # @param type [Symbol] The type of response builder to create (:text or :image).
+    # @param model_id [String] The model_id of response builder to create (Amazon Bedrock model id).
     # @param response [Object] The raw response object, typically an HTTP response.
-    # @param optios [Hash] Additional options for the response builder.
-    def initialize(type, response, options = {})
-      @type = type
+    # @param options [Object] optional attributes to customize the response.
+    def initialize(model_id, response, options = {})
+      @model_id = model_id
       @response = response
       @options = options
     end
@@ -22,16 +28,38 @@ module RubyAmazonBedrock
     # This method uses the @type instance variable to determine which type of response
     # builder to instantiate and return.
     #
-    # @return [ResponseBuilders::Text, ResponseBuilders::Image] Depending on the type,
-    # it returns an instance of either ResponseBuilders::Text or ResponseBuilders::Image.
-    # Returns nil if the type does not match any known response builders.
+    # @return [ResponseBuilders::Class] Depending on the model_id,
+    # it returns an instance of any of the response builder classes.
+    # Returns nil if the model_id does not match any known response builders.
     def create
-      case @type
-      when :text
-        ResponseBuilders::Text.new(@response)
-      when :image
-        ResponseBuilders::Image.new(@response, @options)
-      end
+      builder_class = models_to_builders[@model_id]
+      builder_class.new(@response, @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
+      {
+        'ai21.j2-mid-v1' => ResponseBuilders::Ai21Labs,
+        'ai21.j2-ultra-v1' => ResponseBuilders::Ai21Labs,
+        'amazon.titan-image-generator-v1' => ResponseBuilders::AmazonImage,
+        'amazon.titan-text-lite-v1' => ResponseBuilders::AmazonText,
+        'amazon.titan-text-express-v1' => ResponseBuilders::AmazonText,
+        'anthropic.claude-instant-v1' => ResponseBuilders::Anthropic,
+        'anthropic.claude-v1' => ResponseBuilders::Anthropic,
+        'anthropic.claude-v2' => ResponseBuilders::Anthropic,
+        'cohere.command-light-text-v14' => ResponseBuilders::CohereCommand,
+        'cohere.command-text-v14' => ResponseBuilders::CohereCommand,
+        'cohere.embed-english-v3' => ResponseBuilders::CohereEmbed,
+        'cohere.embed-multilingual-v3' => ResponseBuilders::CohereEmbed,
+        'meta.llama2-13b-chat-v1' => ResponseBuilders::Meta,
+        'meta.llama2-70b-chat-v1' => ResponseBuilders::Meta,
+        'meta.llama3-70b-instruct-v1:0' => ResponseBuilders::Meta,
+        'meta.llama3-8b-instruct-v1:0' => ResponseBuilders::Meta,
+        'stability.stable-diffusion-xl-v0' => ResponseBuilders::StabilityAi,
+        'stability.stable-diffusion-xl-v1' => ResponseBuilders::StabilityAi
+      }
     end
   end
 end

data/ruby-amazon-bedrock.gemspec

@@ -36,4 +36,6 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency "aws-sdk-bedrockruntime", "~> 1.0"
+  spec.add_dependency "aws-sdk-s3", "~> 1.0"
+  spec.add_dependency "base64", "~> 0.2"
 end

data/spec/bedrock_runtime/client_spec.rb

@@ -8,7 +8,8 @@ RSpec.describe RubyAmazonBedrock::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)
+      secret_access_key: ENV.fetch('AWS_SECRET_ACCESS_KEY', nil),
+      profile: ENV.fetch('AWS_PROFILE', nil)
     )
   end
 
@@ -26,6 +27,8 @@ RSpec.describe RubyAmazonBedrock::Client do
       'cohere.embed-multilingual-v3',
       'meta.llama2-13b-chat-v1',
       'meta.llama2-70b-chat-v1',
+      'meta.llama3-70b-instruct-v1:0',
+      'meta.llama3-8b-instruct-v1:0',
       'stability.stable-diffusion-xl-v0',
       'stability.stable-diffusion-xl-v1'
     ]

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

@@ -9,7 +9,7 @@ RSpec.describe RubyAmazonBedrock::PayloadBuilders::Ai21Labs::Base do
   let(:body) do
     {
       prompt: prompt,
-      maxTokenCount: 200,
+      maxTokens: 200,
       temperature: 0,
       topP: 1,
       stopSequences: [],

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

@@ -9,7 +9,7 @@ RSpec.describe RubyAmazonBedrock::PayloadBuilders::Ai21Labs::J2MidV1 do
   let(:body) do
     {
       prompt: prompt,
-      maxTokenCount: 200,
+      maxTokens: 200,
       temperature: 0,
       topP: 1,
       stopSequences: [],

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

@@ -9,7 +9,7 @@ RSpec.describe RubyAmazonBedrock::PayloadBuilders::Ai21Labs::J2UltraV1 do
   let(:body) do
     {
       prompt: prompt,
-      maxTokenCount: 200,
+      maxTokens: 200,
       temperature: 0,
       topP: 1,
       stopSequences: [],

data/spec/bedrock_runtime/payload_builders/meta/llama370b_instruct_v1_spec.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/meta/llama370b_instruct_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Meta::Llama370bInstructV1 do
+  let(:prompt) { 'example_prompt' }
+  let(:options) { {} }
+  let(:body) do
+    {
+      prompt: prompt,
+      max_gen_len: 512,
+      temperature: 0.5,
+      top_p: 0.9
+    }.to_json
+  end
+
+  describe '#build' do
+    it_should_behave_like 'a payload builder'
+
+    context 'with custom parameters' do
+      include_context 'meta parameters'
+      it_should_behave_like 'a payload builder'
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/meta/llama38b_instruct_v1_spec.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/meta/llama38b_instruct_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Meta::Llama38bInstructV1 do
+  let(:prompt) { 'example_prompt' }
+  let(:options) { {} }
+  let(:body) do
+    {
+      prompt: prompt,
+      max_gen_len: 512,
+      temperature: 0.5,
+      top_p: 0.9
+    }.to_json
+  end
+
+  describe '#build' do
+    it_should_behave_like 'a payload builder'
+
+    context 'with custom parameters' do
+      include_context 'meta parameters'
+      it_should_behave_like 'a payload builder'
+    end
+  end
+end

data/spec/bedrock_runtime/payload_factory_spec.rb

@@ -19,6 +19,8 @@ RSpec.describe RubyAmazonBedrock::PayloadFactory do
      "cohere.embed-multilingual-v3",
      "meta.llama2-13b-chat-v1",
      "meta.llama2-70b-chat-v1",
+     "meta.llama3-70b-instruct-v1:0",
+     "meta.llama3-8b-instruct-v1:0",
      "stability.stable-diffusion-xl-v0",
      "stability.stable-diffusion-xl-v1"]
   end
@@ -48,7 +50,7 @@ RSpec.describe RubyAmazonBedrock::PayloadFactory do
 
   describe '#models_to_builders' do
     it 'defines a mapping of model identifiers to builder classes' do
-      expect(models_mapping.keys).to eq(bedrock_models)
+      expect(models_mapping.keys).to contain_exactly(*bedrock_models)
     end
   end
 end

data/spec/bedrock_runtime/response_builders/{text_spec.rb → ai_21_labs_spec.rb}

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
 require 'spec_helper'
-require 'bedrock_runtime/response_builders/text'
+require 'bedrock_runtime/response_builders/ai_21_labs'
 
-RSpec.describe RubyAmazonBedrock::ResponseBuilders::Text do
+RSpec.describe RubyAmazonBedrock::ResponseBuilders::Ai21Labs do
   let(:response_body) { { key1: 'value1', key2: 'value2' }.to_json }
   let(:response) { double('response', body: StringIO.new(response_body)) }
 
@@ -14,8 +14,7 @@ RSpec.describe RubyAmazonBedrock::ResponseBuilders::Text do
       result = subject.build
       expect(result).to be_a(Hash)
       expect(result.keys).to all(be_a(Symbol))
-      expect(result[:key1]).to eq('value1')
-      expect(result[:key2]).to eq('value2')
+      expect(result[:full_response]).to eq(JSON.parse(response_body, symbolize_names: true))
     end
   end
 

data/spec/bedrock_runtime/response_builders/{image_spec.rb → stability_ai_spec.rb}

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
 require 'spec_helper'
-require 'bedrock_runtime/response_builders/image'
+require 'bedrock_runtime/response_builders/stability_ai'
 
-RSpec.describe RubyAmazonBedrock::ResponseBuilders::Image do
+RSpec.describe RubyAmazonBedrock::ResponseBuilders::StabilityAi do
   let(:base64_image) { Base64.encode64(File.binread('spec/fixtures/sample_image.jpg')) }
   let(:response_body) { { artifacts: [{ base64: base64_image }] }.to_json }
   let(:response) { double('response', body: StringIO.new(response_body)) }

data/spec/bedrock_runtime/response_factory_spec.rb

@@ -2,34 +2,94 @@
 
 require 'spec_helper'
 require 'bedrock_runtime/response_factory'
-require 'bedrock_runtime/response_builders/text'
-require 'bedrock_runtime/response_builders/image'
+require 'bedrock_runtime/response_builders/ai_21_labs'
+require 'bedrock_runtime/response_builders/amazon_text'
+require 'bedrock_runtime/response_builders/amazon_image'
+require 'bedrock_runtime/response_builders/anthropic'
+require 'bedrock_runtime/response_builders/cohere_command'
+require 'bedrock_runtime/response_builders/cohere_embed'
+require 'bedrock_runtime/response_builders/meta'
+require 'bedrock_runtime/response_builders/stability_ai'
 
 RSpec.describe RubyAmazonBedrock::ResponseFactory do
   let(:response) { double('response') }
   let(:options) { { some_option: 'value' } }
 
-  context 'when type is :text' do
-    subject { described_class.new(:text, response) }
+  context 'when model_id is from ai_21_labs' do
+    ['ai21.j2-mid-v1', 'ai21.j2-ultra-v1'].each do |model_id|
+      subject { described_class.new(model_id, response) }
 
-    it 'creates a Text response builder' do
-      expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::Text)
+      it 'creates an Ai21Labs response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::Ai21Labs)
+      end
     end
   end
 
-  context 'when type is :image' do
-    subject { described_class.new(:image, response, options) }
+  context 'when model_id is from Amazon text generators' do
+    ['amazon.titan-text-lite-v1', 'amazon.titan-text-express-v1'].each do |model_id|
+      subject { described_class.new(model_id, response) }
 
-    it 'creates an Image response builder' do
-      expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::Image)
+      it 'creates an AmazonText response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::AmazonText)
+      end
     end
   end
 
-  context 'when type is invalid' do
-    subject { described_class.new(:invalid_type, response) }
+  context 'when model_id is from Amazon image generators' do
+    subject { described_class.new('amazon.titan-image-generator-v1', response) }
 
-    it 'returns nil' do
-      expect(subject.create).to be_nil
+    it 'creates an AmazonText response builder' do
+      expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::AmazonImage)
+    end
+  end
+
+  context 'when model_id is from Anthropic generators' do
+    ['anthropic.claude-instant-v1', 'anthropic.claude-v1', 'anthropic.claude-v2'].each do |model_id|
+      subject { described_class.new(model_id, response) }
+
+      it 'creates an AmazonText response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::Anthropic)
+      end
+    end
+  end
+
+  context 'when model_id is from Cohere commnand generators' do
+    ['cohere.command-light-text-v14', 'cohere.command-text-v14'].each do |model_id|
+      subject { described_class.new(model_id, response) }
+
+      it 'creates an AmazonText response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::CohereCommand)
+      end
+    end
+  end
+
+  context 'when model_id is from Cohere embed generators' do
+    ['cohere.embed-english-v3', 'cohere.embed-multilingual-v3'].each do |model_id|
+      subject { described_class.new(model_id, response) }
+
+      it 'creates an AmazonText response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::CohereEmbed)
+      end
+    end
+  end
+
+  context 'when model_id is from Meta generators' do
+    ['meta.llama2-13b-chat-v1', 'meta.llama2-70b-chat-v1'].each do |model_id|
+      subject { described_class.new(model_id, response) }
+
+      it 'creates an AmazonText response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::Meta)
+      end
+    end
+  end
+
+  context 'when model_id is from Stability AI generators' do
+    ['stability.stable-diffusion-xl-v0', 'stability.stable-diffusion-xl-v0'].each do |model_id|
+      subject { described_class.new(model_id, response) }
+
+      it 'creates an AmazonText response builder' do
+        expect(subject.create).to be_a(RubyAmazonBedrock::ResponseBuilders::StabilityAi)
+      end
     end
   end
 end

data/spec/cassettes/models/meta_llama3-70b-instruct-v1_0.yml

@@ -0,0 +1,73 @@
+---
+http_interactions:
+- request:
+    method: post
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/meta.llama3-70b-instruct-v1:0/invoke
+    body:
+      encoding: UTF-8
+      string: '{"prompt":"Give me a short list of steps for creating a Ruby gem","max_gen_len":512,"temperature":0.5,"top_p":0.9}'
+    headers:
+      Accept-Encoding:
+      - ''
+      Amz-Sdk-Invocation-Id:
+      - 32851ae6-1618-400a-9411-11d6d3d67636
+      Content-Type:
+      - application/json
+      Accept:
+      - "*/*"
+      User-Agent:
+      - aws-sdk-ruby3/3.201.4 ua/2.1 api/bedrock_runtime#1.17.0 os/macos#23 md/arm64
+        lang/ruby#3.0.0 md/3.0.0 m/D
+      Host:
+      - bedrock-runtime.us-east-1.amazonaws.com
+      X-Amz-Date:
+      - 20240917T163051Z
+      X-Amz-Content-Sha256:
+      - cf1ec2da83f5824b04de33922f5727968ac482454a9af56c2c3119c6711f2b1e
+      Authorization:
+      - AWS4-HMAC-SHA256 Credential=AKIASKU7DUX5LSIDI7FA/20240917/us-east-1/bedrock/aws4_request,
+        SignedHeaders=accept;amz-sdk-invocation-id;content-type;host;x-amz-content-sha256;x-amz-date,
+        Signature=c2858b5790f78f1428b84dace670018733257a68224824485146d1c0a695f0c6
+      Content-Length:
+      - '114'
+  response:
+    status:
+      code: 200
+      message: OK
+    headers:
+      Date:
+      - Tue, 17 Sep 2024 16:31:00 GMT
+      Content-Type:
+      - application/json
+      Content-Length:
+      - '1244'
+      Connection:
+      - keep-alive
+      X-Amzn-Requestid:
+      - 83c64727-212b-4d5d-a26e-681c991809d3
+      X-Amzn-Bedrock-Invocation-Latency:
+      - '7293'
+      X-Amzn-Bedrock-Output-Token-Count:
+      - '293'
+      X-Amzn-Bedrock-Input-Token-Count:
+      - '12'
+    body:
+      encoding: UTF-8
+      string: '{"generation":".\n\nHere is a short list of steps for creating a Ruby
+        gem:\n\n1. **Create a new gem**: Run `gem new my_gem` (replace \"my_gem\"
+        with your gem''s name) to create a new gem with the basic directory structure.\n2.
+        **Define your gem''s metadata**: Edit the `my_gem.gemspec` file to specify
+        your gem''s name, and other metadata such as version, description, and dependencies.\n3.
+        **Write your gem''s code**: Create a `lib` directory and add your Ruby code
+        to it. This is where you''ll implement the functionality of your gem.\n4.
+        **Test your gem**: Create a `test` directory and add test files to it. Use
+        a testing framework like RSpec or Minitest to write tests for your gem.\n5.
+        **Build and package your gem**: Run `gem build my_gem.gemspec` to build your
+        gem, and then `gem push my_gem-0.1.0.gem` (replace \"0.1.0\" with your gem''s
+        version) to push it to RubyGems.org.\n6. **Publish your gem**: Go to RubyGems.org
+        and create an account if you haven''t already. Then, upload your gem and make
+        it publicly available.\n\nThat''s it! Of course, there are many more details
+        to consider when creating a Ruby gem, but these steps should give you a good
+        starting point.","prompt_token_count":12,"generation_token_count":293,"stop_reason":"stop"}'
+  recorded_at: Tue, 17 Sep 2024 16:31:00 GMT
+recorded_with: VCR 6.2.0

data/spec/cassettes/models/meta_llama3-8b-instruct-v1_0.yml

@@ -0,0 +1,71 @@
+---
+http_interactions:
+- request:
+    method: post
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/meta.llama3-8b-instruct-v1:0/invoke
+    body:
+      encoding: UTF-8
+      string: '{"prompt":"Give me a short list of steps for creating a Ruby gem","max_gen_len":512,"temperature":0.5,"top_p":0.9}'
+    headers:
+      Accept-Encoding:
+      - ''
+      Amz-Sdk-Invocation-Id:
+      - f256772e-a630-485d-b52d-e20311339125
+      Content-Type:
+      - application/json
+      Accept:
+      - "*/*"
+      User-Agent:
+      - aws-sdk-ruby3/3.201.4 ua/2.1 api/bedrock_runtime#1.17.0 os/macos#23 md/arm64
+        lang/ruby#3.0.0 md/3.0.0 m/D
+      Host:
+      - bedrock-runtime.us-east-1.amazonaws.com
+      X-Amz-Date:
+      - 20240917T163109Z
+      X-Amz-Content-Sha256:
+      - cf1ec2da83f5824b04de33922f5727968ac482454a9af56c2c3119c6711f2b1e
+      Authorization:
+      - AWS4-HMAC-SHA256 Credential=AKIASKU7DUX5LSIDI7FA/20240917/us-east-1/bedrock/aws4_request,
+        SignedHeaders=accept;amz-sdk-invocation-id;content-type;host;x-amz-content-sha256;x-amz-date,
+        Signature=036bf9fb08b5c0308655674b5bd9eec2eb8c347f3b2e712585be1def1c780f69
+      Content-Length:
+      - '114'
+  response:
+    status:
+      code: 200
+      message: OK
+    headers:
+      Date:
+      - Tue, 17 Sep 2024 16:31:13 GMT
+      Content-Type:
+      - application/json
+      Content-Length:
+      - '1170'
+      Connection:
+      - keep-alive
+      X-Amzn-Requestid:
+      - 1e88ebb3-f78f-4b57-8926-229f8d4dc625
+      X-Amzn-Bedrock-Invocation-Latency:
+      - '3536'
+      X-Amzn-Bedrock-Output-Token-Count:
+      - '275'
+      X-Amzn-Bedrock-Input-Token-Count:
+      - '12'
+    body:
+      encoding: UTF-8
+      string: '{"generation":".\n\nComment by Brian Hays on January 28, 2013 at 9:13am\n\nHere
+        are the basic steps for creating a Ruby gem:\n\n1. Create a new directory
+        for your gem and navigate to it in your terminal.\n2. Run `gem init` to create
+        a basic gem structure.\n3. Edit the `gemspec` file to add information about
+        your gem, such as its name, version, and dependencies.\n4. Write your gem''s
+        code in the `lib` directory.\n5. Run `gem build` to build your gem.\n6. Run
+        `gem push` to publish your gem to a gem server (such as RubyGems.org).\n\nThat''s
+        the basic process, but there are many other things you can do to customize
+        and improve your gem, such as adding tests, documentation, and a README file.\n\nComment
+        by Brian Hays on January 28, 2013 at 9:14am\n\nAlso, you may want to consider
+        using a gem template tool like Bundler''s `bundle gem` command or the `gem-skeleton`
+        gem to help you get started with creating a new gem. These tools can save
+        you some time and effort by creating a basic gem structure for you. Just run
+        `bundle gem my_gem` or `gem-skeleton my_gem` and you''ll be off and running!","prompt_token_count":12,"generation_token_count":275,"stop_reason":"stop"}'
+  recorded_at: Tue, 17 Sep 2024 16:31:13 GMT
+recorded_with: VCR 6.2.0