Class: Aws::BedrockAgentRuntime::Client

Inherits:
Seahorse::Client::Base show all
Includes:
ClientStubs
Defined in:
gems/aws-sdk-bedrockagentruntime/lib/aws-sdk-bedrockagentruntime/client.rb

Overview

An API client for BedrockAgentRuntime. To construct a client, you need to configure a :region and :credentials.

client = Aws::BedrockAgentRuntime::Client.new(
  region: region_name,
  credentials: credentials,
  # ...
)

For details on configuring region and credentials see the developer guide.

See #initialize for a full list of supported configuration options.

Instance Attribute Summary

Attributes inherited from Seahorse::Client::Base

#config, #handlers

API Operations collapse

Instance Method Summary collapse

Methods included from ClientStubs

#api_requests, #stub_data, #stub_responses

Methods inherited from Seahorse::Client::Base

add_plugin, api, clear_plugins, define, new, #operation_names, plugins, remove_plugin, set_api, set_plugins

Methods included from Seahorse::Client::HandlerBuilder

#handle, #handle_request, #handle_response

Constructor Details

#initialize(options) ⇒ Client

Returns a new instance of Client.

Parameters:

  • options (Hash)

Options Hash (options):

  • :credentials (required, Aws::CredentialProvider) —

    Your AWS credentials. This can be an instance of any one of the following classes:

    • Aws::Credentials - Used for configuring static, non-refreshing credentials.

    • Aws::SharedCredentials - Used for loading static credentials from a shared file, such as ~/.aws/config.

    • Aws::AssumeRoleCredentials - Used when you need to assume a role.

    • Aws::AssumeRoleWebIdentityCredentials - Used when you need to assume a role after providing credentials via the web.

    • Aws::SSOCredentials - Used for loading credentials from AWS SSO using an access token generated from aws login.

    • Aws::ProcessCredentials - Used for loading credentials from a process that outputs to stdout.

    • Aws::InstanceProfileCredentials - Used for loading credentials from an EC2 IMDS on an EC2 instance.

    • Aws::ECSCredentials - Used for loading credentials from instances running in ECS.

    • Aws::CognitoIdentityCredentials - Used for loading credentials from the Cognito Identity service.

    When :credentials are not configured directly, the following locations will be searched for credentials:

    • Aws.config[:credentials]
    • The :access_key_id, :secret_access_key, and :session_token options.
    • ENV['AWS_ACCESS_KEY_ID'], ENV['AWS_SECRET_ACCESS_KEY']
    • ~/.aws/credentials
    • ~/.aws/config
    • EC2/ECS IMDS instance profile - When used by default, the timeouts are very aggressive. Construct and pass an instance of Aws::InstanceProfileCredentails or Aws::ECSCredentials to enable retries and extended timeouts. Instance profile credential fetching can be disabled by setting ENV['AWS_EC2_METADATA_DISABLED'] to true.
  • :region (required, String) —

    The AWS region to connect to. The configured :region is used to determine the service :endpoint. When not passed, a default :region is searched for in the following locations:

    • Aws.config[:region]
    • ENV['AWS_REGION']
    • ENV['AMAZON_REGION']
    • ENV['AWS_DEFAULT_REGION']
    • ~/.aws/credentials
    • ~/.aws/config
  • :access_key_id (String)
  • :active_endpoint_cache (Boolean) — default: false —

    When set to true, a thread polling for endpoints will be running in the background every 60 secs (default). Defaults to false.

  • :adaptive_retry_wait_to_fill (Boolean) — default: true —

    Used only in adaptive retry mode. When true, the request will sleep until there is sufficent client side capacity to retry the request. When false, the request will raise a RetryCapacityNotAvailableError and will not retry instead of sleeping.

  • :client_side_monitoring (Boolean) — default: false —

    When true, client-side metrics will be collected for all API requests from this client.

  • :client_side_monitoring_client_id (String) — default: "" —

    Allows you to provide an identifier for this client which will be attached to all generated client side metrics. Defaults to an empty string.

  • :client_side_monitoring_host (String) — default: "127.0.0.1" —

    Allows you to specify the DNS hostname or IPv4 or IPv6 address that the client side monitoring agent is running on, where client metrics will be published via UDP.

  • :client_side_monitoring_port (Integer) — default: 31000 —

    Required for publishing client metrics. The port that the client side monitoring agent is running on, where client metrics will be published via UDP.

  • :client_side_monitoring_publisher (Aws::ClientSideMonitoring::Publisher) — default: Aws::ClientSideMonitoring::Publisher —

    Allows you to provide a custom client-side monitoring publisher class. By default, will use the Client Side Monitoring Agent Publisher.

  • :convert_params (Boolean) — default: true —

    When true, an attempt is made to coerce request parameters into the required types.

  • :correct_clock_skew (Boolean) — default: true —

    Used only in standard and adaptive retry modes. Specifies whether to apply a clock skew correction and retry requests with skewed client clocks.

  • :defaults_mode (String) — default: "legacy" —

    See DefaultsModeConfiguration for a list of the accepted modes and the configuration defaults that are included.

  • :disable_host_prefix_injection (Boolean) — default: false —

    Set to true to disable SDK automatically adding host prefix to default service endpoint when available.

  • :disable_request_compression (Boolean) — default: false —

    When set to 'true' the request body will not be compressed for supported operations.

  • :endpoint (String, URI::HTTPS, URI::HTTP) —

    Normally you should not configure the :endpoint option directly. This is normally constructed from the :region option. Configuring :endpoint is normally reserved for connecting to test or custom endpoints. The endpoint should be a URI formatted like:

    'http://example.com'
'https://example.com'
'http://example.com:123'
  • :endpoint_cache_max_entries (Integer) — default: 1000 —

    Used for the maximum size limit of the LRU cache storing endpoints data for endpoint discovery enabled operations. Defaults to 1000.

  • :endpoint_cache_max_threads (Integer) — default: 10 —

    Used for the maximum threads in use for polling endpoints to be cached, defaults to 10.

  • :endpoint_cache_poll_interval (Integer) — default: 60 —

    When :endpoint_discovery and :active_endpoint_cache is enabled, Use this option to config the time interval in seconds for making requests fetching endpoints information. Defaults to 60 sec.

  • :endpoint_discovery (Boolean) — default: false —

    When set to true, endpoint discovery will be enabled for operations when available.

  • :event_stream_handler (Proc) —

    When an EventStream or Proc object is provided, it will be used as callback for each chunk of event stream response received along the way.

  • :ignore_configured_endpoint_urls (Boolean) —

    Setting to true disables use of endpoint URLs provided via environment variables and the shared configuration file.

  • :input_event_stream_handler (Proc) —

    When an EventStream or Proc object is provided, it can be used for sending events for the event stream.

  • :log_formatter (Aws::Log::Formatter) — default: Aws::Log::Formatter.default —

    The log formatter.

  • :log_level (Symbol) — default: :info —

    The log level to send messages to the :logger at.

  • :logger (Logger) —

    The Logger instance to send log messages to. If this option is not set, logging will be disabled.

  • :max_attempts (Integer) — default: 3 —

    An integer representing the maximum number attempts that will be made for a single request, including the initial attempt. For example, setting this value to 5 will result in a request being retried up to 4 times. Used in standard and adaptive retry modes.

  • :output_event_stream_handler (Proc) —

    When an EventStream or Proc object is provided, it will be used as callback for each chunk of event stream response received along the way.

  • :profile (String) — default: "default" —

    Used when loading credentials from the shared credentials file at HOME/.aws/credentials. When not specified, 'default' is used.

  • :request_min_compression_size_bytes (Integer) — default: 10240 —

    The minimum size in bytes that triggers compression for request bodies. The value must be non-negative integer value between 0 and 10485780 bytes inclusive.

  • :retry_backoff (Proc) —

    A proc or lambda used for backoff. Defaults to 2**retries * retry_base_delay. This option is only used in the legacy retry mode.

  • :retry_base_delay (Float) — default: 0.3 —

    The base delay in seconds used by the default backoff function. This option is only used in the legacy retry mode.

  • :retry_jitter (Symbol) — default: :none —

    A delay randomiser function used by the default backoff function. Some predefined functions can be referenced by name - :none, :equal, :full, otherwise a Proc that takes and returns a number. This option is only used in the legacy retry mode.

    @see https://www.awsarchitectureblog.com/2015/03/backoff.html

  • :retry_limit (Integer) — default: 3 —

    The maximum number of times to retry failed requests. Only ~ 500 level server errors and certain ~ 400 level client errors are retried. Generally, these are throttling errors, data checksum errors, networking errors, timeout errors, auth errors, endpoint discovery, and errors from expired credentials. This option is only used in the legacy retry mode.

  • :retry_max_delay (Integer) — default: 0 —

    The maximum number of seconds to delay between retries (0 for no limit) used by the default backoff function. This option is only used in the legacy retry mode.

  • :retry_mode (String) — default: "legacy" —

    Specifies which retry algorithm to use. Values are:

    • legacy - The pre-existing retry behavior. This is default value if no retry mode is provided.

    • standard - A standardized set of retry rules across the AWS SDKs. This includes support for retry quotas, which limit the number of unsuccessful retries a client can make.

    • adaptive - An experimental retry mode that includes all the functionality of standard mode along with automatic client side throttling. This is a provisional mode that may change behavior in the future.

  • :sdk_ua_app_id (String) —

    A unique and opaque application ID that is appended to the User-Agent header as app/. It should have a maximum length of 50.

  • :secret_access_key (String)
  • :session_token (String)
  • :stub_responses (Boolean) — default: false —

    Causes the client to return stubbed responses. By default fake responses are generated and returned. You can specify the response data to return or errors to raise by calling ClientStubs#stub_responses. See ClientStubs for more information.

    Please note When response stubbing is enabled, no HTTP requests are made, and retries are disabled.

  • :token_provider (Aws::TokenProvider) —

    A Bearer Token Provider. This can be an instance of any one of the following classes:

    • Aws::StaticTokenProvider - Used for configuring static, non-refreshing tokens.

    • Aws::SSOTokenProvider - Used for loading tokens from AWS SSO using an access token generated from aws login.

    When :token_provider is not configured directly, the Aws::TokenProviderChain will be used to search for tokens configured for your profile in shared configuration files.

  • :use_dualstack_endpoint (Boolean) —

    When set to true, dualstack enabled endpoints (with .aws TLD) will be used if available.

  • :use_fips_endpoint (Boolean) —

    When set to true, fips compatible endpoints will be used if available. When a fips region is used, the region is normalized and this config is set to true.

  • :validate_params (Boolean) — default: true —

    When true, request parameters are validated before sending the request.

  • :endpoint_provider (Aws::BedrockAgentRuntime::EndpointProvider) —

    The endpoint provider used to resolve endpoints. Any object that responds to #resolve_endpoint(parameters) where parameters is a Struct similar to Aws::BedrockAgentRuntime::EndpointParameters

  • :http_continue_timeout (Float) — default: 1 —

    The number of seconds to wait for a 100-continue response before sending the request body. This option has no effect unless the request has "Expect" header set to "100-continue". Defaults to nil which disables this behaviour. This value can safely be set per request on the session.

  • :http_idle_timeout (Float) — default: 5 —

    The number of seconds a connection is allowed to sit idle before it is considered stale. Stale connections are closed and removed from the pool before making a request.

  • :http_open_timeout (Float) — default: 15 —

    The default number of seconds to wait for response data. This value can safely be set per-request on the session.

  • :http_proxy (URI::HTTP, String) —

    A proxy to send requests through. Formatted like 'http://proxy.com:123'.

  • :http_read_timeout (Float) — default: 60 —

    The default number of seconds to wait for response data. This value can safely be set per-request on the session.

  • :http_wire_trace (Boolean) — default: false —

    When true, HTTP debug output will be sent to the :logger.

  • :on_chunk_received (Proc) —

    When a Proc object is provided, it will be used as callback when each chunk of the response body is received. It provides three arguments: the chunk, the number of bytes received, and the total number of bytes in the response (or nil if the server did not send a content-length).

  • :on_chunk_sent (Proc) —

    When a Proc object is provided, it will be used as callback when each chunk of the request body is sent. It provides three arguments: the chunk, the number of bytes read from the body, and the total number of bytes in the body.

  • :raise_response_errors (Boolean) — default: true —

    When true, response errors are raised.

  • :ssl_ca_bundle (String) —

    Full path to the SSL certificate authority bundle file that should be used when verifying peer certificates. If you do not pass :ssl_ca_bundle or :ssl_ca_directory the the system default will be used if available.

  • :ssl_ca_directory (String) —

    Full path of the directory that contains the unbundled SSL certificate authority files for verifying peer certificates. If you do not pass :ssl_ca_bundle or :ssl_ca_directory the the system default will be used if available.

  • :ssl_ca_store (String) —

    Sets the X509::Store to verify peer certificate.

  • :ssl_timeout (Float) —

    Sets the SSL timeout in seconds

  • :ssl_verify_peer (Boolean) — default: true —

    When true, SSL peer certificates are verified when establishing a connection.



420
421
422
 
# File 'gems/aws-sdk-bedrockagentruntime/lib/aws-sdk-bedrockagentruntime/client.rb', line 420

def initialize(*args)
  super
end

Instance Method Details

#invoke_agent(params = {}) ⇒ Types::InvokeAgentResponse

Sends a prompt for the agent to process and respond to. Use return control event type for function calling.

The CLI doesn't support InvokeAgent.

  • To continue the same conversation with an agent, use the same sessionId value in the request.

  • To activate trace enablement, turn enableTrace to true. Trace enablement helps you follow the agent's reasoning process that led it to the information it processed, the actions it took, and the final result it yielded. For more information, see Trace enablement.

  • End a conversation by setting endSession to true.

  • In the sessionState object, you can include attributes for the session or prompt or parameters returned from the action group.

  • Use return control event type for function calling.

The response is returned in the bytes field of the chunk object.

  • The attribution object contains citations for parts of the response.

  • If you set enableTrace to true in the request, you can trace the agent's steps and reasoning process that led it to the response.

  • Errors are also surfaced in the response.

Examples:

EventStream Operation Example


You can process the event once it arrives immediately, or wait until the
full response is complete and iterate through the eventstream enumerator.

To interact with event immediately, you need to register #invoke_agent
with callbacks. Callbacks can be registered for specific events or for all
events, including error events.

Callbacks can be passed into the `:event_stream_handler` option or within a
block statement attached to the #invoke_agent call directly. Hybrid
pattern of both is also supported.

`:event_stream_handler` option takes in either a Proc object or
Aws::BedrockAgentRuntime::EventStreams::ResponseStream object.

Usage pattern a): Callbacks with a block attached to #invoke_agent
  Example for registering callbacks for all event types and an error event

  client.invoke_agent( # params input# ) do |stream|
    stream.on_error_event do |event|
      # catch unmodeled error event in the stream
      raise event
      # => Aws::Errors::EventError
      # event.event_type => :error
      # event.error_code => String
      # event.error_message => String
    end

    stream.on_event do |event|
      # process all events arrive
      puts event.event_type
      ...
    end

  end

Usage pattern b): Pass in `:event_stream_handler` for #invoke_agent

  1) Create a Aws::BedrockAgentRuntime::EventStreams::ResponseStream object
  Example for registering callbacks with specific events

    handler = Aws::BedrockAgentRuntime::EventStreams::ResponseStream.new
    handler.on_access_denied_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::accessDeniedException
    end
    handler.on_bad_gateway_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::badGatewayException
    end
    handler.on_chunk_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::chunk
    end
    handler.on_conflict_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::conflictException
    end
    handler.on_dependency_failed_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::dependencyFailedException
    end
    handler.on_internal_server_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::internalServerException
    end
    handler.on_resource_not_found_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::resourceNotFoundException
    end
    handler.on_return_control_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::returnControl
    end
    handler.on_service_quota_exceeded_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::serviceQuotaExceededException
    end
    handler.on_throttling_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::throttlingException
    end
    handler.on_trace_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::trace
    end
    handler.on_validation_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::validationException
    end

  client.invoke_agent( # params input #, event_stream_handler: handler)

  2) Use a Ruby Proc object
  Example for registering callbacks with specific events

  handler = Proc.new do |stream|
    stream.on_access_denied_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::accessDeniedException
    end
    stream.on_bad_gateway_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::badGatewayException
    end
    stream.on_chunk_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::chunk
    end
    stream.on_conflict_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::conflictException
    end
    stream.on_dependency_failed_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::dependencyFailedException
    end
    stream.on_internal_server_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::internalServerException
    end
    stream.on_resource_not_found_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::resourceNotFoundException
    end
    stream.on_return_control_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::returnControl
    end
    stream.on_service_quota_exceeded_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::serviceQuotaExceededException
    end
    stream.on_throttling_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::throttlingException
    end
    stream.on_trace_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::trace
    end
    stream.on_validation_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::validationException
    end
  end

  client.invoke_agent( # params input #, event_stream_handler: handler)

Usage pattern c): Hybrid pattern of a) and b)

    handler = Aws::BedrockAgentRuntime::EventStreams::ResponseStream.new
    handler.on_access_denied_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::accessDeniedException
    end
    handler.on_bad_gateway_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::badGatewayException
    end
    handler.on_chunk_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::chunk
    end
    handler.on_conflict_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::conflictException
    end
    handler.on_dependency_failed_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::dependencyFailedException
    end
    handler.on_internal_server_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::internalServerException
    end
    handler.on_resource_not_found_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::resourceNotFoundException
    end
    handler.on_return_control_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::returnControl
    end
    handler.on_service_quota_exceeded_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::serviceQuotaExceededException
    end
    handler.on_throttling_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::throttlingException
    end
    handler.on_trace_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::trace
    end
    handler.on_validation_exception_event do |event|
      event # => Aws::BedrockAgentRuntime::Types::validationException
    end

  client.invoke_agent( # params input #, event_stream_handler: handler) do |stream|
    stream.on_error_event do |event|
      # catch unmodeled error event in the stream
      raise event
      # => Aws::Errors::EventError
      # event.event_type => :error
      # event.error_code => String
      # event.error_message => String
    end
  end

You can also iterate through events after the response complete.

Events are available at resp.completion # => Enumerator
For parameter input example, please refer to following request syntax

Request syntax with placeholder values


resp = client.invoke_agent({
  agent_alias_id: "AgentAliasId", # required
  agent_id: "AgentId", # required
  enable_trace: false,
  end_session: false,
  input_text: "InputText",
  session_id: "SessionId", # required
  session_state: {
    invocation_id: "String",
    prompt_session_attributes: {
      "String" => "String",
    },
    return_control_invocation_results: [
      {
        api_result: {
          action_group: "String", # required
          api_path: "ApiPath",
          http_method: "String",
          http_status_code: 1,
          response_body: {
            "String" => {
              body: "String",
            },
          },
          response_state: "FAILURE", # accepts FAILURE, REPROMPT
        },
        function_result: {
          action_group: "String", # required
          function: "String",
          response_body: {
            "String" => {
              body: "String",
            },
          },
          response_state: "FAILURE", # accepts FAILURE, REPROMPT
        },
      },
    ],
    session_attributes: {
      "String" => "String",
    },
  },
})

Response structure


All events are available at resp.completion:
resp.completion #=> Enumerator
resp.completion.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]

For :access_denied_exception event available at #on_access_denied_exception_event callback and response eventstream enumerator:
event.message #=> String

For :bad_gateway_exception event available at #on_bad_gateway_exception_event callback and response eventstream enumerator:
event.message #=> String
event.resource_name #=> String

For :chunk event available at #on_chunk_event callback and response eventstream enumerator:
event.attribution.citations #=> Array
event.attribution.citations[0].generated_response_part.text_response_part.span.end #=> Integer
event.attribution.citations[0].generated_response_part.text_response_part.span.start #=> Integer
event.attribution.citations[0].generated_response_part.text_response_part.text #=> String
event.attribution.citations[0].retrieved_references #=> Array
event.attribution.citations[0].retrieved_references[0].content.text #=> String
event.attribution.citations[0].retrieved_references[0].location.s3_location.uri #=> String
event.attribution.citations[0].retrieved_references[0].location.type #=> String, one of "S3"
event.attribution.citations[0].retrieved_references[0].metadata #=> Hash
event.bytes #=> String

For :conflict_exception event available at #on_conflict_exception_event callback and response eventstream enumerator:
event.message #=> String

For :dependency_failed_exception event available at #on_dependency_failed_exception_event callback and response eventstream enumerator:
event.message #=> String
event.resource_name #=> String

For :internal_server_exception event available at #on_internal_server_exception_event callback and response eventstream enumerator:
event.message #=> String

For :resource_not_found_exception event available at #on_resource_not_found_exception_event callback and response eventstream enumerator:
event.message #=> String

For :return_control event available at #on_return_control_event callback and response eventstream enumerator:
event.invocation_id #=> String
event.invocation_inputs #=> Array
event.invocation_inputs[0].api_invocation_input.action_group #=> String
event.invocation_inputs[0].api_invocation_input.api_path #=> String
event.invocation_inputs[0].api_invocation_input.http_method #=> String
event.invocation_inputs[0].api_invocation_input.parameters #=> Array
event.invocation_inputs[0].api_invocation_input.parameters[0].name #=> String
event.invocation_inputs[0].api_invocation_input.parameters[0].type #=> String
event.invocation_inputs[0].api_invocation_input.parameters[0].value #=> String
event.invocation_inputs[0].api_invocation_input.request_body.content #=> Hash
event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties #=> Array
event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties[0].name #=> String
event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties[0].type #=> String
event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties[0].value #=> String
event.invocation_inputs[0].function_invocation_input.action_group #=> String
event.invocation_inputs[0].function_invocation_input.function #=> String
event.invocation_inputs[0].function_invocation_input.parameters #=> Array
event.invocation_inputs[0].function_invocation_input.parameters[0].name #=> String
event.invocation_inputs[0].function_invocation_input.parameters[0].type #=> String
event.invocation_inputs[0].function_invocation_input.parameters[0].value #=> String

For :service_quota_exceeded_exception event available at #on_service_quota_exceeded_exception_event callback and response eventstream enumerator:
event.message #=> String

For :throttling_exception event available at #on_throttling_exception_event callback and response eventstream enumerator:
event.message #=> String

For :trace event available at #on_trace_event callback and response eventstream enumerator:
event.agent_alias_id #=> String
event.agent_id #=> String
event.agent_version #=> String
event.session_id #=> String
event.trace.failure_trace.failure_reason #=> String
event.trace.failure_trace.trace_id #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.action_group_name #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.api_path #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.function #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters #=> Array
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters[0].name #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters[0].type #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters[0].value #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.request_body.content #=> Hash
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.request_body.content["String"] #=> Array
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.request_body.content["String"][0].name #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.request_body.content["String"][0].type #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.request_body.content["String"][0].value #=> String
event.trace.orchestration_trace.invocation_input.action_group_invocation_input.verb #=> String
event.trace.orchestration_trace.invocation_input.invocation_type #=> String, one of "ACTION_GROUP", "KNOWLEDGE_BASE", "FINISH"
event.trace.orchestration_trace.invocation_input.knowledge_base_lookup_input.knowledge_base_id #=> String
event.trace.orchestration_trace.invocation_input.knowledge_base_lookup_input.text #=> String
event.trace.orchestration_trace.invocation_input.trace_id #=> String
event.trace.orchestration_trace.model_invocation_input.inference_configuration.maximum_length #=> Integer
event.trace.orchestration_trace.model_invocation_input.inference_configuration.stop_sequences #=> Array
event.trace.orchestration_trace.model_invocation_input.inference_configuration.stop_sequences[0] #=> String
event.trace.orchestration_trace.model_invocation_input.inference_configuration.temperature #=> Float
event.trace.orchestration_trace.model_invocation_input.inference_configuration.top_k #=> Integer
event.trace.orchestration_trace.model_invocation_input.inference_configuration.top_p #=> Float
event.trace.orchestration_trace.model_invocation_input.override_lambda #=> String
event.trace.orchestration_trace.model_invocation_input.parser_mode #=> String, one of "DEFAULT", "OVERRIDDEN"
event.trace.orchestration_trace.model_invocation_input.prompt_creation_mode #=> String, one of "DEFAULT", "OVERRIDDEN"
event.trace.orchestration_trace.model_invocation_input.text #=> String
event.trace.orchestration_trace.model_invocation_input.trace_id #=> String
event.trace.orchestration_trace.model_invocation_input.type #=> String, one of "PRE_PROCESSING", "ORCHESTRATION", "KNOWLEDGE_BASE_RESPONSE_GENERATION", "POST_PROCESSING"
event.trace.orchestration_trace.observation.action_group_invocation_output.text #=> String
event.trace.orchestration_trace.observation.final_response.text #=> String
event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references #=> Array
event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].content.text #=> String
event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].location.s3_location.uri #=> String
event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].location.type #=> String, one of "S3"
event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].metadata #=> Hash
event.trace.orchestration_trace.observation.reprompt_response.source #=> String, one of "ACTION_GROUP", "KNOWLEDGE_BASE", "PARSER"
event.trace.orchestration_trace.observation.reprompt_response.text #=> String
event.trace.orchestration_trace.observation.trace_id #=> String
event.trace.orchestration_trace.observation.type #=> String, one of "ACTION_GROUP", "KNOWLEDGE_BASE", "FINISH", "ASK_USER", "REPROMPT"
event.trace.orchestration_trace.rationale.text #=> String
event.trace.orchestration_trace.rationale.trace_id #=> String
event.trace.post_processing_trace.model_invocation_input.inference_configuration.maximum_length #=> Integer
event.trace.post_processing_trace.model_invocation_input.inference_configuration.stop_sequences #=> Array
event.trace.post_processing_trace.model_invocation_input.inference_configuration.stop_sequences[0] #=> String
event.trace.post_processing_trace.model_invocation_input.inference_configuration.temperature #=> Float
event.trace.post_processing_trace.model_invocation_input.inference_configuration.top_k #=> Integer
event.trace.post_processing_trace.model_invocation_input.inference_configuration.top_p #=> Float
event.trace.post_processing_trace.model_invocation_input.override_lambda #=> String
event.trace.post_processing_trace.model_invocation_input.parser_mode #=> String, one of "DEFAULT", "OVERRIDDEN"
event.trace.post_processing_trace.model_invocation_input.prompt_creation_mode #=> String, one of "DEFAULT", "OVERRIDDEN"
event.trace.post_processing_trace.model_invocation_input.text #=> String
event.trace.post_processing_trace.model_invocation_input.trace_id #=> String
event.trace.post_processing_trace.model_invocation_input.type #=> String, one of "PRE_PROCESSING", "ORCHESTRATION", "KNOWLEDGE_BASE_RESPONSE_GENERATION", "POST_PROCESSING"
event.trace.post_processing_trace.model_invocation_output.parsed_response.text #=> String
event.trace.post_processing_trace.model_invocation_output.trace_id #=> String
event.trace.pre_processing_trace.model_invocation_input.inference_configuration.maximum_length #=> Integer
event.trace.pre_processing_trace.model_invocation_input.inference_configuration.stop_sequences #=> Array
event.trace.pre_processing_trace.model_invocation_input.inference_configuration.stop_sequences[0] #=> String
event.trace.pre_processing_trace.model_invocation_input.inference_configuration.temperature #=> Float
event.trace.pre_processing_trace.model_invocation_input.inference_configuration.top_k #=> Integer
event.trace.pre_processing_trace.model_invocation_input.inference_configuration.top_p #=> Float
event.trace.pre_processing_trace.model_invocation_input.override_lambda #=> String
event.trace.pre_processing_trace.model_invocation_input.parser_mode #=> String, one of "DEFAULT", "OVERRIDDEN"
event.trace.pre_processing_trace.model_invocation_input.prompt_creation_mode #=> String, one of "DEFAULT", "OVERRIDDEN"
event.trace.pre_processing_trace.model_invocation_input.text #=> String
event.trace.pre_processing_trace.model_invocation_input.trace_id #=> String
event.trace.pre_processing_trace.model_invocation_input.type #=> String, one of "PRE_PROCESSING", "ORCHESTRATION", "KNOWLEDGE_BASE_RESPONSE_GENERATION", "POST_PROCESSING"
event.trace.pre_processing_trace.model_invocation_output.parsed_response.is_valid #=> Boolean
event.trace.pre_processing_trace.model_invocation_output.parsed_response.rationale #=> String
event.trace.pre_processing_trace.model_invocation_output.trace_id #=> String

For :validation_exception event available at #on_validation_exception_event callback and response eventstream enumerator:
event.message #=> String

resp.content_type #=> String
resp.session_id #=> String

Parameters:

  • params (Hash) (defaults to: {}) —

    ({})

Options Hash (params):

  • :agent_alias_id (required, String) —

    The alias of the agent to use.

  • :agent_id (required, String) —

    The unique identifier of the agent to use.

  • :enable_trace (Boolean) —

    Specifies whether to turn on the trace or not to track the agent's reasoning process. For more information, see Trace enablement.

  • :end_session (Boolean) —

    Specifies whether to end the session with the agent or not.

  • :input_text (String) —

    The prompt text to send the agent.

  • :session_id (required, String) —

    The unique identifier of the session. Use the same value across requests to continue the same conversation.

  • :session_state (Types::SessionState) —

    Contains parameters that specify various attributes of the session. For more information, see Control session context.

Yields:

  • (event_stream_handler)

Returns:

See Also:



884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
 
# File 'gems/aws-sdk-bedrockagentruntime/lib/aws-sdk-bedrockagentruntime/client.rb', line 884

def invoke_agent(params = {}, options = {}, &block)
  params = params.dup
  event_stream_handler = case handler = params.delete(:event_stream_handler)
    when EventStreams::ResponseStream then handler
    when Proc then EventStreams::ResponseStream.new.tap(&handler)
    when nil then EventStreams::ResponseStream.new
    else
      msg = "expected :event_stream_handler to be a block or "\
            "instance of Aws::BedrockAgentRuntime::EventStreams::ResponseStream"\
            ", got `#{handler.inspect}` instead"
      raise ArgumentError, msg
    end

  yield(event_stream_handler) if block_given?

  req = build_request(:invoke_agent, params)

  req.context[:event_stream_handler] = event_stream_handler
  req.handlers.add(Aws::Binary::DecodeHandler, priority: 95)

  req.send_request(options, &block)
end

#retrieve(params = {}) ⇒ Types::RetrieveResponse

Queries a knowledge base and retrieves information from it.

The returned response is a pageable response and is Enumerable. For details on usage see PageableResponse.

Examples:

Request syntax with placeholder values


resp = client.retrieve({
  knowledge_base_id: "KnowledgeBaseId", # required
  next_token: "NextToken",
  retrieval_configuration: {
    vector_search_configuration: { # required
      filter: {
        and_all: [
          {
            # recursive RetrievalFilter
          },
        ],
        equals: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        greater_than: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        greater_than_or_equals: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        in: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        less_than: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        less_than_or_equals: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        not_equals: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        not_in: {
          key: "FilterKey", # required
          value: { # required
          },
        },
        or_all: [
          {
            # recursive RetrievalFilter
          },
        ],
        starts_with: {
          key: "FilterKey", # required
          value: { # required
          },
        },
      },
      number_of_results: 1,
      override_search_type: "HYBRID", # accepts HYBRID, SEMANTIC
    },
  },
  retrieval_query: { # required
    text: "KnowledgeBaseQueryTextString", # required
  },
})

Response structure


resp.next_token #=> String
resp.retrieval_results #=> Array
resp.retrieval_results[0].content.text #=> String
resp.retrieval_results[0].location.s3_location.uri #=> String
resp.retrieval_results[0].location.type #=> String, one of "S3"
resp.retrieval_results[0].metadata #=> Hash
resp.retrieval_results[0].score #=> Float

Parameters:

  • params (Hash) (defaults to: {}) —

    ({})

Options Hash (params):

  • :knowledge_base_id (required, String) —

    The unique identifier of the knowledge base to query.

  • :next_token (String) —

    If there are more results than can fit in the response, the response returns a nextToken. Use this token in the nextToken field of another request to retrieve the next batch of results.

  • :retrieval_configuration (Types::KnowledgeBaseRetrievalConfiguration) —

    Contains configurations for the knowledge base query and retrieval process. For more information, see Query configurations.

  • :retrieval_query (required, Types::KnowledgeBaseQuery) —

    Contains the query to send the knowledge base.

Returns:

See Also:



1022
1023
1024
1025
 
# File 'gems/aws-sdk-bedrockagentruntime/lib/aws-sdk-bedrockagentruntime/client.rb', line 1022

def retrieve(params = {}, options = {})
  req = build_request(:retrieve, params)
  req.send_request(options)
end

#retrieve_and_generate(params = {}) ⇒ Types::RetrieveAndGenerateResponse

Queries a knowledge base and generates responses based on the retrieved results. The response only cites sources that are relevant to the query.

Examples:

Request syntax with placeholder values


resp = client.retrieve_and_generate({
  input: { # required
    text: "RetrieveAndGenerateInputTextString", # required
  },
  retrieve_and_generate_configuration: {
    external_sources_configuration: {
      generation_configuration: {
        prompt_template: {
          text_prompt_template: "TextPromptTemplate",
        },
      },
      model_arn: "BedrockModelArn", # required
      sources: [ # required
        {
          byte_content: {
            content_type: "ContentType", # required
            data: "data", # required
            identifier: "Identifier", # required
          },
          s3_location: {
            uri: "S3Uri", # required
          },
          source_type: "S3", # required, accepts S3, BYTE_CONTENT
        },
      ],
    },
    knowledge_base_configuration: {
      generation_configuration: {
        prompt_template: {
          text_prompt_template: "TextPromptTemplate",
        },
      },
      knowledge_base_id: "KnowledgeBaseId", # required
      model_arn: "BedrockModelArn", # required
      retrieval_configuration: {
        vector_search_configuration: { # required
          filter: {
            and_all: [
              {
                # recursive RetrievalFilter
              },
            ],
            equals: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            greater_than: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            greater_than_or_equals: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            in: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            less_than: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            less_than_or_equals: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            not_equals: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            not_in: {
              key: "FilterKey", # required
              value: { # required
              },
            },
            or_all: [
              {
                # recursive RetrievalFilter
              },
            ],
            starts_with: {
              key: "FilterKey", # required
              value: { # required
              },
            },
          },
          number_of_results: 1,
          override_search_type: "HYBRID", # accepts HYBRID, SEMANTIC
        },
      },
    },
    type: "KNOWLEDGE_BASE", # required, accepts KNOWLEDGE_BASE, EXTERNAL_SOURCES
  },
  session_configuration: {
    kms_key_arn: "KmsKeyArn", # required
  },
  session_id: "SessionId",
})

Response structure


resp.citations #=> Array
resp.citations[0].generated_response_part.text_response_part.span.end #=> Integer
resp.citations[0].generated_response_part.text_response_part.span.start #=> Integer
resp.citations[0].generated_response_part.text_response_part.text #=> String
resp.citations[0].retrieved_references #=> Array
resp.citations[0].retrieved_references[0].content.text #=> String
resp.citations[0].retrieved_references[0].location.s3_location.uri #=> String
resp.citations[0].retrieved_references[0].location.type #=> String, one of "S3"
resp.citations[0].retrieved_references[0].metadata #=> Hash
resp.output.text #=> String
resp.session_id #=> String

Parameters:

  • params (Hash) (defaults to: {}) —

    ({})

Options Hash (params):

Returns:

See Also:



1181
1182
1183
1184
 
# File 'gems/aws-sdk-bedrockagentruntime/lib/aws-sdk-bedrockagentruntime/client.rb', line 1181

def retrieve_and_generate(params = {}, options = {})
  req = build_request(:retrieve_and_generate, params)
  req.send_request(options)
end