zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-20 15:25:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(agent): close MCP sessions wrapped in MCPToolBinding (#18399)

Browse Source
This commit is contained in:
var4dev
2026-08-18 12:15:16 +07:00
committed by GitHub
parent 5ec11f7b07
commit 63a454084e
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
agent/canvas.py
View File

@@ -168,19 +168,22 @@ class Graph:
logging.exception(e)
def close(self):
from common.mcp_tool_call_conn import MCPToolCallSession
from common.mcp_tool_call_conn import MCPToolBinding, MCPToolCallSession
seen = set()
for cpn in self.components.values():
obj = cpn.get("obj")
if obj and hasattr(obj, "tools"):
for tool in obj.tools.values():
if isinstance(tool, MCPToolCallSession) and id(tool) not in seen:
seen.add(id(tool))
session = tool if isinstance(tool, MCPToolCallSession) else (
tool.session if isinstance(tool, MCPToolBinding) else None
)
if isinstance(session, MCPToolCallSession) and id(session) not in seen:
seen.add(id(session))
try:
tool.close_sync(timeout=3)
session.close_sync(timeout=3)
except Exception:
pass
logging.exception("Error closing MCP session for server %s", session._mcp_server.id)
@staticmethod
def _get_component_name(dsl, cid):