fix(langgraph): guard replace_state() against closed StateManager in execute() #10768
Merged
HAL9000
merged 3 commits from 2026-05-05 04:04:13 +00:00
bugfix/m3-langgraph-execute-state-bypass into master
Labels
Clear labels
auto/needs-reevaluation
controller-managed
overdue
auto/blocked-by-deps
auto/ci-timeout
auto/claimed-implementer
auto/claimed-merge
auto/claimed-reviewer
auto/driver-down
auto/invariant-violation
auto/last-attempt-tier-0
auto/last-attempt-tier-1
auto/last-attempt-tier-2
auto/last-attempt-tier-min
Automation Tracking
auto/needs-conflict-resolution
auto/needs-implementer
auto/postmortem
auto/ready-to-merge
auto/restart-throttled
auto/revert
auto/sentinel
auto/stale-inactivity
auto/unstable
Blocked
Needs Feedback
Signed-off: Owner
Signed-off: Scrum Master
Signed-off: Tech Lead
Spike
Controller deferred this PR; awaiting Phase 6+ scope-evaluator or operator re-enablement.
Auto-agents controller manages this PR/issue (see tools/controller/deploy/RUNBOOK.md). Remove this label to abandon controller management.
PR blocked by an open issue dependency. Operator must close the dep (or remove the dependency link) before the merge driver can act. Auto-cleared by merge_drive when no open deps remain.
Most recent merge cycle hit CI timeout. Driver excludes this PR while last merge_cycle row is < 30 min old; label persists thereafter as visible history.
Currently being processed by an implementer worker.
Currently being processed by the merge driver.
Currently being processed by a reviewer worker.
Merge driver heartbeat stale; pipeline halted. Closed automatically on next clean tick.
Detected master commit violating the strict merge invariant. Tracked as an issue (not a PR label); kept here for label completeness.
In-cycle escalation: most recent attempt ran at the Tier 0 slot (`tier-0`). Slot's model defined in .opencode/models/tiers.yaml.
In-cycle escalation: most recent attempt ran at the Tier 1 slot (`tier-1`). Slot's model defined in .opencode/models/tiers.yaml.
In-cycle escalation: most recent attempt ran at the Tier 2 slot (`tier-2`). Slot's model defined in .opencode/models/tiers.yaml. Gated behind IMPLEMENTER_ESCALATION_TIER2_ENABLED.
In-cycle escalation: most recent attempt ran at the Tier -1 slot (`tier-min`). Slot's model defined in .opencode/models/tiers.yaml. Suffix is ``-min`` (not ``--1``) so the Forgejo UI reads naturally.
Tracking issues used by the AI Automation system for agents to communicate and report.
Rebase conflict needs LLM conflict-resolver.
Failing CI needs implementer attention.
Documenting a driver incident or rollback.
Reviewer has APPROVED this PR and no later REQUEST_CHANGES is outstanding. The merge driver requires this label to even consider a PR for merging. Set by the reviewer worker on APPROVE; cleared on REQUEST_CHANGES.
Train repeatedly lost master-tempo races. Driver excludes via merge_cycle until cooldown elapses; label persists as visible history.
Revert PR backing out an invariant violation. Fast-tracked through the merge driver.
Sentinel PR duplicated from upstream into a personal fork by tools/duplicate_prs_to_fork.py for pipeline testing. Lives only in the fork; the canonical pipeline never sees it.
No implementer activity for N days. Flagged for human review. Auto-cleared on next push to head branch.
Repeatedly fails on current master (>= 3 ci-fail-on-rebased-sha releases in 12 h). Excluded from driver until human triage.
A ticket in a blocked state and unable to complete until some other task is completed first.
Bounty
$100
A bounty of $100 for any open-source contributor who provides a MR that solves this issue
Bounty
$1000
A bounty of $1000 for any open-source contributor who provides a MR that solves this issue
Bounty
$10000
A bounty of $10000 for any open-source contributor who provides a MR that solves this issue
Bounty
$20
A bounty of $20 for any open-source contributor who provides a MR that solves this issue
Bounty
$2000
A bounty of $2000 for any open-source contributor who provides a MR that solves this issue
Bounty
$250
A bounty of $250 for any open-source contributor who provides a MR that solves this issue
Bounty
$50
A bounty of $50 for any open-source contributor who provides a MR that solves this issue
Bounty
$500
A bounty of $500 for any open-source contributor who provides a MR that solves this issue
Bounty
$5000
A bounty of $5000 for any open-source contributor who provides a MR that solves this issue
Bounty
$750
A bounty of $750 for any open-source contributor who provides a MR that solves this issue
MoSCoW
Could have
Could have feature in order to satisfy the epic/legendary.
MoSCoW
Must have
Must have feature in order to satisfy the epic/legendary.
MoSCoW
Should have
Should have feature in order to satisfy the epic/legendary.
There are questions in the ticket that can not be completed until the project owner provides clarity.
Points
1
1 man-hours worth of work for an expert with no learning curve.
Points
13
13 man-hours worth of work for an expert with no learning curve.
Points
2
2 man-hours worth of work for an expert with no learning curve.
Points
21
21 man-hours worth of work for an expert with no learning curve.
Points
3
3 man-hours worth of work for an expert with no learning curve.
Points
34
34 man-hours worth of work for an expert with no learning curve.
Points
5
5 man-hours worth of work for an expert with no learning curve.
Points
55
55 man-hours worth of work for an expert with no learning curve.
Points
8
8 man-hours worth of work for an expert with no learning curve.
Points
88
88 man-hours worth of work for an expert with no learning curve.
Priority
Backlog
This ticket has backlogged priority and is not to be worked on yet
Priority
CI Blocker
Critical priority issue that blocks CI/CD pipeline and prevents PR merges
Priority
Critical
The priority is critical
Priority
High
The priority is high
Priority
Low
The priority is low
Priority
Medium
The priority is medium
When an epic or legendary is in review it must be signed off by owner, tech lead, and scrum master before being marked as completed.
When an epic or legendary is in review it must be signed off by owner, tech lead, and scrum master before being marked as completed.
When an epic or legendary is in review it must be signed off by owner, tech lead, and scrum master before being marked as completed.
A ticket for learning a tool or technology that is needed to be able to do future planning and design.
State
Completed
The ticket has been fully implemented, completed, and merged with the source code. This label should only be applied once a ticket is closed.
State
Duplicate
A ticket that represents the same content as an existing ticket.
State
In Progress
A ticket that is actively being developed.
State
In Review
A ticket that has had some code completed to implement but is waiting to pass peer review and is not yet merged in.
State
Paused
This ticket's work started but wasn't finished. It's on hold (likely in a feature branch) and will be resumed later, either due to a blocker or a delay.
State
Unverified
All new tickets start in this state. A developer may set it to show the ticket is unverified. This means we haven't agreed to work on it. It will either move to a verified state or be closed as wontdo.
State
Verified
The issue has been verified by a developer as legitimate. It will be worked on and verified tickets are now considered part of the backlog.
State
Wont Do
This ticket has been decided it wont be done. This may mean the bug has been determined to not be real (cant verify) or the feature is one we have decided we dont want to adopt.
Type
Automation
Any edits or discussion about the AI automated coding system.
Type
Bug
Something that doesnt work as intended.
Type
Discussion
Anytime a ticket represents a discussion about a subject and doesnt fall into one of the other categories.
Type
Documentation
An error or improvement needed in the documentation.
Type
Epic
Any first tier epic. That is, an epic which contains only issues as children and will not have sub-epics.
Type
Feature
Some new functionality not present.
Type
Legendary
A type of Epic which will contain other Epics.
Type
Refactor
A code change that restructures existing code without changing its external behavior.
Type
Support
Someone needs help using the project.
Type
Task
A generic task that doesnt fit into the other type categories.
Type
Testing
Work exclusively focusing on fixing or expanding testing.
No Label
Type
Bug
Projects
Clear projects
No project
Assignees
aditya (Aditya Chhabra)
aleenaumair (Aleena Umair)
brent.edwards (Brent Edwards)
CoreRasurae (Luis Mendes)
drew (Drew Morris)
eugen.thaci (Eugen Thaci)
freemo (Jeffrey Phillips Freeman)
HAL9000 (HAL 9000)
HAL9001 (HAL9001)
hamza.khyari (Hamza Khyari)
hurui200320 (Rui Hu)
justin.morris
khird (Kyle Hird)
org.cleveragents
Clear assignees
No Assignees
Notifications
Due Date
No due date set.
Dependencies
No dependencies set.
Reference: cleveragents/cleveragents-core#10768
Reference in New Issue
Block a user
Blocking a user prevents them from interacting with repositories, such as opening or commenting on pull requests or issues. Learn more about blocking a user.
Delete Branch "bugfix/m3-langgraph-execute-state-bypass"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Summary
This PR fixes a critical bug in
LangGraph.execute()where state assignments were bypassing theis_closedguard enforced byStateManager. The method was directly assigning state and emitting on the state stream without checking if the StateManager had been closed, allowing silent state corruption afterStateManager.close()was called.Changes
Bug Fix
src/cleveragents/langgraph/graph.py: Replaced direct state assignment inLangGraph.execute()withstate_manager.replace_state(), which enforces theis_closedguard and notifies state stream subscribers through the proper API.src/cleveragents/langgraph/state.py: Addedis_closedguard toStateManager.replace_state()— the method now raisesRuntimeError("StateManager is closed")when the manager has been closed, consistent with all other mutation methods (update_state,load_checkpoint,time_travel,reset).Test Coverage
features/tdd_langgraph_execute_closed_state.feature: New Behave BDD feature fileexecute()raises RuntimeError after StateManager.close()execute()succeeds when StateManager is openfeatures/steps/tdd_langgraph_execute_closed_state_steps.py: Step definitions for the new featureRoot Cause
The bug existed because
LangGraph.execute()was directly assigningself.state_manager.state = stateand callingself.state_manager.state_stream.on_next(state)without checking theis_closedflag. This bypassed the protective guard thatStateManager.replace_state()enforces.replace_state()is the semantically correct method for this use case — it atomically replaces the entire state for a fresh execution context, enforces theis_closedguard, and notifies subscribers. Theupdate_state()method is designed for incremental updates (withexecution_counttracking), not for resetting state to a fresh execution context.Testing
Issue Reference
Closes #9994
Automated by CleverAgents Bot
Supervisor: Implementation Pool | Agent: implementation-worker
Fixed LangGraph.execute() to respect the StateManager's is_closed status by: - Checking is_closed before assigning state and raising RuntimeError("StateManager is closed") if closed - Adding a guard to prevent state_stream.on_next() from firing when the StateManager is closed - Introducing Behave BDD tests: - features/tdd_langgraph_execute_closed_state.feature with two scenarios - features/steps/tdd_langgraph_execute_closed_state_steps.py to support the tests - Implemented changes in src/cleveragents/langgraph/graph.py ISSUES CLOSED: #9994Implementation Attempt — Tier 1: haiku — Success
Fixed the PR by replacing direct state assignment with
update_state()inLangGraph.execute():StateUpdateModefromcleveragents.langgraph.stateself.state_manager.state = statewithself.state_manager.update_state(state.to_dict(), mode=StateUpdateMode.REPLACE)is_closedcheck andstate_stream.on_next()call sinceupdate_state()handles bothis_closedguard is properly enforced and prevents state corruption afterStateManager.close()Quality gates status: lint ✓, typecheck ✓
Automated by CleverAgents Bot
Supervisor: Implementation | Agent: implementation-worker
Implementation Attempt — Tier 1: haiku — Success
Fixed the LangGraph.execute() method to use the updated_state returned from update_state() instead of the original state object. This ensures that the state sent to the start stream has the correct execution_count and other updated values.
Changes made:
All quality gates should now pass:
Automated by CleverAgents Bot
Supervisor: Implementation | Agent: implementation-worker
6167b9ab4eto09c58ea81cImplementation Attempt — Tier 1: haiku — Success
Rebased the PR branch onto current master and applied a clean fix:
src/cleveragents/langgraph/state.py: Addedis_closedguard toStateManager.replace_state()— the method now raisesRuntimeError("StateManager is closed")when the manager has been closed, consistent with all other mutation methods (update_state,load_checkpoint,time_travel,reset). This is the root cause fix:LangGraph.execute()delegates toreplace_state(), which previously lacked this guard.features/tdd_langgraph_execute_closed_state.feature: New Behave BDD feature with@tdd_issueand@tdd_issue_9994tags — two scenarios: (1)execute()raises RuntimeError afterStateManager.close(), (2)execute()succeeds when StateManager is open.features/steps/tdd_langgraph_execute_closed_state_steps.py: Step definitions for the new feature.Quality gate status: lint ✓, typecheck ✓, unit_tests ✓ (83 scenarios including 2 new TDD scenarios — all pass), coverage_report ✓
Note: integration_tests and e2e_tests fail with pre-existing SQLite disk I/O errors in test setup (
create_template_db.py), unrelated to this change.Automated by CleverAgents Bot
Supervisor: Implementation | Agent: implementation-worker
Implementation Attempt — Tier 1: haiku — Success
Fixed the CI lint/format failure that was blocking the PR:
features/steps/tdd_langgraph_execute_closed_state_steps.py: Appliedruff formatto fix a single-quote vs double-quote inconsistency in an f-string assertion. The file hadf'got: {context.execute_error}'where ruff expectedf"got: {context.execute_error}". This causednox -s format -- --checkto fail, which in turn caused both theCI / lintandCI / qualityjobs to fail, cascading toCI / status-checkfailure.The core implementation (using
replace_state()withis_closedguard inLangGraph.execute()) was already correct from the previous attempt. The only remaining issue was the formatting violation.Quality gate status: lint ✓, typecheck ✓, format ✓, complexity ✓
Note:
integration_tests,e2e_tests,helm, andpush-validationCI jobs have pre-existing failures unrelated to this PR (SQLite disk I/O errors in test setup and missing CI secrets), as noted in the previous attempt comment.Automated by CleverAgents Bot
Supervisor: Implementation | Agent: implementation-worker
Summary of Changes
state.pynow guards against a closed StateManager in bothupdate_stateandreplace_state, satisfying the TDD scenario.Blocking Issue
fix(langgraph): use update_state() in LangGraph.execute() instead of direct state assignment) does not match the actual changes, which focus on preventing execution afterclose(). Please either updateLangGraph.execute()to usestate_manager.update_state(...)or adjust the PR title to reflect the closed guard fix.Automated by CleverAgents Bot
Supervisor: PR Review | Agent: pr-review-worker
Implementation Attempt — Tier 0: gpt5-mini — Failed
Attempted to fix PR #10768 (fix(langgraph): use update_state() in LangGraph.execute() instead of direct state assignment).
What was done:
9e67909c).Error details:
Automated by CleverAgents Bot
Supervisor: Implementation | Agent: task-implementor
Implementation Attempt — Tier 1: kimi — Success
Addressed the reviewer's
REQUEST_CHANGESfeedback: the PR title says "use update_state() in LangGraph.execute() instead of direct state assignment" but the previous implementation usedreplace_state()instead.Changes made:
src/cleveragents/langgraph/graph.py: InLangGraph.execute(), replacedself.state_manager.replace_state(state)withself.state_manager.update_state(state.to_dict(), mode=StateUpdateMode.REPLACE). This routes the state replacement through the properStateManagerAPI, which enforces theis_closedguard and notifies state stream subscribers correctly.StateUpdateModeto the import fromcleveragents.langgraph.state.start_streammessage to use theupdated_statereturned byupdate_state()(which includes the incrementedexecution_count) rather than the original input state.Quality gate status: lint ✓, format ✓
Note:
typecheckandunit_testscould not be run locally due to environment constraints (typecheck hangs, behave runner hangs during step loading). The previous CI run (commit9e67909c) showed all gates passing; this change is minimal and targeted — onlygraph.pyis modified.Automated by CleverAgents Bot
Supervisor: Implementation | Agent: implementation-worker
fix(langgraph): use update_state() in LangGraph.execute() instead of direct state assignmentto fix(langgraph): guard replace_state() against closed StateManager in execute()648a767c41to6c0196dcd6Implementation Attempt — Tier 1: kimi — Success
Reverted the
update_state(REPLACE)approach (commit648a767c) which broke unit_tests in CI, and restored the correctreplace_state()approach. Also updated the PR title and description to match the actual implementation.Root cause of CI failure:
update_state()always incrementsexecution_countby 1, which caused test failures in the unit_tests suite.replace_state()is the semantically correct method for resetting state to a fresh execution context — it atomically replaces the entire state without incrementingexecution_count.Changes in this commit (
6c0196dc):src/cleveragents/langgraph/graph.py: Reverted toself.state_manager.replace_state(state)inLangGraph.execute(). Removed unusedStateUpdateModeimport.fix(langgraph): guard replace_state() against closed StateManager in execute()replace_state()withis_closedguard, notupdate_state()).The reviewer's concern was that the PR title said "use update_state()" but the implementation used
replace_state(). This is now resolved by updating the PR title to match the implementation.Quality gate status: lint ✓, format ✓
Automated by CleverAgents Bot
Supervisor: Implementation | Agent: implementation-worker
Re-review Summary
This review responds to the previous REQUEST_CHANGES feedback:
Previous Feedback Items Addressed:
replace_state(). The author correctly switched back toreplace_state()after discovering it was semantically appropriate (fresh execution context replacement) whereasupdate_state()(designed for incremental updates with execution_count tracking) broke CI.Overall Assessment:
StateManager.replace_state()now enforces theis_closedguard consistent with all other mutation methods (update_state,load_checkpoint,time_travel,reset)Status: APPROVED
PR Review #2 (Re-review)
Previous REQUEST_CHANGES feedback fully addressed — title now matches
replace_state()implementation. All checklist categories pass. CI: 15/15 green.→ APPROVED