aws-sdk-bedrockagentruntime 1.3.0 → 1.5.0

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -53,6 +53,6 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
 # @!group service
 module Aws::BedrockAgentRuntime
 
-  GEM_VERSION = '1.3.0'
+  GEM_VERSION = '1.5.0'
 
 end

data/sig/client.rbs

@@ -107,6 +107,59 @@ module Aws
                       ?next_token: ::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: {
+                              }
+                            }?,
+                            not_equals: {
+                              key: ::String,
+                              value: {
+                              }
+                            }?,
+                            not_in: {
+                              key: ::String,
+                              value: {
+                              }
+                            }?,
+                            or_all: Array[
+                              untyped,
+                            ]?,
+                            starts_with: {
+                              key: ::String,
+                              value: {
+                              }
+                            }?
+                          }?,
                           number_of_results: ::Integer?,
                           override_search_type: ("HYBRID" | "SEMANTIC")?
                         }
@@ -130,10 +183,68 @@ module Aws
                                    },
                                    ?retrieve_and_generate_configuration: {
                                      knowledge_base_configuration: {
+                                       generation_configuration: {
+                                         prompt_template: {
+                                           text_prompt_template: ::String?
+                                         }?
+                                       }?,
                                        knowledge_base_id: ::String,
                                        model_arn: ::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: {
+                                               }
+                                             }?,
+                                             not_equals: {
+                                               key: ::String,
+                                               value: {
+                                               }
+                                             }?,
+                                             not_in: {
+                                               key: ::String,
+                                               value: {
+                                               }
+                                             }?,
+                                             or_all: Array[
+                                               untyped,
+                                             ]?,
+                                             starts_with: {
+                                               key: ::String,
+                                               value: {
+                                               }
+                                             }?
+                                           }?,
                                            number_of_results: ::Integer?,
                                            override_search_type: ("HYBRID" | "SEMANTIC")?
                                          }

data/sig/types.rbs

@@ -65,6 +65,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:failure_reason]
     end
 
+    class FilterAttribute
+      attr_accessor key: ::String
+      attr_accessor value: untyped
+      SENSITIVE: []
+    end
+
     class FinalResponse
       attr_accessor text: ::String
       SENSITIVE: [:text]
@@ -75,6 +81,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:text_response_part]
     end
 
+    class GenerationConfiguration
+      attr_accessor prompt_template: Types::PromptTemplate
+      SENSITIVE: []
+    end
+
     class InferenceConfiguration
       attr_accessor maximum_length: ::Integer
       attr_accessor stop_sequences: ::Array[::String]
@@ -140,11 +151,13 @@ module Aws::BedrockAgentRuntime
     class KnowledgeBaseRetrievalResult
       attr_accessor content: Types::RetrievalResultContent
       attr_accessor location: Types::RetrievalResultLocation
+      attr_accessor metadata: ::Hash[::String, untyped]
       attr_accessor score: ::Float
-      SENSITIVE: [:content, :location]
+      SENSITIVE: [:content, :location, :metadata]
     end
 
     class KnowledgeBaseRetrieveAndGenerateConfiguration
+      attr_accessor generation_configuration: Types::GenerationConfiguration
       attr_accessor knowledge_base_id: ::String
       attr_accessor model_arn: ::String
       attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
@@ -152,9 +165,10 @@ module Aws::BedrockAgentRuntime
     end
 
     class KnowledgeBaseVectorSearchConfiguration
+      attr_accessor filter: Types::RetrievalFilter
       attr_accessor number_of_results: ::Integer
       attr_accessor override_search_type: ("HYBRID" | "SEMANTIC")
-      SENSITIVE: []
+      SENSITIVE: [:filter]
     end
 
     class ModelInvocationInput
@@ -263,6 +277,11 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class PromptTemplate
+      attr_accessor text_prompt_template: ::String
+      SENSITIVE: [:text_prompt_template]
+    end
+
     class Rationale
       attr_accessor text: ::String
       attr_accessor trace_id: ::String
@@ -286,6 +305,47 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrievalFilter
+      attr_accessor and_all: ::Array[Types::RetrievalFilter]
+      attr_accessor equals: Types::FilterAttribute
+      attr_accessor greater_than: Types::FilterAttribute
+      attr_accessor greater_than_or_equals: Types::FilterAttribute
+      attr_accessor in: Types::FilterAttribute
+      attr_accessor less_than: Types::FilterAttribute
+      attr_accessor less_than_or_equals: Types::FilterAttribute
+      attr_accessor not_equals: Types::FilterAttribute
+      attr_accessor not_in: Types::FilterAttribute
+      attr_accessor or_all: ::Array[Types::RetrievalFilter]
+      attr_accessor starts_with: Types::FilterAttribute
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class AndAll < RetrievalFilter
+      end
+      class Equals < RetrievalFilter
+      end
+      class GreaterThan < RetrievalFilter
+      end
+      class GreaterThanOrEquals < RetrievalFilter
+      end
+      class In < RetrievalFilter
+      end
+      class LessThan < RetrievalFilter
+      end
+      class LessThanOrEquals < RetrievalFilter
+      end
+      class NotEquals < RetrievalFilter
+      end
+      class NotIn < RetrievalFilter
+      end
+      class OrAll < RetrievalFilter
+      end
+      class StartsWith < RetrievalFilter
+      end
+      class Unknown < RetrievalFilter
+      end
+    end
+
     class RetrievalResultContent
       attr_accessor text: ::String
       SENSITIVE: []
@@ -355,7 +415,8 @@ module Aws::BedrockAgentRuntime
     class RetrievedReference
       attr_accessor content: Types::RetrievalResultContent
       attr_accessor location: Types::RetrievalResultLocation
-      SENSITIVE: [:content, :location]
+      attr_accessor metadata: ::Hash[::String, untyped]
+      SENSITIVE: [:content, :location, :metadata]
     end
 
     class ServiceQuotaExceededException

metadata

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