fix: . is missing in file_extension (#8736)

langgenius · Sep 25, 2024 · bf64ff2 · bf64ff2
1 parent 68c7e68
commit bf64ff2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/core/rag/extractor/extract_processor.py b/api/core/rag/extractor/extract_processor.py
@@ -124,7 +124,7 @@ def extract(
  extractor = UnstructuredPPTXExtractor(file_path, unstructured_api_url)
  elif file_extension == ".xml":
  extractor = UnstructuredXmlExtractor(file_path, unstructured_api_url)
- elif file_extension == "epub":
+ elif file_extension == ".epub":
  extractor = UnstructuredEpubExtractor(file_path, unstructured_api_url)
  else:
  # txt
@@ -146,7 +146,7 @@ def extract(
  extractor = WordExtractor(file_path, upload_file.tenant_id, upload_file.created_by)
  elif file_extension == ".csv":
  extractor = CSVExtractor(file_path, autodetect_encoding=True)
- elif file_extension == "epub":
+ elif file_extension == ".epub":
  extractor = UnstructuredEpubExtractor(file_path)
  else:
  # txt