"""企查查 DTO。"""

from pydantic import BaseModel, Field


class QichachaCompanyQueryDTO(BaseModel):
    """企业查询请求。"""

    keyword: str = Field(..., min_length=2, max_length=200, description="企业名称或统一社会信用代码")
    forceRefresh: bool = Field(False, description="是否强制刷新缓存")


class QichachaBatchQueryDTO(BaseModel):
    """批量企业查询请求。"""

    keywords: list[str] = Field(..., min_length=1, max_length=10, description="企业名称或统一社会信用代码列表")
    forceRefresh: bool = Field(False, description="是否强制刷新缓存")