Skip to content

fix(workplace-assistant): release session state after verification - #2886

Merged
ananthsub merged 1 commit into
NVIDIA-NeMo:mainfrom
ananthsub:ananthsub/fix-workplace-session-cleanup
Aug 31, 2026
Merged

fix(workplace-assistant): release session state after verification#2886
ananthsub merged 1 commit into
NVIDIA-NeMo:mainfrom
ananthsub:ananthsub/fix-workplace-session-cleanup

Conversation

@ananthsub

Copy link
Copy Markdown
Contributor

Summary

  • pass the request context into workplace assistant verification so the active session can be identified
  • remove the per-session tool environment after verification, including when scoring raises
  • cover successful and failed verification cleanup

Test plan

  • uv run gym env test --resources-server workplace_assistant (32 passed)
  • pre-commit run --files resources_servers/workplace_assistant/app.py resources_servers/workplace_assistant/tests/test_app.py

Ensure per-rollout tool environments are removed even when scoring fails so the resources server does not retain unbounded session state.

Signed-off-by: Ananth Subramaniam <ansubramania@nvidia.com>
@ananthsub
ananthsub requested a review from cmunley1 August 31, 2026 16:00
@ananthsub ananthsub added the r0.6.0 Auto-cherrypick to release branch. Apply before merge; cherrypick happens after merge. label Aug 31, 2026
@ananthsub
ananthsub requested a review from init-nikhil August 31, 2026 16:27

@init-nikhil init-nikhil left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm.

non-blocking: if the model call fails or times out before /verify is called, it looks like the tool environment created during seed session will still remain?

@ananthsub

Copy link
Copy Markdown
Contributor Author

lgtm.

non-blocking: if the model call fails or times out before /verify is called, it looks like the tool environment created during seed session will still remain?

agreed. for that, we'd need a dedicated cleanup endpoint which is being proposed in #2612

@ananthsub
ananthsub merged commit b1bbb57 into NVIDIA-NeMo:main Aug 31, 2026
33 checks passed
@ananthsub
ananthsub deleted the ananthsub/fix-workplace-session-cleanup branch August 31, 2026 21:37
ananthsub added a commit that referenced this pull request Aug 31, 2026
…on (2886)` into `r0.6.0` (#2902)

beep boop [🤖]: Hi @ananthsub 👋,

    we've cherry picked #2886 into  for you! 🚀

    Please review and approve this cherry pick by your convenience!

Signed-off-by: Ananth Subramaniam <ansubramania@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Ananth Subramaniam <ansubramania@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

r0.6.0 Auto-cherrypick to release branch. Apply before merge; cherrypick happens after merge.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants