From dbecc53c0143b1c2042e2f0cb77f0fd1091d6033 Mon Sep 17 00:00:00 2001 From: FFXN Date: Wed, 28 Jan 2026 11:49:39 +0800 Subject: [PATCH] fix: fix ignore_imports. --- api/.importlinter | 3 --- .../workflow/nodes/document_extractor/node.py | 25 ------------------- 2 files changed, 28 deletions(-) diff --git a/api/.importlinter b/api/.importlinter index 224fcbedbc..080268aeb9 100644 --- a/api/.importlinter +++ b/api/.importlinter @@ -55,7 +55,6 @@ ignore_imports = core.workflow.nodes.llm.llm_utils -> extensions.ext_database core.workflow.nodes.llm.node -> extensions.ext_database core.workflow.nodes.tool.tool_node -> extensions.ext_database - core.workflow.nodes.document_extractor.node -> extensions.ext_database core.workflow.graph_engine.command_channels.redis_channel -> extensions.ext_redis core.workflow.graph_engine.manager -> extensions.ext_redis core.workflow.nodes.knowledge_retrieval.knowledge_retrieval_node -> extensions.ext_redis @@ -120,8 +119,6 @@ ignore_imports = core.workflow.nodes.document_extractor.node -> configs core.workflow.nodes.document_extractor.node -> core.file.file_manager core.workflow.nodes.document_extractor.node -> core.helper.ssrf_proxy - core.workflow.nodes.document_extractor.node -> extensions.ext_database - core.workflow.nodes.document_extractor.node -> extensions.ext_storage core.workflow.nodes.http_request.entities -> configs core.workflow.nodes.http_request.executor -> configs core.workflow.nodes.http_request.executor -> core.file.file_manager diff --git a/api/core/workflow/nodes/document_extractor/node.py b/api/core/workflow/nodes/document_extractor/node.py index 25dd98f48a..14ebd1f9ae 100644 --- a/api/core/workflow/nodes/document_extractor/node.py +++ b/api/core/workflow/nodes/document_extractor/node.py @@ -62,21 +62,6 @@ class DocumentExtractorNode(Node[DocumentExtractorNodeData]): inputs = {"variable_selector": variable_selector} process_data = {"documents": value if isinstance(value, list) else [value]} - # Ensure storage_key is loaded for File objects - files_to_check = value if isinstance(value, list) else [value] - files_needing_storage_key = [ - f for f in files_to_check if isinstance(f, File) and not f.storage_key and f.related_id - ] - if files_needing_storage_key: - from sqlalchemy.orm import Session - - from extensions.ext_database import db - from factories.file_factory import StorageKeyLoader - - with Session(bind=db.engine) as session: - storage_key_loader = StorageKeyLoader(session, tenant_id=self.tenant_id) - storage_key_loader.load_storage_keys(files_needing_storage_key) - try: if isinstance(value, list): extracted_text_list = list(map(_extract_text_from_file, value)) @@ -430,16 +415,6 @@ def _download_file_content(file: File) -> bytes: response.raise_for_status() return response.content else: - # Check if storage_key is set - if not file.storage_key: - raise FileDownloadError(f"File storage_key is missing for file: {file.filename}") - - # Check if file exists before downloading - from extensions.ext_storage import storage - - if not storage.exists(file.storage_key): - raise FileDownloadError(f"File not found in storage: {file.storage_key}") - return file_manager.download(file) except Exception as e: raise FileDownloadError(f"Error downloading file: {str(e)}") from e