aws-sdk-bedrockagentcore 1.55.0

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
pub use crate::operation::invoke_harness::_invoke_harness_input::InvokeHarnessInputBuilder;

pub use crate::operation::invoke_harness::_invoke_harness_output::InvokeHarnessOutputBuilder;

impl crate::operation::invoke_harness::builders::InvokeHarnessInputBuilder {
    /// Sends a request with this input using the given client.
    pub async fn send_with(
        self,
        client: &crate::Client,
    ) -> ::std::result::Result<
        crate::operation::invoke_harness::InvokeHarnessOutput,
        ::aws_smithy_runtime_api::client::result::SdkError<
            crate::operation::invoke_harness::InvokeHarnessError,
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
        >,
    > {
        let mut fluent_builder = client.invoke_harness();
        fluent_builder.inner = self;
        fluent_builder.send().await
    }
}
/// Fluent builder constructing a request to `InvokeHarness`.
///
/// <p>Operation to invoke a Harness.</p>
#[derive(::std::clone::Clone, ::std::fmt::Debug)]
pub struct InvokeHarnessFluentBuilder {
    handle: ::std::sync::Arc<crate::client::Handle>,
    inner: crate::operation::invoke_harness::builders::InvokeHarnessInputBuilder,
    config_override: ::std::option::Option<crate::config::Builder>,
}
impl
    crate::client::customize::internal::CustomizableSend<
        crate::operation::invoke_harness::InvokeHarnessOutput,
        crate::operation::invoke_harness::InvokeHarnessError,
    > for InvokeHarnessFluentBuilder
{
    fn send(
        self,
        config_override: crate::config::Builder,
    ) -> crate::client::customize::internal::BoxFuture<
        crate::client::customize::internal::SendResult<
            crate::operation::invoke_harness::InvokeHarnessOutput,
            crate::operation::invoke_harness::InvokeHarnessError,
        >,
    > {
        ::std::boxed::Box::pin(async move { self.config_override(config_override).send().await })
    }
}
impl InvokeHarnessFluentBuilder {
    /// Creates a new `InvokeHarnessFluentBuilder`.
    pub(crate) fn new(handle: ::std::sync::Arc<crate::client::Handle>) -> Self {
        Self {
            handle,
            inner: ::std::default::Default::default(),
            config_override: ::std::option::Option::None,
        }
    }
    /// Access the InvokeHarness as a reference.
    pub fn as_input(&self) -> &crate::operation::invoke_harness::builders::InvokeHarnessInputBuilder {
        &self.inner
    }
    /// Sends the request and returns the response.
    ///
    /// If an error occurs, an `SdkError` will be returned with additional details that
    /// can be matched against.
    ///
    /// By default, any retryable failures will be retried twice. Retry behavior
    /// is configurable with the [RetryConfig](aws_smithy_types::retry::RetryConfig), which can be
    /// set when configuring the client.
    pub async fn send(
        self,
    ) -> ::std::result::Result<
        crate::operation::invoke_harness::InvokeHarnessOutput,
        ::aws_smithy_runtime_api::client::result::SdkError<
            crate::operation::invoke_harness::InvokeHarnessError,
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
        >,
    > {
        let input = self
            .inner
            .build()
            .map_err(::aws_smithy_runtime_api::client::result::SdkError::construction_failure)?;
        let runtime_plugins = crate::operation::invoke_harness::InvokeHarness::operation_runtime_plugins(
            self.handle.runtime_plugins.clone(),
            &self.handle.conf,
            self.config_override,
        );
        crate::operation::invoke_harness::InvokeHarness::orchestrate(&runtime_plugins, input).await
    }

    /// Consumes this builder, creating a customizable operation that can be modified before being sent.
    pub fn customize(
        self,
    ) -> crate::client::customize::CustomizableOperation<
        crate::operation::invoke_harness::InvokeHarnessOutput,
        crate::operation::invoke_harness::InvokeHarnessError,
        Self,
    > {
        crate::client::customize::CustomizableOperation::new(self)
    }
    pub(crate) fn config_override(mut self, config_override: impl ::std::convert::Into<crate::config::Builder>) -> Self {
        self.set_config_override(::std::option::Option::Some(config_override.into()));
        self
    }

    pub(crate) fn set_config_override(&mut self, config_override: ::std::option::Option<crate::config::Builder>) -> &mut Self {
        self.config_override = config_override;
        self
    }
    /// <p>The ARN of the harness to invoke.</p>
    pub fn harness_arn(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.harness_arn(input.into());
        self
    }
    /// <p>The ARN of the harness to invoke.</p>
    pub fn set_harness_arn(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_harness_arn(input);
        self
    }
    /// <p>The ARN of the harness to invoke.</p>
    pub fn get_harness_arn(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_harness_arn()
    }
    /// <p>The endpoint name to invoke. If omitted, the DEFAULT endpoint is used.</p>
    pub fn qualifier(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.qualifier(input.into());
        self
    }
    /// <p>The endpoint name to invoke. If omitted, the DEFAULT endpoint is used.</p>
    pub fn set_qualifier(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_qualifier(input);
        self
    }
    /// <p>The endpoint name to invoke. If omitted, the DEFAULT endpoint is used.</p>
    pub fn get_qualifier(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_qualifier()
    }
    /// <p>The session ID for the invocation. Use the same session ID across requests to continue a conversation.</p>
    pub fn runtime_session_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.runtime_session_id(input.into());
        self
    }
    /// <p>The session ID for the invocation. Use the same session ID across requests to continue a conversation.</p>
    pub fn set_runtime_session_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_runtime_session_id(input);
        self
    }
    /// <p>The session ID for the invocation. Use the same session ID across requests to continue a conversation.</p>
    pub fn get_runtime_session_id(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_runtime_session_id()
    }
    /// <p>An identifier for the end user making the request. This value is passed through to the runtime container.</p>
    pub fn runtime_user_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.runtime_user_id(input.into());
        self
    }
    /// <p>An identifier for the end user making the request. This value is passed through to the runtime container.</p>
    pub fn set_runtime_user_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_runtime_user_id(input);
        self
    }
    /// <p>An identifier for the end user making the request. This value is passed through to the runtime container.</p>
    pub fn get_runtime_user_id(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_runtime_user_id()
    }
    /// <p>W3C trace context parent header containing version, trace ID, parent span ID, and trace flags.</p>
    pub fn trace_parent(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.trace_parent(input.into());
        self
    }
    /// <p>W3C trace context parent header containing version, trace ID, parent span ID, and trace flags.</p>
    pub fn set_trace_parent(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_trace_parent(input);
        self
    }
    /// <p>W3C trace context parent header containing version, trace ID, parent span ID, and trace flags.</p>
    pub fn get_trace_parent(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_trace_parent()
    }
    /// <p>W3C trace context state header for vendor-specific trace information.</p>
    pub fn trace_state(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.trace_state(input.into());
        self
    }
    /// <p>W3C trace context state header for vendor-specific trace information.</p>
    pub fn set_trace_state(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_trace_state(input);
        self
    }
    /// <p>W3C trace context state header for vendor-specific trace information.</p>
    pub fn get_trace_state(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_trace_state()
    }
    /// <p>Trace ID for maintaining observability through the operation.</p>
    pub fn trace_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.trace_id(input.into());
        self
    }
    /// <p>Trace ID for maintaining observability through the operation.</p>
    pub fn set_trace_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_trace_id(input);
        self
    }
    /// <p>Trace ID for maintaining observability through the operation.</p>
    pub fn get_trace_id(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_trace_id()
    }
    /// <p>W3C Baggage header for user-defined context propagation. Format: key1=value1,key2=value2</p>
    pub fn baggage(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.baggage(input.into());
        self
    }
    /// <p>W3C Baggage header for user-defined context propagation. Format: key1=value1,key2=value2</p>
    pub fn set_baggage(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_baggage(input);
        self
    }
    /// <p>W3C Baggage header for user-defined context propagation. Format: key1=value1,key2=value2</p>
    pub fn get_baggage(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_baggage()
    }
    ///
    /// Appends an item to `messages`.
    ///
    /// To override the contents of this collection use [`set_messages`](Self::set_messages).
    ///
    /// <p>The messages to send to the agent.</p>
    pub fn messages(mut self, input: crate::types::HarnessMessage) -> Self {
        self.inner = self.inner.messages(input);
        self
    }
    /// <p>The messages to send to the agent.</p>
    pub fn set_messages(mut self, input: ::std::option::Option<::std::vec::Vec<crate::types::HarnessMessage>>) -> Self {
        self.inner = self.inner.set_messages(input);
        self
    }
    /// <p>The messages to send to the agent.</p>
    pub fn get_messages(&self) -> &::std::option::Option<::std::vec::Vec<crate::types::HarnessMessage>> {
        self.inner.get_messages()
    }
    /// <p>The model configuration to use for this invocation. If specified, overrides the harness default.</p>
    pub fn model(mut self, input: crate::types::HarnessModelConfiguration) -> Self {
        self.inner = self.inner.model(input);
        self
    }
    /// <p>The model configuration to use for this invocation. If specified, overrides the harness default.</p>
    pub fn set_model(mut self, input: ::std::option::Option<crate::types::HarnessModelConfiguration>) -> Self {
        self.inner = self.inner.set_model(input);
        self
    }
    /// <p>The model configuration to use for this invocation. If specified, overrides the harness default.</p>
    pub fn get_model(&self) -> &::std::option::Option<crate::types::HarnessModelConfiguration> {
        self.inner.get_model()
    }
    ///
    /// Appends an item to `systemPrompt`.
    ///
    /// To override the contents of this collection use [`set_system_prompt`](Self::set_system_prompt).
    ///
    /// <p>The system prompt to use for this invocation. If specified, overrides the harness default.</p>
    pub fn system_prompt(mut self, input: crate::types::HarnessSystemContentBlock) -> Self {
        self.inner = self.inner.system_prompt(input);
        self
    }
    /// <p>The system prompt to use for this invocation. If specified, overrides the harness default.</p>
    pub fn set_system_prompt(mut self, input: ::std::option::Option<::std::vec::Vec<crate::types::HarnessSystemContentBlock>>) -> Self {
        self.inner = self.inner.set_system_prompt(input);
        self
    }
    /// <p>The system prompt to use for this invocation. If specified, overrides the harness default.</p>
    pub fn get_system_prompt(&self) -> &::std::option::Option<::std::vec::Vec<crate::types::HarnessSystemContentBlock>> {
        self.inner.get_system_prompt()
    }
    ///
    /// Appends an item to `tools`.
    ///
    /// To override the contents of this collection use [`set_tools`](Self::set_tools).
    ///
    /// <p>The tools available to the agent for this invocation. If specified, overrides the harness default.</p>
    pub fn tools(mut self, input: crate::types::HarnessTool) -> Self {
        self.inner = self.inner.tools(input);
        self
    }
    /// <p>The tools available to the agent for this invocation. If specified, overrides the harness default.</p>
    pub fn set_tools(mut self, input: ::std::option::Option<::std::vec::Vec<crate::types::HarnessTool>>) -> Self {
        self.inner = self.inner.set_tools(input);
        self
    }
    /// <p>The tools available to the agent for this invocation. If specified, overrides the harness default.</p>
    pub fn get_tools(&self) -> &::std::option::Option<::std::vec::Vec<crate::types::HarnessTool>> {
        self.inner.get_tools()
    }
    ///
    /// Appends an item to `skills`.
    ///
    /// To override the contents of this collection use [`set_skills`](Self::set_skills).
    ///
    /// <p>The skills available to the agent for this invocation. If specified, overrides the harness default.</p>
    pub fn skills(mut self, input: crate::types::HarnessSkill) -> Self {
        self.inner = self.inner.skills(input);
        self
    }
    /// <p>The skills available to the agent for this invocation. If specified, overrides the harness default.</p>
    pub fn set_skills(mut self, input: ::std::option::Option<::std::vec::Vec<crate::types::HarnessSkill>>) -> Self {
        self.inner = self.inner.set_skills(input);
        self
    }
    /// <p>The skills available to the agent for this invocation. If specified, overrides the harness default.</p>
    pub fn get_skills(&self) -> &::std::option::Option<::std::vec::Vec<crate::types::HarnessSkill>> {
        self.inner.get_skills()
    }
    ///
    /// Appends an item to `allowedTools`.
    ///
    /// To override the contents of this collection use [`set_allowed_tools`](Self::set_allowed_tools).
    ///
    /// <p>The tools that the agent is allowed to use for this invocation. If specified, overrides the harness default.</p>
    pub fn allowed_tools(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.allowed_tools(input.into());
        self
    }
    /// <p>The tools that the agent is allowed to use for this invocation. If specified, overrides the harness default.</p>
    pub fn set_allowed_tools(mut self, input: ::std::option::Option<::std::vec::Vec<::std::string::String>>) -> Self {
        self.inner = self.inner.set_allowed_tools(input);
        self
    }
    /// <p>The tools that the agent is allowed to use for this invocation. If specified, overrides the harness default.</p>
    pub fn get_allowed_tools(&self) -> &::std::option::Option<::std::vec::Vec<::std::string::String>> {
        self.inner.get_allowed_tools()
    }
    /// <p>The maximum number of iterations the agent loop can execute. If specified, overrides the harness default.</p>
    pub fn max_iterations(mut self, input: i32) -> Self {
        self.inner = self.inner.max_iterations(input);
        self
    }
    /// <p>The maximum number of iterations the agent loop can execute. If specified, overrides the harness default.</p>
    pub fn set_max_iterations(mut self, input: ::std::option::Option<i32>) -> Self {
        self.inner = self.inner.set_max_iterations(input);
        self
    }
    /// <p>The maximum number of iterations the agent loop can execute. If specified, overrides the harness default.</p>
    pub fn get_max_iterations(&self) -> &::std::option::Option<i32> {
        self.inner.get_max_iterations()
    }
    /// <p>The maximum number of tokens the agent can generate per iteration. If specified, overrides the harness default.</p>
    pub fn max_tokens(mut self, input: i32) -> Self {
        self.inner = self.inner.max_tokens(input);
        self
    }
    /// <p>The maximum number of tokens the agent can generate per iteration. If specified, overrides the harness default.</p>
    pub fn set_max_tokens(mut self, input: ::std::option::Option<i32>) -> Self {
        self.inner = self.inner.set_max_tokens(input);
        self
    }
    /// <p>The maximum number of tokens the agent can generate per iteration. If specified, overrides the harness default.</p>
    pub fn get_max_tokens(&self) -> &::std::option::Option<i32> {
        self.inner.get_max_tokens()
    }
    /// <p>The maximum duration in seconds for the agent loop execution. If specified, overrides the harness default.</p>
    pub fn timeout_seconds(mut self, input: i32) -> Self {
        self.inner = self.inner.timeout_seconds(input);
        self
    }
    /// <p>The maximum duration in seconds for the agent loop execution. If specified, overrides the harness default.</p>
    pub fn set_timeout_seconds(mut self, input: ::std::option::Option<i32>) -> Self {
        self.inner = self.inner.set_timeout_seconds(input);
        self
    }
    /// <p>The maximum duration in seconds for the agent loop execution. If specified, overrides the harness default.</p>
    pub fn get_timeout_seconds(&self) -> &::std::option::Option<i32> {
        self.inner.get_timeout_seconds()
    }
    /// <p>The actor ID for memory operations. Overrides the actor ID configured on the harness.</p>
    pub fn actor_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.actor_id(input.into());
        self
    }
    /// <p>The actor ID for memory operations. Overrides the actor ID configured on the harness.</p>
    pub fn set_actor_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_actor_id(input);
        self
    }
    /// <p>The actor ID for memory operations. Overrides the actor ID configured on the harness.</p>
    pub fn get_actor_id(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_actor_id()
    }
}