ferruzzi commented on code in PR #38693: URL: https://github.com/apache/airflow/pull/38693#discussion_r1552442572
########## docs/apache-airflow-providers-amazon/operators/bedrock.rst: ########## @@ -65,6 +65,44 @@ To invoke an Amazon Titan model you would use: For details on the different formats, see `Inference parameters for foundation models <https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html>`__ +.. _howto/operator:BedrockCustomizeModelOperator: + +Customize an existing Amazon Bedrock Model +========================================== + +To create a fine-tuning job to customize a base model, you can use +:class:`~airflow.providers.amazon.aws.operators.bedrock.BedrockCustomizeModelOperator`. + +Model-customization jobs are asynchronous and the completion time depends on the base model +and the training/validation data size. To monitor the state of the job, you can use the +"model_customization_job_complete" Waiter, the +:class:`~airflow.providers.amazon.aws.sensors.bedrock.BedrockCustomizeModelCompletedSensor` Sensor, +or the :class:`~airflow.providers.amazon.aws.triggersBedrockCustomizeModelCompletedTrigger` Trigger. Review Comment: missed a period in the class path, surprised the build-docs test didn't catch that. Will fix. ########## docs/apache-airflow-providers-amazon/operators/bedrock.rst: ########## @@ -65,6 +65,44 @@ To invoke an Amazon Titan model you would use: For details on the different formats, see `Inference parameters for foundation models <https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html>`__ +.. _howto/operator:BedrockCustomizeModelOperator: + +Customize an existing Amazon Bedrock Model +========================================== + +To create a fine-tuning job to customize a base model, you can use +:class:`~airflow.providers.amazon.aws.operators.bedrock.BedrockCustomizeModelOperator`. + +Model-customization jobs are asynchronous and the completion time depends on the base model +and the training/validation data size. To monitor the state of the job, you can use the +"model_customization_job_complete" Waiter, the +:class:`~airflow.providers.amazon.aws.sensors.bedrock.BedrockCustomizeModelCompletedSensor` Sensor, +or the :class:`~airflow.providers.amazon.aws.triggersBedrockCustomizeModelCompletedTrigger` Trigger. + + +.. exampleinclude:: /../../tests/system/providers/amazon/aws/example_bedrock.py + :language: python + :dedent: 4 + :start-after: [START howto_operator_customize_model] + :end-before: [END howto_operator_customize_model] + + +Sensors +------- + +.. _howto/sensor:BedrockCustomizeModelCompletedSensor: + +Wait for an Amazon Bedrock customize model job +============================================== + +To wait on the state of an AWS CloudFormation stack creation until it reaches a terminal state you can use Review Comment: copypasta :( Will fix -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
