"""
Entity definitions for the Suno recommendation system.
"""

from feast import Entity

# User entity
user = Entity(
    name="user",
    join_keys=["USER_ID"],  # Uppercase - Snowflake default
    description="Suno user identified by numeric user_id",
)

# Hook entity
hook = Entity(
    name="hook",
    join_keys=["HOOK_ID"],  # Uppercase - Snowflake default
    description="Video hook identified by hook_id",
)

# Creator entity
creator = Entity(
    name="creator",
    join_keys=["CREATOR_ID"],  # Uppercase - Snowflake default
    description="Content creator user identified by user id",
)

clip = Entity(
    name="clip",
    join_keys=["CLIP_ID"],  # Uppercase - Snowflake default
    description="Clip identified by clip_id",
)