jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: Use async file operations for error log management in MCPStdioClient (#8012)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-05-13 09:31:30 -03:00 • committed by GitHub
commit 482aa5a7c4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/base/mcp/util.py
View file

@ -4,13 +4,13 @@ import os
import platform import platform
from collections.abc import Awaitable, Callable from collections.abc import Awaitable, Callable
from contextlib import AsyncExitStack from contextlib import AsyncExitStack
from pathlib import Path
from typing import Any, cast from typing import Any, cast
from urllib.parse import urlparse from urllib.parse import urlparse
from uuid import UUID from uuid import UUID
import aiofiles import aiofiles
import httpx import httpx
from anyio import Path
from httpx import codes as httpx_codes from httpx import codes as httpx_codes
from loguru import logger from loguru import logger
from mcp import ClientSession, StdioServerParameters, stdio_client from mcp import ClientSession, StdioServerParameters, stdio_client
@ -271,7 +271,7 @@ class MCPStdioClient:
while True: while True:
await asyncio.sleep(0.05) await asyncio.sleep(0.05)
await tmp.flush() await tmp.flush()
current = Path(errlog_path).stat().st_size current = (await Path(errlog_path).stat()).st_size
if current > last_size: if current > last_size:
async with aiofiles.open(errlog_path, encoding="utf-8") as f: async with aiofiles.open(errlog_path, encoding="utf-8") as f:
await f.seek(last_size) await f.seek(last_size)
@ -326,7 +326,7 @@ class MCPStdioClient:
finally: finally:
# Clean up the temp file # Clean up the temp file
with contextlib.suppress(FileNotFoundError, PermissionError): with contextlib.suppress(FileNotFoundError, PermissionError):
Path(errlog_path).unlink() await Path(errlog_path).unlink()
class MCPSseClient: class MCPSseClient: