• 主页
• airflow.providers.amazon
• airflow.providers.amazon.aws
• airflow.providers.amazon.aws.sensors
• airflow.providers.amazon.aws.sensors.bedrock

airflow.providers.amazon.aws.sensors.bedrock

模块内容

类

BedrockBaseSensor	Amazon Bedrock 的通用传感器行为。
BedrockCustomizeModelCompletedSensor	轮询模型自定义作业的状态，直到它达到最终状态；如果作业失败，则失败。
BedrockProvisionModelThroughputCompletedSensor	轮询预置模型吞吐量作业的状态，直到它达到最终状态；如果作业失败，则失败。
BedrockKnowledgeBaseActiveSensor	轮询知识库状态，直到它达到最终状态；如果创建失败，则失败。
BedrockIngestionJobSensor	轮询摄取作业的状态，直到它达到最终状态；如果创建失败，则失败。

class airflow.providers.amazon.aws.sensors.bedrock.BedrockBaseSensor(deferrable=conf.getboolean('operators', 'default_deferrable', fallback=False), **kwargs)

基类：airflow.providers.amazon.aws.sensors.base_aws.AwsBaseSensor[_GenericBedrockHook]

Amazon Bedrock 的通用传感器行为。

子类必须实现以下方法

• get_state()

子类必须设置以下字段

• INTERMEDIATE_STATES

• FAILURE_STATES

• SUCCESS_STATES

• FAILURE_MESSAGE

参数

deferrable (bool) – 如果为 True，传感器将以可延期模式运行。此模式需要安装 aiobotocore 模块。（默认值：False，但可以在配置文件中通过将 default_deferrable 设置为 True 来覆盖）

INTERMEDIATE_STATES: tuple[str, Ellipsis] = ()

FAILURE_STATES: tuple[str, Ellipsis] = ()

SUCCESS_STATES: tuple[str, Ellipsis] = ()

FAILURE_MESSAGE = ''

aws_hook_class: type[_GenericBedrockHook]

ui_color = '#66c3ff'

poke(context, **kwargs)

在派生此类时覆盖。

abstract get_state()

在子类中实现。

class airflow.providers.amazon.aws.sensors.bedrock.BedrockCustomizeModelCompletedSensor(*, job_name, max_retries=75, poke_interval=120, **kwargs)

基类：BedrockBaseSensor[airflow.providers.amazon.aws.hooks.bedrock.BedrockHook]

轮询模型自定义作业的状态，直到它达到最终状态；如果作业失败，则失败。

另请参阅

有关如何使用此传感器的更多信息，请查看指南： 等待 Amazon Bedrock 自定义模型作业

参数

• job_name (str) – Bedrock 模型自定义作业的名称。

• deferrable – 如果为 True，传感器将以可延期模式运行。此模式需要安装 aiobotocore 模块。（默认值：False，但可以在配置文件中通过将 default_deferrable 设置为 True 来覆盖）

• poke_interval (int) – 轮询作业状态的轮询周期（以秒为单位）。（默认值：120）

• max_retries (int) – 返回当前状态之前的重试次数。（默认值：75）

• aws_conn_id – 用于 AWS 凭证的 Airflow 连接。如果此值为 None 或为空，则使用默认的 boto3 行为。如果在分布式方式下运行 Airflow 并且 aws_conn_id 为 None 或为空，则将使用默认的 boto3 配置（并且必须在每个工作节点上维护）。

• region_name – AWS region_name。如果未指定，则使用默认的 boto3 行为。

• verify – 是否验证 SSL 证书。请参阅：https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html

• botocore_config – botocore 客户端的配置字典（键值对）。请参阅：https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html

INTERMEDIATE_STATES: tuple[str, Ellipsis] = ('InProgress',)

FAILURE_STATES: tuple[str, Ellipsis] = ('Failed', 'Stopping', 'Stopped')

SUCCESS_STATES: tuple[str, Ellipsis] = ('Completed',)

FAILURE_MESSAGE = 'Bedrock 模型自定义作业传感器失败。'

aws_hook_class

template_fields: collections.abc.Sequence[str]

execute(context)

在创建操作符时派生。

上下文与渲染 Jinja 模板时使用的字典相同。

有关更多上下文，请参考 get_template_context。

get_state()

在子类中实现。

class airflow.providers.amazon.aws.sensors.bedrock.BedrockProvisionModelThroughputCompletedSensor(*, model_id, poke_interval=60, max_retries=20, **kwargs)

基类：BedrockBaseSensor[airflow.providers.amazon.aws.hooks.bedrock.BedrockHook]

轮询预置模型吞吐量作业的状态，直到它达到最终状态；如果作业失败，则失败。

另请参阅

有关如何使用此传感器的更多信息，请查看指南：等待 Amazon Bedrock 预配置模型吞吐量作业

参数

• model_id (str) – 预配置吞吐量的 ARN 或名称。

• deferrable – 如果为 True，则传感器将以可延期模式运行。此模式需要安装 aiobotocore 模块。（默认值：False，但可以通过将 default_deferrable 设置为 True 在配置文件中覆盖）

• poke_interval (int) – 以秒为单位的轮询周期，用于检查作业的状态。（默认值：60）

• max_retries (int) – 返回当前状态之前的重试次数（默认值：20）

• aws_conn_id – 用于 AWS 凭证的 Airflow 连接。如果此值为 None 或为空，则使用默认的 boto3 行为。如果在分布式方式下运行 Airflow 并且 aws_conn_id 为 None 或为空，则将使用默认的 boto3 配置（并且必须在每个工作节点上维护）。

• region_name – AWS region_name。如果未指定，则使用默认的 boto3 行为。

• verify – 是否验证 SSL 证书。请参阅：https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html

• botocore_config – botocore 客户端的配置字典（键值对）。请参阅：https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html

INTERMEDIATE_STATES: tuple[str, Ellipsis] = ('Creating', 'Updating')

FAILURE_STATES: tuple[str, Ellipsis] = ('Failed',)

SUCCESS_STATES: tuple[str, Ellipsis] = ('InService',)

FAILURE_MESSAGE = 'Bedrock 预配置模型吞吐量传感器失败。'

aws_hook_class

template_fields: collections.abc.Sequence[str]

get_state()

在子类中实现。

execute(context)

在创建操作符时派生。

上下文与渲染 Jinja 模板时使用的字典相同。

有关更多上下文，请参考 get_template_context。

class airflow.providers.amazon.aws.sensors.bedrock.BedrockKnowledgeBaseActiveSensor(*, knowledge_base_id, poke_interval=5, max_retries=24, **kwargs)

基类: BedrockBaseSensor[airflow.providers.amazon.aws.hooks.bedrock.BedrockAgentHook]

轮询知识库状态，直到它达到最终状态；如果创建失败，则失败。

另请参阅

有关如何使用此传感器的更多信息，请查看指南： 等待 Amazon Bedrock 知识库

参数

• knowledge_base_id (str) – 要获取信息的知识库的唯一标识符。(已模板化)

• deferrable – 如果为 True，则传感器将以可延期模式运行。此模式需要安装 aiobotocore 模块。（默认值：False，但可以通过将 default_deferrable 设置为 True 在配置文件中覆盖）

• poke_interval (int) – 轮询间隔，以秒为单位，检查作业状态。（默认值：5）

• max_retries (int) – 返回当前状态之前的重试次数（默认值：24）

• aws_conn_id – 用于 AWS 凭证的 Airflow 连接。如果此值为 None 或为空，则使用默认的 boto3 行为。如果在分布式方式下运行 Airflow 并且 aws_conn_id 为 None 或为空，则将使用默认的 boto3 配置（并且必须在每个工作节点上维护）。

• region_name – AWS region_name。如果未指定，则使用默认的 boto3 行为。

• verify – 是否验证 SSL 证书。请参阅：https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html

• botocore_config – botocore 客户端的配置字典（键值对）。请参阅：https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html

INTERMEDIATE_STATES: tuple[str, Ellipsis] = ('CREATING', 'UPDATING')

FAILURE_STATES: tuple[str, Ellipsis] = ('DELETING', 'FAILED')

SUCCESS_STATES: tuple[str, Ellipsis] = ('ACTIVE',)

FAILURE_MESSAGE = 'Bedrock 知识库 Active 传感器失败。'

aws_hook_class

template_fields: collections.abc.Sequence[str]

get_state()

在子类中实现。

execute(context)

在创建操作符时派生。

上下文与渲染 Jinja 模板时使用的字典相同。

有关更多上下文，请参考 get_template_context。

class airflow.providers.amazon.aws.sensors.bedrock.BedrockIngestionJobSensor(*, knowledge_base_id, data_source_id, ingestion_job_id, poke_interval=60, max_retries=10, **kwargs)

基类: BedrockBaseSensor[airflow.providers.amazon.aws.hooks.bedrock.BedrockAgentHook]

轮询摄取作业的状态，直到它达到最终状态；如果创建失败，则失败。

另请参阅

有关如何使用此传感器的更多信息，请查看指南： 等待 Amazon Bedrock 摄取作业完成

参数

• knowledge_base_id (str) – 要获取信息的知识库的唯一标识符。(已模板化)

• data_source_id (str) – 摄取作业中数据源的唯一标识符。(已模板化)

• ingestion_job_id (str) – 摄取作业的唯一标识符。(已模板化)

• deferrable – 如果为 True，则传感器将以可延期模式运行。此模式需要安装 aiobotocore 模块。（默认值：False，但可以通过将 default_deferrable 设置为 True 在配置文件中覆盖）

• poke_interval (int) – 以秒为单位的轮询周期，用于检查作业的状态。（默认值：60）

• max_retries (int) – 返回当前状态之前的重试次数（默认值：10）

• aws_conn_id – 用于 AWS 凭证的 Airflow 连接。如果此值为 None 或为空，则使用默认的 boto3 行为。如果在分布式方式下运行 Airflow 并且 aws_conn_id 为 None 或为空，则将使用默认的 boto3 配置（并且必须在每个工作节点上维护）。

• region_name – AWS region_name。如果未指定，则使用默认的 boto3 行为。

• verify – 是否验证 SSL 证书。请参阅：https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html

• botocore_config – botocore 客户端的配置字典（键值对）。请参阅：https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html

INTERMEDIATE_STATES: tuple[str, Ellipsis] = ('STARTING', 'IN_PROGRESS')

FAILURE_STATES: tuple[str, Ellipsis] = ('FAILED',)

SUCCESS_STATES: tuple[str, Ellipsis] = ('COMPLETE',)

FAILURE_MESSAGE = 'Bedrock 摄取作业传感器失败。'

aws_hook_class

template_fields: collections.abc.Sequence[str]

get_state()

在子类中实现。

execute(context)

在创建操作符时派生。

上下文与渲染 Jinja 模板时使用的字典相同。

有关更多上下文，请参考 get_template_context。

上一个 下一个

此条目是否有帮助？