from pydantic import BaseModel


class CompanyInfoSaveRequest(BaseModel):
    session_id: str
    eligibility: str | None = None
    skills_and_experience: str | None = None
    welcome_personality: str | None = None
    key_information: str | None = None


class CompanyInfoSaveResponse(BaseModel):
    status: str
    message: str


class CompanyInfoResponse(BaseModel):
    status: str
    data: dict | None = None  # eligibility, skills_and_experience, welcome_personality, key_information, created_at, updated_at