Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 57 additions & 0 deletions providers/google/docs/operators/cloud/vertex_ai.rst
Original file line number Diff line number Diff line change
Expand Up @@ -664,6 +664,36 @@ The operator returns the cached content response in :ref:`XCom <concepts:xcom>`
Interacting with Vertex AI Feature Store
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

To create feature online store you can use
:class:`~airflow.providers.google.cloud.operators.vertex_ai.feature_store.CreateFeatureOnlineStoreOperator`.
The operator creation results in :ref:`XCom <concepts:xcom>` under ``return_value`` key.

.. exampleinclude:: /../../google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py
:language: python
:dedent: 4
:start-after: [START how_to_cloud_vertex_ai_create_feature_online_store_operator]
:end-before: [END how_to_cloud_vertex_ai_create_feature_online_store_operator]

To create feature store view you can use
:class:`~airflow.providers.google.cloud.operators.vertex_ai.feature_store.CreateFeatureViewOperator`.
The operator creation results in :ref:`XCom <concepts:xcom>` under ``return_value`` key.

.. exampleinclude:: /../../google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py
:language: python
:dedent: 4
:start-after: [START how_to_cloud_vertex_ai_create_feature_view_store_operator]
:end-before: [END how_to_cloud_vertex_ai_create_feature_view_store_operator]

To feature online store you can use
:class:`~airflow.providers.google.cloud.operators.vertex_ai.feature_store.GetFeatureOnlineStoreOperator`.
The operator creation results in :ref:`XCom <concepts:xcom>` under ``return_value`` key.

.. exampleinclude:: /../../google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py
:language: python
:dedent: 4
:start-after: [START how_to_cloud_vertex_ai_get_feature_online_store_operator]
:end-before: [END how_to_cloud_vertex_ai_get_feature_online_store_operator]

To get a feature view sync job you can use
:class:`~airflow.providers.google.cloud.operators.vertex_ai.feature_store.GetFeatureViewSyncOperator`.
The operator returns sync job results in :ref:`XCom <concepts:xcom>` under ``return_value`` key.
Expand Down Expand Up @@ -693,6 +723,33 @@ To check if Feature View Sync succeeded you can use
:start-after: [START how_to_cloud_vertex_ai_feature_store_feature_view_sync_sensor]
:end-before: [END how_to_cloud_vertex_ai_feature_store_feature_view_sync_sensor]

To check feature values data you can use the
:class:`~airflow.providers.google.cloud.sensors.vertex_ai.FetchFeatureValuesOperator`.

.. exampleinclude:: /../../google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py
:language: python
:dedent: 4
:start-after: [START how_to_cloud_vertex_ai_fetch_feature_values_operator]
:end-before: [END how_to_cloud_vertex_ai_fetch_feature_values_operator]

To delete the feature view you can use
:class:`~airflow.providers.google.cloud.sensors.vertex_ai.DeleteFeatureViewOperator`.

.. exampleinclude:: /../../google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py
:language: python
:dedent: 4
:start-after: [START how_to_cloud_vertex_ai_delete_feature_view_operator]
:end-before: [END how_to_cloud_vertex_ai_delete_feature_view_operator]

To delete the feature online store you can use
:class:`~airflow.providers.google.cloud.sensors.vertex_ai.DeleteFeatureOnlineStoreOperator`.

.. exampleinclude:: /../../google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py
:language: python
:dedent: 4
:start-after: [START how_to_cloud_vertex_ai_delete_feature_online_store_operator]
:end-before: [END how_to_cloud_vertex_ai_delete_feature_online_store_operator]

Interacting with Ray on Vertex AI Cluster
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

Expand Down
Loading