hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-09 10:56:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

add doc support in knowledge base for unstructured (#17352)

Browse Source
This commit is contained in:
Jyong
2025-04-02 21:35:01 +08:00
committed by GitHub
parent 33c8cb7b3b
commit 6104b91d3f
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/rag/extractor/unstructured/unstructured_doc_extractor.py
View File

@@ -10,14 +10,11 @@ logger = logging.getLogger(__name__)
class UnstructuredWordExtractor(BaseExtractor):
"""Loader that uses unstructured to load word documents."""
def __init__(
self,
file_path: str,
api_url: str,
):
def __init__(self, file_path: str, api_url: str, api_key: str = ""):
"""Initialize with file path."""
self._file_path = file_path
self._api_url = api_url
self._api_key = api_key
def extract(self) -> list[Document]:
from unstructured.__version__ import __version__ as __unstructured_version__
@@ -41,9 +38,10 @@ class UnstructuredWordExtractor(BaseExtractor):
)
if is_doc:
from unstructured.partition.doc import partition_doc
from unstructured.partition.api import partition_via_api
elements = partition_via_api(filename=self._file_path, api_url=self._api_url, api_key=self._api_key)
elements = partition_doc(filename=self._file_path)
else:
from unstructured.partition.docx import partition_docx