aws-sdk-bedrockagent 1.0.0

data/lib/aws-sdk-bedrockagent/errors.rb

@@ -0,0 +1,155 @@
+# frozen_string_literal: true
+
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws::BedrockAgent
+
+  # When BedrockAgent returns an error response, the Ruby SDK constructs and raises an error.
+  # These errors all extend Aws::BedrockAgent::Errors::ServiceError < {Aws::Errors::ServiceError}
+  #
+  # You can rescue all BedrockAgent errors using ServiceError:
+  #
+  #     begin
+  #       # do stuff
+  #     rescue Aws::BedrockAgent::Errors::ServiceError
+  #       # rescues all BedrockAgent API errors
+  #     end
+  #
+  #
+  # ## Request Context
+  # ServiceError objects have a {Aws::Errors::ServiceError#context #context} method that returns
+  # information about the request that generated the error.
+  # See {Seahorse::Client::RequestContext} for more information.
+  #
+  # ## Error Classes
+  # * {AccessDeniedException}
+  # * {ConflictException}
+  # * {InternalServerException}
+  # * {ResourceNotFoundException}
+  # * {ServiceQuotaExceededException}
+  # * {ThrottlingException}
+  # * {ValidationException}
+  #
+  # Additionally, error classes are dynamically generated for service errors based on the error code
+  # if they are not defined above.
+  module Errors
+
+    extend Aws::Errors::DynamicErrors
+
+    class AccessDeniedException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::AccessDeniedException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
+    class ConflictException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::ConflictException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
+    class InternalServerException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::InternalServerException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
+    class ResourceNotFoundException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::ResourceNotFoundException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
+    class ServiceQuotaExceededException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::ServiceQuotaExceededException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
+    class ThrottlingException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::ThrottlingException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+    end
+
+    class ValidationException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgent::Types::ValidationException] data
+      def initialize(context, message, data = Aws::EmptyStructure.new)
+        super(context, message, data)
+      end
+
+      # @return [String]
+      def message
+        @message || @data[:message]
+      end
+
+      # @return [String]
+      def field_list
+        @data[:field_list]
+      end
+    end
+
+  end
+end

data/lib/aws-sdk-bedrockagent/plugins/endpoints.rb

@@ -0,0 +1,149 @@
+# frozen_string_literal: true
+
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+
+module Aws::BedrockAgent
+  module Plugins
+    class Endpoints < Seahorse::Client::Plugin
+      option(
+        :endpoint_provider,
+        doc_type: 'Aws::BedrockAgent::EndpointProvider',
+        docstring: 'The endpoint provider used to resolve endpoints. Any '\
+                   'object that responds to `#resolve_endpoint(parameters)` '\
+                   'where `parameters` is a Struct similar to '\
+                   '`Aws::BedrockAgent::EndpointParameters`'
+      ) do |cfg|
+        Aws::BedrockAgent::EndpointProvider.new
+      end
+
+      # @api private
+      class Handler < Seahorse::Client::Handler
+        def call(context)
+          unless context[:discovered_endpoint]
+            params = parameters_for_operation(context)
+            endpoint = context.config.endpoint_provider.resolve_endpoint(params)
+
+            context.http_request.endpoint = endpoint.url
+            apply_endpoint_headers(context, endpoint.headers)
+
+            context[:endpoint_params] = params
+            context[:endpoint_properties] = endpoint.properties
+          end
+
+          context[:auth_scheme] =
+            Aws::Endpoints.resolve_auth_scheme(context, endpoint)
+
+          @handler.call(context)
+        end
+
+        private
+
+        def apply_endpoint_headers(context, headers)
+          headers.each do |key, values|
+            value = values
+              .compact
+              .map { |s| Seahorse::Util.escape_header_list_string(s.to_s) }
+              .join(',')
+
+            context.http_request.headers[key] = value
+          end
+        end
+
+        def parameters_for_operation(context)
+          case context.operation_name
+          when :associate_agent_knowledge_base
+            Aws::BedrockAgent::Endpoints::AssociateAgentKnowledgeBase.build(context)
+          when :create_agent
+            Aws::BedrockAgent::Endpoints::CreateAgent.build(context)
+          when :create_agent_action_group
+            Aws::BedrockAgent::Endpoints::CreateAgentActionGroup.build(context)
+          when :create_agent_alias
+            Aws::BedrockAgent::Endpoints::CreateAgentAlias.build(context)
+          when :create_data_source
+            Aws::BedrockAgent::Endpoints::CreateDataSource.build(context)
+          when :create_knowledge_base
+            Aws::BedrockAgent::Endpoints::CreateKnowledgeBase.build(context)
+          when :delete_agent
+            Aws::BedrockAgent::Endpoints::DeleteAgent.build(context)
+          when :delete_agent_action_group
+            Aws::BedrockAgent::Endpoints::DeleteAgentActionGroup.build(context)
+          when :delete_agent_alias
+            Aws::BedrockAgent::Endpoints::DeleteAgentAlias.build(context)
+          when :delete_agent_version
+            Aws::BedrockAgent::Endpoints::DeleteAgentVersion.build(context)
+          when :delete_data_source
+            Aws::BedrockAgent::Endpoints::DeleteDataSource.build(context)
+          when :delete_knowledge_base
+            Aws::BedrockAgent::Endpoints::DeleteKnowledgeBase.build(context)
+          when :disassociate_agent_knowledge_base
+            Aws::BedrockAgent::Endpoints::DisassociateAgentKnowledgeBase.build(context)
+          when :get_agent
+            Aws::BedrockAgent::Endpoints::GetAgent.build(context)
+          when :get_agent_action_group
+            Aws::BedrockAgent::Endpoints::GetAgentActionGroup.build(context)
+          when :get_agent_alias
+            Aws::BedrockAgent::Endpoints::GetAgentAlias.build(context)
+          when :get_agent_knowledge_base
+            Aws::BedrockAgent::Endpoints::GetAgentKnowledgeBase.build(context)
+          when :get_agent_version
+            Aws::BedrockAgent::Endpoints::GetAgentVersion.build(context)
+          when :get_data_source
+            Aws::BedrockAgent::Endpoints::GetDataSource.build(context)
+          when :get_ingestion_job
+            Aws::BedrockAgent::Endpoints::GetIngestionJob.build(context)
+          when :get_knowledge_base
+            Aws::BedrockAgent::Endpoints::GetKnowledgeBase.build(context)
+          when :list_agent_action_groups
+            Aws::BedrockAgent::Endpoints::ListAgentActionGroups.build(context)
+          when :list_agent_aliases
+            Aws::BedrockAgent::Endpoints::ListAgentAliases.build(context)
+          when :list_agent_knowledge_bases
+            Aws::BedrockAgent::Endpoints::ListAgentKnowledgeBases.build(context)
+          when :list_agent_versions
+            Aws::BedrockAgent::Endpoints::ListAgentVersions.build(context)
+          when :list_agents
+            Aws::BedrockAgent::Endpoints::ListAgents.build(context)
+          when :list_data_sources
+            Aws::BedrockAgent::Endpoints::ListDataSources.build(context)
+          when :list_ingestion_jobs
+            Aws::BedrockAgent::Endpoints::ListIngestionJobs.build(context)
+          when :list_knowledge_bases
+            Aws::BedrockAgent::Endpoints::ListKnowledgeBases.build(context)
+          when :list_tags_for_resource
+            Aws::BedrockAgent::Endpoints::ListTagsForResource.build(context)
+          when :prepare_agent
+            Aws::BedrockAgent::Endpoints::PrepareAgent.build(context)
+          when :start_ingestion_job
+            Aws::BedrockAgent::Endpoints::StartIngestionJob.build(context)
+          when :tag_resource
+            Aws::BedrockAgent::Endpoints::TagResource.build(context)
+          when :untag_resource
+            Aws::BedrockAgent::Endpoints::UntagResource.build(context)
+          when :update_agent
+            Aws::BedrockAgent::Endpoints::UpdateAgent.build(context)
+          when :update_agent_action_group
+            Aws::BedrockAgent::Endpoints::UpdateAgentActionGroup.build(context)
+          when :update_agent_alias
+            Aws::BedrockAgent::Endpoints::UpdateAgentAlias.build(context)
+          when :update_agent_knowledge_base
+            Aws::BedrockAgent::Endpoints::UpdateAgentKnowledgeBase.build(context)
+          when :update_data_source
+            Aws::BedrockAgent::Endpoints::UpdateDataSource.build(context)
+          when :update_knowledge_base
+            Aws::BedrockAgent::Endpoints::UpdateKnowledgeBase.build(context)
+          end
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :build, priority: 75)
+      end
+    end
+  end
+end

data/lib/aws-sdk-bedrockagent/resource.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws::BedrockAgent
+
+  class Resource
+
+    # @param options ({})
+    # @option options [Client] :client
+    def initialize(options = {})
+      @client = options[:client] || Client.new(options)
+    end
+
+    # @return [Client]
+    def client
+      @client
+    end
+
+  end
+end