rl-rock 1.3.7 → 1.3.9

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
@@ -1287,7 +1332,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): Promise<UploadResponse>;
+    uploadByPath(sourcePath: string, targetPath: string, options?: UploadOptions): Promise<UploadResponse>;
     /**
      * Get OSS STS credentials from sandbox
      */
@@ -1297,15 +1342,29 @@ 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): Promise<DownloadFileResponse>;
+    private downloadViaOss;
     /**
      * Upload file via OSS (internal method)
+     * @param timeout - Optional timeout in milliseconds
      */
     private uploadViaOss;
     /**
      * Setup OSS bucket with STS credentials
+     * @param timeout - Optional timeout in milliseconds (defaults to ROCK_OSS_TIMEOUT env var or 300000ms)
      */
     private setupOss;
     close(): Promise<void>;
@@ -2425,4 +2484,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
@@ -1287,7 +1332,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): Promise<UploadResponse>;
+    uploadByPath(sourcePath: string, targetPath: string, options?: UploadOptions): Promise<UploadResponse>;
     /**
      * Get OSS STS credentials from sandbox
      */
@@ -1297,15 +1342,29 @@ 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): Promise<DownloadFileResponse>;
+    private downloadViaOss;
     /**
      * Upload file via OSS (internal method)
+     * @param timeout - Optional timeout in milliseconds
      */
     private uploadViaOss;
     /**
      * Setup OSS bucket with STS credentials
+     * @param timeout - Optional timeout in milliseconds (defaults to ROCK_OSS_TIMEOUT env var or 300000ms)
      */
     private setupOss;
     close(): Promise<void>;
@@ -2425,4 +2484,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(),
@@ -601,6 +602,9 @@ var envVars = {
   get ROCK_OSS_BUCKET_REGION() {
     return getEnv("ROCK_OSS_BUCKET_REGION");
   },
+  get ROCK_OSS_TIMEOUT() {
+    return parseInt(getEnv("ROCK_OSS_TIMEOUT", "300000"), 10);
+  },
   // Pip
   get ROCK_PIP_INDEX_URL() {
     return getEnv("ROCK_PIP_INDEX_URL", "https://pypi.org/simple/");
@@ -2295,9 +2299,12 @@ var Sandbox = class extends AbstractSandbox {
   async upload(request) {
     return this.uploadByPath(request.sourcePath, request.targetPath);
   }
-  async uploadByPath(sourcePath, targetPath, uploadMode = "auto") {
+  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 {
@@ -2310,7 +2317,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);
+        return this.uploadViaOss(sourcePath, targetPath, timeout, onProgress);
       }
       const fileBuffer = await fs.readFile(sourcePath);
       const fileName = sourcePath.split("/").pop() ?? "file";
@@ -2359,9 +2366,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) {
+  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,
@@ -2369,15 +2428,8 @@ 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();
+        await this.setupOss(timeout);
       }
       if (!this.ossBucket) {
         return { success: false, message: "Failed to setup OSS bucket" };
@@ -2396,23 +2448,33 @@ var Sandbox = class extends AbstractSandbox {
       if (uploadResult.exitCode !== 0) {
         return { success: false, message: `Sandbox to OSS upload failed: ${uploadResult.output}` };
       }
-      const result = await this.ossBucket.get(objectName, localPath);
+      const ossTimeout = timeout ?? envVars.ROCK_OSS_TIMEOUT;
+      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) {
+  async uploadViaOss(sourcePath, targetPath, timeout, onProgress) {
     try {
       if (this.ossBucket === null || this.isTokenExpired()) {
-        await this.setupOss();
+        await this.setupOss(timeout);
       }
       if (!this.ossBucket) {
         return { success: false, message: "Failed to setup OSS bucket" };
@@ -2420,8 +2482,39 @@ var Sandbox = class extends AbstractSandbox {
       const timestamp = Date.now();
       const fileName = sourcePath.split("/").pop() ?? "file";
       const objectName = `${timestamp}-${fileName}`;
-      await this.ossBucket.put(objectName, sourcePath);
+      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;
+      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 });
@@ -2435,13 +2528,16 @@ var Sandbox = class extends AbstractSandbox {
   }
   /**
    * Setup OSS bucket with STS credentials
+   * @param timeout - Optional timeout in milliseconds (defaults to ROCK_OSS_TIMEOUT env var or 300000ms)
    */
-  async setupOss() {
+  async setupOss(timeout) {
     const credentials = await this.getOssStsCredentials();
     const OSS = (await import('ali-oss')).default;
+    const ossTimeout = timeout ?? envVars.ROCK_OSS_TIMEOUT;
     this.ossBucket = new OSS({
       secure: true,
       // Use HTTPS for OSS connections
+      timeout: ossTimeout,
       region: envVars.ROCK_OSS_BUCKET_REGION ?? "",
       accessKeyId: credentials.accessKeyId,
       accessKeySecret: credentials.accessKeySecret,
@@ -3424,6 +3520,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;