aws-sdk-bedrockagentruntime 1.31.0 → 1.33.0

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -55,7 +55,7 @@ module Aws::BedrockAgentRuntime
   autoload :Endpoints, 'aws-sdk-bedrockagentruntime/endpoints'
   autoload :EventStreams, 'aws-sdk-bedrockagentruntime/event_streams'
 
-  GEM_VERSION = '1.31.0'
+  GEM_VERSION = '1.33.0'
 
 end
 

data/sig/client.rbs

@@ -265,6 +265,215 @@ module Aws
                        ) ?{ (*untyped) -> void } -> _InvokeFlowResponseSuccess
                      | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeFlowResponseSuccess
 
+      interface _InvokeInlineAgentResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::InvokeInlineAgentResponse]
+        def completion: () -> Types::InlineAgentResponseStream
+        def content_type: () -> ::String
+        def session_id: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#invoke_inline_agent-instance_method
+      def invoke_inline_agent: (
+                                 ?action_groups: Array[
+                                   {
+                                     action_group_executor: {
+                                       custom_control: ("RETURN_CONTROL")?,
+                                       lambda: ::String?
+                                     }?,
+                                     action_group_name: ::String,
+                                     api_schema: {
+                                       payload: ::String?,
+                                       s3: {
+                                         s3_bucket_name: ::String?,
+                                         s3_object_key: ::String?
+                                       }?
+                                     }?,
+                                     description: ::String?,
+                                     function_schema: {
+                                       functions: Array[
+                                         {
+                                           description: ::String?,
+                                           name: ::String,
+                                           parameters: Hash[::String, {
+                                               description: ::String?,
+                                               required: bool?,
+                                               type: ("string" | "number" | "integer" | "boolean" | "array")
+                                             }]?,
+                                           require_confirmation: ("ENABLED" | "DISABLED")?
+                                         },
+                                       ]?
+                                     }?,
+                                     parent_action_group_signature: ("AMAZON.UserInput" | "AMAZON.CodeInterpreter")?
+                                   },
+                                 ],
+                                 ?customer_encryption_key_arn: ::String,
+                                 ?enable_trace: bool,
+                                 ?end_session: bool,
+                                 foundation_model: ::String,
+                                 ?guardrail_configuration: {
+                                   guardrail_identifier: ::String,
+                                   guardrail_version: ::String
+                                 },
+                                 ?idle_session_ttl_in_seconds: ::Integer,
+                                 ?inline_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?,
+                                   prompt_session_attributes: Hash[::String, ::String]?,
+                                   return_control_invocation_results: Array[
+                                     {
+                                       api_result: {
+                                         action_group: ::String,
+                                         api_path: ::String?,
+                                         confirmation_state: ("CONFIRM" | "DENY")?,
+                                         http_method: ::String?,
+                                         http_status_code: ::Integer?,
+                                         response_body: Hash[::String, {
+                                             body: ::String?
+                                           }]?,
+                                         response_state: ("FAILURE" | "REPROMPT")?
+                                       }?,
+                                       function_result: {
+                                         action_group: ::String,
+                                         confirmation_state: ("CONFIRM" | "DENY")?,
+                                         function: ::String?,
+                                         response_body: Hash[::String, {
+                                             body: ::String?
+                                           }]?,
+                                         response_state: ("FAILURE" | "REPROMPT")?
+                                       }?
+                                     },
+                                   ]?,
+                                   session_attributes: Hash[::String, ::String]?
+                                 },
+                                 ?input_text: ::String,
+                                 instruction: ::String,
+                                 ?knowledge_bases: Array[
+                                   {
+                                     description: ::String,
+                                     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_override_configuration: {
+                                   override_lambda: ::String?,
+                                   prompt_configurations: Array[
+                                     {
+                                       base_prompt_template: ::String?,
+                                       inference_configuration: {
+                                         maximum_length: ::Integer?,
+                                         stop_sequences: Array[::String]?,
+                                         temperature: ::Float?,
+                                         top_k: ::Integer?,
+                                         top_p: ::Float?
+                                       }?,
+                                       parser_mode: ("DEFAULT" | "OVERRIDDEN")?,
+                                       prompt_creation_mode: ("DEFAULT" | "OVERRIDDEN")?,
+                                       prompt_state: ("ENABLED" | "DISABLED")?,
+                                       prompt_type: ("PRE_PROCESSING" | "ORCHESTRATION" | "KNOWLEDGE_BASE_RESPONSE_GENERATION" | "POST_PROCESSING" | "ROUTING_CLASSIFIER")?
+                                     },
+                                   ]
+                                 },
+                                 session_id: ::String
+                               ) ?{ (*untyped) -> void } -> _InvokeInlineAgentResponseSuccess
+                             | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeInlineAgentResponseSuccess
+
+      interface _OptimizePromptResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::OptimizePromptResponse]
+        def optimized_prompt: () -> Types::OptimizedPromptStream
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#optimize_prompt-instance_method
+      def optimize_prompt: (
+                             input: {
+                               text_prompt: {
+                                 text: ::String
+                               }?
+                             },
+                             target_model_id: ::String
+                           ) ?{ (*untyped) -> void } -> _OptimizePromptResponseSuccess
+                         | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _OptimizePromptResponseSuccess
+
       interface _RetrieveResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::RetrieveResponse]
         def next_token: () -> ::String

data/sig/types.rbs

@@ -8,12 +8,40 @@
 module Aws::BedrockAgentRuntime
   module Types
 
+    class APISchema
+      attr_accessor payload: ::String
+      attr_accessor s3: Types::S3Identifier
+      attr_accessor unknown: untyped
+      SENSITIVE: [:payload]
+
+      class Payload < APISchema
+      end
+      class S3 < APISchema
+      end
+      class Unknown < APISchema
+      end
+    end
+
     class AccessDeniedException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
       SENSITIVE: []
     end
 
+    class ActionGroupExecutor
+      attr_accessor custom_control: ("RETURN_CONTROL")
+      attr_accessor lambda: ::String
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class CustomControl < ActionGroupExecutor
+      end
+      class Lambda < ActionGroupExecutor
+      end
+      class Unknown < ActionGroupExecutor
+      end
+    end
+
     class ActionGroupInvocationInput
       attr_accessor action_group_name: ::String
       attr_accessor api_path: ::String
@@ -31,6 +59,22 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:text]
     end
 
+    class AgentActionGroup
+      attr_accessor action_group_executor: Types::ActionGroupExecutor
+      attr_accessor action_group_name: ::String
+      attr_accessor api_schema: Types::APISchema
+      attr_accessor description: ::String
+      attr_accessor function_schema: Types::FunctionSchema
+      attr_accessor parent_action_group_signature: ("AMAZON.UserInput" | "AMAZON.CodeInterpreter")
+      SENSITIVE: [:action_group_name, :description]
+    end
+
+    class AnalyzePromptEvent
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
     class ApiInvocationInput
       attr_accessor action_group: ::String
       attr_accessor action_invocation_type: ("RESULT" | "USER_CONFIRMATION" | "USER_CONFIRMATION_AND_RESULT")
@@ -315,6 +359,14 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class FunctionDefinition
+      attr_accessor description: ::String
+      attr_accessor name: ::String
+      attr_accessor parameters: ::Hash[::String, Types::ParameterDetail]
+      attr_accessor require_confirmation: ("ENABLED" | "DISABLED")
+      SENSITIVE: [:name]
+    end
+
     class FunctionInvocationInput
       attr_accessor action_group: ::String
       attr_accessor action_invocation_type: ("RESULT" | "USER_CONFIRMATION" | "USER_CONFIRMATION_AND_RESULT")
@@ -339,6 +391,17 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class FunctionSchema
+      attr_accessor functions: ::Array[Types::FunctionDefinition]
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class Functions < FunctionSchema
+      end
+      class Unknown < FunctionSchema
+      end
+    end
+
     class GeneratedResponsePart
       attr_accessor text_response_part: Types::TextResponsePart
       SENSITIVE: [:text_response_part]
@@ -382,6 +445,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class GuardrailConfigurationWithArn
+      attr_accessor guardrail_identifier: ::String
+      attr_accessor guardrail_version: ::String
+      SENSITIVE: []
+    end
+
     class GuardrailContentFilter
       attr_accessor action: ("BLOCKED")
       attr_accessor confidence: ("NONE" | "LOW" | "MEDIUM" | "HIGH")
@@ -468,6 +537,42 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class InlineAgentFilePart
+      attr_accessor files: ::Array[Types::OutputFile]
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class InlineAgentPayloadPart
+      attr_accessor attribution: Types::Attribution
+      attr_accessor bytes: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: [:bytes]
+    end
+
+    class InlineAgentReturnControlPayload
+      attr_accessor invocation_id: ::String
+      attr_accessor invocation_inputs: ::Array[Types::InvocationInputMember]
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class InlineAgentTracePart
+      attr_accessor session_id: ::String
+      attr_accessor trace: Types::Trace
+      attr_accessor event_type: untyped
+      SENSITIVE: [:trace]
+    end
+
+    class InlineSessionState
+      attr_accessor files: ::Array[Types::InputFile]
+      attr_accessor invocation_id: ::String
+      attr_accessor prompt_session_attributes: ::Hash[::String, ::String]
+      attr_accessor return_control_invocation_results: ::Array[Types::InvocationResultMember]
+      attr_accessor session_attributes: ::Hash[::String, ::String]
+      SENSITIVE: []
+    end
+
     class InputFile
       attr_accessor name: ::String
       attr_accessor source: Types::FileSource
@@ -475,6 +580,17 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class InputPrompt
+      attr_accessor text_prompt: Types::TextPrompt
+      attr_accessor unknown: untyped
+      SENSITIVE: [:text_prompt]
+
+      class TextPrompt < InputPrompt
+      end
+      class Unknown < InputPrompt
+      end
+    end
+
     class InternalServerException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -551,6 +667,37 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class InvokeInlineAgentRequest
+      attr_accessor action_groups: ::Array[Types::AgentActionGroup]
+      attr_accessor customer_encryption_key_arn: ::String
+      attr_accessor enable_trace: bool
+      attr_accessor end_session: bool
+      attr_accessor foundation_model: ::String
+      attr_accessor guardrail_configuration: Types::GuardrailConfigurationWithArn
+      attr_accessor idle_session_ttl_in_seconds: ::Integer
+      attr_accessor inline_session_state: Types::InlineSessionState
+      attr_accessor input_text: ::String
+      attr_accessor instruction: ::String
+      attr_accessor knowledge_bases: ::Array[Types::KnowledgeBase]
+      attr_accessor prompt_override_configuration: Types::PromptOverrideConfiguration
+      attr_accessor session_id: ::String
+      SENSITIVE: [:input_text, :instruction, :prompt_override_configuration]
+    end
+
+    class InvokeInlineAgentResponse
+      attr_accessor completion: Types::InlineAgentResponseStream
+      attr_accessor content_type: ::String
+      attr_accessor session_id: ::String
+      SENSITIVE: []
+    end
+
+    class KnowledgeBase
+      attr_accessor description: ::String
+      attr_accessor knowledge_base_id: ::String
+      attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
+      SENSITIVE: [:description]
+    end
+
     class KnowledgeBaseConfiguration
       attr_accessor knowledge_base_id: ::String
       attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
@@ -634,7 +781,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor prompt_creation_mode: ("DEFAULT" | "OVERRIDDEN")
       attr_accessor text: ::String
       attr_accessor trace_id: ::String
-      attr_accessor type: ("PRE_PROCESSING" | "ORCHESTRATION" | "KNOWLEDGE_BASE_RESPONSE_GENERATION" | "POST_PROCESSING")
+      attr_accessor type: ("PRE_PROCESSING" | "ORCHESTRATION" | "KNOWLEDGE_BASE_RESPONSE_GENERATION" | "POST_PROCESSING" | "ROUTING_CLASSIFIER")
       SENSITIVE: [:text]
     end
 
@@ -649,6 +796,34 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:reprompt_response]
     end
 
+    class OptimizePromptRequest
+      attr_accessor input: Types::InputPrompt
+      attr_accessor target_model_id: ::String
+      SENSITIVE: []
+    end
+
+    class OptimizePromptResponse
+      attr_accessor optimized_prompt: Types::OptimizedPromptStream
+      SENSITIVE: []
+    end
+
+    class OptimizedPrompt
+      attr_accessor text_prompt: Types::TextPrompt
+      attr_accessor unknown: untyped
+      SENSITIVE: [:text_prompt]
+
+      class TextPrompt < OptimizedPrompt
+      end
+      class Unknown < OptimizedPrompt
+      end
+    end
+
+    class OptimizedPromptEvent
+      attr_accessor optimized_prompt: Types::OptimizedPrompt
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
     class OrchestrationConfiguration
       attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
       attr_accessor inference_config: Types::InferenceConfig
@@ -701,6 +876,13 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class ParameterDetail
+      attr_accessor description: ::String
+      attr_accessor required: bool
+      attr_accessor type: ("string" | "number" | "integer" | "boolean" | "array")
+      SENSITIVE: []
+    end
+
     class PayloadPart
       attr_accessor attribution: Types::Attribution
       attr_accessor bytes: ::String
@@ -763,6 +945,22 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class PromptConfiguration
+      attr_accessor base_prompt_template: ::String
+      attr_accessor inference_configuration: Types::InferenceConfiguration
+      attr_accessor parser_mode: ("DEFAULT" | "OVERRIDDEN")
+      attr_accessor prompt_creation_mode: ("DEFAULT" | "OVERRIDDEN")
+      attr_accessor prompt_state: ("ENABLED" | "DISABLED")
+      attr_accessor prompt_type: ("PRE_PROCESSING" | "ORCHESTRATION" | "KNOWLEDGE_BASE_RESPONSE_GENERATION" | "POST_PROCESSING" | "ROUTING_CLASSIFIER")
+      SENSITIVE: [:base_prompt_template]
+    end
+
+    class PromptOverrideConfiguration
+      attr_accessor override_lambda: ::String
+      attr_accessor prompt_configurations: ::Array[Types::PromptConfiguration]
+      SENSITIVE: []
+    end
+
     class PromptTemplate
       attr_accessor text_prompt_template: ::String
       SENSITIVE: [:text_prompt_template]
@@ -959,6 +1157,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class S3Identifier
+      attr_accessor s3_bucket_name: ::String
+      attr_accessor s3_object_key: ::String
+      SENSITIVE: []
+    end
+
     class S3ObjectDoc
       attr_accessor uri: ::String
       SENSITIVE: []
@@ -999,6 +1203,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class TextPrompt
+      attr_accessor text: ::String
+      SENSITIVE: []
+    end
+
     class TextResponsePart
       attr_accessor span: Types::Span
       attr_accessor text: ::String
@@ -1060,6 +1269,14 @@ module Aws::BedrockAgentRuntime
       def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :conflict_exception, :dependency_failed_exception, :flow_completion_event, :flow_output_event, :flow_trace_event, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :validation_exception]
     end
 
+    class InlineAgentResponseStream < 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]
+    end
+
+    class OptimizedPromptStream < Enumerator[untyped, untyped]
+      def event_types: () -> [:access_denied_exception, :analyze_prompt_event, :bad_gateway_exception, :dependency_failed_exception, :internal_server_exception, :optimized_prompt_event, :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, :files, :internal_server_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
     end

metadata

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