airflow.providers.amazon.aws.triggers.bedrock
¶
Module Contents¶
Classes¶
Trigger when a Bedrock model customization job is complete. |
|
Trigger when a Bedrock Knowledge Base reaches the ACTIVE state. |
|
Trigger when a provisioned throughput job is complete. |
|
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.
- 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.
- 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.
- 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.