aws_sdk_bedrockagentruntime/operation/generate_query/
builders.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
pub use crate::operation::generate_query::_generate_query_output::GenerateQueryOutputBuilder;

pub use crate::operation::generate_query::_generate_query_input::GenerateQueryInputBuilder;

impl crate::operation::generate_query::builders::GenerateQueryInputBuilder {
    /// Sends a request with this input using the given client.
    pub async fn send_with(
        self,
        client: &crate::Client,
    ) -> ::std::result::Result<
        crate::operation::generate_query::GenerateQueryOutput,
        ::aws_smithy_runtime_api::client::result::SdkError<
            crate::operation::generate_query::GenerateQueryError,
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
        >,
    > {
        let mut fluent_builder = client.generate_query();
        fluent_builder.inner = self;
        fluent_builder.send().await
    }
}
/// Fluent builder constructing a request to `GenerateQuery`.
///
/// <p>Generates an SQL query from a natural language query. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/knowledge-base-generate-query.html">Generate a query for structured data</a> in the Amazon Bedrock User Guide.</p>
#[derive(::std::clone::Clone, ::std::fmt::Debug)]
pub struct GenerateQueryFluentBuilder {
    handle: ::std::sync::Arc<crate::client::Handle>,
    inner: crate::operation::generate_query::builders::GenerateQueryInputBuilder,
    config_override: ::std::option::Option<crate::config::Builder>,
}
impl
    crate::client::customize::internal::CustomizableSend<
        crate::operation::generate_query::GenerateQueryOutput,
        crate::operation::generate_query::GenerateQueryError,
    > for GenerateQueryFluentBuilder
{
    fn send(
        self,
        config_override: crate::config::Builder,
    ) -> crate::client::customize::internal::BoxFuture<
        crate::client::customize::internal::SendResult<
            crate::operation::generate_query::GenerateQueryOutput,
            crate::operation::generate_query::GenerateQueryError,
        >,
    > {
        ::std::boxed::Box::pin(async move { self.config_override(config_override).send().await })
    }
}
impl GenerateQueryFluentBuilder {
    /// Creates a new `GenerateQueryFluentBuilder`.
    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 GenerateQuery as a reference.
    pub fn as_input(&self) -> &crate::operation::generate_query::builders::GenerateQueryInputBuilder {
        &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::generate_query::GenerateQueryOutput,
        ::aws_smithy_runtime_api::client::result::SdkError<
            crate::operation::generate_query::GenerateQueryError,
            ::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::generate_query::GenerateQuery::operation_runtime_plugins(
            self.handle.runtime_plugins.clone(),
            &self.handle.conf,
            self.config_override,
        );
        crate::operation::generate_query::GenerateQuery::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::generate_query::GenerateQueryOutput,
        crate::operation::generate_query::GenerateQueryError,
        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>Specifies information about a natural language query to transform into SQL.</p>
    pub fn query_generation_input(mut self, input: crate::types::QueryGenerationInput) -> Self {
        self.inner = self.inner.query_generation_input(input);
        self
    }
    /// <p>Specifies information about a natural language query to transform into SQL.</p>
    pub fn set_query_generation_input(mut self, input: ::std::option::Option<crate::types::QueryGenerationInput>) -> Self {
        self.inner = self.inner.set_query_generation_input(input);
        self
    }
    /// <p>Specifies information about a natural language query to transform into SQL.</p>
    pub fn get_query_generation_input(&self) -> &::std::option::Option<crate::types::QueryGenerationInput> {
        self.inner.get_query_generation_input()
    }
    /// <p>Specifies configurations for transforming the natural language query into SQL.</p>
    pub fn transformation_configuration(mut self, input: crate::types::TransformationConfiguration) -> Self {
        self.inner = self.inner.transformation_configuration(input);
        self
    }
    /// <p>Specifies configurations for transforming the natural language query into SQL.</p>
    pub fn set_transformation_configuration(mut self, input: ::std::option::Option<crate::types::TransformationConfiguration>) -> Self {
        self.inner = self.inner.set_transformation_configuration(input);
        self
    }
    /// <p>Specifies configurations for transforming the natural language query into SQL.</p>
    pub fn get_transformation_configuration(&self) -> &::std::option::Option<crate::types::TransformationConfiguration> {
        self.inner.get_transformation_configuration()
    }
}