airflow.providers.amazon.aws.hooks.bedrock

Module Contents

Classes

BedrockHook

Interact with Amazon Bedrock.

BedrockRuntimeHook

Interact with the Amazon Bedrock Runtime.

BedrockAgentHook

Interact with the Amazon Agents for Bedrock API.

BedrockAgentRuntimeHook

Interact with the Amazon Agents for Bedrock API.

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.

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.

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

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

Interact with the Amazon Agents for Bedrock API.

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

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

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

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

Interact with the Amazon Agents for Bedrock API.

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

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

client_type = 'bedrock-agent-runtime'[source]

Was this entry helpful?