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