airflow.providers.amazon.aws.triggers.bedrock

模块内容

BedrockCustomizeModelCompletedTrigger

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

BedrockKnowledgeBaseActiveTrigger

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

BedrockProvisionModelThroughputCompletedTrigger

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

BedrockIngestionJobTrigger

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

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

基类: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()[源代码]

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

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

基类: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()[源代码]

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

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

基类: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()[源代码]

在子类中重写以返回正确的 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)[源代码]

基类: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()[源代码]

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

此条目是否有帮助?