AdithyaSK HF Staff commited on
Commit
19f2fe7
·
verified ·
1 Parent(s): b02d6d5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. server/opencode_environment.py +11 -7
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
- if handle.session is not None:
321
- try:
322
  aborted = bool(handle.session.abort())
323
- except Exception as exc: # noqa: BLE001
324
- return json.dumps({
325
- "rollout_id": rollout_id, "aborted": False, "reason": str(exc),
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="serve",
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: