Add UNSTRUCTURED_API_KEY env support (#4369)

2024-05-20 13:14:17 +08:00
parent 3a51f2a778
commit b5204111da
4 changed files with 8 additions and 3 deletions
--- a/api/core/rag/extractor/unstructured/unstructured_ppt_extractor.py
+++ b/api/core/rag/extractor/unstructured/unstructured_ppt_extractor.py
@@ -17,16 +17,18 @@ class UnstructuredPPTExtractor(BaseExtractor):
    def __init__(
            self,
            file_path: str,
-            api_url: 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.partition.api import partition_via_api

-        elements = partition_via_api(filename=self._file_path, api_url=self._api_url)
+        elements = partition_via_api(filename=self._file_path, api_url=self._api_url, api_key=self._api_key)
        text_by_page = {}
        for element in elements:
            page = element.metadata.page_number