aws-sdk-bedrockagentruntime 1.15.0 → 1.18.0

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -33,7 +33,7 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
 # structure.
 #
 #     bedrock_agent_runtime = Aws::BedrockAgentRuntime::Client.new
-#     resp = bedrock_agent_runtime.invoke_agent(params)
+#     resp = bedrock_agent_runtime.delete_agent_memory(params)
 #
 # See {Client} for more information.
 #
@@ -53,6 +53,6 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
 # @!group service
 module Aws::BedrockAgentRuntime
 
-  GEM_VERSION = '1.15.0'
+  GEM_VERSION = '1.18.0'
 
 end

data/sig/client.rbs

@@ -76,10 +76,38 @@ module Aws
                   | (?Hash[Symbol, untyped]) -> instance
 
 
+      interface _DeleteAgentMemoryResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::DeleteAgentMemoryResponse]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#delete_agent_memory-instance_method
+      def delete_agent_memory: (
+                                 agent_alias_id: ::String,
+                                 agent_id: ::String,
+                                 ?memory_id: ::String
+                               ) -> _DeleteAgentMemoryResponseSuccess
+                             | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DeleteAgentMemoryResponseSuccess
+
+      interface _GetAgentMemoryResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::GetAgentMemoryResponse]
+        def memory_contents: () -> ::Array[Types::Memory]
+        def next_token: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#get_agent_memory-instance_method
+      def get_agent_memory: (
+                              agent_alias_id: ::String,
+                              agent_id: ::String,
+                              ?max_items: ::Integer,
+                              memory_id: ::String,
+                              memory_type: ("SESSION_SUMMARY"),
+                              ?next_token: ::String
+                            ) -> _GetAgentMemoryResponseSuccess
+                          | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _GetAgentMemoryResponseSuccess
+
       interface _InvokeAgentResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::InvokeAgentResponse]
         def completion: () -> Types::ResponseStream
         def content_type: () -> ::String
+        def memory_id: () -> ::String
         def session_id: () -> ::String
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#invoke_agent-instance_method
@@ -89,9 +117,100 @@ module Aws
                           ?enable_trace: bool,
                           ?end_session: bool,
                           ?input_text: ::String,
+                          ?memory_id: ::String,
                           session_id: ::String,
                           ?session_state: {
+                            files: Array[
+                              {
+                                name: ::String,
+                                source: {
+                                  byte_content: {
+                                    data: ::String,
+                                    media_type: ::String
+                                  }?,
+                                  s3_location: {
+                                    uri: ::String
+                                  }?,
+                                  source_type: ("S3" | "BYTE_CONTENT")
+                                },
+                                use_case: ("CODE_INTERPRETER" | "CHAT")
+                              },
+                            ]?,
                             invocation_id: ::String?,
+                            knowledge_base_configurations: Array[
+                              {
+                                knowledge_base_id: ::String,
+                                retrieval_configuration: {
+                                  vector_search_configuration: {
+                                    filter: {
+                                      and_all: Array[
+                                        untyped,
+                                      ]?,
+                                      equals: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      greater_than: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      greater_than_or_equals: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      in: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      less_than: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      less_than_or_equals: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      list_contains: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      not_equals: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      not_in: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      or_all: Array[
+                                        untyped,
+                                      ]?,
+                                      starts_with: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?,
+                                      string_contains: {
+                                        key: ::String,
+                                        value: {
+                                        }
+                                      }?
+                                    }?,
+                                    number_of_results: ::Integer?,
+                                    override_search_type: ("HYBRID" | "SEMANTIC")?
+                                  }
+                                }
+                              },
+                            ]?,
                             prompt_session_attributes: Hash[::String, ::String]?,
                             return_control_invocation_results: Array[
                               {
@@ -120,6 +239,27 @@ module Aws
                         ) ?{ (*untyped) -> void } -> _InvokeAgentResponseSuccess
                       | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeAgentResponseSuccess
 
+      interface _InvokeFlowResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::InvokeFlowResponse]
+        def response_stream: () -> Types::FlowResponseStream
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#invoke_flow-instance_method
+      def invoke_flow: (
+                         flow_alias_identifier: ::String,
+                         flow_identifier: ::String,
+                         inputs: Array[
+                           {
+                             content: {
+                               document: {
+                               }?
+                             },
+                             node_name: ::String,
+                             node_output_name: ::String
+                           },
+                         ]
+                       ) ?{ (*untyped) -> void } -> _InvokeFlowResponseSuccess
+                     | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeFlowResponseSuccess
+
       interface _RetrieveResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::RetrieveResponse]
         def next_token: () -> ::String
@@ -274,6 +414,11 @@ module Aws
                                        }?,
                                        knowledge_base_id: ::String,
                                        model_arn: ::String,
+                                       orchestration_configuration: {
+                                         query_transformation_configuration: {
+                                           type: ("QUERY_DECOMPOSITION")
+                                         }
+                                       }?,
                                        retrieval_configuration: {
                                          vector_search_configuration: {
                                            filter: {

data/sig/types.rbs

@@ -17,7 +17,9 @@ module Aws::BedrockAgentRuntime
     class ActionGroupInvocationInput
       attr_accessor action_group_name: ::String
       attr_accessor api_path: ::String
+      attr_accessor execution_type: ("LAMBDA" | "RETURN_CONTROL")
       attr_accessor function: ::String
+      attr_accessor invocation_id: ::String
       attr_accessor parameters: ::Array[Types::Parameter]
       attr_accessor request_body: Types::RequestBody
       attr_accessor verb: ::String
@@ -79,12 +81,32 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:data, :identifier]
     end
 
+    class ByteContentFile
+      attr_accessor data: ::String
+      attr_accessor media_type: ::String
+      SENSITIVE: [:data]
+    end
+
     class Citation
       attr_accessor generated_response_part: Types::GeneratedResponsePart
       attr_accessor retrieved_references: ::Array[Types::RetrievedReference]
       SENSITIVE: []
     end
 
+    class CodeInterpreterInvocationInput
+      attr_accessor code: ::String
+      attr_accessor files: ::Array[::String]
+      SENSITIVE: []
+    end
+
+    class CodeInterpreterInvocationOutput
+      attr_accessor execution_error: ::String
+      attr_accessor execution_output: ::String
+      attr_accessor execution_timeout: bool
+      attr_accessor files: ::Array[::String]
+      SENSITIVE: []
+    end
+
     class ConflictException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -96,6 +118,16 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class DeleteAgentMemoryRequest
+      attr_accessor agent_alias_id: ::String
+      attr_accessor agent_id: ::String
+      attr_accessor memory_id: ::String
+      SENSITIVE: []
+    end
+
+    class DeleteAgentMemoryResponse < Aws::EmptyStructure
+    end
+
     class DependencyFailedException
       attr_accessor message: ::String
       attr_accessor resource_name: ::String
@@ -131,6 +163,19 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:failure_reason]
     end
 
+    class FilePart
+      attr_accessor files: ::Array[Types::OutputFile]
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class FileSource
+      attr_accessor byte_content: Types::ByteContentFile
+      attr_accessor s3_location: Types::S3ObjectFile
+      attr_accessor source_type: ("S3" | "BYTE_CONTENT")
+      SENSITIVE: []
+    end
+
     class FilterAttribute
       attr_accessor key: ::String
       attr_accessor value: untyped
@@ -142,6 +187,49 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:text]
     end
 
+    class FlowCompletionEvent
+      attr_accessor completion_reason: ("SUCCESS")
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class FlowInput
+      attr_accessor content: Types::FlowInputContent
+      attr_accessor node_name: ::String
+      attr_accessor node_output_name: ::String
+      SENSITIVE: [:content]
+    end
+
+    class FlowInputContent
+      attr_accessor document: untyped
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class Document < FlowInputContent
+      end
+      class Unknown < FlowInputContent
+      end
+    end
+
+    class FlowOutputContent
+      attr_accessor document: untyped
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class Document < FlowOutputContent
+      end
+      class Unknown < FlowOutputContent
+      end
+    end
+
+    class FlowOutputEvent
+      attr_accessor content: Types::FlowOutputContent
+      attr_accessor node_name: ::String
+      attr_accessor node_type: ("FlowInputNode" | "FlowOutputNode" | "LambdaFunctionNode" | "KnowledgeBaseNode" | "PromptNode" | "ConditionNode" | "LexNode")
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
     class FunctionInvocationInput
       attr_accessor action_group: ::String
       attr_accessor function: ::String
@@ -177,6 +265,22 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class GetAgentMemoryRequest
+      attr_accessor agent_alias_id: ::String
+      attr_accessor agent_id: ::String
+      attr_accessor max_items: ::Integer
+      attr_accessor memory_id: ::String
+      attr_accessor memory_type: ("SESSION_SUMMARY")
+      attr_accessor next_token: ::String
+      SENSITIVE: []
+    end
+
+    class GetAgentMemoryResponse
+      attr_accessor memory_contents: ::Array[Types::Memory]
+      attr_accessor next_token: ::String
+      SENSITIVE: []
+    end
+
     class GuardrailAssessment
       attr_accessor content_policy: Types::GuardrailContentPolicyAssessment
       attr_accessor sensitive_information_policy: Types::GuardrailSensitiveInformationPolicyAssessment
@@ -277,6 +381,13 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class InputFile
+      attr_accessor name: ::String
+      attr_accessor source: Types::FileSource
+      attr_accessor use_case: ("CODE_INTERPRETER" | "CHAT")
+      SENSITIVE: []
+    end
+
     class InternalServerException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -285,7 +396,8 @@ module Aws::BedrockAgentRuntime
 
     class InvocationInput
       attr_accessor action_group_invocation_input: Types::ActionGroupInvocationInput
-      attr_accessor invocation_type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH")
+      attr_accessor code_interpreter_invocation_input: Types::CodeInterpreterInvocationInput
+      attr_accessor invocation_type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH" | "ACTION_GROUP_CODE_INTERPRETER")
       attr_accessor knowledge_base_lookup_input: Types::KnowledgeBaseLookupInput
       attr_accessor trace_id: ::String
       SENSITIVE: []
@@ -325,6 +437,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor enable_trace: bool
       attr_accessor end_session: bool
       attr_accessor input_text: ::String
+      attr_accessor memory_id: ::String
       attr_accessor session_id: ::String
       attr_accessor session_state: Types::SessionState
       SENSITIVE: [:input_text]
@@ -333,10 +446,29 @@ module Aws::BedrockAgentRuntime
     class InvokeAgentResponse
       attr_accessor completion: Types::ResponseStream
       attr_accessor content_type: ::String
+      attr_accessor memory_id: ::String
       attr_accessor session_id: ::String
       SENSITIVE: []
     end
 
+    class InvokeFlowRequest
+      attr_accessor flow_alias_identifier: ::String
+      attr_accessor flow_identifier: ::String
+      attr_accessor inputs: ::Array[Types::FlowInput]
+      SENSITIVE: []
+    end
+
+    class InvokeFlowResponse
+      attr_accessor response_stream: Types::FlowResponseStream
+      SENSITIVE: []
+    end
+
+    class KnowledgeBaseConfiguration
+      attr_accessor knowledge_base_id: ::String
+      attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
+      SENSITIVE: []
+    end
+
     class KnowledgeBaseLookupInput
       attr_accessor knowledge_base_id: ::String
       attr_accessor text: ::String
@@ -370,6 +502,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor generation_configuration: Types::GenerationConfiguration
       attr_accessor knowledge_base_id: ::String
       attr_accessor model_arn: ::String
+      attr_accessor orchestration_configuration: Types::OrchestrationConfiguration
       attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
       SENSITIVE: []
     end
@@ -381,6 +514,31 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:filter]
     end
 
+    class Memory
+      attr_accessor session_summary: Types::MemorySessionSummary
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class SessionSummary < Memory
+      end
+      class Unknown < Memory
+      end
+    end
+
+    class MemorySessionSummary
+      attr_accessor memory_id: ::String
+      attr_accessor session_expiry_time: ::Time
+      attr_accessor session_id: ::String
+      attr_accessor session_start_time: ::Time
+      attr_accessor summary_text: ::String
+      SENSITIVE: []
+    end
+
+    class Metadata
+      attr_accessor usage: Types::Usage
+      SENSITIVE: [:usage]
+    end
+
     class ModelInvocationInput
       attr_accessor inference_configuration: Types::InferenceConfiguration
       attr_accessor override_lambda: ::String
@@ -394,6 +552,7 @@ module Aws::BedrockAgentRuntime
 
     class Observation
       attr_accessor action_group_invocation_output: Types::ActionGroupInvocationOutput
+      attr_accessor code_interpreter_invocation_output: Types::CodeInterpreterInvocationOutput
       attr_accessor final_response: Types::FinalResponse
       attr_accessor knowledge_base_lookup_output: Types::KnowledgeBaseLookupOutput
       attr_accessor reprompt_response: Types::RepromptResponse
@@ -402,18 +561,33 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:reprompt_response]
     end
 
+    class OrchestrationConfiguration
+      attr_accessor query_transformation_configuration: Types::QueryTransformationConfiguration
+      SENSITIVE: []
+    end
+
+    class OrchestrationModelInvocationOutput
+      attr_accessor metadata: Types::Metadata
+      attr_accessor raw_response: Types::RawResponse
+      attr_accessor trace_id: ::String
+      SENSITIVE: [:metadata, :raw_response]
+    end
+
     class OrchestrationTrace
       attr_accessor invocation_input: Types::InvocationInput
       attr_accessor model_invocation_input: Types::ModelInvocationInput
+      attr_accessor model_invocation_output: Types::OrchestrationModelInvocationOutput
       attr_accessor observation: Types::Observation
       attr_accessor rationale: Types::Rationale
       attr_accessor unknown: untyped
-      SENSITIVE: [:invocation_input, :model_invocation_input, :observation, :rationale]
+      SENSITIVE: [:invocation_input, :model_invocation_input, :model_invocation_output, :observation, :rationale]
 
       class InvocationInput < OrchestrationTrace
       end
       class ModelInvocationInput < OrchestrationTrace
       end
+      class ModelInvocationOutput < OrchestrationTrace
+      end
       class Observation < OrchestrationTrace
       end
       class Rationale < OrchestrationTrace
@@ -422,6 +596,13 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class OutputFile
+      attr_accessor bytes: ::String
+      attr_accessor name: ::String
+      attr_accessor type: ::String
+      SENSITIVE: [:bytes]
+    end
+
     class Parameter
       attr_accessor name: ::String
       attr_accessor type: ::String
@@ -497,12 +678,22 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class QueryTransformationConfiguration
+      attr_accessor type: ("QUERY_DECOMPOSITION")
+      SENSITIVE: []
+    end
+
     class Rationale
       attr_accessor text: ::String
       attr_accessor trace_id: ::String
       SENSITIVE: [:text]
     end
 
+    class RawResponse
+      attr_accessor content: ::String
+      SENSITIVE: []
+    end
+
     class RepromptResponse
       attr_accessor source: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "PARSER")
       attr_accessor text: ::String
@@ -567,14 +758,23 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class RetrievalResultConfluenceLocation
+      attr_accessor url: ::String
+      SENSITIVE: []
+    end
+
     class RetrievalResultContent
       attr_accessor text: ::String
       SENSITIVE: []
     end
 
     class RetrievalResultLocation
+      attr_accessor confluence_location: Types::RetrievalResultConfluenceLocation
       attr_accessor s3_location: Types::RetrievalResultS3Location
-      attr_accessor type: ("S3")
+      attr_accessor salesforce_location: Types::RetrievalResultSalesforceLocation
+      attr_accessor share_point_location: Types::RetrievalResultSharePointLocation
+      attr_accessor type: ("S3" | "WEB" | "CONFLUENCE" | "SALESFORCE" | "SHAREPOINT")
+      attr_accessor web_location: Types::RetrievalResultWebLocation
       SENSITIVE: []
     end
 
@@ -583,6 +783,21 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrievalResultSalesforceLocation
+      attr_accessor url: ::String
+      SENSITIVE: []
+    end
+
+    class RetrievalResultSharePointLocation
+      attr_accessor url: ::String
+      SENSITIVE: []
+    end
+
+    class RetrievalResultWebLocation
+      attr_accessor url: ::String
+      SENSITIVE: []
+    end
+
     class RetrieveAndGenerateConfiguration
       attr_accessor external_sources_configuration: Types::ExternalSourcesRetrieveAndGenerateConfiguration
       attr_accessor knowledge_base_configuration: Types::KnowledgeBaseRetrieveAndGenerateConfiguration
@@ -654,6 +869,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class S3ObjectFile
+      attr_accessor uri: ::String
+      SENSITIVE: []
+    end
+
     class ServiceQuotaExceededException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -661,7 +881,9 @@ module Aws::BedrockAgentRuntime
     end
 
     class SessionState
+      attr_accessor files: ::Array[Types::InputFile]
       attr_accessor invocation_id: ::String
+      attr_accessor knowledge_base_configurations: ::Array[Types::KnowledgeBaseConfiguration]
       attr_accessor prompt_session_attributes: ::Hash[::String, ::String]
       attr_accessor return_control_invocation_results: ::Array[Types::InvocationResultMember]
       attr_accessor session_attributes: ::Hash[::String, ::String]
@@ -727,14 +949,24 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:trace]
     end
 
+    class Usage
+      attr_accessor input_tokens: ::Integer
+      attr_accessor output_tokens: ::Integer
+      SENSITIVE: []
+    end
+
     class ValidationException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
       SENSITIVE: []
     end
 
+    class FlowResponseStream < Enumerator[untyped, untyped]
+      def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :conflict_exception, :dependency_failed_exception, :flow_completion_event, :flow_output_event, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :validation_exception]
+    end
+
     class ResponseStream < Enumerator[untyped, untyped]
-      def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :internal_server_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
+      def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :files, :internal_server_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagentruntime
 version: !ruby/object:Gem::Version
-  version: 1.15.0
+  version: 1.18.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-07-02 00:00:00.000000000 Z
+date: 2024-08-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core