aws-sdk-bedrockagentruntime 1.15.0 → 1.16.0

data/lib/aws-sdk-bedrockagentruntime/event_streams.rb

@@ -35,6 +35,10 @@ module Aws::BedrockAgentRuntime
         @event_emitter.on(:dependency_failed_exception, block) if block_given?
       end
 
+      def on_files_event(&block)
+        @event_emitter.on(:files, block) if block_given?
+      end
+
       def on_internal_server_exception_event(&block)
         @event_emitter.on(:internal_server_exception, block) if block_given?
       end
@@ -81,6 +85,7 @@ module Aws::BedrockAgentRuntime
         on_chunk_event(&block)
         on_conflict_exception_event(&block)
         on_dependency_failed_exception_event(&block)
+        on_files_event(&block)
         on_internal_server_exception_event(&block)
         on_resource_not_found_exception_event(&block)
         on_return_control_event(&block)
@@ -98,6 +103,90 @@ module Aws::BedrockAgentRuntime
       attr_reader :event_emitter
 
     end
+    class FlowResponseStream
+
+      def initialize
+        @event_emitter = Aws::EventEmitter.new
+      end
+
+      def on_access_denied_exception_event(&block)
+        @event_emitter.on(:access_denied_exception, block) if block_given?
+      end
+
+      def on_bad_gateway_exception_event(&block)
+        @event_emitter.on(:bad_gateway_exception, block) if block_given?
+      end
+
+      def on_conflict_exception_event(&block)
+        @event_emitter.on(:conflict_exception, block) if block_given?
+      end
+
+      def on_dependency_failed_exception_event(&block)
+        @event_emitter.on(:dependency_failed_exception, block) if block_given?
+      end
+
+      def on_flow_completion_event_event(&block)
+        @event_emitter.on(:flow_completion_event, block) if block_given?
+      end
+
+      def on_flow_output_event_event(&block)
+        @event_emitter.on(:flow_output_event, block) if block_given?
+      end
+
+      def on_internal_server_exception_event(&block)
+        @event_emitter.on(:internal_server_exception, block) if block_given?
+      end
+
+      def on_resource_not_found_exception_event(&block)
+        @event_emitter.on(:resource_not_found_exception, block) if block_given?
+      end
+
+      def on_service_quota_exceeded_exception_event(&block)
+        @event_emitter.on(:service_quota_exceeded_exception, block) if block_given?
+      end
+
+      def on_throttling_exception_event(&block)
+        @event_emitter.on(:throttling_exception, block) if block_given?
+      end
+
+      def on_validation_exception_event(&block)
+        @event_emitter.on(:validation_exception, block) if block_given?
+      end
+
+      def on_error_event(&block)
+        @event_emitter.on(:error, block) if block_given?
+      end
+
+      def on_initial_response_event(&block)
+        @event_emitter.on(:initial_response, block) if block_given?
+      end
+
+      def on_unknown_event(&block)
+        @event_emitter.on(:unknown_event, block) if block_given?
+      end
+
+      def on_event(&block)
+        on_access_denied_exception_event(&block)
+        on_bad_gateway_exception_event(&block)
+        on_conflict_exception_event(&block)
+        on_dependency_failed_exception_event(&block)
+        on_flow_completion_event_event(&block)
+        on_flow_output_event_event(&block)
+        on_internal_server_exception_event(&block)
+        on_resource_not_found_exception_event(&block)
+        on_service_quota_exceeded_exception_event(&block)
+        on_throttling_exception_event(&block)
+        on_validation_exception_event(&block)
+        on_error_event(&block)
+        on_initial_response_event(&block)
+        on_unknown_event(&block)
+      end
+
+      # @api private
+      # @return Aws::EventEmitter
+      attr_reader :event_emitter
+
+    end
 
   end
 end

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

@@ -58,8 +58,14 @@ module Aws::BedrockAgentRuntime
 
         def parameters_for_operation(context)
           case context.operation_name
+          when :delete_agent_memory
+            Aws::BedrockAgentRuntime::Endpoints::DeleteAgentMemory.build(context)
+          when :get_agent_memory
+            Aws::BedrockAgentRuntime::Endpoints::GetAgentMemory.build(context)
           when :invoke_agent
             Aws::BedrockAgentRuntime::Endpoints::InvokeAgent.build(context)
+          when :invoke_flow
+            Aws::BedrockAgentRuntime::Endpoints::InvokeFlow.build(context)
           when :retrieve
             Aws::BedrockAgentRuntime::Endpoints::Retrieve.build(context)
           when :retrieve_and_generate