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

refactor: simplify stream error handler branching

Browse Source 
Set log level, display message, and error code upfront based on
is_transient, then use them once — removes the if/else duplication.
This commit is contained in:
Zamil Majdy
2026-03-17 13:32:29 +07:00
parent feeed4645c
commit 0614b22a72
1 changed files with 15 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
autogpt_platform/backend/backend/copilot/sdk/service.py
View File

@@ -1039,40 +1039,32 @@ async def stream_chat_completion_sdk(
# frontend gets a clean finish.
err_str = str(stream_err)
is_transient = is_transient_api_error(err_str)
if is_transient:
logger.warning(
"%s Transient stream error from SDK: %s",
log_prefix,
stream_err,
exc_info=True,
)
else:
logger.error(
"%s Stream error from SDK: %s",
log_prefix,
stream_err,
exc_info=True,
)
ended_with_stream_error = True
log = logger.warning if is_transient else logger.error
display = (
FRIENDLY_TRANSIENT_MSG
if is_transient
else f"SDK stream error: {stream_err}"
)
code = (
"transient_api_error"
if is_transient
else "sdk_stream_error"
)
log(
"%s Stream error from SDK: %s",
log_prefix,
stream_err,
exc_info=True,
)
ended_with_stream_error = True
session.messages.append(
ChatMessage(
role="assistant",
content=f"{COPILOT_ERROR_PREFIX} {display}",
)
)
yield StreamError(
errorText=display,
code=(
"transient_api_error"
if is_transient
else "sdk_stream_error"
),
)
yield StreamError(errorText=display, code=code)
break
logger.info(