Skip to content

AI Mask

AI Mask is a transformation component that uses the Databricks ai_mask() function to invoke generative AI to mask specified entities in text in your input table. This function uses a Databricks chat model serving endpoint made available by Databricks Foundation Model APIs.

The output is a string where the specified information (label) is masked.


Make sure you have read and understand the Requirements set out by Databricks before using this component.


Name = string

A human-readable name for the component.

Column = drop-down

Select the column that holds data you wish to mask.

Mask Labels = column editor

Add mask labels as text strings. Each label represents a type of information to be masked. For example, adding a mask label of name prompts the component to mask any names in a given row.

Include Input Columns = boolean

  • Yes: Includes both your input column and the newly created masked column. This will also include those input columns not selected in Column.
  • No: Only includes the new, masked column.

Snowflake Databricks Amazon Redshift