github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(backend): always yield session_id even on stream failure

Browse Source 
Move session creation from execute_copilot into run() so the
session_id is yielded to the user even if the downstream stream
raises an exception, preventing orphaned sessions.
This commit is contained in:
Zamil Majdy
2026-03-17 00:32:22 +07:00
parent cbc16e43a6
commit 040d1e851c
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
autogpt_platform/backend/backend/blocks/autopilot.py
View File

@@ -292,11 +292,20 @@ class AutoPilotBlock(Block):
yield "error", "Cannot run autopilot without an authenticated user."
return
from backend.copilot.model import create_chat_session
# Create session eagerly so the user always gets the session_id,
# even if the downstream stream fails (avoids orphaned sessions).
sid = input_data.session_id
if not sid:
session = await create_chat_session(execution_context.user_id)
sid = session.session_id
try:
response, tool_calls, history, sid, usage = await self.execute_copilot(
response, tool_calls, history, _, usage = await self.execute_copilot(
prompt=input_data.prompt,
system_context=input_data.system_context,
session_id=input_data.session_id,
session_id=sid,
max_recursion_depth=input_data.max_recursion_depth,
user_id=execution_context.user_id,
)
@@ -307,4 +316,5 @@ class AutoPilotBlock(Block):
yield "session_id", sid
yield "token_usage", usage
except Exception as e:
yield "session_id", sid
yield "error", str(e)