From 9f91e2aff7bbd2652754711cd53a2c19025f7a12 Mon Sep 17 00:00:00 2001
From: braisedpork1964 <497494458@qq.com>
Date: Thu, 23 Jul 2026 03:16:25 +0000
Subject: [PATCH] pass cluster_name as an explicit sandbox create arg

---
 xtuner/v1/rl/agent_loop/sandbox_agent_loop/sandbox.py | 2 ++
 xtuner/v1/rl/agent_loop/sandbox_agent_loop/schemas.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/xtuner/v1/rl/agent_loop/sandbox_agent_loop/sandbox.py b/xtuner/v1/rl/agent_loop/sandbox_agent_loop/sandbox.py
index a3fd579c1..d55399df2 100644
--- a/xtuner/v1/rl/agent_loop/sandbox_agent_loop/sandbox.py
+++ b/xtuner/v1/rl/agent_loop/sandbox_agent_loop/sandbox.py
@@ -939,6 +939,8 @@ async def _acquire_ready(self, spec: SandboxSpec, *, record: StageRecord | None
                 record.metadata["sandbox_create_attempts"] = attempt
             try:
                 create_kwargs: dict[str, Any] = {}
+                if spec.cluster_name:
+                    create_kwargs["cluster_name"] = spec.cluster_name
                 if spec.key:
                     create_kwargs["key"] = spec.key
                 if spec.env_vars:
diff --git a/xtuner/v1/rl/agent_loop/sandbox_agent_loop/schemas.py b/xtuner/v1/rl/agent_loop/sandbox_agent_loop/schemas.py
index 6f6f62296..7065d290a 100644
--- a/xtuner/v1/rl/agent_loop/sandbox_agent_loop/schemas.py
+++ b/xtuner/v1/rl/agent_loop/sandbox_agent_loop/schemas.py
@@ -229,6 +229,7 @@ class SandboxSpec(BaseModel):
     env_vars: dict[str, str] = Field(default_factory=dict)
     resources: dict[str, Any] = Field(default_factory=dict)
     key: str | None = None
+    cluster_name: str | None = None
 
 
 class AgentSpec(BaseModel):