from pydantic import BaseModel

class EnterpriseTaskCreate(BaseModel):
    title: str
    category: str = ""
    sub_category: str = ""
    description: str = ""
    mode: str = "grab"
    budget_min: int = 0
    budget_max: int = 0
    deadline: str = ""
    publisher_name: str = ""



class TaskReviewRequest(BaseModel):
    action: str  # accept | reject
    comment: str = ""



class ReferralCreate(BaseModel):
    provider_id: str = ""
    provider_name: str = ""
    opc_id: str = ""
    opc_name: str = ""



