dify/api/core/file/tool_file_parser.py

from collections.abc import Callable
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from core.tools.tool_file_manager import ToolFileManager

_tool_file_manager_factory: Callable[[], "ToolFileManager"] | None = None


def set_tool_file_manager_factory(factory: Callable[[], "ToolFileManager"]) -> None:
    global _tool_file_manager_factory
    _tool_file_manager_factory = factory