apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-04 20:05:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix(gateway): ensure proper closure of event streams in agent invocation

Browse source
This commit is contained in:
Anish Sarkar 2026-05-28 00:25:12 +05:30
parent 708e3a9120
commit f2d82234d4
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
surfsense_backend/app/gateway/agent_invoke.py
View file

@ -79,7 +79,12 @@ async def call_agent_for_gateway(
disabled_tools=sorted(DEFAULT_HITL_TOOL_NAMES), disabled_tools=sorted(DEFAULT_HITL_TOOL_NAMES),
request_id=request_id or "gateway", 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") record_gateway_turn_latency(0, platform="telegram")
finally: finally:
release_thread_lock(thread.id) release_thread_lock(thread.id)