data_type.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from enum import Enum
  2. class DirectDataType(Enum):
  3. """
  4. DirectDataType enum contains data types that contain raw data directly.
  5. """
  6. TEXT = "text"
  7. class IndirectDataType(Enum):
  8. """
  9. IndirectDataType enum contains data types that contain references to data stored elsewhere.
  10. """
  11. YOUTUBE_VIDEO = "youtube_video"
  12. PDF_FILE = "pdf_file"
  13. WEB_PAGE = "web_page"
  14. SITEMAP = "sitemap"
  15. XML = "xml"
  16. DOCX = "docx"
  17. DOCS_SITE = "docs_site"
  18. NOTION = "notion"
  19. CSV = "csv"
  20. MDX = "mdx"
  21. IMAGES = "images"
  22. class SpecialDataType(Enum):
  23. """
  24. SpecialDataType enum contains data types that are neither direct nor indirect, or simply require special attention.
  25. """
  26. QNA_PAIR = "qna_pair"
  27. class DataType(Enum):
  28. TEXT = DirectDataType.TEXT.value
  29. YOUTUBE_VIDEO = IndirectDataType.YOUTUBE_VIDEO.value
  30. PDF_FILE = IndirectDataType.PDF_FILE.value
  31. WEB_PAGE = IndirectDataType.WEB_PAGE.value
  32. SITEMAP = IndirectDataType.SITEMAP.value
  33. XML = IndirectDataType.XML.value
  34. DOCX = IndirectDataType.DOCX.value
  35. DOCS_SITE = IndirectDataType.DOCS_SITE.value
  36. NOTION = IndirectDataType.NOTION.value
  37. CSV = IndirectDataType.CSV.value
  38. MDX = IndirectDataType.MDX.value
  39. QNA_PAIR = SpecialDataType.QNA_PAIR.value
  40. IMAGES = IndirectDataType.IMAGES.value