from pydantic import BaseModel
from typing import Optional

class TransportModeBaseDTO(BaseModel):
    mode_key: str
    mode_description: Optional[str]
    company_id: str
    tenant_id: str

class TransportModeCreateDTO(TransportModeBaseDTO):
    pass

class TransportModeResponseDTO(TransportModeBaseDTO):
    class Config:
        orm_mode = True