ruby-amazon-bedrock 0.2.2 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e34184a2246b1054a64d6d29bc7bb1144487a508c6d27f717d8f92edb4cb32a2
-  data.tar.gz: ae40e48b1107bf163bbb30e4c058d660784f1bf9cc6fcafa0bfa9843b526dd49
+  metadata.gz: 06f4b8ff875c2d3d8b104c026621108f24e2ea60dd4ef07e85b4874b346b95af
+  data.tar.gz: 5fae3ad77abbab1f45ce187ade385da0ab0d3a64d7a8c38e22283a08ed23078b
 SHA512:
-  metadata.gz: fcdfeb1017d3bf2cefeb1bf77b68a85bd36bd3a7906651b1a56f8ff2544309ba98eee08fc1f04ce80ed22b738302fd38e7811087f56a976c6eeea1f7cfae0010
-  data.tar.gz: f19b95ef5965317c53eda21c96c69e421abdb34ce441cc850731663aa8ed35bbc67000d02a34ebc2f192ce7fbe81e0d18afe3ab59d0dbea38e32c858625fcc89
+  metadata.gz: cd0bd9610262b0222d9680386cfcd46b070582c8f0053737020c35509504a31c2cac4e94668debed36fc9f3f5c4f45396f0082c57314277572124d24ce2bffb4
+  data.tar.gz: 44c080409f214e186841ff78a073a84749bb84cc978d2d206045f0a143b9533bf652356da17e4405bbdf3ae59337a0752f762af02ec2125ee1e84750a1c394a9

data/.circleci/config.yml

@@ -50,8 +50,6 @@ workflows:
           matrix:
             parameters:
               ruby-image:
-                - cimg/ruby:2.6-node
-                - cimg/ruby:2.7-node
                 - cimg/ruby:3.0-node
                 - cimg/ruby:3.1-node
                 - cimg/ruby:3.2-node

data/.env.sample

@@ -0,0 +1,5 @@
+AWS_REGION: 'us-east-1'
+
+# You don't need valid key data *unless you are creating new
+AWS_ACCESS_KEY_ID: 'aaaaaaaaaaaaaaaaaaaa'
+AWS_SECRET_ACCESS_KEY: 'aaaaaaaaaaaaaaaaaaaa'

data/.gitignore

@@ -4,11 +4,11 @@
 /coverage/
 /doc/
 /pkg/
-/spec/reports/
+	/spec/reports/
 /tmp/
 .env
 *.gem
-image.jpg
+*.jpg
 
 # rspec failure tracking
 .rspec_status

data/CHANGELOG.md

@@ -1,3 +1,15 @@
+## [0.2.5] - 2024-09-17
+
+- Support for llama38b_instruct_v1
+- Support for llama370b_instruct_v1
+- Explicitly add base64 gem to dependencies
+- Support the use of AWS Named Profiles for authentication
+
+## [0.2.3] - 2024-01-12
+
+- Fix A121 Labs bug for maxTokens parameter
+- Allow Image generator to store images in S3 after being created
+
 ## [0.2.2] - 2024-01-10
 
 - Fix bug for Amazon Titan Text Lite

data/Gemfile

@@ -9,6 +9,7 @@ gem "bundler-audit", "~> 0.8"
 gem "dotenv", "~> 2.7"
 gem "pry", "~> 0.13"
 gem "rake", "~> 13.0"
+gem "rexml", ">= 3.3.6"
 gem "rspec", "~> 3.0"
 gem "rubocop", "~> 1.21"
 gem "vcr", "~> 6.0"

data/Gemfile.lock

@@ -1,96 +1,110 @@
 PATH
   remote: .
   specs:
-    ruby-amazon-bedrock (0.2.2)
+    ruby-amazon-bedrock (0.2.4)
       aws-sdk-bedrockruntime (~> 1.0)
+      aws-sdk-s3 (~> 1.0)
+      base64 (~> 0.2)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    addressable (2.8.5)
-      public_suffix (>= 2.0.2, < 6.0)
+    addressable (2.8.7)
+      public_suffix (>= 2.0.2, < 7.0)
     ast (2.4.2)
     aws-eventstream (1.3.0)
-    aws-partitions (1.860.0)
-    aws-sdk-bedrockruntime (1.4.0)
-      aws-sdk-core (~> 3, >= 3.188.0)
-      aws-sigv4 (~> 1.1)
-    aws-sdk-core (3.189.0)
+    aws-partitions (1.963.0)
+    aws-sdk-bedrockruntime (1.17.0)
+      aws-sdk-core (~> 3, >= 3.201.0)
+      aws-sigv4 (~> 1.5)
+    aws-sdk-core (3.201.4)
       aws-eventstream (~> 1, >= 1.3.0)
       aws-partitions (~> 1, >= 1.651.0)
       aws-sigv4 (~> 1.8)
       jmespath (~> 1, >= 1.6.1)
-    aws-sigv4 (1.8.0)
+    aws-sdk-kms (1.88.0)
+      aws-sdk-core (~> 3, >= 3.201.0)
+      aws-sigv4 (~> 1.5)
+    aws-sdk-s3 (1.157.0)
+      aws-sdk-core (~> 3, >= 3.201.0)
+      aws-sdk-kms (~> 1)
+      aws-sigv4 (~> 1.5)
+    aws-sigv4 (1.9.1)
       aws-eventstream (~> 1, >= 1.0.2)
+    base64 (0.2.0)
+    bigdecimal (3.1.8)
     bundler-audit (0.9.1)
       bundler (>= 1.2.0, < 3)
       thor (~> 1.0)
     coderay (1.1.3)
-    crack (0.4.5)
+    crack (1.0.0)
+      bigdecimal
       rexml
-    diff-lcs (1.5.0)
+    diff-lcs (1.5.1)
     dotenv (2.8.1)
-    hashdiff (1.0.1)
+    hashdiff (1.1.1)
     jmespath (1.6.2)
-    json (2.6.3)
+    json (2.7.2)
     language_server-protocol (3.17.0.3)
-    method_source (1.0.0)
-    parallel (1.23.0)
-    parser (3.2.2.4)
+    method_source (1.1.0)
+    parallel (1.26.1)
+    parser (3.3.4.2)
       ast (~> 2.4.1)
       racc
     pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
-    public_suffix (5.0.4)
-    racc (1.7.3)
+    public_suffix (5.1.1)
+    racc (1.8.1)
     rainbow (3.1.1)
-    rake (13.1.0)
-    regexp_parser (2.8.2)
-    rexml (3.2.6)
-    rspec (3.12.0)
-      rspec-core (~> 3.12.0)
-      rspec-expectations (~> 3.12.0)
-      rspec-mocks (~> 3.12.0)
-    rspec-core (3.12.2)
-      rspec-support (~> 3.12.0)
-    rspec-expectations (3.12.3)
+    rake (13.2.1)
+    regexp_parser (2.9.2)
+    rexml (3.3.7)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.0)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.12.0)
-    rspec-mocks (3.12.6)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.12.0)
-    rspec-support (3.12.1)
-    rubocop (1.57.2)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.1)
+    rubocop (1.65.1)
       json (~> 2.3)
       language_server-protocol (>= 3.17.0)
       parallel (~> 1.10)
-      parser (>= 3.2.2.4)
+      parser (>= 3.3.0.2)
       rainbow (>= 2.2.2, < 4.0)
-      regexp_parser (>= 1.8, < 3.0)
+      regexp_parser (>= 2.4, < 3.0)
       rexml (>= 3.2.5, < 4.0)
-      rubocop-ast (>= 1.28.1, < 2.0)
+      rubocop-ast (>= 1.31.1, < 2.0)
       ruby-progressbar (~> 1.7)
       unicode-display_width (>= 2.4.0, < 3.0)
-    rubocop-ast (1.30.0)
-      parser (>= 3.2.1.0)
+    rubocop-ast (1.32.0)
+      parser (>= 3.3.1.0)
     ruby-progressbar (1.13.0)
-    thor (1.3.0)
+    thor (1.3.1)
     unicode-display_width (2.5.0)
     vcr (6.2.0)
-    webmock (3.19.1)
+    webmock (3.23.1)
       addressable (>= 2.8.0)
       crack (>= 0.3.2)
       hashdiff (>= 0.4.0, < 2.0.0)
 
 PLATFORMS
   arm64-darwin-22
+  arm64-darwin-23
 
 DEPENDENCIES
   bundler-audit (~> 0.8)
   dotenv (~> 2.7)
   pry (~> 0.13)
   rake (~> 13.0)
+  rexml (>= 3.3.6)
   rspec (~> 3.0)
   rubocop (~> 1.21)
   ruby-amazon-bedrock!
@@ -98,4 +112,4 @@ DEPENDENCIES
   webmock (~> 3.12)
 
 BUNDLED WITH
-   2.4.22
+   2.4.14

data/README.md

@@ -14,7 +14,7 @@ Amazon Bedrock is a fully managed service that makes FMs from leading AI startup
 Add the following line to your application's Gemfile:
 
 ```ruby
-gem "ruby-amazon-bedrock", "~> 0.2.2"
+gem "ruby-amazon-bedrock", "~> 0.2.4"
 ```
 
 And then execute:
@@ -28,7 +28,7 @@ $ bundle install
 Or install with:
 
 ```bash
-$ gem install ruby-amazon-bedrock -v 0.2.1
+$ gem install ruby-amazon-bedrock -v 0.2.3
 ```
 
 and require with:
@@ -59,20 +59,28 @@ Instantiate a client by passing your AWS IAM credentials:
 client = RubyAmazonBedrock::Client.new(
   region: "AWS_REGION",
   access_key_id: "AWS_ACCESS_KEY_ID",
-  access_token: "AWS_SECRET_ACCESS_KEY"
+  secret_access_key: "AWS_SECRET_ACCESS_KEY"
 )
 ```
 
-## With Configuration
+Or copy the `.env.sample` to a `.env` file with your account's credentials and use the `ENV` variable. NOTE: This step is REQUIRED in order to run the tests with `rspec`
 
 ```ruby
-RubyAmazonBedrock.configure do |config|
-  config.region = ENV.fetch('AWS_REGION', nil)
-  config.access_key_id = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
-  config.secret_access_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
-end
+client = RubyAmazonBedrock::Client.new(
+  region: ENV["AWS_REGION"],
+  access_key_id: ENV["AWS_ACCESS_KEY_ID"],
+  secret_access_key: ENV["AWS_SECRET_ACCESS_KEY"]
+)
+```
+
+### AWS Named Profiles
 
-client = RubyAmazonBedrock::Client.new
+You can also use [AWS Named Profiles](https://docs.aws.amazon.com/cli/latest/userguide/cli-configure-files.html#cli-configure-files-format-profile) by passing the `profile` keywoard argument. When using a named profile, specyfing the `region`, `access_key_id` and `access_token` won't be required.
+
+```ruby
+client = RubyAmazonBedrock::Client.new(
+  profile: "AWS_PROFILE"
+)
 ```
 
 ## Options
@@ -231,7 +239,7 @@ _*Claude Instant 1.2*_
 Supports: Question answering, information extraction, removing PII, content generation, multiple choice classification, Roleplay, comparing text, summarization, document Q&A with citation
 
 ```ruby
-client.invoke_model('anthropic.claude-instant-v1', 'What is a neural network?')
+client.invoke_model(id: 'anthropic.claude-instant-v1', prompt: 'What is a neural network?')
 
 # Response
 {:completion=>

data/lib/amazon_bedrock/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyAmazonBedrock
-  VERSION = "0.2.2"
+  VERSION = "0.2.4"
 end

data/lib/amazon_bedrock.rb

@@ -13,7 +13,7 @@ module RubyAmazonBedrock
 
   # Configuration class for setting up AWS credentials and region.
   class Configuration
-    attr_accessor :region, :access_key_id, :secret_access_key
+    attr_accessor :region, :access_key_id, :secret_access_key, :profile
 
     # Initializes a new Configuration instance, loading values from
     # environment variables or setting them to nil by default so the
@@ -22,6 +22,7 @@ module RubyAmazonBedrock
       @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)
+      @profile = ENV.fetch('AWS_PROFILE', nil)
     end
   end
 

data/lib/bedrock_runtime/client.rb

@@ -14,14 +14,20 @@ module RubyAmazonBedrock
     # Initializes the AWS BedrockRuntime client.
     #
     # @note The AWS credentials and region are fetched from the environment variables.
-    def initialize(region: nil, access_key_id: nil, secret_access_key: nil)
+    def initialize(region: nil, access_key_id: nil, secret_access_key: nil, profile: nil)
       config = RubyAmazonBedrock.configuration || RubyAmazonBedrock::Configuration.new
 
-      @client = Aws::BedrockRuntime::Client.new(
-        region: region || config.region,
-        access_key_id: access_key_id || config.access_key_id,
-        secret_access_key: secret_access_key || config.secret_access_key
-      )
+      @client = if profile
+                  Aws::BedrockRuntime::Client.new(
+                    profile: profile
+                  )
+                else
+                  Aws::BedrockRuntime::Client.new(
+                    region: region || config.region,
+                    access_key_id: access_key_id || config.access_key_id,
+                    secret_access_key: secret_access_key || config.secret_access_key
+                  )
+                end
     end
 
     # Invokes a model using the Bedrock Runtime client.
@@ -39,8 +45,7 @@ module RubyAmazonBedrock
       payload_builder_class = RubyAmazonBedrock::PayloadFactory.new(id, prompt, options).create
       response = @client.invoke_model(payload_builder_class.build)
 
-      response_builder_class = RubyAmazonBedrock::ResponseFactory.new(payload_builder_class.type, response,
-                                                                      options).create
+      response_builder_class = RubyAmazonBedrock::ResponseFactory.new(id, response, options).create
       response_builder_class.build
     end
   end

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

@@ -35,7 +35,7 @@ module RubyAmazonBedrock
             accept: '*/*',
             body: {
               prompt: @prompt,
-              maxTokenCount: parameters[:maxTokenCount],
+              maxTokens: parameters[:maxTokens],
               temperature: parameters[:temperature],
               topP: parameters[:topP],
               stopSequences: parameters[:stopSequences],
@@ -52,7 +52,7 @@ module RubyAmazonBedrock
 
         def parameters # rubocop:disable Metrics/CyclomaticComplexity
           {
-            maxTokenCount: @options[:max_tokens] || 200,
+            maxTokens: @options[:max_tokens] || 200,
             stopSequences: @options[:stop_sequences] || [],
             temperature: @options[:temperature] || 0,
             topP: @options[:top_p] || 1,

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

@@ -20,7 +20,7 @@ module RubyAmazonBedrock
         #   - :body [String] A JSON string that includes the following details:
         #     - :inputText [String] The input text to be processed by the model.
         #     - :textGenerationConfig [Hash] A hash containing configuration parameters for text generation:
-        #       - :maxTokenCount [Integer] The maximum number of tokens to generate.
+        #       - :maxTokens [Integer] The maximum number of tokens to generate.
         #       - :stopSequences [Array<String>] An array of strings that, when encountered, will signal the end
         # 				of generation.
         #       - :temperature [Float] A parameter controlling the randomness in the response generation.

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

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

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

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

data/lib/bedrock_runtime/payload_factory.rb

@@ -14,6 +14,8 @@ 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/meta/llama370b_instruct_v1'
+require_relative 'payload_builders/meta/llama38b_instruct_v1'
 require_relative 'payload_builders/stability_ai/stable_diffusion_xl_v0'
 require_relative 'payload_builders/stability_ai/stable_diffusion_xl_v1'
 
@@ -64,6 +66,8 @@ module RubyAmazonBedrock
         'cohere.embed-multilingual-v3' => PayloadBuilders::Cohere::EmbedMultilingualV3,
         'meta.llama2-13b-chat-v1' => PayloadBuilders::Meta::Llama213bChatV1,
         'meta.llama2-70b-chat-v1' => PayloadBuilders::Meta::Llama270bChatV1,
+        'meta.llama3-70b-instruct-v1:0' => PayloadBuilders::Meta::Llama370bInstructV1,
+        'meta.llama3-8b-instruct-v1:0' => PayloadBuilders::Meta::Llama38bInstructV1,
         'stability.stable-diffusion-xl-v0' => PayloadBuilders::StabilityAi::StableDiffusionXlV0,
         'stability.stable-diffusion-xl-v1' => PayloadBuilders::StabilityAi::StableDiffusionXlV1
       }

data/lib/bedrock_runtime/response_builders/ai_21_labs.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module ResponseBuilders
+    # The Ai21Labs class is responsible for parsing and building a structured response from a raw text response.
+    # It converts the HTTP response for an AI21 Labs model into a structured format to make it easier to access
+    # the response data.
+    class Ai21Labs
+      # 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[:completions]&.first&.dig(:data, :text),
+          full_response: response
+        }
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/response_builders/{image.rb → amazon_image.rb}

@@ -1,13 +1,15 @@
 # frozen_string_literal: true
 
+require 'aws-sdk-s3'
 require 'base64'
+require 'digest'
 
 module RubyAmazonBedrock
   module ResponseBuilders
-    # The Image class is responsible for handling and processing image data received in a response.
+    # The AmazonImage 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 Image
-      # Initializes a new instance of the Image class.
+    class AmazonImage
+      # Initializes a new instance of the AmazonImage class.
       #
       # @param response [Object] The raw response object which is an HTTP response.
       # The response should contain the image data in Base64 format.
@@ -16,6 +18,7 @@ module RubyAmazonBedrock
       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.
@@ -26,15 +29,20 @@ module RubyAmazonBedrock
       #
       # @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
-        }
+          file_path: @file_path,
+          s3_url: s3_url
+        }.compact
       rescue StandardError => e
         {
           result: :failure,
@@ -42,8 +50,35 @@ module RubyAmazonBedrock
         }
       end
 
-      def model_id
-        # noop
+      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

data/lib/bedrock_runtime/response_builders/amazon_text.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module ResponseBuilders
+    # The AmazonText class is responsible for parsing and building a structured response from a raw text response.
+    # It converts the HTTP response for an Amazon Titan model into a structured format to make it easier to access
+    # the response data.
+    class AmazonText
+      # 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[:results]&.first&.dig(:outputText),
+          full_response: response
+        }
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/response_builders/anthropic.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module ResponseBuilders
+    # The Anthropic class is responsible for parsing and building a structured response from a raw text response.
+    # It converts the HTTP response for an Anthropic model into a structured format to make it easier to access
+    # the response data.
+    class Anthropic
+      # 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[:completion],
+          full_response: response
+        }
+      end
+    end
+  end
+end

data/lib/bedrock_runtime/response_builders/cohere_command.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module RubyAmazonBedrock
+  module ResponseBuilders
+    # The CohereCommand class is responsible for parsing and building a structured response from a raw text response.
+    # It converts the HTTP response for an Cohere Command model into a structured format to make it easier to access
+    # the response data.
+    class CohereCommand
+      # 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[:generations]&.first&.[](:text),
+          full_response: response
+        }
+      end
+    end
+  end
+end