Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files
server/opencode_environment.py
CHANGED
|
@@ -317,13 +317,17 @@ class OpenCodeEnvironment(MCPEnvironment):
|
|
| 317 |
if handle is None:
|
| 318 |
return json.dumps({"rollout_id": rollout_id, "aborted": False, "reason": "unknown"})
|
| 319 |
aborted = False
|
| 320 |
-
|
| 321 |
-
|
| 322 |
aborted = bool(handle.session.abort())
|
| 323 |
-
|
| 324 |
-
|
| 325 |
-
|
| 326 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 327 |
return json.dumps({"rollout_id": rollout_id, "aborted": aborted})
|
| 328 |
|
| 329 |
@mcp.tool
|
|
@@ -598,7 +602,7 @@ class OpenCodeEnvironment(MCPEnvironment):
|
|
| 598 |
sandbox_backend=self._E2BSandboxBackend(),
|
| 599 |
mode=mode,
|
| 600 |
verifier=None,
|
| 601 |
-
driver="
|
| 602 |
)
|
| 603 |
handle.session = factory.create(task=task)
|
| 604 |
try:
|
|
|
|
| 317 |
if handle is None:
|
| 318 |
return json.dumps({"rollout_id": rollout_id, "aborted": False, "reason": "unknown"})
|
| 319 |
aborted = False
|
| 320 |
+
try:
|
| 321 |
+
if handle.session is not None and handle.session.driver == "serve":
|
| 322 |
aborted = bool(handle.session.abort())
|
| 323 |
+
elif handle.session is not None:
|
| 324 |
+
# CLI driver: kill the sandbox to stop opencode mid-run.
|
| 325 |
+
handle.session.sandbox.kill()
|
| 326 |
+
aborted = True
|
| 327 |
+
except Exception as exc: # noqa: BLE001
|
| 328 |
+
return json.dumps({
|
| 329 |
+
"rollout_id": rollout_id, "aborted": False, "reason": str(exc),
|
| 330 |
+
})
|
| 331 |
return json.dumps({"rollout_id": rollout_id, "aborted": aborted})
|
| 332 |
|
| 333 |
@mcp.tool
|
|
|
|
| 602 |
sandbox_backend=self._E2BSandboxBackend(),
|
| 603 |
mode=mode,
|
| 604 |
verifier=None,
|
| 605 |
+
driver="cli", # serve is a state store, not an agent runner
|
| 606 |
)
|
| 607 |
handle.session = factory.create(task=task)
|
| 608 |
try:
|