mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-02 19:55:18 +02:00
fix(gateway): ensure proper closure of event streams in agent invocation
This commit is contained in:
parent
708e3a9120
commit
f2d82234d4
1 changed files with 6 additions and 1 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue