airflow.providers.amazon.aws.triggers.bedrock

Module Contents

Classes

BedrockCustomizeModelCompletedTrigger

Trigger when a Bedrock model customization job is complete.

BedrockKnowledgeBaseActiveTrigger

Trigger when a Bedrock Knowledge Base reaches the ACTIVE state.

BedrockProvisionModelThroughputCompletedTrigger

Trigger when a provisioned throughput job is complete.

BedrockIngestionJobTrigger

Trigger when a Bedrock ingestion job reaches the COMPLETE state.

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

Bases: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

Trigger when a Bedrock model customization job is complete.

Parameters
  • job_name (str) – The name of the Bedrock model customization job.

  • waiter_delay (int) – The amount of time in seconds to wait between attempts. (default: 120)

  • waiter_max_attempts (int) – The maximum number of attempts to be made. (default: 75)

  • aws_conn_id (str | None) – The Airflow connection used for AWS credentials.

hook()[source]

Override in subclasses to return the right hook.

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

Bases: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

Trigger when a Bedrock Knowledge Base reaches the ACTIVE state.

Parameters
  • knowledge_base_id (str) – The unique identifier of the knowledge base for which to get information.

  • waiter_delay (int) – The amount of time in seconds to wait between attempts. (default: 5)

  • waiter_max_attempts (int) – The maximum number of attempts to be made. (default: 24)

  • aws_conn_id (str | None) – The Airflow connection used for AWS credentials.

hook()[source]

Override in subclasses to return the right hook.

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

Bases: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

Trigger when a provisioned throughput job is complete.

Parameters
  • provisioned_model_id (str) – The ARN or name of the provisioned throughput.

  • waiter_delay (int) – The amount of time in seconds to wait between attempts. (default: 120)

  • waiter_max_attempts (int) – The maximum number of attempts to be made. (default: 75)

  • aws_conn_id (str | None) – The Airflow connection used for AWS credentials.

hook()[source]

Override in subclasses to return the right 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]

Bases: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

Trigger when a Bedrock ingestion job reaches the COMPLETE state.

Parameters
  • knowledge_base_id (str) – The unique identifier of the knowledge base for which to get information.

  • data_source_id (str) – The unique identifier of the data source in the ingestion job.

  • ingestion_job_id (str) – The unique identifier of the ingestion job.

  • waiter_delay (int) – The amount of time in seconds to wait between attempts. (default: 60)

  • waiter_max_attempts (int) – The maximum number of attempts to be made. (default: 10)

  • aws_conn_id (str | None) – The Airflow connection used for AWS credentials.

hook()[source]

Override in subclasses to return the right hook.

Was this entry helpful?