helpers.rb in ruby-amazon-bedrock-0.2.3

- old
+ new

@@ -1,9 +1,7 @@
 # frozen_string_literal: true
 
-require 'pry'
-
 RSpec.shared_examples 'a payload builder' do
   it 'validates the payload structure' do
     payload_builder = described_class.new(prompt, options)
     payload = payload_builder.build
 
@@ -26,10 +24,10 @@
     }
   end
   let(:body) do
     {
       prompt: prompt,
-      maxTokenCount: 100,
+      maxTokens: 100,
       temperature: 0.5,
       topP: 2,
       stopSequences: %w[stop1 stop2],
       countPenalty: { scale: 1 },
       presencePenalty: { scale: 1 },