rl-rock 1.2.2 → 1.2.3

package/dist/index.mjs

@@ -86,14 +86,21 @@ var ChmodRequestSchema = z.object({
   mode: z.string().default("755"),
   recursive: z.boolean().default(false)
 });
+var HeaderFieldsSchema = {
+  cluster: z.string().optional(),
+  requestId: z.string().optional(),
+  eagleeyeTraceid: z.string().optional()
+};
 var SandboxResponseSchema = z.object({
   code: z.nativeEnum(Codes).optional(),
   exitCode: z.number().optional(),
-  failureReason: z.string().optional()
+  failureReason: z.string().optional(),
+  ...HeaderFieldsSchema
 });
 var IsAliveResponseSchema = z.object({
   isAlive: z.boolean(),
-  message: z.string().default("")
+  message: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var SandboxStatusResponseSchema = z.object({
   sandboxId: z.string().optional(),
@@ -110,54 +117,68 @@ var SandboxStatusResponseSchema = z.object({
   namespace: z.string().optional(),
   cpus: z.number().optional(),
   memory: z.string().optional(),
-  state: z.unknown().optional()
+  state: z.unknown().optional(),
+  ...HeaderFieldsSchema
 });
 var CommandResponseSchema = z.object({
   stdout: z.string().default(""),
   stderr: z.string().default(""),
-  exitCode: z.number().optional()
+  exitCode: z.number().optional(),
+  ...HeaderFieldsSchema
 });
 var WriteFileResponseSchema = z.object({
   success: z.boolean().default(false),
-  message: z.string().default("")
+  message: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var ReadFileResponseSchema = z.object({
-  content: z.string().default("")
+  content: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var UploadResponseSchema = z.object({
   success: z.boolean().default(false),
   message: z.string().default(""),
-  fileName: z.string().optional()
+  fileName: z.string().optional(),
+  ...HeaderFieldsSchema
 });
 var ObservationSchema = z.object({
   output: z.string().default(""),
   exitCode: z.number().optional(),
   failureReason: z.string().default(""),
-  expectString: z.string().default("")
+  expectString: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var CreateSessionResponseSchema = z.object({
   output: z.string().default(""),
-  sessionType: z.literal("bash").default("bash")
+  sessionType: z.literal("bash").default("bash"),
+  ...HeaderFieldsSchema
 });
 var CloseSessionResponseSchema = z.object({
-  sessionType: z.literal("bash").default("bash")
+  sessionType: z.literal("bash").default("bash"),
+  ...HeaderFieldsSchema
+});
+var CloseResponseSchema = z.object({
+  ...HeaderFieldsSchema
 });
-var CloseResponseSchema = z.object({});
 var ChownResponseSchema = z.object({
   success: z.boolean().default(false),
-  message: z.string().default("")
+  message: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var ChmodResponseSchema = z.object({
   success: z.boolean().default(false),
-  message: z.string().default("")
+  message: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var ExecuteBashSessionResponseSchema = z.object({
   success: z.boolean().default(false),
-  message: z.string().default("")
+  message: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 var OssSetupResponseSchema = z.object({
   success: z.boolean().default(false),
-  message: z.string().default("")
+  message: z.string().default(""),
+  ...HeaderFieldsSchema
 });
 
 // src/common/constants.ts
@@ -273,6 +294,7 @@ var HttpUtils = class {
    * Send POST request
    * Automatically converts request body from camelCase to snake_case
    * Automatically converts response from snake_case to camelCase
+   * Returns structured response with headers
    */
   static async post(url, headers, data, readTimeout) {
     const client = this.createClient({
@@ -282,7 +304,18 @@ var HttpUtils = class {
     const snakeData = objectToSnake(data);
     try {
       const response = await client.post(url, snakeData);
-      return objectToCamel(response.data);
+      const result = objectToCamel(response.data);
+      const responseHeaders = {};
+      for (const [key, value] of Object.entries(response.headers)) {
+        if (typeof value === "string") {
+          responseHeaders[key.toLowerCase()] = value;
+        }
+      }
+      return {
+        status: result?.status ?? "Success",
+        result,
+        headers: responseHeaders
+      };
     } catch (error) {
       if (error instanceof AxiosError) {
         throw new Error(`Failed to POST ${url}: ${error.message}`);
@@ -293,12 +326,24 @@ var HttpUtils = class {
   /**
    * Send GET request
    * Automatically converts response from snake_case to camelCase
+   * Returns structured response with headers
    */
   static async get(url, headers) {
     const client = this.createClient({ headers });
     try {
       const response = await client.get(url);
-      return objectToCamel(response.data);
+      const result = objectToCamel(response.data);
+      const responseHeaders = {};
+      for (const [key, value] of Object.entries(response.headers)) {
+        if (typeof value === "string") {
+          responseHeaders[key.toLowerCase()] = value;
+        }
+      }
+      return {
+        status: result?.status ?? "Success",
+        result,
+        headers: responseHeaders
+      };
     } catch (error) {
       if (error instanceof AxiosError) {
         throw new Error(`Failed to GET ${url}: ${error.message}`);
@@ -316,6 +361,7 @@ var HttpUtils = class {
    * Send multipart/form-data request
    * Automatically converts form data keys to snake_case
    * Automatically converts response from snake_case to camelCase
+   * Returns structured response with headers
    */
   static async postMultipart(url, headers, data, files) {
     const formData = new FormData();
@@ -352,7 +398,18 @@ var HttpUtils = class {
     });
     try {
       const response = await client.post(url, formData);
-      return objectToCamel(response.data);
+      const result = objectToCamel(response.data);
+      const responseHeaders = {};
+      for (const [key, value] of Object.entries(response.headers)) {
+        if (typeof value === "string") {
+          responseHeaders[key.toLowerCase()] = value;
+        }
+      }
+      return {
+        status: result?.status ?? "Success",
+        result,
+        headers: responseHeaders
+      };
     } catch (error) {
       if (error instanceof AxiosError) {
         throw new Error(`Failed to POST multipart ${url}: ${error.message}`);
@@ -563,7 +620,7 @@ var EnvHubClient = class {
         this.headers,
         payload
       );
-      return createRockEnvInfo(response);
+      return createRockEnvInfo(response.result);
     } catch (e) {
       throw new EnvHubError(`Failed to register environment: ${e}`);
     }
@@ -580,7 +637,7 @@ var EnvHubClient = class {
         this.headers,
         payload
       );
-      return createRockEnvInfo(response);
+      return createRockEnvInfo(response.result);
     } catch (e) {
       throw new EnvHubError(`Failed to get environment ${envName}: ${e}`);
     }
@@ -600,7 +657,7 @@ var EnvHubClient = class {
         this.headers,
         payload
       );
-      const envsData = response.envs ?? [];
+      const envsData = response.result?.envs ?? [];
       return envsData.map((envData) => createRockEnvInfo(envData));
     } catch (e) {
       throw new EnvHubError(`Failed to list environments: ${e}`);
@@ -633,7 +690,7 @@ var EnvHubClient = class {
         url,
         this.headers
       );
-      return response;
+      return response.result;
     } catch (e) {
       throw new EnvHubError(`Failed to health check: ${e}`);
     }
@@ -1082,7 +1139,7 @@ var Process = class {
     const scriptPath = `/tmp/${name}`;
     try {
       logger5.info(`[${sandboxId}] Uploading script to ${scriptPath}`);
-      const writeResult = await this.sandbox.write_file({
+      const writeResult = await this.sandbox.writeFile({
         content: scriptContent,
         path: scriptPath
       });
@@ -1341,6 +1398,16 @@ var Sandbox = class extends AbstractSandbox {
       headers["X-Namespace"] = this.config.namespace;
     }
   }
+  /**
+   * Extract header-derived fields from response headers
+   */
+  extractHeaderFields(headers) {
+    return {
+      cluster: headers["x-rock-gateway-target-cluster"] || this.config.cluster || "N/A",
+      requestId: headers["x-request-id"] || headers["request-id"] || "N/A",
+      eagleeyeTraceid: headers["eagleeye-traceid"] || "N/A"
+    };
+  }
   // Lifecycle methods
   async start() {
     logger8.info("Starting sandbox...");
@@ -1424,7 +1491,11 @@ var Sandbox = class extends AbstractSandbox {
     if (response.status !== "Success") {
       throw new Error(`Failed to get status: ${JSON.stringify(response)}`);
     }
-    return response.result;
+    const result = response.result;
+    result.cluster = response.headers["x-rock-gateway-target-cluster"] || this.config.cluster || "N/A";
+    result.requestId = response.headers["x-request-id"] || response.headers["request-id"] || "N/A";
+    result.eagleeyeTraceid = response.headers["eagleeye-traceid"] || "N/A";
+    return result;
   }
   // Command execution
   async execute(command) {
@@ -1446,7 +1517,9 @@ var Sandbox = class extends AbstractSandbox {
       if (response.status !== "Success") {
         throw new Error(`Failed to execute command: ${JSON.stringify(response)}`);
       }
-      return response.result;
+      const result = response.result;
+      Object.assign(result, this.extractHeaderFields(response.headers));
+      return result;
     } catch (e) {
       throw new Error(`Failed to execute command: ${e}`);
     }
@@ -1468,7 +1541,9 @@ var Sandbox = class extends AbstractSandbox {
       if (response.status !== "Success") {
         throw new Error(`Failed to create session: ${JSON.stringify(response)}`);
       }
-      return response.result;
+      const result = response.result;
+      Object.assign(result, this.extractHeaderFields(response.headers));
+      return result;
     } catch (e) {
       throw new Error(`Failed to create session: ${e}`);
     }
@@ -1489,7 +1564,9 @@ var Sandbox = class extends AbstractSandbox {
       if (response.status !== "Success") {
         throw new Error(`Failed to close session: ${JSON.stringify(response)}`);
       }
-      return response.result ?? { sessionType: "bash" };
+      const result = response.result ?? { sessionType: "bash" };
+      Object.assign(result, this.extractHeaderFields(response.headers));
+      return result;
     } catch (e) {
       throw new Error(`Failed to close session: ${e}`);
     }
@@ -1535,7 +1612,9 @@ var Sandbox = class extends AbstractSandbox {
       if (response.status !== "Success") {
         throw new Error(`Failed to execute command: ${JSON.stringify(response)}`);
       }
-      return response.result;
+      const result = response.result;
+      Object.assign(result, this.extractHeaderFields(response.headers));
+      return result;
     } catch (e) {
       throw new Error(`Failed to run in session: ${e}`);
     }
@@ -1574,7 +1653,10 @@ var Sandbox = class extends AbstractSandbox {
         output: "Failed to submit command, nohup failed to extract PID",
         exitCode: 1,
         failureReason: "PID extraction failed",
-        expectString: ""
+        expectString: "",
+        cluster: response.cluster,
+        requestId: response.requestId,
+        eagleeyeTraceid: response.eagleeyeTraceid
       };
     }
     const success = await this.waitForProcessCompletion(pid, tmpSession, waitTimeout, waitInterval);
@@ -1583,7 +1665,10 @@ var Sandbox = class extends AbstractSandbox {
         output: `Command executed in nohup mode. Output file: ${tmpFile}`,
         exitCode: success ? 0 : 1,
         failureReason: success ? "" : "Process did not complete successfully",
-        expectString: ""
+        expectString: "",
+        cluster: response.cluster,
+        requestId: response.requestId,
+        eagleeyeTraceid: response.eagleeyeTraceid
       };
     }
     const readCmd = responseLimitedBytesInNohup ? `head -c ${responseLimitedBytesInNohup} ${tmpFile}` : `cat ${tmpFile}`;
@@ -1595,7 +1680,10 @@ var Sandbox = class extends AbstractSandbox {
       output: outputResult.output,
       exitCode: success ? 0 : 1,
       failureReason: success ? "" : "Process did not complete successfully",
-      expectString: ""
+      expectString: "",
+      cluster: outputResult.cluster,
+      requestId: outputResult.requestId,
+      eagleeyeTraceid: outputResult.eagleeyeTraceid
     };
   }
   async waitForProcessCompletion(pid, session, waitTimeout, waitInterval) {
@@ -1621,7 +1709,7 @@ var Sandbox = class extends AbstractSandbox {
     return false;
   }
   // File operations
-  async write_file(request) {
+  async writeFile(request) {
     const url = `${this.url}/write_file`;
     const headers = this.buildHeaders();
     const data = {
@@ -1630,12 +1718,13 @@ var Sandbox = class extends AbstractSandbox {
       sandboxId: this.sandboxId
     };
     const response = await HttpUtils.post(url, headers, data);
+    const headerFields = this.extractHeaderFields(response.headers);
     if (response.status !== "Success") {
-      return { success: false, message: `Failed to write file ${request.path}` };
+      return { success: false, message: `Failed to write file ${request.path}`, ...headerFields };
     }
-    return { success: true, message: `Successfully write content to file ${request.path}` };
+    return { success: true, message: `Successfully write content to file ${request.path}`, ...headerFields };
   }
-  async read_file(request) {
+  async readFile(request) {
     const url = `${this.url}/read_file`;
     const headers = this.buildHeaders();
     const data = {
@@ -1649,7 +1738,10 @@ var Sandbox = class extends AbstractSandbox {
       headers,
       data
     );
-    return { content: response.result?.content ?? "" };
+    return {
+      content: response.result?.content ?? "",
+      ...this.extractHeaderFields(response.headers)
+    };
   }
   // Upload
   async upload(request) {
@@ -1661,7 +1753,7 @@ var Sandbox = class extends AbstractSandbox {
     try {
       const fs = await import('fs');
       if (!fs.existsSync(sourcePath)) {
-        return { success: false, message: `File not found: ${sourcePath}` };
+        return { success: false, message: `File not found: ${sourcePath}`, cluster: "N/A", requestId: "N/A", eagleeyeTraceid: "N/A" };
       }
       const fileBuffer = fs.readFileSync(sourcePath);
       const fileName = sourcePath.split("/").pop() ?? "file";
@@ -1671,12 +1763,13 @@ var Sandbox = class extends AbstractSandbox {
         { targetPath, sandboxId: this.sandboxId ?? "" },
         { file: [fileName, fileBuffer, "application/octet-stream"] }
       );
+      const headerFields = this.extractHeaderFields(response.headers);
       if (response.status !== "Success") {
-        return { success: false, message: "Upload failed" };
+        return { success: false, message: "Upload failed", ...headerFields };
       }
-      return { success: true, message: `Successfully uploaded file ${fileName} to ${targetPath}` };
+      return { success: true, message: `Successfully uploaded file ${fileName} to ${targetPath}`, ...headerFields };
     } catch (e) {
-      return { success: false, message: `Upload failed: ${e}` };
+      return { success: false, message: `Upload failed: ${e}`, cluster: "N/A", requestId: "N/A", eagleeyeTraceid: "N/A" };
     }
   }
   // Close
@@ -1982,6 +2075,6 @@ var ModelService = class {
 // src/index.ts
 var VERSION = "1.2.1";
 
-export { BadRequestRockError, BashActionSchema, ChmodRequestSchema, ChmodResponseSchema, ChownRequestSchema, ChownResponseSchema, CloseResponseSchema, CloseSessionRequestSchema, CloseSessionResponseSchema, Codes, CommandResponseSchema, CommandRockError, CommandSchema, Constants, CreateBashSessionRequestSchema, CreateSessionResponseSchema, Deploy, EnvHubClient, EnvHubClientConfigSchema, EnvHubError, ExecuteBashSessionResponseSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, ModelService, Network, ObservationSchema, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, Process, ReadFileRequestSchema, ReadFileResponseSchema, ReasonPhrases, RockEnv, RockEnvInfoSchema, RockException, RunMode, Sandbox, SandboxConfigSchema, SandboxGroup, SandboxGroupConfigSchema, SandboxResponseSchema, SandboxStatusResponseSchema, SpeedupType, UploadRequestSchema, UploadResponseSchema, VERSION, WriteFileRequestSchema, WriteFileResponseSchema, arunWithRetry, createRockEnvInfo, createSandboxConfig, createSandboxGroupConfig, deprecated, deprecatedClass, extractNohupPid as extractNohupPidFromSandbox, fromRockException, getEnv, getReasonPhrase, getRequiredEnv, isBrowser, isClientError, isCommandError, isEnvSet, isError, isNode, isServerError, isSuccess, make, raiseForCode, retryAsync, sleep, withRetry, withTimeLogging };
+export { BadRequestRockError, BashActionSchema, ChmodRequestSchema, ChmodResponseSchema, ChownRequestSchema, ChownResponseSchema, CloseResponseSchema, CloseSessionRequestSchema, CloseSessionResponseSchema, Codes, CommandResponseSchema, CommandRockError, CommandSchema, Constants, CreateBashSessionRequestSchema, CreateSessionResponseSchema, Deploy, EnvHubClient, EnvHubClientConfigSchema, EnvHubError, ExecuteBashSessionResponseSchema, HeaderFieldsSchema, HttpUtils, InternalServerRockError, InvalidParameterRockException, IsAliveResponseSchema, LinuxFileSystem, LinuxRemoteUser, ModelClient, ModelService, Network, ObservationSchema, OssSetupResponseSchema, PID_PREFIX, PID_SUFFIX, Process, ReadFileRequestSchema, ReadFileResponseSchema, ReasonPhrases, RockEnv, RockEnvInfoSchema, RockException, RunMode, Sandbox, SandboxConfigSchema, SandboxGroup, SandboxGroupConfigSchema, SandboxResponseSchema, SandboxStatusResponseSchema, SpeedupType, UploadRequestSchema, UploadResponseSchema, VERSION, WriteFileRequestSchema, WriteFileResponseSchema, arunWithRetry, createRockEnvInfo, createSandboxConfig, createSandboxGroupConfig, deprecated, deprecatedClass, extractNohupPid as extractNohupPidFromSandbox, fromRockException, getEnv, getReasonPhrase, getRequiredEnv, isBrowser, isClientError, isCommandError, isEnvSet, isError, isNode, isServerError, isSuccess, make, raiseForCode, retryAsync, sleep, withRetry, withTimeLogging };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map