Sha256: 17ee10fcc999420785c0e7f6cd20e6b7bdd4b2efd853d3496c5b99bcd5ea005e
Contents?: true
Size: 951 Bytes
Versions: 3
Compression:
Stored size: 951 Bytes
Contents
# 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
Version data entries
3 entries across 3 versions & 1 rubygems