rl-rock 1.2.5 → 1.2.7

package/dist/index.mjs

@@ -1,13 +1,14 @@
 import path, { join, resolve, basename } from 'path';
 import { fileURLToPath } from 'url';
-import { existsSync, mkdirSync, readFileSync, statSync, mkdtempSync, rmSync, appendFileSync } from 'fs';
+import { existsSync, mkdirSync, readFileSync, statSync, mkdtempSync, rmSync } from 'fs';
 import { z } from 'zod';
-import axios, { AxiosError } from 'axios';
+import axios from 'axios';
 import https from 'https';
 import { objectToCamel as objectToCamel$1, objectToSnake as objectToSnake$1 } from 'ts-case-convert';
-import { tmpdir } from 'os';
 import winston from 'winston';
+import { tmpdir, homedir } from 'os';
 import { randomUUID } from 'crypto';
+import { readFile, appendFile } from 'fs/promises';
 
 // node_modules/.pnpm/tsup@8.5.1_typescript@5.9.3/node_modules/tsup/assets/esm_shims.js
 var getFilename = () => fileURLToPath(import.meta.url);
@@ -174,14 +175,6 @@ var RunMode = {
   NORMAL: "normal",
   NOHUP: "nohup"
 };
-var Constants = class {
-  static BASE_URL_PRODUCT = "";
-  static BASE_URL_ALIYUN = "";
-  static BASE_URL_INNER = "";
-  static BASE_URL_PRE = "";
-  static BASE_URL_LOCAL = "";
-  static REQUEST_TIMEOUT_SECONDS = 180;
-};
 var PID_PREFIX = "__ROCK_PID_START__";
 var PID_SUFFIX = "__ROCK_PID_END__";
 
@@ -259,6 +252,10 @@ var objectToCamel = objectToCamel$1;
 var objectToSnake = objectToSnake$1;
 
 // src/utils/http.ts
+var sharedHttpsAgent = new https.Agent({
+  rejectUnauthorized: true,
+  keepAlive: true
+});
 var HttpUtils = class {
   static defaultTimeout = 3e5;
   // 5 minutes
@@ -273,9 +270,7 @@ var HttpUtils = class {
         "Content-Type": "application/json",
         ...config?.headers
       },
-      httpsAgent: new https.Agent({
-        rejectUnauthorized: true
-      })
+      httpsAgent: sharedHttpsAgent
     });
   }
   /**
@@ -304,9 +299,6 @@ var HttpUtils = class {
       }
       return httpResponse;
     } catch (error) {
-      if (error instanceof AxiosError) {
-        throw new Error(`Failed to POST ${url}: ${error.message}`);
-      }
       throw error;
     }
   }
@@ -331,9 +323,6 @@ var HttpUtils = class {
       }
       return httpResponse;
     } catch (error) {
-      if (error instanceof AxiosError) {
-        throw new Error(`Failed to GET ${url}: ${error.message}`);
-      }
       throw error;
     }
   }
@@ -413,9 +402,6 @@ var HttpUtils = class {
       }
       return httpResponse;
     } catch (error) {
-      if (error instanceof AxiosError) {
-        throw new Error(`Failed to POST multipart ${url}: ${error.message}`);
-      }
       throw error;
     }
   }
@@ -499,43 +485,12 @@ function withRetry(fn, options = {}) {
   });
 }
 
-// src/utils/deprecated.ts
-function deprecated(reason = "") {
-  return function(target, propertyKey, descriptor) {
-    const originalMethod = descriptor.value;
-    descriptor.value = function(...args) {
-      console.warn(
-        `${String(propertyKey)} is deprecated. ${reason}`
-      );
-      return originalMethod.apply(this, args);
-    };
-    return descriptor;
-  };
-}
-function deprecatedClass(reason = "") {
-  return function(constructor) {
-    return class extends constructor {
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      constructor(...args) {
-        console.warn(`${constructor.name} is deprecated. ${reason}`);
-        super(...args);
-      }
-    };
-  };
-}
-
 // src/utils/system.ts
 function isNode() {
   return typeof process !== "undefined" && process.versions != null && process.versions.node != null;
 }
-function isBrowser() {
-  return typeof globalThis !== "undefined" && "window" in globalThis && typeof globalThis.window !== "undefined";
-}
 function getEnv(key, defaultValue) {
-  if (isNode()) {
-    return process.env[key] ?? defaultValue;
-  }
-  return defaultValue;
+  return process.env[key] ?? defaultValue;
 }
 function getRequiredEnv(key) {
   const value = getEnv(key);
@@ -545,10 +500,7 @@ function getRequiredEnv(key) {
   return value;
 }
 function isEnvSet(key) {
-  if (isNode()) {
-    return key in process.env;
-  }
-  return false;
+  return key in process.env;
 }
 var envVars = {
   // Logging
@@ -561,25 +513,117 @@ var envVars = {
   get ROCK_LOGGING_LEVEL() {
     return getEnv("ROCK_LOGGING_LEVEL", "INFO");
   },
+  // Service
+  get ROCK_SERVICE_STATUS_DIR() {
+    return getEnv("ROCK_SERVICE_STATUS_DIR", "/data/service_status");
+  },
+  get ROCK_SCHEDULER_STATUS_DIR() {
+    return getEnv("ROCK_SCHEDULER_STATUS_DIR", "/data/scheduler_status");
+  },
+  // Config
+  get ROCK_CONFIG() {
+    return getEnv("ROCK_CONFIG");
+  },
+  get ROCK_CONFIG_DIR_NAME() {
+    return getEnv("ROCK_CONFIG_DIR_NAME", "rock-conf");
+  },
   // Base URLs
   get ROCK_BASE_URL() {
     return getEnv("ROCK_BASE_URL", "http://localhost:8080");
   },
+  get ROCK_WORKER_ROCKLET_PORT() {
+    const val = getEnv("ROCK_WORKER_ROCKLET_PORT");
+    return val ? parseInt(val, 10) : void 0;
+  },
   get ROCK_SANDBOX_STARTUP_TIMEOUT_SECONDS() {
     return parseInt(getEnv("ROCK_SANDBOX_STARTUP_TIMEOUT_SECONDS", "180"), 10);
   },
+  get ROCK_CODE_SANDBOX_BASE_URL() {
+    return getEnv("ROCK_CODE_SANDBOX_BASE_URL", "");
+  },
   // EnvHub
   get ROCK_ENVHUB_BASE_URL() {
     return getEnv("ROCK_ENVHUB_BASE_URL", "http://localhost:8081");
   },
+  get ROCK_ENVHUB_DEFAULT_DOCKER_IMAGE() {
+    return getEnv("ROCK_ENVHUB_DEFAULT_DOCKER_IMAGE", "python:3.11");
+  },
+  get ROCK_ENVHUB_DB_URL() {
+    return getEnv(
+      "ROCK_ENVHUB_DB_URL",
+      `sqlite:///${join(homedir(), ".rock", "rock_envs.db")}`
+    );
+  },
+  // Auto clear
+  get ROCK_DEFAULT_AUTO_CLEAR_TIME_MINUTES() {
+    return parseInt(getEnv("ROCK_DEFAULT_AUTO_CLEAR_TIME_MINUTES", "360"), 10);
+  },
+  // Ray
+  get ROCK_RAY_NAMESPACE() {
+    return getEnv("ROCK_RAY_NAMESPACE", "xrl-sandbox");
+  },
+  get ROCK_SANDBOX_EXPIRE_TIME_KEY() {
+    return getEnv("ROCK_SANDBOX_EXPIRE_TIME_KEY", "expire_time");
+  },
+  get ROCK_SANDBOX_AUTO_CLEAR_TIME_KEY() {
+    return getEnv("ROCK_SANDBOX_AUTO_CLEAR_TIME_KEY", "auto_clear_time");
+  },
+  // Timezone
+  get ROCK_TIME_ZONE() {
+    return getEnv("ROCK_TIME_ZONE", "Asia/Shanghai");
+  },
+  // OSS
+  get ROCK_OSS_ENABLE() {
+    return getEnv("ROCK_OSS_ENABLE", "false")?.toLowerCase() === "true";
+  },
+  get ROCK_OSS_BUCKET_ENDPOINT() {
+    return getEnv("ROCK_OSS_BUCKET_ENDPOINT");
+  },
+  get ROCK_OSS_BUCKET_NAME() {
+    return getEnv("ROCK_OSS_BUCKET_NAME");
+  },
+  get ROCK_OSS_BUCKET_REGION() {
+    return getEnv("ROCK_OSS_BUCKET_REGION");
+  },
   // Pip
   get ROCK_PIP_INDEX_URL() {
-    return getEnv("ROCK_PIP_INDEX_URL", "https://mirrors.aliyun.com/pypi/simple/");
+    return getEnv("ROCK_PIP_INDEX_URL", "https://pypi.org/simple/");
+  },
+  // Monitor
+  get ROCK_MONITOR_ENABLE() {
+    return getEnv("ROCK_MONITOR_ENABLE", "false")?.toLowerCase() === "true";
+  },
+  // Project
+  get ROCK_PROJECT_ROOT() {
+    return getEnv("ROCK_PROJECT_ROOT", process.cwd());
+  },
+  get ROCK_WORKER_ENV_TYPE() {
+    return getEnv("ROCK_WORKER_ENV_TYPE", "local");
+  },
+  get ROCK_PYTHON_ENV_PATH() {
+    return getEnv("ROCK_PYTHON_ENV_PATH", process.cwd());
+  },
+  // Admin
+  get ROCK_ADMIN_ENV() {
+    return getEnv("ROCK_ADMIN_ENV", "dev");
+  },
+  get ROCK_ADMIN_ROLE() {
+    return getEnv("ROCK_ADMIN_ROLE", "write");
+  },
+  // CLI
+  get ROCK_CLI_LOAD_PATHS() {
+    return getEnv("ROCK_CLI_LOAD_PATHS", "");
+  },
+  get ROCK_CLI_DEFAULT_CONFIG_PATH() {
+    return getEnv("ROCK_CLI_DEFAULT_CONFIG_PATH", join(homedir(), ".rock", "config.ini"));
   },
   // Model Service
   get ROCK_MODEL_SERVICE_DATA_DIR() {
     return getEnv("ROCK_MODEL_SERVICE_DATA_DIR", "/data/logs");
   },
+  get ROCK_MODEL_SERVICE_TRAJ_APPEND_MODE() {
+    return getEnv("ROCK_MODEL_SERVICE_TRAJ_APPEND_MODE", "false")?.toLowerCase() === "true";
+  },
   // RuntimeEnv
   get ROCK_RTENV_PYTHON_V31114_INSTALL_CMD() {
     return getEnv(
@@ -608,11 +652,167 @@ var envVars = {
       return [];
     }
   },
+  get ROCK_AGENT_IFLOW_CLI_INSTALL_CMD() {
+    return getEnv("ROCK_AGENT_IFLOW_CLI_INSTALL_CMD", "npm i -g @iflow-ai/iflow-cli@latest");
+  },
   get ROCK_MODEL_SERVICE_INSTALL_CMD() {
     return getEnv("ROCK_MODEL_SERVICE_INSTALL_CMD", "pip install rl_rock[model-service]");
-  }};
+  },
+  // Doccuum
+  get ROCK_DOCUUM_INSTALL_URL() {
+    return getEnv(
+      "ROCK_DOCUUM_INSTALL_URL",
+      "https://raw.githubusercontent.com/stepchowfun/docuum/main/install.sh"
+    );
+  },
+  // ========== Sandbox Defaults ==========
+  // Sandbox configuration defaults - allow users to override via environment variables
+  get ROCK_DEFAULT_IMAGE() {
+    return getEnv("ROCK_DEFAULT_IMAGE", "python:3.11");
+  },
+  get ROCK_DEFAULT_MEMORY() {
+    return getEnv("ROCK_DEFAULT_MEMORY", "8g");
+  },
+  get ROCK_DEFAULT_CPUS() {
+    return parseFloat(getEnv("ROCK_DEFAULT_CPUS", "2"));
+  },
+  get ROCK_DEFAULT_CLUSTER() {
+    return getEnv("ROCK_DEFAULT_CLUSTER", "zb");
+  },
+  get ROCK_DEFAULT_AUTO_CLEAR_SECONDS() {
+    return parseInt(getEnv("ROCK_DEFAULT_AUTO_CLEAR_SECONDS", "300"), 10);
+  },
+  // ========== SandboxGroup Defaults ==========
+  get ROCK_DEFAULT_GROUP_SIZE() {
+    return parseInt(getEnv("ROCK_DEFAULT_GROUP_SIZE", "2"), 10);
+  },
+  get ROCK_DEFAULT_START_CONCURRENCY() {
+    return parseInt(getEnv("ROCK_DEFAULT_START_CONCURRENCY", "2"), 10);
+  },
+  get ROCK_DEFAULT_START_RETRY_TIMES() {
+    return parseInt(getEnv("ROCK_DEFAULT_START_RETRY_TIMES", "3"), 10);
+  },
+  // ========== Client Timeouts (in seconds) ==========
+  get ROCK_DEFAULT_ARUN_TIMEOUT() {
+    return parseInt(getEnv("ROCK_DEFAULT_ARUN_TIMEOUT", "300"), 10);
+  },
+  get ROCK_DEFAULT_NOHUP_WAIT_TIMEOUT() {
+    return parseInt(getEnv("ROCK_DEFAULT_NOHUP_WAIT_TIMEOUT", "300"), 10);
+  },
+  get ROCK_DEFAULT_NOHUP_WAIT_INTERVAL() {
+    return parseInt(getEnv("ROCK_DEFAULT_NOHUP_WAIT_INTERVAL", "10"), 10);
+  },
+  get ROCK_DEFAULT_STATUS_CHECK_INTERVAL() {
+    return parseInt(getEnv("ROCK_DEFAULT_STATUS_CHECK_INTERVAL", "3"), 10);
+  }
+};
+var levels = {
+  error: 0,
+  warn: 1,
+  info: 2,
+  http: 3,
+  debug: 4
+};
+var colors = {
+  error: "red",
+  warn: "yellow",
+  info: "green",
+  http: "magenta",
+  debug: "cyan"
+};
+winston.addColors(colors);
+var consoleFormat = winston.format.combine(
+  winston.format.timestamp({ format: "YYYY-MM-DD HH:mm:ss.SSS" }),
+  winston.format.colorize({ all: true }),
+  winston.format.printf((info) => {
+    const { timestamp, level, message, ...meta } = info;
+    const metaStr = Object.keys(meta).length ? JSON.stringify(meta) : "";
+    return `${timestamp} ${level}: ${message} ${metaStr}`;
+  })
+);
+var fileFormat = winston.format.combine(
+  winston.format.timestamp({ format: "YYYY-MM-DD HH:mm:ss.SSS" }),
+  winston.format.json()
+);
+function getLogLevel() {
+  const level = envVars.ROCK_LOGGING_LEVEL.toLowerCase();
+  if (level in levels) {
+    return level;
+  }
+  return "info";
+}
+var loggerContainer = new winston.Container();
+function initLogger(name = "rock", fileName) {
+  if (loggerContainer.has(name)) {
+    return loggerContainer.get(name);
+  }
+  const transports = [];
+  const logLevel = getLogLevel();
+  const logPath = envVars.ROCK_LOGGING_PATH;
+  const logFileName = envVars.ROCK_LOGGING_FILE_NAME;
+  if (logPath) {
+    if (!existsSync(logPath)) {
+      mkdirSync(logPath, { recursive: true });
+    }
+    transports.push(
+      new winston.transports.File({
+        filename: join(logPath, logFileName),
+        format: fileFormat,
+        level: logLevel
+      })
+    );
+  } else {
+    transports.push(
+      new winston.transports.Console({
+        format: consoleFormat,
+        level: logLevel
+      })
+    );
+  }
+  const logger13 = loggerContainer.add(name, {
+    levels,
+    defaultMeta: { service: name },
+    transports
+  });
+  return logger13;
+}
+initLogger("rock");
 
-// src/envhub/schema.ts
+// src/utils/deprecated.ts
+var warnedKeys = /* @__PURE__ */ new Set();
+function getLogger() {
+  return initLogger("rock.deprecated");
+}
+function warnOnce(key, message) {
+  if (warnedKeys.has(key)) {
+    return;
+  }
+  getLogger().warn(message);
+  warnedKeys.add(key);
+}
+function deprecated(reason = "") {
+  return function(target, propertyKey, descriptor) {
+    const originalMethod = descriptor.value;
+    const key = String(propertyKey);
+    descriptor.value = function(...args) {
+      warnOnce(key, `${key} is deprecated. ${reason}`);
+      return originalMethod.apply(this, args);
+    };
+    return descriptor;
+  };
+}
+function deprecatedClass(reason = "") {
+  return function(constructor) {
+    const key = constructor.name;
+    return class extends constructor {
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      constructor(...args) {
+        warnOnce(key, `${key} is deprecated. ${reason}`);
+        super(...args);
+      }
+    };
+  };
+}
 var EnvHubClientConfigSchema = z.object({
   baseUrl: z.string().default(envVars.ROCK_ENVHUB_BASE_URL)
 });
@@ -741,104 +941,51 @@ var EnvHubClient = class {
     }
   }
 };
-var levels = {
-  error: 0,
-  warn: 1,
-  info: 2,
-  http: 3,
-  debug: 4
-};
-var colors = {
-  error: "red",
-  warn: "yellow",
-  info: "green",
-  http: "magenta",
-  debug: "cyan"
-};
-winston.addColors(colors);
-var consoleFormat = winston.format.combine(
-  winston.format.timestamp({ format: "YYYY-MM-DD HH:mm:ss.SSS" }),
-  winston.format.colorize({ all: true }),
-  winston.format.printf((info) => {
-    const { timestamp, level, message, ...meta } = info;
-    const metaStr = Object.keys(meta).length ? JSON.stringify(meta) : "";
-    return `${timestamp} ${level}: ${message} ${metaStr}`;
-  })
-);
-var fileFormat = winston.format.combine(
-  winston.format.timestamp({ format: "YYYY-MM-DD HH:mm:ss.SSS" }),
-  winston.format.json()
-);
-function getLogLevel() {
-  const level = envVars.ROCK_LOGGING_LEVEL.toLowerCase();
-  if (level in levels) {
-    return level;
-  }
-  return "info";
-}
-var loggerCache = /* @__PURE__ */ new Map();
-function initLogger(name = "rock", fileName) {
-  if (loggerCache.has(name)) {
-    return loggerCache.get(name);
-  }
-  const transports = [];
-  const logLevel = getLogLevel();
-  const logPath = envVars.ROCK_LOGGING_PATH;
-  const logFileName = envVars.ROCK_LOGGING_FILE_NAME;
-  if (logPath) {
-    if (!existsSync(logPath)) {
-      mkdirSync(logPath, { recursive: true });
-    }
-    transports.push(
-      new winston.transports.File({
-        filename: join(logPath, logFileName),
-        format: fileFormat,
-        level: logLevel
-      })
-    );
-  } else {
-    transports.push(
-      new winston.transports.Console({
-        format: consoleFormat,
-        level: logLevel
-      })
-    );
-  }
-  const logger13 = winston.createLogger({
-    levels,
-    defaultMeta: { service: name },
-    transports
-  });
-  loggerCache.set(name, logger13);
-  return logger13;
-}
-initLogger("rock");
 
 // src/envs/rock_env.ts
 var logger = initLogger("rock.envs");
-var RockEnv = class {
+var RockEnv = class _RockEnv {
   envId;
   sandboxId = null;
   isClosed = false;
-  client;
   constructor(config) {
     this.envId = config.envId;
-    this.client = axios.create({
-      baseURL: envVars.ROCK_BASE_URL,
-      timeout: 3e5,
-      headers: { "Content-Type": "application/json" }
-    });
+  }
+  /**
+   * Create and initialize a RockEnv instance
+   *
+   * @param config - Environment configuration
+   * @returns Initialized RockEnv instance
+   */
+  static async create(config) {
+    const env = new _RockEnv(config);
     try {
-      this.initializeEnvironment();
+      await env.initializeEnvironment();
     } catch (e) {
       throw new Error(`Failed to initialize environment: ${e}`);
     }
+    return env;
+  }
+  /**
+   * Get the sandbox ID
+   */
+  getSandboxId() {
+    return this.sandboxId;
   }
   /**
    * Initialize environment instance
    */
-  initializeEnvironment() {
+  async initializeEnvironment() {
     logger.debug(`Initializing environment: ${this.envId}`);
+    const response = await HttpUtils.post(
+      `${envVars.ROCK_BASE_URL}/apis/v1/envs/gem/make`,
+      { "Content-Type": "application/json" },
+      { envId: this.envId }
+    );
+    this.sandboxId = response.result?.sandboxId ?? null;
+    if (!this.sandboxId) {
+      throw new Error("Failed to get environment instance ID");
+    }
   }
   /**
    * Execute an action step
@@ -848,19 +995,12 @@ var RockEnv = class {
    */
   async step(action) {
     this.ensureNotClosed();
-    const params = {
-      sandbox_id: this.sandboxId,
-      action
-    };
-    try {
-      const response = await this.client.post(
-        "/apis/v1/envs/gem/step",
-        params
-      );
-      return this.parseStepResult(response.data);
-    } catch (e) {
-      throw new Error(`Failed to execute step with action ${action}: ${e}`);
-    }
+    const response = await HttpUtils.post(
+      `${envVars.ROCK_BASE_URL}/apis/v1/envs/gem/step`,
+      { "Content-Type": "application/json" },
+      { sandboxId: this.sandboxId, action }
+    );
+    return this.parseStepResult(response.result);
   }
   /**
    * Reset environment to initial state
@@ -870,19 +1010,16 @@ var RockEnv = class {
    */
   async reset(seed) {
     this.ensureNotClosed();
-    const params = { sandbox_id: this.sandboxId };
+    const params = { sandboxId: this.sandboxId };
     if (seed !== void 0) {
       params.seed = seed;
     }
-    try {
-      const response = await this.client.post(
-        "/apis/v1/envs/gem/reset",
-        params
-      );
-      return this.parseResetResult(response.data);
-    } catch (e) {
-      throw new Error(`Failed to reset environment: ${e}`);
-    }
+    const response = await HttpUtils.post(
+      `${envVars.ROCK_BASE_URL}/apis/v1/envs/gem/reset`,
+      { "Content-Type": "application/json" },
+      params
+    );
+    return this.parseResetResult(response.result);
   }
   /**
    * Close environment and clean up resources
@@ -892,9 +1029,11 @@ var RockEnv = class {
       return;
     }
     try {
-      await this.client.post("/apis/v1/envs/gem/close", {
-        sandbox_id: this.sandboxId
-      });
+      await HttpUtils.post(
+        `${envVars.ROCK_BASE_URL}/apis/v1/envs/gem/close`,
+        { "Content-Type": "application/json" },
+        { sandboxId: this.sandboxId }
+      );
     } catch (e) {
       throw new Error(`Failed to close environment: ${e}`);
     } finally {
@@ -906,6 +1045,9 @@ var RockEnv = class {
    * Parse step result from API response
    */
   parseStepResult(data) {
+    if (!data) {
+      throw new Error("Invalid step result: no data");
+    }
     return [
       data.observation,
       data.reward,
@@ -918,6 +1060,9 @@ var RockEnv = class {
    * Parse reset result from API response
    */
   parseResetResult(data) {
+    if (!data) {
+      throw new Error("Invalid reset result: no data");
+    }
     return [data.observation, data.info];
   }
   /**
@@ -931,30 +1076,30 @@ var RockEnv = class {
 };
 
 // src/envs/registration.ts
-function make(envId, options) {
-  return new RockEnv({ envId, ...options });
+async function make(envId, options) {
+  return RockEnv.create({ envId, ...options });
 }
 var BaseConfigSchema = z.object({
-  baseUrl: z.string().default(envVars.ROCK_BASE_URL),
+  baseUrl: z.string().default(() => envVars.ROCK_BASE_URL),
   xrlAuthorization: z.string().optional(),
   extraHeaders: z.record(z.string()).default({})
 });
 var SandboxConfigSchema = BaseConfigSchema.extend({
-  image: z.string().default("python:3.11"),
-  autoClearSeconds: z.number().default(300),
+  image: z.string().default(() => envVars.ROCK_DEFAULT_IMAGE),
+  autoClearSeconds: z.number().default(() => envVars.ROCK_DEFAULT_AUTO_CLEAR_SECONDS),
   routeKey: z.string().optional(),
-  startupTimeout: z.number().default(envVars.ROCK_SANDBOX_STARTUP_TIMEOUT_SECONDS),
-  memory: z.string().default("8g"),
-  cpus: z.number().default(2),
+  startupTimeout: z.number().default(() => envVars.ROCK_SANDBOX_STARTUP_TIMEOUT_SECONDS),
+  memory: z.string().default(() => envVars.ROCK_DEFAULT_MEMORY),
+  cpus: z.number().default(() => envVars.ROCK_DEFAULT_CPUS),
   userId: z.string().optional(),
   experimentId: z.string().optional(),
-  cluster: z.string().default("zb"),
+  cluster: z.string().default(() => envVars.ROCK_DEFAULT_CLUSTER),
   namespace: z.string().optional()
 });
 var SandboxGroupConfigSchema = SandboxConfigSchema.extend({
-  size: z.number().default(2),
-  startConcurrency: z.number().default(2),
-  startRetryTimes: z.number().default(3)
+  size: z.number().default(() => envVars.ROCK_DEFAULT_GROUP_SIZE),
+  startConcurrency: z.number().default(() => envVars.ROCK_DEFAULT_START_CONCURRENCY),
+  startRetryTimes: z.number().default(() => envVars.ROCK_DEFAULT_START_RETRY_TIMES)
 });
 function createSandboxConfig(config) {
   return SandboxConfigSchema.parse(config ?? {});
@@ -1027,14 +1172,6 @@ var Deploy = class {
   }
 };
 
-// src/sandbox/types.ts
-var SpeedupType = /* @__PURE__ */ ((SpeedupType2) => {
-  SpeedupType2["APT"] = "apt";
-  SpeedupType2["PIP"] = "pip";
-  SpeedupType2["GITHUB"] = "github";
-  return SpeedupType2;
-})(SpeedupType || {});
-
 // src/utils/shell.ts
 function shellQuote(str) {
   if (str === "") {
@@ -1297,6 +1434,12 @@ var LinuxFileSystem = class extends FileSystem {
 
 // src/sandbox/network.ts
 var logger4 = initLogger("rock.sandbox.network");
+var SpeedupType = /* @__PURE__ */ ((SpeedupType2) => {
+  SpeedupType2["APT"] = "apt";
+  SpeedupType2["PIP"] = "pip";
+  SpeedupType2["GITHUB"] = "github";
+  return SpeedupType2;
+})(SpeedupType || {});
 var Network = class {
   sandbox;
   constructor(sandbox) {
@@ -1936,13 +2079,6 @@ var Sandbox = class extends AbstractSandbox {
     } = options;
     const sessionName = session ?? "default";
     if (mode === "normal") {
-      try {
-        await this.createSession({ session: sessionName, startupSource: [], envEnable: false });
-      } catch (e) {
-        if (String(e).includes("already exists")) ; else {
-          throw e;
-        }
-      }
       return this.runInSession({ command: cmd, session: sessionName, timeout });
     }
     return this.arunWithNohup(cmd, options);
@@ -1982,13 +2118,12 @@ var Sandbox = class extends AbstractSandbox {
       outputFile
     } = options;
     const timestamp = Date.now();
-    const tmpSession = session ?? `bash-${timestamp}`;
-    try {
+    let tmpSession;
+    if (session === void 0 || session === null) {
+      tmpSession = `bash-${timestamp}`;
       await this.createSession({ session: tmpSession, startupSource: [], envEnable: false });
-    } catch (e) {
-      if (!String(e).includes("already exists")) {
-        throw e;
-      }
+    } else {
+      tmpSession = session;
     }
     const tmpFile = outputFile ?? `/tmp/tmp_${timestamp}.out`;
     const nohupCommand = `nohup ${cmd} < /dev/null > ${tmpFile} 2>&1 & echo __ROCK_PID_START__$!__ROCK_PID_END__;disown`;
@@ -2091,11 +2226,13 @@ var Sandbox = class extends AbstractSandbox {
     const url = `${this.url}/upload`;
     const headers = this.buildHeaders();
     try {
-      const fs = await import('fs');
-      if (!fs.existsSync(sourcePath)) {
+      const fs = await import('fs/promises');
+      try {
+        await fs.access(sourcePath);
+      } catch {
         return { success: false, message: `File not found: ${sourcePath}` };
       }
-      const fileBuffer = fs.readFileSync(sourcePath);
+      const fileBuffer = await fs.readFile(sourcePath);
       const fileName = sourcePath.split("/").pop() ?? "file";
       const response = await HttpUtils.postMultipart(
         url,
@@ -2203,7 +2340,7 @@ var ModelClient = class {
     const content = this.constructResponse(lastResponse, index);
     const lastResponseLine = await this.readLastResponseLine();
     if (lastResponseLine === null) {
-      this.appendResponse(content);
+      await this.appendResponse(content);
       return;
     }
     const { meta } = this.parseResponseLine(lastResponseLine);
@@ -2215,7 +2352,7 @@ var ModelClient = class {
       logger9.debug(`response index ${index} already exists, skip`);
       return;
     }
-    this.appendResponse(content);
+    await this.appendResponse(content);
   }
   /**
    * Pop request from log file
@@ -2247,8 +2384,13 @@ var ModelClient = class {
         logger9.debug(`Last request is not the index ${index} we want, waiting...`);
         await sleep(1e3);
       } catch (e) {
-        if (e instanceof Error && e.message.includes("aborted")) {
-          throw e;
+        if (e instanceof Error) {
+          if (e.message.includes("aborted")) {
+            throw e;
+          }
+          if (e.message.includes("Invalid request line format")) {
+            throw e;
+          }
         }
         logger9.debug(`Error reading request: ${e}, waiting...`);
         await sleep(1e3);
@@ -2276,7 +2418,7 @@ var ModelClient = class {
         await sleep(1e3);
         continue;
       }
-      const content = readFileSync(this.logFile, "utf-8");
+      const content = await readFile(this.logFile, "utf-8");
       const lines = content.split("\n").filter((l) => l.trim());
       if (lines.length === 0) {
         logger9.debug(`Log file ${this.logFile} is empty, waiting for the first request...`);
@@ -2290,21 +2432,31 @@ var ModelClient = class {
     if (lineContent.includes(SESSION_END_MARKER)) {
       return { requestJson: SESSION_END_MARKER, meta: {} };
     }
-    const parts = lineContent.split(REQUEST_END_MARKER);
-    const metaJson = parts[1] ?? "";
-    const requestJson = parts[0]?.split(REQUEST_START_MARKER)[1] ?? "";
-    const meta = JSON.parse(metaJson);
-    return { requestJson, meta };
+    try {
+      const parts = lineContent.split(REQUEST_END_MARKER);
+      const metaJson = parts[1] ?? "";
+      const requestJson = parts[0]?.split(REQUEST_START_MARKER)[1] ?? "";
+      const meta = JSON.parse(metaJson);
+      return { requestJson, meta };
+    } catch (e) {
+      logger9.error(`Failed to parse request line: ${lineContent}, error: ${e}`);
+      throw new Error(`Invalid request line format: ${e}`);
+    }
   }
   parseResponseLine(lineContent) {
-    const parts = lineContent.split(RESPONSE_END_MARKER);
-    const metaJson = parts[1] ?? "";
-    const responseJson = parts[0]?.split(RESPONSE_START_MARKER)[1] ?? "";
-    const meta = JSON.parse(metaJson);
-    return { responseJson, meta };
+    try {
+      const parts = lineContent.split(RESPONSE_END_MARKER);
+      const metaJson = parts[1] ?? "";
+      const responseJson = parts[0]?.split(RESPONSE_START_MARKER)[1] ?? "";
+      const meta = JSON.parse(metaJson);
+      return { responseJson, meta };
+    } catch (e) {
+      logger9.error(`Failed to parse response line: ${lineContent}, error: ${e}`);
+      throw new Error(`Invalid response line format: ${e}`);
+    }
   }
   async readLastRequestLine() {
-    const content = readFileSync(this.logFile, "utf-8");
+    const content = await readFile(this.logFile, "utf-8");
     const lines = content.split("\n").filter((l) => l.trim());
     for (let i = lines.length - 1; i >= 0; i--) {
       const line = lines[i];
@@ -2315,7 +2467,7 @@ var ModelClient = class {
     throw new Error(`No request found in log file ${this.logFile}`);
   }
   async readLastResponseLine() {
-    const content = readFileSync(this.logFile, "utf-8");
+    const content = await readFile(this.logFile, "utf-8");
     const lines = content.split("\n").filter((l) => l.trim());
     for (let i = lines.length - 1; i >= 0; i--) {
       const line = lines[i];
@@ -2325,8 +2477,8 @@ var ModelClient = class {
     }
     return null;
   }
-  appendResponse(content) {
-    appendFileSync(this.logFile, content);
+  async appendResponse(content) {
+    await appendFile(this.logFile, content);
   }
   constructResponse(lastResponse, index) {
     const meta = {
@@ -3039,6 +3191,6 @@ function getVersion() {
 }
 var VERSION = getVersion();
 
-export { Agent, AgentBashCommandSchema, AgentConfigSchema, BadRequestRockError, BashActionSchema, ChmodRequestSchema, ChmodResponseSchema, ChownRequestSchema, ChownResponseSchema, CloseResponseSchema, CloseSessionRequestSchema, CloseSessionResponseSchema, Codes, CommandResponseSchema, CommandRockError, CommandSchema, Constants, CreateBashSessionRequestSchema, CreateSessionResponseSchema, DefaultAgent, DefaultAgentConfigSchema, Deploy, EnvHubClient, EnvHubClientConfigSchema, EnvHubError, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, ModelService, ModelServiceConfigSchema, NODE_DEFAULT_VERSION, Network, NodeRuntimeEnv, NodeRuntimeEnvConfigSchema, ObservationSchema, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, Process, PythonRuntimeEnv, PythonRuntimeEnvConfigSchema, ReadFileRequestSchema, ReadFileResponseSchema, ReasonPhrases, RockAgentConfigSchema, RockEnv, RockEnvInfoSchema, RockException, RunMode, RuntimeEnv, RuntimeEnvConfigSchema, Sandbox, SandboxConfigSchema, SandboxGroup, SandboxGroupConfigSchema, SandboxResponseSchema, SandboxStatusResponseSchema, SpeedupType, UploadRequestSchema, UploadResponseSchema, VERSION, WriteFileRequestSchema, WriteFileResponseSchema, arunWithRetry, createRockEnvInfo, createRuntimeEnvId, createSandboxConfig, createSandboxGroupConfig, deprecated, deprecatedClass, extractNohupPid as extractNohupPidFromSandbox, fromRockException, getDefaultPipIndexUrl, getEnv, getReasonPhrase, getRequiredEnv, isBrowser, isClientError, isCommandError, isEnvSet, isError, isNode, isServerError, isSuccess, make, raiseForCode, retryAsync, sleep, withRetry, withTimeLogging };
+export { Agent, AgentBashCommandSchema, AgentConfigSchema, BadRequestRockError, BashActionSchema, ChmodRequestSchema, ChmodResponseSchema, ChownRequestSchema, ChownResponseSchema, CloseResponseSchema, CloseSessionRequestSchema, CloseSessionResponseSchema, Codes, CommandResponseSchema, CommandRockError, CommandSchema, CreateBashSessionRequestSchema, CreateSessionResponseSchema, DefaultAgent, DefaultAgentConfigSchema, Deploy, EnvHubClient, EnvHubClientConfigSchema, EnvHubError, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, ModelService, ModelServiceConfigSchema, NODE_DEFAULT_VERSION, Network, NodeRuntimeEnv, NodeRuntimeEnvConfigSchema, ObservationSchema, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, Process, PythonRuntimeEnv, PythonRuntimeEnvConfigSchema, ReadFileRequestSchema, ReadFileResponseSchema, ReasonPhrases, RockAgentConfigSchema, RockEnv, RockEnvInfoSchema, RockException, RunMode, RuntimeEnv, RuntimeEnvConfigSchema, Sandbox, SandboxConfigSchema, SandboxGroup, SandboxGroupConfigSchema, SandboxResponseSchema, SandboxStatusResponseSchema, SpeedupType, UploadRequestSchema, UploadResponseSchema, VERSION, WriteFileRequestSchema, WriteFileResponseSchema, arunWithRetry, createRockEnvInfo, createRuntimeEnvId, createSandboxConfig, createSandboxGroupConfig, deprecated, deprecatedClass, extractNohupPid as extractNohupPidFromSandbox, fromRockException, getDefaultPipIndexUrl, getEnv, getReasonPhrase, getRequiredEnv, isClientError, isCommandError, isEnvSet, isError, isNode, isServerError, isSuccess, make, raiseForCode, retryAsync, sleep, withRetry, withTimeLogging };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map