rl-rock 1.3.8 → 1.10.0-beta.0

package/dist/index.d.mts

@@ -159,6 +159,14 @@ type ReadFileRequest = z.infer<typeof ReadFileRequestSchema>;
  */
 declare const UploadModeSchema: z.ZodEnum<["auto", "direct", "oss"]>;
 type UploadMode = z.infer<typeof UploadModeSchema>;
+/**
+ * Download mode enum
+ * - auto: Automatically choose download method based on file size and OSS availability
+ * - direct: Force direct HTTP download via readFile API
+ * - oss: Force OSS download
+ */
+declare const DownloadModeSchema: z.ZodEnum<["auto", "direct", "oss"]>;
+type DownloadMode = z.infer<typeof DownloadModeSchema>;
 /**
  * Upload file request
  * Note: uploadMode defaults to 'auto' in the implementation, not in the schema
@@ -222,6 +230,43 @@ declare const ChmodRequestSchema: z.ZodObject<{
     mode?: string | undefined;
 }>;
 type ChmodRequest = z.infer<typeof ChmodRequestSchema>;
+/**
+ * Progress phase for upload operations
+ * - upload-to-oss: Uploading from local to OSS
+ * - download-to-sandbox: Downloading from OSS to sandbox (via wget)
+ */
+type UploadPhase = 'upload-to-oss' | 'download-to-sandbox';
+/**
+ * Progress phase for download operations
+ * - upload-to-oss-from-sandbox: Uploading from sandbox to OSS (via ossutil)
+ * - download-to-local: Downloading from OSS to local
+ */
+type DownloadPhase = 'upload-to-oss-from-sandbox' | 'download-to-local';
+/**
+ * Progress information callback
+ * @param phase - Current phase of the transfer
+ * @param percent - Progress percentage (0-100), or -1 if not available
+ */
+interface ProgressInfo {
+    phase: UploadPhase | DownloadPhase;
+    percent: number;
+}
+/**
+ * Upload options
+ */
+interface UploadOptions {
+    uploadMode?: UploadMode;
+    timeout?: number;
+    onProgress?: (info: ProgressInfo) => void;
+}
+/**
+ * Download options
+ */
+interface DownloadOptions {
+    downloadMode?: DownloadMode;
+    timeout?: number;
+    onProgress?: (info: ProgressInfo) => void;
+}
 
 /**
  * Response types
@@ -938,6 +983,7 @@ declare const SandboxConfigSchema: z.ZodObject<{
     experimentId: z.ZodOptional<z.ZodString>;
     cluster: z.ZodDefault<z.ZodString>;
     namespace: z.ZodOptional<z.ZodString>;
+    autoDeleteSeconds: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
 }, "strip", z.ZodTypeAny, {
     image: string;
     cpus: number;
@@ -952,6 +998,7 @@ declare const SandboxConfigSchema: z.ZodObject<{
     namespace?: string | undefined;
     xrlAuthorization?: string | undefined;
     routeKey?: string | undefined;
+    autoDeleteSeconds?: number | null | undefined;
 }, {
     image?: string | undefined;
     userId?: string | undefined;
@@ -966,6 +1013,7 @@ declare const SandboxConfigSchema: z.ZodObject<{
     autoClearSeconds?: number | undefined;
     routeKey?: string | undefined;
     startupTimeout?: number | undefined;
+    autoDeleteSeconds?: number | null | undefined;
 }>;
 type SandboxConfig = z.infer<typeof SandboxConfigSchema>;
 /**
@@ -986,6 +1034,7 @@ declare const SandboxGroupConfigSchema: z.ZodObject<{
     experimentId: z.ZodOptional<z.ZodString>;
     cluster: z.ZodDefault<z.ZodString>;
     namespace: z.ZodOptional<z.ZodString>;
+    autoDeleteSeconds: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
 } & {
     size: z.ZodDefault<z.ZodNumber>;
     startConcurrency: z.ZodDefault<z.ZodNumber>;
@@ -1007,6 +1056,7 @@ declare const SandboxGroupConfigSchema: z.ZodObject<{
     namespace?: string | undefined;
     xrlAuthorization?: string | undefined;
     routeKey?: string | undefined;
+    autoDeleteSeconds?: number | null | undefined;
 }, {
     image?: string | undefined;
     userId?: string | undefined;
@@ -1021,6 +1071,7 @@ declare const SandboxGroupConfigSchema: z.ZodObject<{
     autoClearSeconds?: number | undefined;
     routeKey?: string | undefined;
     startupTimeout?: number | undefined;
+    autoDeleteSeconds?: number | null | undefined;
     size?: number | undefined;
     startConcurrency?: number | undefined;
     startRetryTimes?: number | undefined;
@@ -1267,7 +1318,9 @@ declare class Sandbox extends AbstractSandbox {
     start(): Promise<void>;
     stop(): Promise<void>;
     isAlive(): Promise<IsAliveResponse>;
-    getStatus(): Promise<SandboxStatusResponse>;
+    getStatus(options?: {
+        includeAllStates?: boolean;
+    }): Promise<SandboxStatusResponse>;
     execute(command: Command): Promise<CommandResponse>;
     createSession(request: CreateBashSessionRequest): Promise<CreateSessionResponse>;
     closeSession(request: CloseSessionRequest): Promise<CloseSessionResponse>;
@@ -1287,7 +1340,7 @@ declare class Sandbox extends AbstractSandbox {
     writeFile(request: WriteFileRequest): Promise<WriteFileResponse>;
     readFile(request: ReadFileRequest): Promise<ReadFileResponse>;
     upload(request: UploadRequest): Promise<UploadResponse>;
-    uploadByPath(sourcePath: string, targetPath: string, uploadMode?: UploadMode, timeout?: number): Promise<UploadResponse>;
+    uploadByPath(sourcePath: string, targetPath: string, options?: UploadOptions): Promise<UploadResponse>;
     /**
      * Get OSS STS credentials from sandbox
      */
@@ -1297,9 +1350,21 @@ declare class Sandbox extends AbstractSandbox {
      */
     isTokenExpired(): boolean;
     /**
-     * Download file from sandbox via OSS
+     * Download file from sandbox
+     * @param remotePath - File path in sandbox
+     * @param localPath - Local file path
+     * @param downloadMode - Download mode: 'auto' (default), 'direct', or 'oss'
+     * @param timeout - Optional timeout in milliseconds for OSS mode
+     */
+    downloadFile(remotePath: string, localPath: string, options?: DownloadOptions): Promise<DownloadFileResponse>;
+    /**
+     * Download file directly via readFile API
+     */
+    private downloadDirect;
+    /**
+     * Download file via OSS as intermediary
      */
-    downloadFile(remotePath: string, localPath: string, timeout?: number): Promise<DownloadFileResponse>;
+    private downloadViaOss;
     /**
      * Upload file via OSS (internal method)
      * @param timeout - Optional timeout in milliseconds
@@ -1310,6 +1375,9 @@ declare class Sandbox extends AbstractSandbox {
      * @param timeout - Optional timeout in milliseconds (defaults to ROCK_OSS_TIMEOUT env var or 300000ms)
      */
     private setupOss;
+    restart(): Promise<void>;
+    private waitForAlive;
+    private parseErrorMessageFromStatus;
     close(): Promise<void>;
     toString(): string;
 }
@@ -2427,4 +2495,4 @@ declare class DefaultAgent extends Agent {
  */
 declare const VERSION: string;
 
-export { Agent, type AgentBashCommand, AgentBashCommandSchema, type AgentConfig, AgentConfigSchema, BadRequestRockError, type BaseConfig, type BashAction, BashActionSchema, type ChmodRequest, ChmodRequestSchema, type ChmodResponse, ChmodResponseSchema, type ChownRequest, ChownRequestSchema, type ChownResponse, ChownResponseSchema, type CloseResponse, CloseResponseSchema, type CloseSessionRequest, CloseSessionRequestSchema, type CloseSessionResponse, CloseSessionResponseSchema, Codes, type Command, type CommandResponse, CommandResponseSchema, CommandRockError, CommandSchema, type CreateBashSessionRequest, CreateBashSessionRequestSchema, type CreateSessionResponse, CreateSessionResponseSchema, DefaultAgent, type DefaultAgentConfig, DefaultAgentConfigSchema, Deploy, type DownloadFileResponse, DownloadFileResponseSchema, EnvHubClient, type EnvHubClientConfig, EnvHubClientConfigSchema, EnvHubError, type ExecuteBashSessionResponse, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, type IsAliveResponse, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, type ModelClientConfig, ModelService, type ModelServiceConfig, ModelServiceConfigSchema, NODE_DEFAULT_VERSION, Network, NodeRuntimeEnv, type NodeRuntimeEnvConfig, NodeRuntimeEnvConfigSchema, type Observation, ObservationSchema, type OssCredentials, OssCredentialsSchema, type OssSetupResponse, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, type PollOptions, Process, PythonRuntimeEnv, type PythonRuntimeEnvConfig, PythonRuntimeEnvConfigSchema, type ReadFileRequest, ReadFileRequestSchema, type ReadFileResponse, ReadFileResponseSchema, ReasonPhrases, type ResetResult, type RockAgentConfig, RockAgentConfigSchema, RockEnv, type RockEnvConfig, type RockEnvInfo, RockEnvInfoSchema, RockException, RunMode, type RunModeType, RuntimeEnv, type RuntimeEnvConfig, RuntimeEnvConfigSchema, type RuntimeEnvId, Sandbox, type SandboxConfig, SandboxConfigSchema, SandboxGroup, type SandboxGroupConfig, SandboxGroupConfigSchema, type SandboxLike, type SandboxResponse, SandboxResponseSchema, type RunModeType as SandboxRunModeType, type SandboxStatusResponse, SandboxStatusResponseSchema, SpeedupType, type StepResult, type UploadMode, UploadModeSchema, type UploadRequest, UploadRequestSchema, type UploadResponse, UploadResponseSchema, VERSION, type WriteFileRequest, WriteFileRequestSchema, type WriteFileResponse, 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 };
+export { Agent, type AgentBashCommand, AgentBashCommandSchema, type AgentConfig, AgentConfigSchema, BadRequestRockError, type BaseConfig, type BashAction, BashActionSchema, type ChmodRequest, ChmodRequestSchema, type ChmodResponse, ChmodResponseSchema, type ChownRequest, ChownRequestSchema, type ChownResponse, ChownResponseSchema, type CloseResponse, CloseResponseSchema, type CloseSessionRequest, CloseSessionRequestSchema, type CloseSessionResponse, CloseSessionResponseSchema, Codes, type Command, type CommandResponse, CommandResponseSchema, CommandRockError, CommandSchema, type CreateBashSessionRequest, CreateBashSessionRequestSchema, type CreateSessionResponse, CreateSessionResponseSchema, DefaultAgent, type DefaultAgentConfig, DefaultAgentConfigSchema, Deploy, type DownloadFileResponse, DownloadFileResponseSchema, type DownloadMode, DownloadModeSchema, type DownloadOptions, type DownloadPhase, EnvHubClient, type EnvHubClientConfig, EnvHubClientConfigSchema, EnvHubError, type ExecuteBashSessionResponse, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, type IsAliveResponse, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, type ModelClientConfig, ModelService, type ModelServiceConfig, ModelServiceConfigSchema, NODE_DEFAULT_VERSION, Network, NodeRuntimeEnv, type NodeRuntimeEnvConfig, NodeRuntimeEnvConfigSchema, type Observation, ObservationSchema, type OssCredentials, OssCredentialsSchema, type OssSetupResponse, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, type PollOptions, Process, type ProgressInfo, PythonRuntimeEnv, type PythonRuntimeEnvConfig, PythonRuntimeEnvConfigSchema, type ReadFileRequest, ReadFileRequestSchema, type ReadFileResponse, ReadFileResponseSchema, ReasonPhrases, type ResetResult, type RockAgentConfig, RockAgentConfigSchema, RockEnv, type RockEnvConfig, type RockEnvInfo, RockEnvInfoSchema, RockException, RunMode, type RunModeType, RuntimeEnv, type RuntimeEnvConfig, RuntimeEnvConfigSchema, type RuntimeEnvId, Sandbox, type SandboxConfig, SandboxConfigSchema, SandboxGroup, type SandboxGroupConfig, SandboxGroupConfigSchema, type SandboxLike, type SandboxResponse, SandboxResponseSchema, type RunModeType as SandboxRunModeType, type SandboxStatusResponse, SandboxStatusResponseSchema, SpeedupType, type StepResult, type UploadMode, UploadModeSchema, type UploadOptions, type UploadPhase, type UploadRequest, UploadRequestSchema, type UploadResponse, UploadResponseSchema, VERSION, type WriteFileRequest, WriteFileRequestSchema, type WriteFileResponse, 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 };

package/dist/index.d.ts

@@ -159,6 +159,14 @@ type ReadFileRequest = z.infer<typeof ReadFileRequestSchema>;
  */
 declare const UploadModeSchema: z.ZodEnum<["auto", "direct", "oss"]>;
 type UploadMode = z.infer<typeof UploadModeSchema>;
+/**
+ * Download mode enum
+ * - auto: Automatically choose download method based on file size and OSS availability
+ * - direct: Force direct HTTP download via readFile API
+ * - oss: Force OSS download
+ */
+declare const DownloadModeSchema: z.ZodEnum<["auto", "direct", "oss"]>;
+type DownloadMode = z.infer<typeof DownloadModeSchema>;
 /**
  * Upload file request
  * Note: uploadMode defaults to 'auto' in the implementation, not in the schema
@@ -222,6 +230,43 @@ declare const ChmodRequestSchema: z.ZodObject<{
     mode?: string | undefined;
 }>;
 type ChmodRequest = z.infer<typeof ChmodRequestSchema>;
+/**
+ * Progress phase for upload operations
+ * - upload-to-oss: Uploading from local to OSS
+ * - download-to-sandbox: Downloading from OSS to sandbox (via wget)
+ */
+type UploadPhase = 'upload-to-oss' | 'download-to-sandbox';
+/**
+ * Progress phase for download operations
+ * - upload-to-oss-from-sandbox: Uploading from sandbox to OSS (via ossutil)
+ * - download-to-local: Downloading from OSS to local
+ */
+type DownloadPhase = 'upload-to-oss-from-sandbox' | 'download-to-local';
+/**
+ * Progress information callback
+ * @param phase - Current phase of the transfer
+ * @param percent - Progress percentage (0-100), or -1 if not available
+ */
+interface ProgressInfo {
+    phase: UploadPhase | DownloadPhase;
+    percent: number;
+}
+/**
+ * Upload options
+ */
+interface UploadOptions {
+    uploadMode?: UploadMode;
+    timeout?: number;
+    onProgress?: (info: ProgressInfo) => void;
+}
+/**
+ * Download options
+ */
+interface DownloadOptions {
+    downloadMode?: DownloadMode;
+    timeout?: number;
+    onProgress?: (info: ProgressInfo) => void;
+}
 
 /**
  * Response types
@@ -938,6 +983,7 @@ declare const SandboxConfigSchema: z.ZodObject<{
     experimentId: z.ZodOptional<z.ZodString>;
     cluster: z.ZodDefault<z.ZodString>;
     namespace: z.ZodOptional<z.ZodString>;
+    autoDeleteSeconds: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
 }, "strip", z.ZodTypeAny, {
     image: string;
     cpus: number;
@@ -952,6 +998,7 @@ declare const SandboxConfigSchema: z.ZodObject<{
     namespace?: string | undefined;
     xrlAuthorization?: string | undefined;
     routeKey?: string | undefined;
+    autoDeleteSeconds?: number | null | undefined;
 }, {
     image?: string | undefined;
     userId?: string | undefined;
@@ -966,6 +1013,7 @@ declare const SandboxConfigSchema: z.ZodObject<{
     autoClearSeconds?: number | undefined;
     routeKey?: string | undefined;
     startupTimeout?: number | undefined;
+    autoDeleteSeconds?: number | null | undefined;
 }>;
 type SandboxConfig = z.infer<typeof SandboxConfigSchema>;
 /**
@@ -986,6 +1034,7 @@ declare const SandboxGroupConfigSchema: z.ZodObject<{
     experimentId: z.ZodOptional<z.ZodString>;
     cluster: z.ZodDefault<z.ZodString>;
     namespace: z.ZodOptional<z.ZodString>;
+    autoDeleteSeconds: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
 } & {
     size: z.ZodDefault<z.ZodNumber>;
     startConcurrency: z.ZodDefault<z.ZodNumber>;
@@ -1007,6 +1056,7 @@ declare const SandboxGroupConfigSchema: z.ZodObject<{
     namespace?: string | undefined;
     xrlAuthorization?: string | undefined;
     routeKey?: string | undefined;
+    autoDeleteSeconds?: number | null | undefined;
 }, {
     image?: string | undefined;
     userId?: string | undefined;
@@ -1021,6 +1071,7 @@ declare const SandboxGroupConfigSchema: z.ZodObject<{
     autoClearSeconds?: number | undefined;
     routeKey?: string | undefined;
     startupTimeout?: number | undefined;
+    autoDeleteSeconds?: number | null | undefined;
     size?: number | undefined;
     startConcurrency?: number | undefined;
     startRetryTimes?: number | undefined;
@@ -1267,7 +1318,9 @@ declare class Sandbox extends AbstractSandbox {
     start(): Promise<void>;
     stop(): Promise<void>;
     isAlive(): Promise<IsAliveResponse>;
-    getStatus(): Promise<SandboxStatusResponse>;
+    getStatus(options?: {
+        includeAllStates?: boolean;
+    }): Promise<SandboxStatusResponse>;
     execute(command: Command): Promise<CommandResponse>;
     createSession(request: CreateBashSessionRequest): Promise<CreateSessionResponse>;
     closeSession(request: CloseSessionRequest): Promise<CloseSessionResponse>;
@@ -1287,7 +1340,7 @@ declare class Sandbox extends AbstractSandbox {
     writeFile(request: WriteFileRequest): Promise<WriteFileResponse>;
     readFile(request: ReadFileRequest): Promise<ReadFileResponse>;
     upload(request: UploadRequest): Promise<UploadResponse>;
-    uploadByPath(sourcePath: string, targetPath: string, uploadMode?: UploadMode, timeout?: number): Promise<UploadResponse>;
+    uploadByPath(sourcePath: string, targetPath: string, options?: UploadOptions): Promise<UploadResponse>;
     /**
      * Get OSS STS credentials from sandbox
      */
@@ -1297,9 +1350,21 @@ declare class Sandbox extends AbstractSandbox {
      */
     isTokenExpired(): boolean;
     /**
-     * Download file from sandbox via OSS
+     * Download file from sandbox
+     * @param remotePath - File path in sandbox
+     * @param localPath - Local file path
+     * @param downloadMode - Download mode: 'auto' (default), 'direct', or 'oss'
+     * @param timeout - Optional timeout in milliseconds for OSS mode
+     */
+    downloadFile(remotePath: string, localPath: string, options?: DownloadOptions): Promise<DownloadFileResponse>;
+    /**
+     * Download file directly via readFile API
+     */
+    private downloadDirect;
+    /**
+     * Download file via OSS as intermediary
      */
-    downloadFile(remotePath: string, localPath: string, timeout?: number): Promise<DownloadFileResponse>;
+    private downloadViaOss;
     /**
      * Upload file via OSS (internal method)
      * @param timeout - Optional timeout in milliseconds
@@ -1310,6 +1375,9 @@ declare class Sandbox extends AbstractSandbox {
      * @param timeout - Optional timeout in milliseconds (defaults to ROCK_OSS_TIMEOUT env var or 300000ms)
      */
     private setupOss;
+    restart(): Promise<void>;
+    private waitForAlive;
+    private parseErrorMessageFromStatus;
     close(): Promise<void>;
     toString(): string;
 }
@@ -2427,4 +2495,4 @@ declare class DefaultAgent extends Agent {
  */
 declare const VERSION: string;
 
-export { Agent, type AgentBashCommand, AgentBashCommandSchema, type AgentConfig, AgentConfigSchema, BadRequestRockError, type BaseConfig, type BashAction, BashActionSchema, type ChmodRequest, ChmodRequestSchema, type ChmodResponse, ChmodResponseSchema, type ChownRequest, ChownRequestSchema, type ChownResponse, ChownResponseSchema, type CloseResponse, CloseResponseSchema, type CloseSessionRequest, CloseSessionRequestSchema, type CloseSessionResponse, CloseSessionResponseSchema, Codes, type Command, type CommandResponse, CommandResponseSchema, CommandRockError, CommandSchema, type CreateBashSessionRequest, CreateBashSessionRequestSchema, type CreateSessionResponse, CreateSessionResponseSchema, DefaultAgent, type DefaultAgentConfig, DefaultAgentConfigSchema, Deploy, type DownloadFileResponse, DownloadFileResponseSchema, EnvHubClient, type EnvHubClientConfig, EnvHubClientConfigSchema, EnvHubError, type ExecuteBashSessionResponse, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, type IsAliveResponse, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, type ModelClientConfig, ModelService, type ModelServiceConfig, ModelServiceConfigSchema, NODE_DEFAULT_VERSION, Network, NodeRuntimeEnv, type NodeRuntimeEnvConfig, NodeRuntimeEnvConfigSchema, type Observation, ObservationSchema, type OssCredentials, OssCredentialsSchema, type OssSetupResponse, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, type PollOptions, Process, PythonRuntimeEnv, type PythonRuntimeEnvConfig, PythonRuntimeEnvConfigSchema, type ReadFileRequest, ReadFileRequestSchema, type ReadFileResponse, ReadFileResponseSchema, ReasonPhrases, type ResetResult, type RockAgentConfig, RockAgentConfigSchema, RockEnv, type RockEnvConfig, type RockEnvInfo, RockEnvInfoSchema, RockException, RunMode, type RunModeType, RuntimeEnv, type RuntimeEnvConfig, RuntimeEnvConfigSchema, type RuntimeEnvId, Sandbox, type SandboxConfig, SandboxConfigSchema, SandboxGroup, type SandboxGroupConfig, SandboxGroupConfigSchema, type SandboxLike, type SandboxResponse, SandboxResponseSchema, type RunModeType as SandboxRunModeType, type SandboxStatusResponse, SandboxStatusResponseSchema, SpeedupType, type StepResult, type UploadMode, UploadModeSchema, type UploadRequest, UploadRequestSchema, type UploadResponse, UploadResponseSchema, VERSION, type WriteFileRequest, WriteFileRequestSchema, type WriteFileResponse, 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 };
+export { Agent, type AgentBashCommand, AgentBashCommandSchema, type AgentConfig, AgentConfigSchema, BadRequestRockError, type BaseConfig, type BashAction, BashActionSchema, type ChmodRequest, ChmodRequestSchema, type ChmodResponse, ChmodResponseSchema, type ChownRequest, ChownRequestSchema, type ChownResponse, ChownResponseSchema, type CloseResponse, CloseResponseSchema, type CloseSessionRequest, CloseSessionRequestSchema, type CloseSessionResponse, CloseSessionResponseSchema, Codes, type Command, type CommandResponse, CommandResponseSchema, CommandRockError, CommandSchema, type CreateBashSessionRequest, CreateBashSessionRequestSchema, type CreateSessionResponse, CreateSessionResponseSchema, DefaultAgent, type DefaultAgentConfig, DefaultAgentConfigSchema, Deploy, type DownloadFileResponse, DownloadFileResponseSchema, type DownloadMode, DownloadModeSchema, type DownloadOptions, type DownloadPhase, EnvHubClient, type EnvHubClientConfig, EnvHubClientConfigSchema, EnvHubError, type ExecuteBashSessionResponse, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, type IsAliveResponse, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, type ModelClientConfig, ModelService, type ModelServiceConfig, ModelServiceConfigSchema, NODE_DEFAULT_VERSION, Network, NodeRuntimeEnv, type NodeRuntimeEnvConfig, NodeRuntimeEnvConfigSchema, type Observation, ObservationSchema, type OssCredentials, OssCredentialsSchema, type OssSetupResponse, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, type PollOptions, Process, type ProgressInfo, PythonRuntimeEnv, type PythonRuntimeEnvConfig, PythonRuntimeEnvConfigSchema, type ReadFileRequest, ReadFileRequestSchema, type ReadFileResponse, ReadFileResponseSchema, ReasonPhrases, type ResetResult, type RockAgentConfig, RockAgentConfigSchema, RockEnv, type RockEnvConfig, type RockEnvInfo, RockEnvInfoSchema, RockException, RunMode, type RunModeType, RuntimeEnv, type RuntimeEnvConfig, RuntimeEnvConfigSchema, type RuntimeEnvId, Sandbox, type SandboxConfig, SandboxConfigSchema, SandboxGroup, type SandboxGroupConfig, SandboxGroupConfigSchema, type SandboxLike, type SandboxResponse, SandboxResponseSchema, type RunModeType as SandboxRunModeType, type SandboxStatusResponse, SandboxStatusResponseSchema, SpeedupType, type StepResult, type UploadMode, UploadModeSchema, type UploadOptions, type UploadPhase, type UploadRequest, UploadRequestSchema, type UploadResponse, UploadResponseSchema, VERSION, type WriteFileRequest, WriteFileRequestSchema, type WriteFileResponse, 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 };

package/dist/index.js

@@ -80,6 +80,7 @@ var ReadFileRequestSchema = zod.z.object({
   errors: zod.z.string().optional()
 });
 var UploadModeSchema = zod.z.enum(["auto", "direct", "oss"]);
+var DownloadModeSchema = zod.z.enum(["auto", "direct", "oss"]);
 var UploadRequestSchema = zod.z.object({
   sourcePath: zod.z.string(),
   targetPath: zod.z.string(),
@@ -1113,7 +1114,8 @@ var SandboxConfigSchema = BaseConfigSchema.extend({
   userId: zod.z.string().optional(),
   experimentId: zod.z.string().optional(),
   cluster: zod.z.string().default(() => envVars.ROCK_DEFAULT_CLUSTER),
-  namespace: zod.z.string().optional()
+  namespace: zod.z.string().optional(),
+  autoDeleteSeconds: zod.z.number().min(0, "autoDeleteSeconds must be >= 0").nullish()
 });
 var SandboxGroupConfigSchema = SandboxConfigSchema.extend({
   size: zod.z.number().default(() => envVars.ROCK_DEFAULT_GROUP_SIZE),
@@ -1994,7 +1996,8 @@ var Sandbox = class extends AbstractSandbox {
       autoClearTimeMinutes: this.config.autoClearSeconds / 60,
       startupTimeout: this.config.startupTimeout,
       memory: this.config.memory,
-      cpus: this.config.cpus
+      cpus: this.config.cpus,
+      autoDeleteSeconds: this.config.autoDeleteSeconds
     };
     logger8.debug(`Calling start_async API: ${url}`);
     logger8.debug(`Request data: ${JSON.stringify(data)}`);
@@ -2013,30 +2016,7 @@ var Sandbox = class extends AbstractSandbox {
     this.hostName = response.result?.hostName ?? null;
     this.hostIp = response.result?.hostIp ?? null;
     logger8.info(`Sandbox ID: ${this.sandboxId}`);
-    await sleep(2e3);
-    const startTime = Date.now();
-    const checkTimeout = 1e4;
-    const checkInterval = 3e3;
-    while (Date.now() - startTime < this.config.startupTimeout * 1e3) {
-      try {
-        logger8.info(`Checking status... (elapsed: ${Math.round((Date.now() - startTime) / 1e3)}s)`);
-        const statusPromise = this.getStatus();
-        const timeoutPromise = new Promise(
-          (_, reject) => setTimeout(() => reject(new Error("Status check timeout")), checkTimeout)
-        );
-        const status = await Promise.race([statusPromise, timeoutPromise]);
-        if (status && status.isAlive) {
-          logger8.info("Sandbox is alive");
-          return;
-        }
-      } catch (e) {
-        logger8.debug(`Status check failed (will retry): ${e}`);
-      }
-      await sleep(checkInterval);
-    }
-    throw new InternalServerRockError(
-      `Failed to start sandbox within ${this.config.startupTimeout}s, sandbox: ${this.toString()}`
-    );
+    await this.waitForAlive();
   }
   async stop() {
     if (!this.sandboxId) return;
@@ -2059,8 +2039,11 @@ var Sandbox = class extends AbstractSandbox {
       throw new Error(`Failed to get is alive: ${e}`);
     }
   }
-  async getStatus() {
-    const url = `${this.url}/get_status?sandbox_id=${this.sandboxId}`;
+  async getStatus(options) {
+    let url = `${this.url}/get_status?sandbox_id=${this.sandboxId}`;
+    if (options?.includeAllStates) {
+      url += "&include_all_states=true";
+    }
     const headers = this.buildHeaders();
     const response = await HttpUtils.get(url, headers);
     if (response.status !== "Success") {
@@ -2298,9 +2281,12 @@ var Sandbox = class extends AbstractSandbox {
   async upload(request) {
     return this.uploadByPath(request.sourcePath, request.targetPath);
   }
-  async uploadByPath(sourcePath, targetPath, uploadMode = "auto", timeout) {
+  async uploadByPath(sourcePath, targetPath, options) {
     const url = `${this.url}/upload`;
     const headers = this.buildHeaders();
+    const uploadMode = options?.uploadMode ?? "auto";
+    const timeout = options?.timeout;
+    const onProgress = options?.onProgress;
     try {
       const fs = await import('fs/promises');
       try {
@@ -2313,7 +2299,7 @@ var Sandbox = class extends AbstractSandbox {
       const ossEnabled = envVars.ROCK_OSS_ENABLE;
       const ossThreshold = 1024 * 1024;
       if (uploadMode === "oss" || uploadMode === "auto" && ossEnabled && fileSize > ossThreshold) {
-        return this.uploadViaOss(sourcePath, targetPath, timeout);
+        return this.uploadViaOss(sourcePath, targetPath, timeout, onProgress);
       }
       const fileBuffer = await fs.readFile(sourcePath);
       const fileName = sourcePath.split("/").pop() ?? "file";
@@ -2362,9 +2348,61 @@ var Sandbox = class extends AbstractSandbox {
     }
   }
   /**
-   * Download file from sandbox via OSS
+   * Download file from sandbox
+   * @param remotePath - File path in sandbox
+   * @param localPath - Local file path
+   * @param downloadMode - Download mode: 'auto' (default), 'direct', or 'oss'
+   * @param timeout - Optional timeout in milliseconds for OSS mode
    */
-  async downloadFile(remotePath, localPath, timeout) {
+  async downloadFile(remotePath, localPath, options) {
+    const downloadMode = options?.downloadMode ?? "auto";
+    const timeout = options?.timeout;
+    const onProgress = options?.onProgress;
+    if (!remotePath || remotePath.trim() === "") {
+      return { success: false, message: "Remote path is required" };
+    }
+    const checkResult = await this.execute({ command: ["test", "-f", remotePath], timeout: 60 });
+    if (checkResult.exitCode !== 0) {
+      return { success: false, message: `Remote file does not exist: ${remotePath}` };
+    }
+    if (downloadMode === "direct") {
+      return this.downloadDirect(remotePath, localPath);
+    }
+    if (downloadMode === "oss") {
+      return this.downloadViaOss(remotePath, localPath, timeout, onProgress);
+    }
+    const sizeResult = await this.execute({ command: ["stat", "-c", "%s", remotePath], timeout: 60 });
+    if (sizeResult.exitCode !== 0) {
+      return this.downloadDirect(remotePath, localPath);
+    }
+    const fileSize = parseInt(sizeResult.stdout.trim(), 10);
+    const ossThreshold = 1024 * 1024;
+    const ossEnabled = envVars.ROCK_OSS_ENABLE;
+    if (ossEnabled && fileSize >= ossThreshold) {
+      return this.downloadViaOss(remotePath, localPath, timeout, onProgress);
+    }
+    return this.downloadDirect(remotePath, localPath);
+  }
+  /**
+   * Download file directly via readFile API
+   */
+  async downloadDirect(remotePath, localPath) {
+    try {
+      const fs = await import('fs/promises');
+      const path = await import('path');
+      const response = await this.readFile({ path: remotePath });
+      const parentDir = path.dirname(localPath);
+      await fs.mkdir(parentDir, { recursive: true });
+      await fs.writeFile(localPath, response.content, "utf-8");
+      return { success: true, message: `Successfully downloaded ${remotePath} to ${localPath}` };
+    } catch (e) {
+      return { success: false, message: `Direct download failed: ${e}` };
+    }
+  }
+  /**
+   * Download file via OSS as intermediary
+   */
+  async downloadViaOss(remotePath, localPath, timeout, onProgress) {
     if (!envVars.ROCK_OSS_ENABLE) {
       return {
         success: false,
@@ -2372,13 +2410,6 @@ var Sandbox = class extends AbstractSandbox {
       };
     }
     try {
-      if (!remotePath || remotePath.trim() === "") {
-        return { success: false, message: "Remote path is required" };
-      }
-      const checkResult = await this.execute({ command: ["test", "-f", remotePath], timeout: 60 });
-      if (checkResult.exitCode !== 0) {
-        return { success: false, message: `Remote file does not exist: ${remotePath}` };
-      }
       if (this.ossBucket === null || this.isTokenExpired()) {
         await this.setupOss(timeout);
       }
@@ -2400,21 +2431,29 @@ var Sandbox = class extends AbstractSandbox {
         return { success: false, message: `Sandbox to OSS upload failed: ${uploadResult.output}` };
       }
       const ossTimeout = timeout ?? envVars.ROCK_OSS_TIMEOUT;
-      const result = await this.ossBucket.get(objectName, localPath, { timeout: ossTimeout });
+      const result = await this.ossBucket.get(objectName, localPath, {
+        timeout: ossTimeout,
+        progress: (p) => {
+          onProgress?.({
+            phase: "download-to-local",
+            percent: Math.round(p * 100)
+          });
+        }
+      });
       try {
         await this.ossBucket.delete(objectName);
       } catch {
       }
       return { success: true, message: `Successfully downloaded ${remotePath} to ${localPath}` };
     } catch (e) {
-      return { success: false, message: `Download failed: ${e}` };
+      return { success: false, message: `OSS download failed: ${e}` };
     }
   }
   /**
    * Upload file via OSS (internal method)
    * @param timeout - Optional timeout in milliseconds
    */
-  async uploadViaOss(sourcePath, targetPath, timeout) {
+  async uploadViaOss(sourcePath, targetPath, timeout, onProgress) {
     try {
       if (this.ossBucket === null || this.isTokenExpired()) {
         await this.setupOss(timeout);
@@ -2425,9 +2464,39 @@ var Sandbox = class extends AbstractSandbox {
       const timestamp = Date.now();
       const fileName = sourcePath.split("/").pop() ?? "file";
       const objectName = `${timestamp}-${fileName}`;
+      const fs = await import('fs/promises');
+      const stats = await fs.stat(sourcePath);
+      const fileSize = stats.size;
+      const multipartThreshold = 1024 * 1024;
       const ossTimeout = timeout ?? envVars.ROCK_OSS_TIMEOUT;
-      await this.ossBucket.put(objectName, sourcePath, { timeout: ossTimeout });
+      if (fileSize >= multipartThreshold) {
+        await this.ossBucket.multipartUpload(objectName, sourcePath, {
+          timeout: ossTimeout,
+          partSize: multipartThreshold,
+          // 1MB per part
+          progress: (p) => {
+            onProgress?.({
+              phase: "upload-to-oss",
+              percent: Math.round(p * 100)
+            });
+          }
+        });
+      } else {
+        await this.ossBucket.put(objectName, sourcePath, {
+          timeout: ossTimeout,
+          progress: (p) => {
+            onProgress?.({
+              phase: "upload-to-oss",
+              percent: Math.round(p * 100)
+            });
+          }
+        });
+      }
       const signedUrl = this.ossBucket.signatureUrl(objectName, { expires: 600 });
+      onProgress?.({
+        phase: "download-to-sandbox",
+        percent: -1
+      });
       const downloadCmd = `wget -c -O '${targetPath}' '${signedUrl}'`;
       await this.arun(downloadCmd, { mode: "nohup", waitTimeout: 600 });
       const checkResult = await this.execute({ command: ["test", "-f", targetPath], timeout: 60 });
@@ -2468,6 +2537,74 @@ var Sandbox = class extends AbstractSandbox {
       // 5 minutes
     });
   }
+  async restart() {
+    if (!this.sandboxId) {
+      throw new Error("sandbox_id is not set, cannot restart");
+    }
+    const url = `${this.url}/restart`;
+    const headers = this.buildHeaders();
+    const data = { sandboxId: this.sandboxId };
+    const response = await HttpUtils.post(url, headers, data);
+    logger8.debug(`Restart sandbox response: ${JSON.stringify(response)}`);
+    if (response.status !== "Success") {
+      const code = response.result?.code;
+      raiseForCode(code, `Failed to restart sandbox: ${JSON.stringify(response)}`);
+      throw new Error(`Failed to restart sandbox: ${JSON.stringify(response)}`);
+    }
+    await this.waitForAlive({ includeAllStates: true });
+  }
+  async waitForAlive(options) {
+    await sleep(2e3);
+    const startTime = Date.now();
+    const checkTimeout = 1e4;
+    const checkInterval = 3e3;
+    while (Date.now() - startTime < this.config.startupTimeout * 1e3) {
+      let timeoutId;
+      try {
+        logger8.info(`Checking status... (elapsed: ${Math.round((Date.now() - startTime) / 1e3)}s)`);
+        const statusPromise = this.getStatus(options);
+        const timeoutPromise = new Promise((_, reject) => {
+          timeoutId = setTimeout(() => reject(new Error("Status check timeout")), checkTimeout);
+        });
+        const status = await Promise.race([statusPromise, timeoutPromise]);
+        if (status && status.isAlive) {
+          logger8.info("Sandbox is alive");
+          return;
+        }
+        if (options?.includeAllStates && status) {
+          const errorMsg = this.parseErrorMessageFromStatus(status.status);
+          if (errorMsg) {
+            throw new InternalServerRockError(
+              `Failed to restart sandbox because ${errorMsg}, sandbox: ${this.toString()}`
+            );
+          }
+        }
+      } catch (e) {
+        if (e instanceof InternalServerRockError) {
+          throw e;
+        }
+        logger8.debug(`Status check failed (will retry): ${e}`);
+      } finally {
+        clearTimeout(timeoutId);
+      }
+      await sleep(checkInterval);
+    }
+    throw new InternalServerRockError(
+      `Failed to start sandbox within ${this.config.startupTimeout}s, sandbox: ${this.toString()}`
+    );
+  }
+  parseErrorMessageFromStatus(status) {
+    if (!status) return null;
+    for (const [stage, details] of Object.entries(status)) {
+      if (details && typeof details === "object") {
+        const d = details;
+        if (d.status === "failed" || d.status === "timeout") {
+          return `${stage}: ${d.message ?? "No message provided"}`;
+        }
+      }
+    }
+    return null;
+  }
   // Close
   async close() {
     await this.stop();
@@ -3433,6 +3570,7 @@ exports.DefaultAgent = DefaultAgent;
 exports.DefaultAgentConfigSchema = DefaultAgentConfigSchema;
 exports.Deploy = Deploy;
 exports.DownloadFileResponseSchema = DownloadFileResponseSchema;
+exports.DownloadModeSchema = DownloadModeSchema;
 exports.EnvHubClient = EnvHubClient;
 exports.EnvHubClientConfigSchema = EnvHubClientConfigSchema;
 exports.EnvHubError = EnvHubError;