from snowflake.snowpark.session import Session
from snowflake.snowpark.types import BooleanType, StringType
from user_agents import parse
def create_is_user_agent_mobile_function(session: Session):
    def is_user_agent_mobile(user_agent_value: str) -> bool:
        return parse(user_agent_value).is_mobile

    session.udf.register(
        func=is_user_agent_mobile,
        return_type=BooleanType(),
        input_types=[StringType()],
        name="is_user_agent_mobile",
        is_permanent=True,
        stage_location="@SUNO_CODE",
        packages=["user-agents"],
        replace=True,
    )