From f2d82234d4af21cd3e7e904a3eb5524a0bad1dbc Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Thu, 28 May 2026 00:25:12 +0530
Subject: [PATCH] fix(gateway): ensure proper closure of event streams in agent
 invocation

---
 surfsense_backend/app/gateway/agent_invoke.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/surfsense_backend/app/gateway/agent_invoke.py b/surfsense_backend/app/gateway/agent_invoke.py
index 4faf3711f..b876d1977 100644
--- a/surfsense_backend/app/gateway/agent_invoke.py
+++ b/surfsense_backend/app/gateway/agent_invoke.py
@@ -79,7 +79,12 @@ async def call_agent_for_gateway(
             disabled_tools=sorted(DEFAULT_HITL_TOOL_NAMES),
             request_id=request_id or "gateway",
         )
-        await translator.translate(_events_from_sse(stream))
+        events = _events_from_sse(stream)
+        try:
+            await translator.translate(events)
+        finally:
+            await events.aclose()
+            await stream.aclose()
         record_gateway_turn_latency(0, platform="telegram")
     finally:
         release_thread_lock(thread.id)