fix: Use anyio for file operations in a few places (#5167)

Use anyio for file operations in a few places

src/backend/base/langflow/components/data/api_request.py

@@ -11,6 +11,7 @@ from zoneinfo import ZoneInfo
 
 import httpx
 import validators
+from aiofile import async_open
 
 from langflow.base.curl.parse import parse_context
 from langflow.custom import Component
@@ -193,8 +194,8 @@ class APIRequestComponent(Component):
                 mode = "wb" if is_binary else "w"
                 encoding = response.encoding if mode == "w" else None
                 if file_path:
-                    with file_path.open(mode, encoding=encoding) as f:
-                        f.write(response.content if is_binary else response.text)
+                    async with async_open(file_path, mode, encoding=encoding) as f:
+                        await f.write(response.content if is_binary else response.text)
 
                 if include_httpx_metadata:
                     metadata.update(

src/backend/base/langflow/custom/directory_reader/directory_reader.py

@@ -3,6 +3,7 @@ import asyncio
 import zlib
 from pathlib import Path
 
+import anyio
 from aiofile import async_open
 from loguru import logger
 
@@ -107,18 +108,18 @@ class DirectoryReader:
 
     async def aread_file_content(self, file_path):
         """Read and return the content of a file."""
-        file_path_ = Path(file_path)
-        if not file_path_.is_file():
+        file_path_ = anyio.Path(file_path)
+        if not await file_path_.is_file():
             return None
         try:
-            async with async_open(file_path_, encoding="utf-8") as file:
+            async with async_open(str(file_path_), encoding="utf-8") as file:
                 # UnicodeDecodeError: 'charmap' codec can't decode byte 0x9d in position 3069:
                 # character maps to <undefined>
                 return await file.read()
         except UnicodeDecodeError:
             # This is happening in Windows, so we need to open the file in binary mode
             # The file is always just a python file, so we can safely read it as utf-8
-            async with async_open(file_path_, "rb") as f:
+            async with async_open(str(file_path_), "rb") as f:
                 return (await f.read()).decode("utf-8")
 
     def get_files(self):

src/backend/base/langflow/initial_setup/setup.py

@@ -8,6 +8,7 @@ from datetime import datetime, timezone
 from pathlib import Path
 from uuid import UUID
 
+import anyio
 import orjson
 from aiofile import async_open
 from emoji import demojize, purely_emoji
@@ -544,13 +545,13 @@ async def load_flows_from_directory() -> None:
             msg = "Superuser not found in the database"
             raise NoResultFound(msg)
         user_id = user.id
-        flows_path_ = Path(flows_path)
-        files = [f for f in flows_path_.iterdir() if f.is_file()]
+        flows_path_ = anyio.Path(flows_path)
+        files = [f async for f in flows_path_.iterdir() if await f.is_file()]
         for file_path in files:
             if file_path.suffix != ".json":
                 continue
             logger.info(f"Loading flow from file: {file_path.name}")
-            async with async_open(file_path, "r", encoding="utf-8") as f:
+            async with async_open(str(file_path), "r", encoding="utf-8") as f:
                 content = await f.read()
             flow = orjson.loads(content)
             no_json_name = file_path.stem

src/backend/base/langflow/main.py

@@ -8,6 +8,7 @@ from http import HTTPStatus
 from pathlib import Path
 from urllib.parse import urlencode
 
+import anyio
 from fastapi import FastAPI, HTTPException, Request, Response, status
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import FileResponse, JSONResponse
@@ -260,9 +261,9 @@ def setup_static_files(app: FastAPI, static_files_dir: Path) -> None:
 
     @app.exception_handler(404)
     async def custom_404_handler(_request, _exc):
-        path = static_files_dir / "index.html"
+        path = anyio.Path(static_files_dir) / "index.html"
 
-        if not path.exists():
+        if not await path.exists():
             msg = f"File at path {path} does not exist."
             raise RuntimeError(msg)
         return FileResponse(path)

src/backend/base/langflow/services/storage/local.py

@@ -1,5 +1,4 @@
-from pathlib import Path
-
+import anyio
 from aiofile import async_open
 from loguru import logger
 
@@ -12,7 +11,7 @@ class LocalStorageService(StorageService):
     def __init__(self, session_service, settings_service) -> None:
         """Initialize the local storage service with session and settings services."""
         super().__init__(session_service, settings_service)
-        self.data_dir = Path(settings_service.settings.config_dir)
+        self.data_dir = anyio.Path(settings_service.settings.config_dir)
         self.set_ready()
 
     def build_full_path(self, flow_id: str, file_name: str) -> str:
@@ -30,11 +29,11 @@ class LocalStorageService(StorageService):
         :raises PermissionError: If there is no permission to write the file.
         """
         folder_path = self.data_dir / flow_id
-        folder_path.mkdir(parents=True, exist_ok=True)
+        await folder_path.mkdir(parents=True, exist_ok=True)
         file_path = folder_path / file_name
 
         try:
-            async with async_open(file_path, "wb") as f:
+            async with async_open(str(file_path), "wb") as f:
                 await f.write(data)
             logger.info(f"File {file_name} saved successfully in flow {flow_id}.")
         except Exception:
@@ -50,12 +49,12 @@ class LocalStorageService(StorageService):
         :raises FileNotFoundError: If the file does not exist.
         """
         file_path = self.data_dir / flow_id / file_name
-        if not file_path.exists():
+        if not await file_path.exists():
             logger.warning(f"File {file_name} not found in flow {flow_id}.")
             msg = f"File {file_name} not found in flow {flow_id}"
             raise FileNotFoundError(msg)
 
-        async with async_open(file_path, "rb") as f:
+        async with async_open(str(file_path), "rb") as f:
             content = await f.read()
 
         logger.debug(f"File {file_name} retrieved successfully from flow {flow_id}.")
@@ -69,12 +68,12 @@ class LocalStorageService(StorageService):
         :raises FileNotFoundError: If the flow directory does not exist.
         """
         folder_path = self.data_dir / flow_id
-        if not folder_path.exists() or not folder_path.is_dir():
+        if not await folder_path.exists() or not await folder_path.is_dir():
             logger.warning(f"Flow {flow_id} directory does not exist.")
             msg = f"Flow {flow_id} directory does not exist."
             raise FileNotFoundError(msg)
 
-        files = [file.name for file in folder_path.iterdir() if file.is_file()]
+        files = [file.name async for file in folder_path.iterdir() if await file.is_file()]
         logger.info(f"Listed {len(files)} files in flow {flow_id}.")
         return files
 
@@ -85,8 +84,8 @@ class LocalStorageService(StorageService):
         :param file_name: The name of the file to be deleted.
         """
         file_path = self.data_dir / flow_id / file_name
-        if file_path.exists():
-            file_path.unlink()
+        if await file_path.exists():
+            await file_path.unlink()
             logger.info(f"File {file_name} deleted successfully from flow {flow_id}.")
         else:
             logger.warning(f"Attempted to delete non-existent file {file_name} in flow {flow_id}.")