Class DbtToFeastMapper

sdk/python/feast/dbt/mapper.py:170–480

Maps dbt models to Feast objects. Supports creating DataSource, Entity, and FeatureView objects from dbt model metadata. Example:: mapper = DbtToFeastMapper(data_source_type="bigquery") data_source = mapper.create_data_source(model) feature_view = mapper.c

class DbtToFeastMapper:
    """
    Maps dbt models to Feast objects.

    Supports creating DataSource, Entity, and FeatureView objects from
    dbt model metadata.

    Example::

        mapper = DbtToFeastMapper(data_source_type="bigquery")
        data_source = mapper.create_data_source(model)
        feature_view = mapper.create_feature_view(
            model, data_source, entity_column="driver_id"
        )

    Args:
        data_source_type: Type of data source ('bigquery', 'snowflake', 'file')
        timestamp_field: Default timestamp field name
        ttl_days: Default TTL in days for feature views
    """

    def __init__(
        self,
        data_source_type: str = "bigquery",
        timestamp_field: str = "event_timestamp",
        ttl_days: int = 1,
    ):
        self.data_source_type = data_source_type.lower()
        self.timestamp_field = timestamp_field
        self.ttl_days = ttl_days

    def _infer_entity_value_type(self, model: DbtModel, entity_col: str) -> ValueType:
        """Infer entity ValueType from dbt model column type."""
        for column in model.columns:
            if column.name == entity_col:
                feast_type = map_dbt_type_to_feast_type(column.data_type)
                return feast_type_to_value_type(feast_type)
        return ValueType.UNKNOWN

    def create_data_source(
        self,
        model: DbtModel,
        timestamp_field: Optional[str] = None,
        created_timestamp_column: Optional[str] = None,
    ) -> Any:
        """
        Create a Feast DataSource from a dbt model.

        Args:
            model: The DbtModel to create a DataSource from
            timestamp_field: Override the default timestamp field
            created_timestamp_column: Column for created timestamp (dedup)

        Returns:
            A Feast DataSource (BigQuerySource, SnowflakeSource, or FileSource)

        Raises:
            ValueError: If data_source_type is not supported