lib/bedrock_runtime/client.rb in ruby-amazon-bedrock-0.2.2 vs lib/bedrock_runtime/client.rb in ruby-amazon-bedrock-0.2.3

@@ -37,11 +37,10 @@
     # 	)
     def invoke_model(id:, prompt:, options: {})
       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
 end