cbh/api/message/dto.py

from enum import Enum


class AgentRequestType(Enum):
    WorkerSearch = 1
    FacilitySearch = 2
    LocationSearch = 3
    ShiftStatistics = 4
    AverageStatistics = 5
    CountStatistics = 6
    LocationStatistics = 7
    General = 8



class Author(Enum):
    User = "user"
    Assistant = "assistant"


class EntityType(Enum):
    Worker = 1
    Facility = 2
    Shift = 3


class LocationType(Enum):
    city = 1
    state = 2