# In dbt_import.py:191-197
if entity_column not in column_names:
    click.echo(warning)
    continue
# No type checking!

entity_col = next((c for c in model.columns if c.name == entity_column), None)
if entity_col:
    normalized_type = entity_col.data_type.upper()
    valid_entity_types = ['STRING', 'TEXT', 'VARCHAR', 'INT', 'INT32', 'INT64', 'INTEGER', 'BIGINT', 'UUID']
    
    if not any(t in normalized_type for t in valid_entity_types):
        click.echo(
            f"{Fore.YELLOW}Warning: Entity column '{entity_column}' has type "
            f"'{entity_col.data_type}' which may not be suitable for entity keys."
            f" Recommended types: STRING, INT64{Style.RESET_ALL}"
        )