test-wuying-agentbay-sdk 0.13.0-beta.20251210113609 → 0.13.0-beta.20251210135226

package/dist/index.cjs

@@ -5379,6 +5379,92 @@ var _Code = class _Code {
   constructor(session) {
     this.session = session;
   }
+  /**
+   * Parses the backend JSON response into a structured CodeExecutionResult
+   * @param data Raw JSON string from backend
+   */
+  parseBackendResponse(data) {
+    let raw;
+    try {
+      raw = JSON.parse(data);
+      if (typeof raw === "string") {
+        try {
+          raw = JSON.parse(raw);
+        } catch (e) {
+        }
+      }
+    } catch (e) {
+      return {
+        requestId: "",
+        // Will be filled later
+        success: true,
+        result: data,
+        logs: { stdout: [data], stderr: [] },
+        results: [{ text: data, isMainResult: true }]
+      };
+    }
+    const logs = {
+      stdout: raw.stdout || [],
+      stderr: raw.stderr || []
+    };
+    const results = [];
+    if (Array.isArray(raw.result)) {
+      for (const itemStr of raw.result) {
+        try {
+          const itemMap = typeof itemStr === "string" ? JSON.parse(itemStr) : itemStr;
+          const item = {};
+          if (itemMap.isMainResult) item.isMainResult = true;
+          if (itemMap.is_main_result) item.isMainResult = true;
+          if (itemMap["text/plain"]) item.text = itemMap["text/plain"];
+          if (itemMap["text/html"]) item.html = itemMap["text/html"];
+          if (itemMap["text/markdown"]) item.markdown = itemMap["text/markdown"];
+          if (itemMap["image/png"]) item.png = itemMap["image/png"];
+          if (itemMap["image/jpeg"]) item.jpeg = itemMap["image/jpeg"];
+          if (itemMap["image/svg+xml"]) item.svg = itemMap["image/svg+xml"];
+          if (itemMap["text/latex"]) item.latex = itemMap["text/latex"];
+          if (itemMap["application/json"]) item.json = itemMap["application/json"];
+          if (itemMap["application/vnd.vegalite.v4+json"]) {
+            item.chart = itemMap["application/vnd.vegalite.v4+json"];
+          } else if (itemMap["application/vnd.vegalite.v5+json"]) {
+            item.chart = itemMap["application/vnd.vegalite.v5+json"];
+          } else if (itemMap["application/vnd.vega.v5+json"]) {
+            item.chart = itemMap["application/vnd.vega.v5+json"];
+          }
+          results.push(item);
+        } catch (e) {
+        }
+      }
+    }
+    let error;
+    if (raw.executionError) {
+      error = {
+        name: "ExecutionError",
+        value: raw.executionError,
+        traceback: ""
+      };
+    }
+    let resultText = "";
+    const mainRes = results.find((r) => r.isMainResult && r.text);
+    if (mainRes && mainRes.text) {
+      resultText = mainRes.text;
+    } else if (results.length > 0 && results[0].text) {
+      resultText = results[0].text;
+    } else if (logs.stdout.length > 0) {
+      resultText = logs.stdout.join("");
+    }
+    return {
+      requestId: "",
+      // Will be filled by caller
+      success: !raw.executionError,
+      result: resultText,
+      errorMessage: raw.executionError,
+      logs,
+      results,
+      error,
+      executionTime: raw.executionTime || raw.execution_time,
+      executionCount: raw.executionCount || raw.execution_count
+    };
+  }
   /**
    * Execute code in the specified language with a timeout.
    * Corresponds to Python's run_code() method
@@ -5397,6 +5483,9 @@ var _Code = class _Code {
    * if (result.success) {
    *   const codeResult = await result.session.code.runCode('print("Hello")', "python");
    *   console.log(codeResult.result);
+   *   if (codeResult.results) {
+   *      // Access rich output like images or charts
+   *   }
    *   await result.session.delete();
    * }
    * ```
@@ -5416,23 +5505,34 @@ var _Code = class _Code {
         language,
         timeout_s: timeoutS
       };
-      const response = await this.session.callMcpTool(
-        "run_code",
-        args
-      );
-      if (!response.success) {
-        return {
-          requestId: response.requestId,
-          success: false,
-          result: "",
-          errorMessage: response.errorMessage
-        };
+      const response = await this.session.callMcpTool("run_code", args);
+      let codeResult;
+      if (response.success) {
+        codeResult = this.parseBackendResponse(response.data);
+      } else {
+        try {
+          if (response.errorMessage && response.errorMessage.trim().startsWith("{")) {
+            codeResult = this.parseBackendResponse(response.errorMessage);
+            codeResult.success = false;
+          } else {
+            return {
+              requestId: response.requestId,
+              success: false,
+              result: "",
+              errorMessage: response.errorMessage
+            };
+          }
+        } catch (e6) {
+          return {
+            requestId: response.requestId,
+            success: false,
+            result: "",
+            errorMessage: response.errorMessage
+          };
+        }
       }
-      return {
-        requestId: response.requestId,
-        success: true,
-        result: response.data || ""
-      };
+      codeResult.requestId = response.requestId;
+      return codeResult;
     } catch (error) {
       return {
         requestId: "",
@@ -5610,7 +5710,7 @@ var _Command = class _Command {
               errorMessage: stderr || result.errorMessage
             };
           }
-        } catch (e6) {
+        } catch (e7) {
         }
         return {
           requestId: result.requestId,
@@ -10665,6 +10765,12 @@ var _Session = class _Session {
         if (toolName === "run_code" && actualResult) {
           const dataStr = typeof actualResult === "string" ? actualResult : JSON.stringify(actualResult);
           logCodeExecutionOutput(requestId, dataStr);
+          return {
+            success: true,
+            data: dataStr,
+            errorMessage: "",
+            requestId: ""
+          };
         }
         return {
           success: true,
@@ -10699,6 +10805,23 @@ var _Session = class _Session {
           };
         }
         const data = response.body.data;
+        const reqId = extractRequestId(response) || "";
+        if (toolName === "run_code") {
+          let dataStr = "";
+          if (data && Array.isArray(data.content) && data.content.length > 0 && data.content[0].text) {
+            dataStr = data.content[0].text;
+          } else {
+            dataStr = typeof response.body.data === "string" ? response.body.data : JSON.stringify(response.body.data);
+          }
+          logCodeExecutionOutput(reqId, dataStr);
+          const isError = data.isError === true;
+          return {
+            success: !isError,
+            data: dataStr,
+            errorMessage: isError ? dataStr : "",
+            requestId: reqId
+          };
+        }
         if (data.isError) {
           const errorContent = data.content || [];
           const errorMessage = errorContent.map((item) => item.text || "Unknown error").join("; ");
@@ -10706,7 +10829,7 @@ var _Session = class _Session {
             success: false,
             data: "",
             errorMessage,
-            requestId: extractRequestId(response) || ""
+            requestId: reqId
           };
         }
         const content = data.content || [];
@@ -10714,11 +10837,6 @@ var _Session = class _Session {
         if (content.length > 0 && content[0].text !== void 0) {
           textContent = content[0].text;
         }
-        const reqId = extractRequestId(response) || "";
-        if (toolName === "run_code" && data) {
-          const dataStr = typeof response.body.data === "string" ? response.body.data : JSON.stringify(response.body.data);
-          logCodeExecutionOutput(reqId, dataStr);
-        }
         return {
           success: true,
           data: textContent,