from polyfactory.factories.pydantic_factory import ModelFactory

from app.adapters.tiktok.types import Audience, AudienceFile


class AudienceFactory(ModelFactory[Audience]):
    __model__ = Audience


class AudienceFileFactory(ModelFactory[AudienceFile]):
    __model__ = AudienceFile