aws-sdk-bedrockagentruntime 1.36.0 → 1.38.0

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

@@ -51,6 +51,7 @@ module Aws::BedrockAgentRuntime
     BadGatewayException = Shapes::StructureShape.new(name: 'BadGatewayException')
     BasePromptTemplate = Shapes::StringShape.new(name: 'BasePromptTemplate')
     BedrockModelArn = Shapes::StringShape.new(name: 'BedrockModelArn')
+    BedrockModelConfigurations = Shapes::StructureShape.new(name: 'BedrockModelConfigurations')
     BedrockRerankingConfiguration = Shapes::StructureShape.new(name: 'BedrockRerankingConfiguration')
     BedrockRerankingConfigurationNumberOfResultsInteger = Shapes::IntegerShape.new(name: 'BedrockRerankingConfigurationNumberOfResultsInteger')
     BedrockRerankingModelArn = Shapes::StringShape.new(name: 'BedrockRerankingModelArn')
@@ -140,6 +141,11 @@ module Aws::BedrockAgentRuntime
     FunctionResult = Shapes::StructureShape.new(name: 'FunctionResult')
     FunctionSchema = Shapes::UnionShape.new(name: 'FunctionSchema')
     Functions = Shapes::ListShape.new(name: 'Functions')
+    GenerateQueryRequest = Shapes::StructureShape.new(name: 'GenerateQueryRequest')
+    GenerateQueryResponse = Shapes::StructureShape.new(name: 'GenerateQueryResponse')
+    GeneratedQueries = Shapes::ListShape.new(name: 'GeneratedQueries')
+    GeneratedQuery = Shapes::StructureShape.new(name: 'GeneratedQuery')
+    GeneratedQueryType = Shapes::StringShape.new(name: 'GeneratedQueryType')
     GeneratedResponsePart = Shapes::StructureShape.new(name: 'GeneratedResponsePart')
     GenerationConfiguration = Shapes::StructureShape.new(name: 'GenerationConfiguration')
     GetAgentMemoryRequest = Shapes::StructureShape.new(name: 'GetAgentMemoryRequest')
@@ -190,10 +196,12 @@ module Aws::BedrockAgentRuntime
     InlineAgentResponseStream = Shapes::StructureShape.new(name: 'InlineAgentResponseStream')
     InlineAgentReturnControlPayload = Shapes::StructureShape.new(name: 'InlineAgentReturnControlPayload')
     InlineAgentTracePart = Shapes::StructureShape.new(name: 'InlineAgentTracePart')
+    InlineBedrockModelConfigurations = Shapes::StructureShape.new(name: 'InlineBedrockModelConfigurations')
     InlineSessionState = Shapes::StructureShape.new(name: 'InlineSessionState')
     InputFile = Shapes::StructureShape.new(name: 'InputFile')
     InputFiles = Shapes::ListShape.new(name: 'InputFiles')
     InputPrompt = Shapes::UnionShape.new(name: 'InputPrompt')
+    InputQueryType = Shapes::StringShape.new(name: 'InputQueryType')
     InputText = Shapes::StringShape.new(name: 'InputText')
     Instruction = Shapes::StringShape.new(name: 'Instruction')
     Integer = Shapes::IntegerShape.new(name: 'Integer')
@@ -211,6 +219,7 @@ module Aws::BedrockAgentRuntime
     InvokeInlineAgentResponse = Shapes::StructureShape.new(name: 'InvokeInlineAgentResponse')
     KmsKeyArn = Shapes::StringShape.new(name: 'KmsKeyArn')
     KnowledgeBase = Shapes::StructureShape.new(name: 'KnowledgeBase')
+    KnowledgeBaseArn = Shapes::StringShape.new(name: 'KnowledgeBaseArn')
     KnowledgeBaseConfiguration = Shapes::StructureShape.new(name: 'KnowledgeBaseConfiguration')
     KnowledgeBaseConfigurations = Shapes::ListShape.new(name: 'KnowledgeBaseConfigurations')
     KnowledgeBaseId = Shapes::StringShape.new(name: 'KnowledgeBaseId')
@@ -247,6 +256,8 @@ module Aws::BedrockAgentRuntime
     MimeType = Shapes::StringShape.new(name: 'MimeType')
     ModelIdentifier = Shapes::StringShape.new(name: 'ModelIdentifier')
     ModelInvocationInput = Shapes::StructureShape.new(name: 'ModelInvocationInput')
+    ModelNotReadyException = Shapes::StructureShape.new(name: 'ModelNotReadyException')
+    ModelPerformanceConfiguration = Shapes::StructureShape.new(name: 'ModelPerformanceConfiguration')
     Name = Shapes::StringShape.new(name: 'Name')
     NextToken = Shapes::StringShape.new(name: 'NextToken')
     NodeInputName = Shapes::StringShape.new(name: 'NodeInputName')
@@ -279,6 +290,8 @@ module Aws::BedrockAgentRuntime
     Payload = Shapes::StringShape.new(name: 'Payload')
     PayloadPart = Shapes::StructureShape.new(name: 'PayloadPart')
     PayloadType = Shapes::StringShape.new(name: 'PayloadType')
+    PerformanceConfigLatency = Shapes::StringShape.new(name: 'PerformanceConfigLatency')
+    PerformanceConfiguration = Shapes::StructureShape.new(name: 'PerformanceConfiguration')
     PostProcessingModelInvocationOutput = Shapes::StructureShape.new(name: 'PostProcessingModelInvocationOutput')
     PostProcessingParsedResponse = Shapes::StructureShape.new(name: 'PostProcessingParsedResponse')
     PostProcessingTrace = Shapes::UnionShape.new(name: 'PostProcessingTrace')
@@ -294,7 +307,10 @@ module Aws::BedrockAgentRuntime
     PromptText = Shapes::StringShape.new(name: 'PromptText')
     PromptType = Shapes::StringShape.new(name: 'PromptType')
     PropertyParameters = Shapes::StructureShape.new(name: 'PropertyParameters')
+    QueryGenerationInput = Shapes::StructureShape.new(name: 'QueryGenerationInput')
+    QueryGenerationInputTextString = Shapes::StringShape.new(name: 'QueryGenerationInputTextString')
     QueryTransformationConfiguration = Shapes::StructureShape.new(name: 'QueryTransformationConfiguration')
+    QueryTransformationMode = Shapes::StringShape.new(name: 'QueryTransformationMode')
     QueryTransformationType = Shapes::StringShape.new(name: 'QueryTransformationType')
     RAGStopSequences = Shapes::ListShape.new(name: 'RAGStopSequences')
     RAGStopSequencesMemberString = Shapes::StringShape.new(name: 'RAGStopSequencesMemberString')
@@ -333,7 +349,12 @@ module Aws::BedrockAgentRuntime
     RetrievalFilterList = Shapes::ListShape.new(name: 'RetrievalFilterList')
     RetrievalResultConfluenceLocation = Shapes::StructureShape.new(name: 'RetrievalResultConfluenceLocation')
     RetrievalResultContent = Shapes::StructureShape.new(name: 'RetrievalResultContent')
+    RetrievalResultContentColumn = Shapes::StructureShape.new(name: 'RetrievalResultContentColumn')
+    RetrievalResultContentColumnType = Shapes::StringShape.new(name: 'RetrievalResultContentColumnType')
+    RetrievalResultContentRow = Shapes::ListShape.new(name: 'RetrievalResultContentRow')
+    RetrievalResultContentType = Shapes::StringShape.new(name: 'RetrievalResultContentType')
     RetrievalResultCustomDocumentLocation = Shapes::StructureShape.new(name: 'RetrievalResultCustomDocumentLocation')
+    RetrievalResultKendraDocumentLocation = Shapes::StructureShape.new(name: 'RetrievalResultKendraDocumentLocation')
     RetrievalResultLocation = Shapes::StructureShape.new(name: 'RetrievalResultLocation')
     RetrievalResultLocationType = Shapes::StringShape.new(name: 'RetrievalResultLocationType')
     RetrievalResultMetadata = Shapes::MapShape.new(name: 'RetrievalResultMetadata')
@@ -342,6 +363,7 @@ module Aws::BedrockAgentRuntime
     RetrievalResultS3Location = Shapes::StructureShape.new(name: 'RetrievalResultS3Location')
     RetrievalResultSalesforceLocation = Shapes::StructureShape.new(name: 'RetrievalResultSalesforceLocation')
     RetrievalResultSharePointLocation = Shapes::StructureShape.new(name: 'RetrievalResultSharePointLocation')
+    RetrievalResultSqlLocation = Shapes::StructureShape.new(name: 'RetrievalResultSqlLocation')
     RetrievalResultWebLocation = Shapes::StructureShape.new(name: 'RetrievalResultWebLocation')
     RetrieveAndGenerateConfiguration = Shapes::StructureShape.new(name: 'RetrieveAndGenerateConfiguration')
     RetrieveAndGenerateInput = Shapes::StructureShape.new(name: 'RetrieveAndGenerateInput')
@@ -391,6 +413,9 @@ module Aws::BedrockAgentRuntime
     TextPromptTemplate = Shapes::StringShape.new(name: 'TextPromptTemplate')
     TextPromptTextString = Shapes::StringShape.new(name: 'TextPromptTextString')
     TextResponsePart = Shapes::StructureShape.new(name: 'TextResponsePart')
+    TextToSqlConfiguration = Shapes::StructureShape.new(name: 'TextToSqlConfiguration')
+    TextToSqlConfigurationType = Shapes::StringShape.new(name: 'TextToSqlConfigurationType')
+    TextToSqlKnowledgeBaseConfiguration = Shapes::StructureShape.new(name: 'TextToSqlKnowledgeBaseConfiguration')
     ThrottlingException = Shapes::StructureShape.new(name: 'ThrottlingException')
     TopK = Shapes::IntegerShape.new(name: 'TopK')
     TopP = Shapes::FloatShape.new(name: 'TopP')
@@ -398,6 +423,7 @@ module Aws::BedrockAgentRuntime
     TraceId = Shapes::StringShape.new(name: 'TraceId')
     TraceKnowledgeBaseId = Shapes::StringShape.new(name: 'TraceKnowledgeBaseId')
     TracePart = Shapes::StructureShape.new(name: 'TracePart')
+    TransformationConfiguration = Shapes::StructureShape.new(name: 'TransformationConfiguration')
     Type = Shapes::StringShape.new(name: 'Type')
     Usage = Shapes::StructureShape.new(name: 'Usage')
     ValidationException = Shapes::StructureShape.new(name: 'ValidationException')
@@ -516,6 +542,9 @@ module Aws::BedrockAgentRuntime
     BadGatewayException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
     BadGatewayException.struct_class = Types::BadGatewayException
 
+    BedrockModelConfigurations.add_member(:performance_config, Shapes::ShapeRef.new(shape: PerformanceConfiguration, location_name: "performanceConfig"))
+    BedrockModelConfigurations.struct_class = Types::BedrockModelConfigurations
+
     BedrockRerankingConfiguration.add_member(:model_configuration, Shapes::ShapeRef.new(shape: BedrockRerankingModelConfiguration, required: true, location_name: "modelConfiguration"))
     BedrockRerankingConfiguration.add_member(:number_of_results, Shapes::ShapeRef.new(shape: BedrockRerankingConfigurationNumberOfResultsInteger, location_name: "numberOfResults"))
     BedrockRerankingConfiguration.struct_class = Types::BedrockRerankingConfiguration
@@ -590,6 +619,7 @@ module Aws::BedrockAgentRuntime
     DeleteAgentMemoryRequest.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, required: true, location: "uri", location_name: "agentAliasId"))
     DeleteAgentMemoryRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
     DeleteAgentMemoryRequest.add_member(:memory_id, Shapes::ShapeRef.new(shape: MemoryId, location: "querystring", location_name: "memoryId"))
+    DeleteAgentMemoryRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location: "querystring", location_name: "sessionId"))
     DeleteAgentMemoryRequest.struct_class = Types::DeleteAgentMemoryRequest
 
     DeleteAgentMemoryResponse.struct_class = Types::DeleteAgentMemoryResponse
@@ -608,6 +638,7 @@ module Aws::BedrockAgentRuntime
     ExternalSourcesGenerationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
     ExternalSourcesGenerationConfiguration.add_member(:guardrail_configuration, Shapes::ShapeRef.new(shape: GuardrailConfiguration, location_name: "guardrailConfiguration"))
     ExternalSourcesGenerationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
+    ExternalSourcesGenerationConfiguration.add_member(:performance_config, Shapes::ShapeRef.new(shape: PerformanceConfiguration, location_name: "performanceConfig"))
     ExternalSourcesGenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
     ExternalSourcesGenerationConfiguration.struct_class = Types::ExternalSourcesGenerationConfiguration
 
@@ -777,12 +808,26 @@ module Aws::BedrockAgentRuntime
 
     Functions.member = Shapes::ShapeRef.new(shape: FunctionDefinition)
 
+    GenerateQueryRequest.add_member(:query_generation_input, Shapes::ShapeRef.new(shape: QueryGenerationInput, required: true, location_name: "queryGenerationInput"))
+    GenerateQueryRequest.add_member(:transformation_configuration, Shapes::ShapeRef.new(shape: TransformationConfiguration, required: true, location_name: "transformationConfiguration"))
+    GenerateQueryRequest.struct_class = Types::GenerateQueryRequest
+
+    GenerateQueryResponse.add_member(:queries, Shapes::ShapeRef.new(shape: GeneratedQueries, location_name: "queries"))
+    GenerateQueryResponse.struct_class = Types::GenerateQueryResponse
+
+    GeneratedQueries.member = Shapes::ShapeRef.new(shape: GeneratedQuery)
+
+    GeneratedQuery.add_member(:sql, Shapes::ShapeRef.new(shape: String, location_name: "sql"))
+    GeneratedQuery.add_member(:type, Shapes::ShapeRef.new(shape: GeneratedQueryType, location_name: "type"))
+    GeneratedQuery.struct_class = Types::GeneratedQuery
+
     GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
     GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
 
     GenerationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
     GenerationConfiguration.add_member(:guardrail_configuration, Shapes::ShapeRef.new(shape: GuardrailConfiguration, location_name: "guardrailConfiguration"))
     GenerationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
+    GenerationConfiguration.add_member(:performance_config, Shapes::ShapeRef.new(shape: PerformanceConfiguration, location_name: "performanceConfig"))
     GenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
     GenerationConfiguration.struct_class = Types::GenerationConfiguration
 
@@ -923,6 +968,9 @@ module Aws::BedrockAgentRuntime
     InlineAgentTracePart.add_member(:trace, Shapes::ShapeRef.new(shape: Trace, location_name: "trace"))
     InlineAgentTracePart.struct_class = Types::InlineAgentTracePart
 
+    InlineBedrockModelConfigurations.add_member(:performance_config, Shapes::ShapeRef.new(shape: PerformanceConfiguration, location_name: "performanceConfig"))
+    InlineBedrockModelConfigurations.struct_class = Types::InlineBedrockModelConfigurations
+
     InlineSessionState.add_member(:files, Shapes::ShapeRef.new(shape: InputFiles, location_name: "files"))
     InlineSessionState.add_member(:invocation_id, Shapes::ShapeRef.new(shape: String, location_name: "invocationId"))
     InlineSessionState.add_member(:prompt_session_attributes, Shapes::ShapeRef.new(shape: PromptSessionAttributesMap, location_name: "promptSessionAttributes"))
@@ -974,6 +1022,7 @@ module Aws::BedrockAgentRuntime
 
     InvokeAgentRequest.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, required: true, location: "uri", location_name: "agentAliasId"))
     InvokeAgentRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
+    InvokeAgentRequest.add_member(:bedrock_model_configurations, Shapes::ShapeRef.new(shape: BedrockModelConfigurations, location_name: "bedrockModelConfigurations"))
     InvokeAgentRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
     InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
     InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, location_name: "inputText"))
@@ -996,6 +1045,7 @@ module Aws::BedrockAgentRuntime
     InvokeFlowRequest.add_member(:flow_alias_identifier, Shapes::ShapeRef.new(shape: FlowAliasIdentifier, required: true, location: "uri", location_name: "flowAliasIdentifier"))
     InvokeFlowRequest.add_member(:flow_identifier, Shapes::ShapeRef.new(shape: FlowIdentifier, required: true, location: "uri", location_name: "flowIdentifier"))
     InvokeFlowRequest.add_member(:inputs, Shapes::ShapeRef.new(shape: FlowInputs, required: true, location_name: "inputs"))
+    InvokeFlowRequest.add_member(:model_performance_configuration, Shapes::ShapeRef.new(shape: ModelPerformanceConfiguration, location_name: "modelPerformanceConfiguration"))
     InvokeFlowRequest.struct_class = Types::InvokeFlowRequest
 
     InvokeFlowResponse.add_member(:response_stream, Shapes::ShapeRef.new(shape: FlowResponseStream, required: true, eventstream: true, location_name: "responseStream"))
@@ -1004,6 +1054,7 @@ module Aws::BedrockAgentRuntime
     InvokeFlowResponse[:payload_member] = InvokeFlowResponse.member(:response_stream)
 
     InvokeInlineAgentRequest.add_member(:action_groups, Shapes::ShapeRef.new(shape: AgentActionGroups, location_name: "actionGroups"))
+    InvokeInlineAgentRequest.add_member(:bedrock_model_configurations, Shapes::ShapeRef.new(shape: InlineBedrockModelConfigurations, location_name: "bedrockModelConfigurations"))
     InvokeInlineAgentRequest.add_member(:customer_encryption_key_arn, Shapes::ShapeRef.new(shape: KmsKeyArn, location_name: "customerEncryptionKeyArn"))
     InvokeInlineAgentRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
     InvokeInlineAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
@@ -1118,6 +1169,12 @@ module Aws::BedrockAgentRuntime
     ModelInvocationInput.add_member(:type, Shapes::ShapeRef.new(shape: PromptType, location_name: "type"))
     ModelInvocationInput.struct_class = Types::ModelInvocationInput
 
+    ModelNotReadyException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
+    ModelNotReadyException.struct_class = Types::ModelNotReadyException
+
+    ModelPerformanceConfiguration.add_member(:performance_config, Shapes::ShapeRef.new(shape: PerformanceConfiguration, location_name: "performanceConfig"))
+    ModelPerformanceConfiguration.struct_class = Types::ModelPerformanceConfiguration
+
     Observation.add_member(:action_group_invocation_output, Shapes::ShapeRef.new(shape: ActionGroupInvocationOutput, location_name: "actionGroupInvocationOutput"))
     Observation.add_member(:agent_collaborator_invocation_output, Shapes::ShapeRef.new(shape: AgentCollaboratorInvocationOutput, location_name: "agentCollaboratorInvocationOutput"))
     Observation.add_member(:code_interpreter_invocation_output, Shapes::ShapeRef.new(shape: CodeInterpreterInvocationOutput, location_name: "codeInterpreterInvocationOutput"))
@@ -1158,6 +1215,7 @@ module Aws::BedrockAgentRuntime
 
     OrchestrationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
     OrchestrationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
+    OrchestrationConfiguration.add_member(:performance_config, Shapes::ShapeRef.new(shape: PerformanceConfiguration, location_name: "performanceConfig"))
     OrchestrationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
     OrchestrationConfiguration.add_member(:query_transformation_configuration, Shapes::ShapeRef.new(shape: QueryTransformationConfiguration, location_name: "queryTransformationConfiguration"))
     OrchestrationConfiguration.struct_class = Types::OrchestrationConfiguration
@@ -1209,6 +1267,9 @@ module Aws::BedrockAgentRuntime
     PayloadPart.add_member(:bytes, Shapes::ShapeRef.new(shape: PartBody, location_name: "bytes"))
     PayloadPart.struct_class = Types::PayloadPart
 
+    PerformanceConfiguration.add_member(:latency, Shapes::ShapeRef.new(shape: PerformanceConfigLatency, location_name: "latency"))
+    PerformanceConfiguration.struct_class = Types::PerformanceConfiguration
+
     PostProcessingModelInvocationOutput.add_member(:metadata, Shapes::ShapeRef.new(shape: Metadata, location_name: "metadata"))
     PostProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PostProcessingParsedResponse, location_name: "parsedResponse"))
     PostProcessingModelInvocationOutput.add_member(:raw_response, Shapes::ShapeRef.new(shape: RawResponse, location_name: "rawResponse"))
@@ -1267,6 +1328,10 @@ module Aws::BedrockAgentRuntime
     PropertyParameters.add_member(:properties, Shapes::ShapeRef.new(shape: ParameterList, location_name: "properties"))
     PropertyParameters.struct_class = Types::PropertyParameters
 
+    QueryGenerationInput.add_member(:text, Shapes::ShapeRef.new(shape: QueryGenerationInputTextString, required: true, location_name: "text"))
+    QueryGenerationInput.add_member(:type, Shapes::ShapeRef.new(shape: InputQueryType, required: true, location_name: "type"))
+    QueryGenerationInput.struct_class = Types::QueryGenerationInput
+
     QueryTransformationConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: QueryTransformationType, required: true, location_name: "type"))
     QueryTransformationConfiguration.struct_class = Types::QueryTransformationConfiguration
 
@@ -1348,6 +1413,7 @@ module Aws::BedrockAgentRuntime
     ResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
     ResponseStream.add_member(:files, Shapes::ShapeRef.new(shape: FilePart, event: true, location_name: "files"))
     ResponseStream.add_member(:internal_server_exception, Shapes::ShapeRef.new(shape: InternalServerException, location_name: "internalServerException"))
+    ResponseStream.add_member(:model_not_ready_exception, Shapes::ShapeRef.new(shape: ModelNotReadyException, location_name: "modelNotReadyException"))
     ResponseStream.add_member(:resource_not_found_exception, Shapes::ShapeRef.new(shape: ResourceNotFoundException, location_name: "resourceNotFoundException"))
     ResponseStream.add_member(:return_control, Shapes::ShapeRef.new(shape: ReturnControlPayload, event: true, location_name: "returnControl"))
     ResponseStream.add_member(:service_quota_exceeded_exception, Shapes::ShapeRef.new(shape: ServiceQuotaExceededException, location_name: "serviceQuotaExceededException"))
@@ -1391,17 +1457,32 @@ module Aws::BedrockAgentRuntime
     RetrievalResultConfluenceLocation.add_member(:url, Shapes::ShapeRef.new(shape: String, location_name: "url"))
     RetrievalResultConfluenceLocation.struct_class = Types::RetrievalResultConfluenceLocation
 
-    RetrievalResultContent.add_member(:text, Shapes::ShapeRef.new(shape: String, required: true, location_name: "text"))
+    RetrievalResultContent.add_member(:byte_content, Shapes::ShapeRef.new(shape: String, location_name: "byteContent"))
+    RetrievalResultContent.add_member(:row, Shapes::ShapeRef.new(shape: RetrievalResultContentRow, location_name: "row"))
+    RetrievalResultContent.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
+    RetrievalResultContent.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultContentType, location_name: "type"))
     RetrievalResultContent.struct_class = Types::RetrievalResultContent
 
+    RetrievalResultContentColumn.add_member(:column_name, Shapes::ShapeRef.new(shape: String, location_name: "columnName"))
+    RetrievalResultContentColumn.add_member(:column_value, Shapes::ShapeRef.new(shape: String, location_name: "columnValue"))
+    RetrievalResultContentColumn.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultContentColumnType, location_name: "type"))
+    RetrievalResultContentColumn.struct_class = Types::RetrievalResultContentColumn
+
+    RetrievalResultContentRow.member = Shapes::ShapeRef.new(shape: RetrievalResultContentColumn)
+
     RetrievalResultCustomDocumentLocation.add_member(:id, Shapes::ShapeRef.new(shape: String, location_name: "id"))
     RetrievalResultCustomDocumentLocation.struct_class = Types::RetrievalResultCustomDocumentLocation
 
+    RetrievalResultKendraDocumentLocation.add_member(:uri, Shapes::ShapeRef.new(shape: String, location_name: "uri"))
+    RetrievalResultKendraDocumentLocation.struct_class = Types::RetrievalResultKendraDocumentLocation
+
     RetrievalResultLocation.add_member(:confluence_location, Shapes::ShapeRef.new(shape: RetrievalResultConfluenceLocation, location_name: "confluenceLocation"))
     RetrievalResultLocation.add_member(:custom_document_location, Shapes::ShapeRef.new(shape: RetrievalResultCustomDocumentLocation, location_name: "customDocumentLocation"))
+    RetrievalResultLocation.add_member(:kendra_document_location, Shapes::ShapeRef.new(shape: RetrievalResultKendraDocumentLocation, location_name: "kendraDocumentLocation"))
     RetrievalResultLocation.add_member(:s3_location, Shapes::ShapeRef.new(shape: RetrievalResultS3Location, location_name: "s3Location"))
     RetrievalResultLocation.add_member(:salesforce_location, Shapes::ShapeRef.new(shape: RetrievalResultSalesforceLocation, location_name: "salesforceLocation"))
     RetrievalResultLocation.add_member(:share_point_location, Shapes::ShapeRef.new(shape: RetrievalResultSharePointLocation, location_name: "sharePointLocation"))
+    RetrievalResultLocation.add_member(:sql_location, Shapes::ShapeRef.new(shape: RetrievalResultSqlLocation, location_name: "sqlLocation"))
     RetrievalResultLocation.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultLocationType, required: true, location_name: "type"))
     RetrievalResultLocation.add_member(:web_location, Shapes::ShapeRef.new(shape: RetrievalResultWebLocation, location_name: "webLocation"))
     RetrievalResultLocation.struct_class = Types::RetrievalResultLocation
@@ -1418,6 +1499,9 @@ module Aws::BedrockAgentRuntime
     RetrievalResultSharePointLocation.add_member(:url, Shapes::ShapeRef.new(shape: String, location_name: "url"))
     RetrievalResultSharePointLocation.struct_class = Types::RetrievalResultSharePointLocation
 
+    RetrievalResultSqlLocation.add_member(:query, Shapes::ShapeRef.new(shape: String, location_name: "query"))
+    RetrievalResultSqlLocation.struct_class = Types::RetrievalResultSqlLocation
+
     RetrievalResultWebLocation.add_member(:url, Shapes::ShapeRef.new(shape: String, location_name: "url"))
     RetrievalResultWebLocation.struct_class = Types::RetrievalResultWebLocation
 
@@ -1570,6 +1654,13 @@ module Aws::BedrockAgentRuntime
     TextResponsePart.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
     TextResponsePart.struct_class = Types::TextResponsePart
 
+    TextToSqlConfiguration.add_member(:knowledge_base_configuration, Shapes::ShapeRef.new(shape: TextToSqlKnowledgeBaseConfiguration, location_name: "knowledgeBaseConfiguration"))
+    TextToSqlConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: TextToSqlConfigurationType, required: true, location_name: "type"))
+    TextToSqlConfiguration.struct_class = Types::TextToSqlConfiguration
+
+    TextToSqlKnowledgeBaseConfiguration.add_member(:knowledge_base_arn, Shapes::ShapeRef.new(shape: KnowledgeBaseArn, required: true, location_name: "knowledgeBaseArn"))
+    TextToSqlKnowledgeBaseConfiguration.struct_class = Types::TextToSqlKnowledgeBaseConfiguration
+
     ThrottlingException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
     ThrottlingException.struct_class = Types::ThrottlingException
 
@@ -1600,6 +1691,10 @@ module Aws::BedrockAgentRuntime
     TracePart.add_member(:trace, Shapes::ShapeRef.new(shape: Trace, location_name: "trace"))
     TracePart.struct_class = Types::TracePart
 
+    TransformationConfiguration.add_member(:mode, Shapes::ShapeRef.new(shape: QueryTransformationMode, required: true, location_name: "mode"))
+    TransformationConfiguration.add_member(:text_to_sql_configuration, Shapes::ShapeRef.new(shape: TextToSqlConfiguration, location_name: "textToSqlConfiguration"))
+    TransformationConfiguration.struct_class = Types::TransformationConfiguration
+
     Usage.add_member(:input_tokens, Shapes::ShapeRef.new(shape: Integer, location_name: "inputTokens"))
     Usage.add_member(:output_tokens, Shapes::ShapeRef.new(shape: Integer, location_name: "outputTokens"))
     Usage.struct_class = Types::Usage
@@ -1657,6 +1752,23 @@ module Aws::BedrockAgentRuntime
         o.errors << Shapes::ShapeRef.new(shape: ServiceQuotaExceededException)
       end)
 
+      api.add_operation(:generate_query, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "GenerateQuery"
+        o.http_method = "POST"
+        o.http_request_uri = "/generateQuery"
+        o.input = Shapes::ShapeRef.new(shape: GenerateQueryRequest)
+        o.output = Shapes::ShapeRef.new(shape: GenerateQueryResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ConflictException)
+        o.errors << Shapes::ShapeRef.new(shape: ResourceNotFoundException)
+        o.errors << Shapes::ShapeRef.new(shape: ValidationException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+        o.errors << Shapes::ShapeRef.new(shape: DependencyFailedException)
+        o.errors << Shapes::ShapeRef.new(shape: BadGatewayException)
+        o.errors << Shapes::ShapeRef.new(shape: ThrottlingException)
+        o.errors << Shapes::ShapeRef.new(shape: AccessDeniedException)
+        o.errors << Shapes::ShapeRef.new(shape: ServiceQuotaExceededException)
+      end)
+
       api.add_operation(:get_agent_memory, Seahorse::Model::Operation.new.tap do |o|
         o.name = "GetAgentMemory"
         o.http_method = "GET"
@@ -1686,6 +1798,7 @@ module Aws::BedrockAgentRuntime
         o.http_request_uri = "/agents/{agentId}/agentAliases/{agentAliasId}/sessions/{sessionId}/text"
         o.input = Shapes::ShapeRef.new(shape: InvokeAgentRequest)
         o.output = Shapes::ShapeRef.new(shape: InvokeAgentResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ModelNotReadyException)
         o.errors << Shapes::ShapeRef.new(shape: ConflictException)
         o.errors << Shapes::ShapeRef.new(shape: ResourceNotFoundException)
         o.errors << Shapes::ShapeRef.new(shape: ValidationException)

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

@@ -32,6 +32,7 @@ module Aws::BedrockAgentRuntime
   # * {ConflictException}
   # * {DependencyFailedException}
   # * {InternalServerException}
+  # * {ModelNotReadyException}
   # * {ResourceNotFoundException}
   # * {ServiceQuotaExceededException}
   # * {ThrottlingException}
@@ -128,6 +129,21 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class ModelNotReadyException < ServiceError
+
+      # @param [Seahorse::Client::RequestContext] context
+      # @param [String] message
+      # @param [Aws::BedrockAgentRuntime::Types::ModelNotReadyException] 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

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

@@ -43,6 +43,10 @@ module Aws::BedrockAgentRuntime
         @event_emitter.on(:internal_server_exception, block) if block_given?
       end
 
+      def on_model_not_ready_exception_event(&block)
+        @event_emitter.on(:model_not_ready_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
@@ -87,6 +91,7 @@ module Aws::BedrockAgentRuntime
         on_dependency_failed_exception_event(&block)
         on_files_event(&block)
         on_internal_server_exception_event(&block)
+        on_model_not_ready_exception_event(&block)
         on_resource_not_found_exception_event(&block)
         on_return_control_event(&block)
         on_service_quota_exceeded_exception_event(&block)