lib/amazon_bedrock.rb in ruby-amazon-bedrock-0.2.3 vs lib/amazon_bedrock.rb in ruby-amazon-bedrock-0.2.4

@@ -11,18 +11,19 @@
   class UnknownModelError < StandardError; end
   class Error < StandardError; end
 
   # Configuration class for setting up AWS credentials and region.
   class Configuration
-    attr_accessor :region, :access_key_id, :secret_access_key
+    attr_accessor :region, :access_key_id, :secret_access_key, :profile
 
     # Initializes a new Configuration instance, loading values from
     # environment variables or setting them to nil by default so the
     # values can be dynamically set.
     def initialize
       @region = ENV.fetch('AWS_REGION', nil)
       @access_key_id = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
       @secret_access_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
+      @profile = ENV.fetch('AWS_PROFILE', nil)
     end
   end
 
   class << self
     # Returns the current configuration instance or creates a new one.