__init__.py 977 B

12345678910111213141516171819202122
  1. from configs import dify_config
  2. HIDDEN_VALUE = "[__HIDDEN__]"
  3. UUID_NIL = "00000000-0000-0000-0000-000000000000"
  4. IMAGE_EXTENSIONS = ["jpg", "jpeg", "png", "webp", "gif", "svg"]
  5. IMAGE_EXTENSIONS.extend([ext.upper() for ext in IMAGE_EXTENSIONS])
  6. VIDEO_EXTENSIONS = ["mp4", "mov", "mpeg", "mpga"]
  7. VIDEO_EXTENSIONS.extend([ext.upper() for ext in VIDEO_EXTENSIONS])
  8. AUDIO_EXTENSIONS = ["mp3", "m4a", "wav", "webm", "amr"]
  9. AUDIO_EXTENSIONS.extend([ext.upper() for ext in AUDIO_EXTENSIONS])
  10. if dify_config.ETL_TYPE == "Unstructured":
  11. DOCUMENT_EXTENSIONS = ["txt", "markdown", "md", "pdf", "html", "htm", "xlsx", "xls"]
  12. DOCUMENT_EXTENSIONS.extend(("docx", "csv", "eml", "msg", "pptx", "ppt", "xml", "epub"))
  13. DOCUMENT_EXTENSIONS.extend([ext.upper() for ext in DOCUMENT_EXTENSIONS])
  14. else:
  15. DOCUMENT_EXTENSIONS = ["txt", "markdown", "md", "pdf", "html", "htm", "xlsx", "xls", "docx", "csv"]
  16. DOCUMENT_EXTENSIONS.extend([ext.upper() for ext in DOCUMENT_EXTENSIONS])