Skip to content

Polar integration#1349

Open
HaoZhang534 wants to merge 2 commits into
NVIDIA-NeMo:mainfrom
HaoZhang534:polar
Open

Polar integration#1349
HaoZhang534 wants to merge 2 commits into
NVIDIA-NeMo:mainfrom
HaoZhang534:polar

Conversation

@HaoZhang534
Copy link
Copy Markdown

Summary

This PR adds a Polar Responses API agent adapter for Gym under responses_api_agents/polar.

The adapter converts Gym /run requests into the single-row SWE-Gym JSONL format consumed by ProRL-Agent-Server/examples/swegym_slime_grpo/simple_inference.py, executes Polar inference, and converts the result JSON back into a Gym NeMoGymResponse.

Main additions:

  • Add Polar agent server implementation with /responses and /run support.
  • Support both direct inference against an existing OpenAI-compatible SGLang server and SLURM-backed inference through the Polar submit script.
  • Preserve tool calls, tool outputs, text outputs, token ids, logprobs, usage, and Polar metadata in the returned response.
  • Add a Polar Gym config, example SWE-Gym row, README usage docs, and unit tests.

Testing

.venv/bin/pytest responses_api_agents/polar/tests/test_app.py

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented May 16, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@HaoZhang534 HaoZhang534 changed the title Polar Polar integration May 16, 2026
@cwing-nvidia cwing-nvidia added the env-integration Integrate with existing environments or gyms external to NeMo Gym label May 27, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

env-integration Integrate with existing environments or gyms external to NeMo Gym

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants