nlp.py 516 B

123456789101112131415161718192021222324
  1. from enum import Enum
  2. from typing import List, Optional
  3. from pydantic import BaseModel
  4. class RoomSize(str, Enum):
  5. small = "small"
  6. medium = "medium"
  7. large = "large"
  8. unknown = ""
  9. class NLPResponseBase(BaseModel):
  10. Message: str
  11. class MeetingInfoResponse(NLPResponseBase):
  12. AcceptableStartTime: Optional[str]
  13. AcceptableEndTime: Optional[str]
  14. MeetingDurationSeconds: Optional[int]
  15. MeetingRoomSize: Optional[RoomSize]
  16. Topic: Optional[str]
  17. Participants: Optional[List[str]]