openai/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-09-05 15:18:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Preserve failed logical goal result

Browse Source
This commit is contained in:
Ahmed Ibrahim
2026-06-08 19:13:23 -07:00
parent 2ca9882d11
commit ed2fb85cba
1 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
sdk/python/src/openai_codex/_goal.py
View File

@@ -238,6 +238,7 @@ class _GoalStreamCursor:
state: _GoalOperationState
started: Turn | None = None
last_completed: TurnCompletedNotification | None = None
failed_completion: TurnCompletedNotification | None = None
status: ThreadGoalStatus | None = None
active: bool = False
cleared: bool = False
@@ -259,8 +260,14 @@ class _GoalStreamCursor:
self.active = False
self.last_completed = payload
if payload.turn.status == TurnStatus.interrupted:
return [self._completion(notification.method, payload)], True
return [
self._completion(
notification.method,
self.failed_completion or payload,
)
], True
if payload.turn.status == TurnStatus.failed:
self.failed_completion = payload
if self.cleared or _terminal_goal_status(self.status):
self.state.finish()
return [self._completion(notification.method, payload)], True
@@ -271,7 +278,12 @@ class _GoalStreamCursor:
)
if self.cleared or _terminal_goal_status(self.status):
self.state.finish()
return [self._completion(notification.method, payload)], True
return [
self._completion(
notification.method,
self.failed_completion or payload,
)
], True
return [], False
events = [_logical_notification(notification, logical_turn_id)]
@@ -290,7 +302,12 @@ class _GoalStreamCursor:
and (self.cleared or _terminal_goal_status(self.status))
):
self.state.finish()
events.append(self._completion("turn/completed", self.last_completed))
events.append(
self._completion(
"turn/completed",
self.failed_completion or self.last_completed,
)
)
return events, True
return events, False