From a326c18b7443eae64fc6514c2c326e63d29ef879 Mon Sep 17 00:00:00 2001 From: Danny Avila Date: Sun, 13 Apr 2025 11:09:01 -0400 Subject: [PATCH] refactor: Prevent memory leaks by nullifying boundModel.client in disposeClient function --- api/server/cleanup.js | 3 +++ api/server/controllers/agents/client.js | 2 ++ 2 files changed, 5 insertions(+) diff --git a/api/server/cleanup.js b/api/server/cleanup.js index 6a9abb8180..6d5b77196a 100644 --- a/api/server/cleanup.js +++ b/api/server/cleanup.js @@ -238,6 +238,9 @@ function disposeClient(client) { client.run.Graph.streamBuffer = null; client.run.Graph.clientOptions = null; client.run.Graph.graphState = null; + if (client.run.Graph.boundModel?.client) { + client.run.Graph.boundModel.client = null; + } client.run.Graph.boundModel = null; client.run.Graph.systemMessage = null; client.run.Graph.reasoningKey = null; diff --git a/api/server/controllers/agents/client.js b/api/server/controllers/agents/client.js index a0de98c5f1..8a128bcdba 100644 --- a/api/server/controllers/agents/client.js +++ b/api/server/controllers/agents/client.js @@ -787,6 +787,8 @@ class AgentClient extends BaseClient { [Callback.TOOL_ERROR]: logToolError, }, }); + + config.signal = null; }; await runAgent(this.options.agent, initialMessages);