airflow.providers.amazon.aws.triggers.bedrock¶

类¶

`BedrockCustomizeModelCompletedTrigger`	当 Bedrock 模型定制作业完成时触发。
`BedrockKnowledgeBaseActiveTrigger`	当 Bedrock 知识库达到 ACTIVE 状态时触发。
`BedrockProvisionModelThroughputCompletedTrigger`	当预置吞吐量作业完成时触发。
`BedrockIngestionJobTrigger`	当 Bedrock 数据摄取作业达到 COMPLETE 状态时触发。
`BedrockBaseBatchInferenceTrigger`	当批量推理作业完成时触发。
`BedrockBatchInferenceCompletedTrigger`	当批量推理作业完成时触发。
`BedrockBatchInferenceScheduledTrigger`	当批量推理作业已调度时触发。

模块内容¶

class airflow.providers.amazon.aws.triggers.bedrock.BedrockCustomizeModelCompletedTrigger(*, job_name, waiter_delay=120, waiter_max_attempts=75, aws_conn_id=None)[source]¶

继承自: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

当 Bedrock 模型定制作业完成时触发。

参数：:

job_name (str) – Bedrock 模型定制作业的名称。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 120)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 75)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。

hook()[source]¶

在子类中重写以返回正确的 hook。

class airflow.providers.amazon.aws.triggers.bedrock.BedrockKnowledgeBaseActiveTrigger(*, knowledge_base_id, waiter_delay=5, waiter_max_attempts=24, aws_conn_id=None)[source]¶

继承自: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

当 Bedrock 知识库达到 ACTIVE 状态时触发。

参数：:

knowledge_base_id (str) – 要获取信息的知识库的唯一标识符。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 5)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 24)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。

hook()[source]¶

在子类中重写以返回正确的 hook。

class airflow.providers.amazon.aws.triggers.bedrock.BedrockProvisionModelThroughputCompletedTrigger(*, provisioned_model_id, waiter_delay=120, waiter_max_attempts=75, aws_conn_id=None)[source]¶

继承自: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

当预置吞吐量作业完成时触发。

参数：:

provisioned_model_id (str) – 预置吞吐量的 ARN 或名称。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 120)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 75)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。

hook()[source]¶

在子类中重写以返回正确的 hook。

class airflow.providers.amazon.aws.triggers.bedrock.BedrockIngestionJobTrigger(*, knowledge_base_id, data_source_id, ingestion_job_id, waiter_delay=60, waiter_max_attempts=10, aws_conn_id=None)[source]¶

继承自: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

当 Bedrock 数据摄取作业达到 COMPLETE 状态时触发。

参数：:

knowledge_base_id (str) – 要获取信息的知识库的唯一标识符。
data_source_id (str) – 数据摄取作业中数据源的唯一标识符。
ingestion_job_id (str) – 数据摄取作业的唯一标识符。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 60)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 10)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。

hook()[source]¶

在子类中重写以返回正确的 hook。

class airflow.providers.amazon.aws.triggers.bedrock.BedrockBaseBatchInferenceTrigger(*, job_arn, waiter_name=NOTSET, waiter_delay=120, waiter_max_attempts=75, aws_conn_id=None)[source]¶

继承自: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

当批量推理作业完成时触发。

参数：:

job_arn (str) – 批量推理作业的 Amazon 资源名称 (ARN)。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 120)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 75)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。

hook()[source]¶

在子类中重写以返回正确的 hook。

class airflow.providers.amazon.aws.triggers.bedrock.BedrockBatchInferenceCompletedTrigger(*, job_arn, waiter_delay=120, waiter_max_attempts=75, aws_conn_id=None)[source]¶

继承自: BedrockBaseBatchInferenceTrigger

当批量推理作业完成时触发。

参数：:

job_arn (str) – 批量推理作业的 Amazon 资源名称 (ARN)。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 120)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 75)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。

class airflow.providers.amazon.aws.triggers.bedrock.BedrockBatchInferenceScheduledTrigger(*, job_arn, waiter_delay=120, waiter_max_attempts=75, aws_conn_id=None)[source]¶

继承自: BedrockBaseBatchInferenceTrigger

当批量推理作业已调度时触发。

参数：:

job_arn (str) – 批量推理作业的 Amazon 资源名称 (ARN)。
waiter_delay (int) – 每次尝试之间等待的时间间隔（秒）。 (默认值: 120)
waiter_max_attempts (int) – 最大尝试次数。 (默认值: 75)
aws_conn_id (str | None) – 用于 AWS 凭据的 Airflow 连接。