ソースを参照

remove unstructured api key check (#18989)

tags/1.4.0
Jyong 6ヶ月前
コミット
b8bb45b106
コミッターのメールアドレスに関連付けられたアカウントが存在しません
1個のファイルの変更3行の追加3行の削除
  1. 3
    3
      api/core/workflow/nodes/document_extractor/node.py

+ 3
- 3
api/core/workflow/nodes/document_extractor/node.py ファイルの表示

@@ -223,8 +223,8 @@ def _extract_text_from_doc(file_content: bytes) -> str:
"""
from unstructured.partition.api import partition_via_api

if not (dify_config.UNSTRUCTURED_API_URL and dify_config.UNSTRUCTURED_API_KEY):
raise TextExtractionError("UNSTRUCTURED_API_URL and UNSTRUCTURED_API_KEY must be set")
if not dify_config.UNSTRUCTURED_API_URL:
raise TextExtractionError("UNSTRUCTURED_API_URL must be set")

try:
with tempfile.NamedTemporaryFile(suffix=".doc", delete=False) as temp_file:
@@ -235,7 +235,7 @@ def _extract_text_from_doc(file_content: bytes) -> str:
file=file,
metadata_filename=temp_file.name,
api_url=dify_config.UNSTRUCTURED_API_URL,
api_key=dify_config.UNSTRUCTURED_API_KEY,
api_key=dify_config.UNSTRUCTURED_API_KEY, # type: ignore
)
os.unlink(temp_file.name)
return "\n".join([getattr(element, "text", "") for element in elements])

読み込み中…
キャンセル
保存