__init__.py 1.0 KB

123456789101112131415161718192021222324
  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", "xml", "epub"))
  13. if dify_config.UNSTRUCTURED_API_URL:
  14. DOCUMENT_EXTENSIONS.append("ppt")
  15. DOCUMENT_EXTENSIONS.extend([ext.upper() for ext in DOCUMENT_EXTENSIONS])
  16. else:
  17. DOCUMENT_EXTENSIONS = ["txt", "markdown", "md", "pdf", "html", "htm", "xlsx", "xls", "docx", "csv"]
  18. DOCUMENT_EXTENSIONS.extend([ext.upper() for ext in DOCUMENT_EXTENSIONS])