ruby-amazon-bedrock 0.1.1

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

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/anthropic/base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Anthropic::Base do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "\n\nHuman: #{input}\n\nAssistant:",
+      max_tokens_to_sample: 300,
+      temperature: 0.5,
+      top_k: 250,
+      top_p: 1,
+      stop_sequences: [
+        '\n\nHuman'
+      ],
+      anthropic_version: 'bedrock-2023-05-31'
+    }.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/anthropic/claude_instant_v1_spec.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/anthropic/claude_instant_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Anthropic::ClaudeInstantV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "\n\nHuman: #{input}\n\nAssistant:",
+      max_tokens_to_sample: 300,
+      temperature: 0.5,
+      top_k: 250,
+      top_p: 1,
+      stop_sequences: [
+        '\n\nHuman'
+      ],
+      anthropic_version: 'bedrock-2023-05-31'
+    }.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('anthropic.claude-instant-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/anthropic/claude_v1_spec.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/anthropic/claude_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Anthropic::ClaudeV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "\n\nHuman: #{input}\n\nAssistant:",
+      max_tokens_to_sample: 300,
+      temperature: 0.5,
+      top_k: 250,
+      top_p: 1,
+      stop_sequences: [
+        '\n\nHuman'
+      ],
+      anthropic_version: 'bedrock-2023-05-31'
+    }.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('anthropic.claude-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/anthropic/claude_v2_spec.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/anthropic/claude_v2'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Anthropic::ClaudeV2 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "\n\nHuman: #{input}\n\nAssistant:",
+      max_tokens_to_sample: 300,
+      temperature: 0.5,
+      top_k: 250,
+      top_p: 1,
+      stop_sequences: [
+        '\n\nHuman'
+      ],
+      anthropic_version: 'bedrock-2023-05-31'
+    }.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('anthropic.claude-v2')
+      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/base_spec.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Base do
+  describe '#initialize' do
+    it 'creates a new instance with input and options' do
+      input = 'example_input'
+      options = { key: 'value' }
+
+      payload_builder = described_class.new(input, options)
+
+      expect(payload_builder.instance_variable_get(:@input)).to eq(input)
+      expect(payload_builder.instance_variable_get(:@options)).to eq(options)
+    end
+  end
+
+  describe '#build' do
+    it 'raises NotImplementedError' do
+      payload_builder = described_class.new('example_input')
+
+      expect { payload_builder.build }.to raise_error(NotImplementedError)
+    end
+  end
+
+  describe '#model_id' do
+    it 'raises NotImplementedError' do
+      payload_builder = described_class.new('example_input')
+
+      expect { payload_builder.model_id }.to raise_error(NotImplementedError)
+    end
+  end
+end

data/spec/bedrock_runtime/payload_builders/cohere/command_base_spec.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/cohere/command_base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Cohere::CommandBase do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "#{input}:",
+      max_tokens: 100,
+      temperature: 0.8
+    }.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/cohere/command_light_text_v14_spec.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/cohere/command_light_text_v14'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Cohere::CommandLightTextV14 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "#{input}:",
+      max_tokens: 100,
+      temperature: 0.8
+    }.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('cohere.command-light-text-v14')
+      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/cohere/command_text_v14_spec.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/cohere/command_text_v14'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Cohere::CommandTextV14 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: "#{input}:",
+      max_tokens: 100,
+      temperature: 0.8
+    }.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('cohere.command-text-v14')
+      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/cohere/embed_base_spec.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/cohere/embed_base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Cohere::EmbedBase do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      texts: [input],
+      input_type: 'search_document'
+    }.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/cohere/embed_english_v3_spec.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/cohere/embed_english_v3'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Cohere::EmbedEnglishV3 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      texts: [input],
+      input_type: 'search_document'
+    }.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('cohere.embed-english-v3')
+      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/cohere/embed_multilingual_v3_spec.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/cohere/embed_multilingual_v3'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Cohere::EmbedMultilingualV3 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      texts: [input],
+      input_type: 'search_document'
+    }.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('cohere.embed-multilingual-v3')
+      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/meta/base_spec.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/meta/base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Meta::Base do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: input,
+      max_gen_len: 512,
+      temperature: 0.2,
+      top_p: 0.9
+    }.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/meta/llama213b_chat_v1_spec.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/meta/llama213b_chat_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Meta::Llama213bChatV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: input,
+      max_gen_len: 512,
+      temperature: 0.2,
+      top_p: 0.9
+    }.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('meta.llama2-13b-chat-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/meta/llama270b_chat_v1_spec.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/meta/llama270b_chat_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::Meta::Llama270bChatV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      prompt: input,
+      max_gen_len: 512,
+      temperature: 0.2,
+      top_p: 0.9
+    }.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('meta.llama2-70b-chat-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/stability_ai/base_spec.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/stability_ai/base'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::StabilityAi::Base do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      text_prompts: [
+        { text: input }
+      ],
+      cfg_scale: 10,
+      seed: 0,
+      steps: 50
+    }.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/stability_ai/stable_diffusion_xl_v0_spec.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::StabilityAi::StableDiffusionXlV0 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      text_prompts: [
+        { text: input }
+      ],
+      cfg_scale: 10,
+      seed: 0,
+      steps: 50
+    }.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('stability.stable-diffusion-xl-v0')
+      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/stability_ai/stable_diffusion_xl_v1_spec.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1'
+
+RSpec.describe RubyAmazonBedrock::PayloadBuilders::StabilityAi::StableDiffusionXlV1 do
+  let(:input) { 'example_input' }
+  let(:options) { { key: 'value' } }
+  let(:body) do
+    {
+      text_prompts: [
+        { text: input }
+      ],
+      cfg_scale: 10,
+      seed: 0,
+      steps: 50
+    }.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('stability.stable-diffusion-xl-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_factory_spec.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require 'bedrock_runtime/payload_factory'
+require 'spec_helper'
+
+require 'pry'
+
+RSpec.describe RubyAmazonBedrock::PayloadFactory do
+  let(:bedrock_models) do
+    ["ai21labs.j2-mid-v1",
+     "ai21labs.j2-ultra-v1",
+     "amazon.titan-image-generator-v1",
+     "amazon.titan-text-lite-v1",
+     "amazon.titan-text-express-v1",
+     "anthropic.claude-v1",
+     "anthropic.claude-instant-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"]
+  end
+  let(:models_mapping) { described_class.new(nil, nil).models_to_builders }
+
+  describe '#create' do
+    context 'when model_id is a valid Bedrock model' do
+      it 'creates a payload using the appropriate builder' do
+        model_id = bedrock_models.sample
+        input = 'test input'
+        builder = described_class.new(model_id, input).create
+
+        expect(builder).to be_a(models_mapping[model_id])
+      end
+    end
+
+    context 'when model_id is not a valid Bedrock model' do
+      it 'raises an UnknownModelError' do
+        model_id = 'unknown_model'
+        input = 'test input'
+        payload_factory = described_class.new(model_id, input)
+
+        expect { payload_factory.create }.to raise_error(RubyAmazonBedrock::UnknownModelError, "Unknown modelId: #{model_id}")
+      end
+    end
+  end
+
+  describe '#models_to_builders' do
+    it 'defines a mapping of model identifiers to builder classes' do
+      expect(models_mapping.keys).to eq(bedrock_models)
+    end
+  end
+end

data/spec/bedrock_spec.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+RSpec.describe Ruby::Amazon::Bedrock do
+  it "has a version number" do
+    expect(Ruby::Amazon::Bedrock::VERSION).not_to be nil
+  end
+end

data/spec/cassettes/models/amazon_titan-text-express-v1.yml

@@ -0,0 +1,59 @@
+---
+http_interactions:
+- request:
+    method: post
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/amazon.titan-text-express-v1/invoke
+    body:
+      encoding: UTF-8
+      string: '{"inputText":"Give me a short list of steps for creating a Ruby gem","textGenerationConfig":{"maxTokenCount":4096,"stopSequences":[],"temperature":0,"topP":1}}'
+    headers:
+      Accept-Encoding:
+      - ''
+      Content-Type:
+      - application/json
+      Accept:
+      - "*/*"
+      User-Agent:
+      - aws-sdk-ruby3/3.189.0 ua/2.0 api/bedrock_runtime#1.4.0 os/macos#22 md/arm64
+        lang/ruby#3.2.2 md/3.2.2 cfg/retry-mode#legacy
+      Host:
+      - bedrock-runtime.us-east-1.amazonaws.com
+      Content-Length:
+      - '159'
+  response:
+    status:
+      code: 200
+      message: OK
+    headers:
+      Date:
+      - Wed, 06 Dec 2023 05:48:40 GMT
+      Content-Type:
+      - application/json
+      Content-Length:
+      - '936'
+      Connection:
+      - keep-alive
+      X-Amzn-Requestid:
+      - bfa4ea73-8745-4e6f-857c-cb568c567f0c
+      X-Amzn-Bedrock-Invocation-Latency:
+      - '7087'
+      X-Amzn-Bedrock-Output-Token-Count:
+      - '234'
+      X-Amzn-Bedrock-Input-Token-Count:
+      - '12'
+    body:
+      encoding: UTF-8
+      string: '{"inputTextTokenCount":12,"results":[{"tokenCount":234,"outputText":"\nHere
+        is a short list of steps for creating a Ruby gem:\n1. Create a new directory
+        for your gem.\n2. Initialize a new Ruby gem using the ''gem'' command.\n3.
+        Add a ''Gemfile'' to your gem directory.\n4. Write the code for your gem in
+        the ''lib/your_gem_name'' directory.\n5. Add a ''Rakefile'' to your gem directory.\n6.
+        Use the ''gem build'' command to build your gem.\n7. Use the ''gem install''
+        command to install your gem on your system.\n8. Publish your gem to the RubyGems
+        repository using the ''gem push'' command.\n9. Use the ''gem'' command to
+        install your gem from the RubyGems repository.\n10. Test your gem using the
+        ''rake'' command.\n11. Document your gem using the ''rdoc'' command.\n12.
+        Release a new version of your gem using the ''gem release'' command.\n13.
+        Update your gem with new features and bug fixes using the ''gem update'' command.","completionReason":"FINISH"}]}'
+  recorded_at: Wed, 06 Dec 2023 05:48:40 GMT
+recorded_with: VCR 6.2.0