airflow.providers.amazon.aws.hooks.bedrock

Module Contents

Classes

BedrockHook

Interact with Amazon Bedrock.

BedrockRuntimeHook

Interact with the Amazon Bedrock Runtime.
class airflow.providers.amazon.aws.hooks.bedrock.BedrockHook(*args, **kwargs)[source]

Bases: airflow.providers.amazon.aws.hooks.base_aws.AwsBaseHook

Interact with Amazon Bedrock.

Provide thin wrapper around boto3.client("bedrock").

Additional arguments (such as aws_conn_id) may be specified and are passed down to the underlying AwsBaseHook.

See also

client_type = 'bedrock'[source]
class airflow.providers.amazon.aws.hooks.bedrock.BedrockRuntimeHook(*args, **kwargs)[source]

Bases: airflow.providers.amazon.aws.hooks.base_aws.AwsBaseHook

Interact with the Amazon Bedrock Runtime.

Provide thin wrapper around boto3.client("bedrock-runtime").

Additional arguments (such as aws_conn_id) may be specified and are passed down to the underlying AwsBaseHook.

See also

client_type = 'bedrock-runtime'[source]

Was this entry helpful?