aws-sdk-bedrockagentruntime 1.36.0 → 1.38.0

data/sig/client.rbs

@@ -85,10 +85,33 @@ module Aws
       def delete_agent_memory: (
                                  agent_alias_id: ::String,
                                  agent_id: ::String,
-                                 ?memory_id: ::String
+                                 ?memory_id: ::String,
+                                 ?session_id: ::String
                                ) -> _DeleteAgentMemoryResponseSuccess
                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _DeleteAgentMemoryResponseSuccess
 
+      interface _GenerateQueryResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::GenerateQueryResponse]
+        def queries: () -> ::Array[Types::GeneratedQuery]
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#generate_query-instance_method
+      def generate_query: (
+                            query_generation_input: {
+                              text: ::String,
+                              type: ("TEXT")
+                            },
+                            transformation_configuration: {
+                              mode: ("TEXT_TO_SQL"),
+                              text_to_sql_configuration: {
+                                knowledge_base_configuration: {
+                                  knowledge_base_arn: ::String
+                                }?,
+                                type: ("KNOWLEDGE_BASE")
+                              }?
+                            }
+                          ) -> _GenerateQueryResponseSuccess
+                        | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _GenerateQueryResponseSuccess
+
       interface _GetAgentMemoryResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::GetAgentMemoryResponse]
         def memory_contents: () -> ::Array[Types::Memory]
@@ -116,6 +139,11 @@ module Aws
       def invoke_agent: (
                           agent_alias_id: ::String,
                           agent_id: ::String,
+                          ?bedrock_model_configurations: {
+                            performance_config: {
+                              latency: ("standard" | "optimized")?
+                            }?
+                          },
                           ?enable_trace: bool,
                           ?end_session: bool,
                           ?input_text: ::String,
@@ -316,7 +344,12 @@ module Aws
                              node_name: ::String,
                              node_output_name: ::String
                            },
-                         ]
+                         ],
+                         ?model_performance_configuration: {
+                           performance_config: {
+                             latency: ("standard" | "optimized")?
+                           }?
+                         }
                        ) ?{ (*untyped) -> void } -> _InvokeFlowResponseSuccess
                      | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeFlowResponseSuccess
 
@@ -360,6 +393,11 @@ module Aws
                                      parent_action_group_signature: ("AMAZON.UserInput" | "AMAZON.CodeInterpreter")?
                                    },
                                  ],
+                                 ?bedrock_model_configurations: {
+                                   performance_config: {
+                                     latency: ("standard" | "optimized")?
+                                   }?
+                                 },
                                  ?customer_encryption_key_arn: ::String,
                                  ?enable_trace: bool,
                                  ?end_session: bool,
@@ -764,6 +802,9 @@ module Aws
                                              top_p: ::Float?
                                            }?
                                          }?,
+                                         performance_config: {
+                                           latency: ("standard" | "optimized")?
+                                         }?,
                                          prompt_template: {
                                            text_prompt_template: ::String?
                                          }?
@@ -799,6 +840,9 @@ module Aws
                                              top_p: ::Float?
                                            }?
                                          }?,
+                                         performance_config: {
+                                           latency: ("standard" | "optimized")?
+                                         }?,
                                          prompt_template: {
                                            text_prompt_template: ::String?
                                          }?
@@ -816,6 +860,9 @@ module Aws
                                              top_p: ::Float?
                                            }?
                                          }?,
+                                         performance_config: {
+                                           latency: ("standard" | "optimized")?
+                                         }?,
                                          prompt_template: {
                                            text_prompt_template: ::String?
                                          }?,
@@ -965,6 +1012,9 @@ module Aws
                                                     top_p: ::Float?
                                                   }?
                                                 }?,
+                                                performance_config: {
+                                                  latency: ("standard" | "optimized")?
+                                                }?,
                                                 prompt_template: {
                                                   text_prompt_template: ::String?
                                                 }?
@@ -1000,6 +1050,9 @@ module Aws
                                                     top_p: ::Float?
                                                   }?
                                                 }?,
+                                                performance_config: {
+                                                  latency: ("standard" | "optimized")?
+                                                }?,
                                                 prompt_template: {
                                                   text_prompt_template: ::String?
                                                 }?
@@ -1017,6 +1070,9 @@ module Aws
                                                     top_p: ::Float?
                                                   }?
                                                 }?,
+                                                performance_config: {
+                                                  latency: ("standard" | "optimized")?
+                                                }?,
                                                 prompt_template: {
                                                   text_prompt_template: ::String?
                                                 }?,

data/sig/errors.rbs

@@ -28,6 +28,9 @@ module Aws
       class InternalServerException < ::Aws::Errors::ServiceError
         def message: () -> ::String
       end
+      class ModelNotReadyException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
       class ResourceNotFoundException < ::Aws::Errors::ServiceError
         def message: () -> ::String
       end

data/sig/types.rbs

@@ -151,6 +151,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class BedrockModelConfigurations
+      attr_accessor performance_config: Types::PerformanceConfiguration
+      SENSITIVE: []
+    end
+
     class BedrockRerankingConfiguration
       attr_accessor model_configuration: Types::BedrockRerankingModelConfiguration
       attr_accessor number_of_results: ::Integer
@@ -255,6 +260,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor agent_alias_id: ::String
       attr_accessor agent_id: ::String
       attr_accessor memory_id: ::String
+      attr_accessor session_id: ::String
       SENSITIVE: []
     end
 
@@ -279,6 +285,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
       attr_accessor guardrail_configuration: Types::GuardrailConfiguration
       attr_accessor inference_config: Types::InferenceConfig
+      attr_accessor performance_config: Types::PerformanceConfiguration
       attr_accessor prompt_template: Types::PromptTemplate
       SENSITIVE: []
     end
@@ -497,6 +504,23 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class GenerateQueryRequest
+      attr_accessor query_generation_input: Types::QueryGenerationInput
+      attr_accessor transformation_configuration: Types::TransformationConfiguration
+      SENSITIVE: [:query_generation_input]
+    end
+
+    class GenerateQueryResponse
+      attr_accessor queries: ::Array[Types::GeneratedQuery]
+      SENSITIVE: []
+    end
+
+    class GeneratedQuery
+      attr_accessor sql: ::String
+      attr_accessor type: ("REDSHIFT_SQL")
+      SENSITIVE: []
+    end
+
     class GeneratedResponsePart
       attr_accessor text_response_part: Types::TextResponsePart
       SENSITIVE: [:text_response_part]
@@ -506,6 +530,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
       attr_accessor guardrail_configuration: Types::GuardrailConfiguration
       attr_accessor inference_config: Types::InferenceConfig
+      attr_accessor performance_config: Types::PerformanceConfiguration
       attr_accessor prompt_template: Types::PromptTemplate
       SENSITIVE: []
     end
@@ -671,6 +696,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:trace]
     end
 
+    class InlineBedrockModelConfigurations
+      attr_accessor performance_config: Types::PerformanceConfiguration
+      SENSITIVE: []
+    end
+
     class InlineSessionState
       attr_accessor files: ::Array[Types::InputFile]
       attr_accessor invocation_id: ::String
@@ -745,6 +775,7 @@ module Aws::BedrockAgentRuntime
     class InvokeAgentRequest
       attr_accessor agent_alias_id: ::String
       attr_accessor agent_id: ::String
+      attr_accessor bedrock_model_configurations: Types::BedrockModelConfigurations
       attr_accessor enable_trace: bool
       attr_accessor end_session: bool
       attr_accessor input_text: ::String
@@ -769,6 +800,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor flow_alias_identifier: ::String
       attr_accessor flow_identifier: ::String
       attr_accessor inputs: ::Array[Types::FlowInput]
+      attr_accessor model_performance_configuration: Types::ModelPerformanceConfiguration
       SENSITIVE: []
     end
 
@@ -779,6 +811,7 @@ module Aws::BedrockAgentRuntime
 
     class InvokeInlineAgentRequest
       attr_accessor action_groups: ::Array[Types::AgentActionGroup]
+      attr_accessor bedrock_model_configurations: Types::InlineBedrockModelConfigurations
       attr_accessor customer_encryption_key_arn: ::String
       attr_accessor enable_trace: bool
       attr_accessor end_session: bool
@@ -917,6 +950,17 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:text]
     end
 
+    class ModelNotReadyException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class ModelPerformanceConfiguration
+      attr_accessor performance_config: Types::PerformanceConfiguration
+      SENSITIVE: []
+    end
+
     class Observation
       attr_accessor action_group_invocation_output: Types::ActionGroupInvocationOutput
       attr_accessor agent_collaborator_invocation_output: Types::AgentCollaboratorInvocationOutput
@@ -960,6 +1004,7 @@ module Aws::BedrockAgentRuntime
     class OrchestrationConfiguration
       attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
       attr_accessor inference_config: Types::InferenceConfig
+      attr_accessor performance_config: Types::PerformanceConfiguration
       attr_accessor prompt_template: Types::PromptTemplate
       attr_accessor query_transformation_configuration: Types::QueryTransformationConfiguration
       SENSITIVE: []
@@ -1023,6 +1068,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:bytes]
     end
 
+    class PerformanceConfiguration
+      attr_accessor latency: ("standard" | "optimized")
+      SENSITIVE: []
+    end
+
     class PostProcessingModelInvocationOutput
       attr_accessor metadata: Types::Metadata
       attr_accessor parsed_response: Types::PostProcessingParsedResponse
@@ -1104,6 +1154,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class QueryGenerationInput
+      attr_accessor text: ::String
+      attr_accessor type: ("TEXT")
+      SENSITIVE: []
+    end
+
     class QueryTransformationConfiguration
       attr_accessor type: ("QUERY_DECOMPOSITION")
       SENSITIVE: []
@@ -1255,7 +1311,17 @@ module Aws::BedrockAgentRuntime
     end
 
     class RetrievalResultContent
+      attr_accessor byte_content: ::String
+      attr_accessor row: ::Array[Types::RetrievalResultContentColumn]
       attr_accessor text: ::String
+      attr_accessor type: ("TEXT" | "IMAGE" | "ROW")
+      SENSITIVE: [:row]
+    end
+
+    class RetrievalResultContentColumn
+      attr_accessor column_name: ::String
+      attr_accessor column_value: ::String
+      attr_accessor type: ("BLOB" | "BOOLEAN" | "DOUBLE" | "NULL" | "LONG" | "STRING")
       SENSITIVE: []
     end
 
@@ -1264,13 +1330,20 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrievalResultKendraDocumentLocation
+      attr_accessor uri: ::String
+      SENSITIVE: []
+    end
+
     class RetrievalResultLocation
       attr_accessor confluence_location: Types::RetrievalResultConfluenceLocation
       attr_accessor custom_document_location: Types::RetrievalResultCustomDocumentLocation
+      attr_accessor kendra_document_location: Types::RetrievalResultKendraDocumentLocation
       attr_accessor s3_location: Types::RetrievalResultS3Location
       attr_accessor salesforce_location: Types::RetrievalResultSalesforceLocation
       attr_accessor share_point_location: Types::RetrievalResultSharePointLocation
-      attr_accessor type: ("S3" | "WEB" | "CONFLUENCE" | "SALESFORCE" | "SHAREPOINT" | "CUSTOM")
+      attr_accessor sql_location: Types::RetrievalResultSqlLocation
+      attr_accessor type: ("S3" | "WEB" | "CONFLUENCE" | "SALESFORCE" | "SHAREPOINT" | "CUSTOM" | "KENDRA" | "SQL")
       attr_accessor web_location: Types::RetrievalResultWebLocation
       SENSITIVE: []
     end
@@ -1290,6 +1363,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrievalResultSqlLocation
+      attr_accessor query: ::String
+      SENSITIVE: []
+    end
+
     class RetrievalResultWebLocation
       attr_accessor url: ::String
       SENSITIVE: []
@@ -1480,6 +1558,17 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class TextToSqlConfiguration
+      attr_accessor knowledge_base_configuration: Types::TextToSqlKnowledgeBaseConfiguration
+      attr_accessor type: ("KNOWLEDGE_BASE")
+      SENSITIVE: []
+    end
+
+    class TextToSqlKnowledgeBaseConfiguration
+      attr_accessor knowledge_base_arn: ::String
+      SENSITIVE: []
+    end
+
     class ThrottlingException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -1527,6 +1616,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:collaborator_name, :trace]
     end
 
+    class TransformationConfiguration
+      attr_accessor mode: ("TEXT_TO_SQL")
+      attr_accessor text_to_sql_configuration: Types::TextToSqlConfiguration
+      SENSITIVE: []
+    end
+
     class Usage
       attr_accessor input_tokens: ::Integer
       attr_accessor output_tokens: ::Integer
@@ -1571,7 +1666,7 @@ module Aws::BedrockAgentRuntime
     end
 
     class ResponseStream < Enumerator[untyped, untyped]
-      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]
+      def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :files, :internal_server_exception, :model_not_ready_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
     end
 
     class RetrieveAndGenerateStreamResponseOutput < Enumerator[untyped, untyped]

metadata

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