test-wuying-agentbay-sdk 0.13.0-beta.20251212160413 → 0.13.0-beta.20251215095254

package/dist/index.mjs

@@ -41,6 +41,10 @@ import {
   GetAdbLinkResponse,
   GetAdbLinkResponseBody,
   GetAdbLinkResponseBodyData,
+  GetAndLoadInternalContextRequest,
+  GetAndLoadInternalContextResponse,
+  GetAndLoadInternalContextResponseBody,
+  GetAndLoadInternalContextResponseBodyData,
   GetCdpLinkRequest,
   GetCdpLinkResponse,
   GetCdpLinkResponseBody,
@@ -109,7 +113,7 @@ import {
   SyncContextRequest,
   SyncContextResponse,
   SyncContextResponseBody
-} from "./chunk-S7SG7CLI.mjs";
+} from "./chunk-VEHDE3LU.mjs";
 import {
   __commonJS,
   __dirname,
@@ -693,7 +697,7 @@ var _Client = class _Client extends (OpenApi.default || OpenApi) {
   }
   /**
    * Delete Persistent Context
-   * 
+   *
    * @param request - ClearContextRequest
    * @param runtime - runtime options for this request RuntimeOptions
    * @returns ClearContextResponse
@@ -725,7 +729,7 @@ var _Client = class _Client extends (OpenApi.default || OpenApi) {
   }
   /**
    * Delete Persistent Context
-   * 
+   *
    * @param request - ClearContextRequest
    * @returns ClearContextResponse
    */
@@ -1255,6 +1259,42 @@ var _Client = class _Client extends (OpenApi.default || OpenApi) {
     const runtime = new $dara.RuntimeOptions({});
     return await this.getMcpResourceWithOptions(request, runtime);
   }
+  /**
+   * Get internal context request
+   *
+   * @param request - GetAndLoadInternalContextRequest
+   * @param runtime - runtime options for this request RuntimeOptions
+   * @returns GetAndLoadInternalContextResponse
+   */
+  async getAndLoadInternalContextWithOptions(request, runtime) {
+    request.validate();
+    let query = OpenApiUtil.query(request.toMap());
+    let req = new $OpenApiUtil.OpenApiRequest({
+      query: OpenApiUtil.query(query)
+    });
+    let params = new $OpenApiUtil.Params({
+      action: "GetAndLoadInternalContext",
+      version: "2025-05-06",
+      protocol: "HTTPS",
+      pathname: "/",
+      method: "GET",
+      authType: "Anonymous",
+      style: "RPC",
+      reqBodyType: "formData",
+      bodyType: "json"
+    });
+    return $dara.cast(await this.doRPCRequest(params.action, params.version, params.protocol, params.method, params.authType, params.bodyType, req, runtime), new GetAndLoadInternalContextResponse({}));
+  }
+  /**
+   * Get internal context request
+   *
+   * @param request - GetAndLoadInternalContextRequest
+   * @returns GetAndLoadInternalContextResponse
+   */
+  async getAndLoadInternalContext(request) {
+    let runtime = new $dara.RuntimeOptions({});
+    return await this.getAndLoadInternalContextWithOptions(request, runtime);
+  }
   /**
    * Get context list
    *
@@ -1274,6 +1314,12 @@ var _Client = class _Client extends (OpenApi.default || OpenApi) {
     if (!$dara.isNull(request.nextToken)) {
       body["NextToken"] = request.nextToken;
     }
+    if (!$dara.isNull(request.sessionId)) {
+      body["SessionId"] = request.sessionId;
+    }
+    if (!$dara.isNull(request.type)) {
+      body["ContextType"] = request.type;
+    }
     const req = new $OpenApiUtil.OpenApiRequest({
       body: OpenApiUtil.parseToMap(body)
     });
@@ -2006,7 +2052,6 @@ init_esm_shims();
 var dotenv = __toESM(require_main());
 import * as fs2 from "fs";
 import * as path2 from "path";
-var BROWSER_RECORD_PATH = "/home/wuying/record";
 function defaultConfig() {
   return {
     endpoint: "wuyingai.cn-shanghai.aliyuncs.com",
@@ -2701,7 +2746,8 @@ var _ContextService = class _ContextService {
       const request = new ListContextsRequest({
         authorization: `Bearer ${this.agentBay.getAPIKey()}`,
         maxResults,
-        nextToken: params?.nextToken
+        nextToken: params?.nextToken,
+        sessionId: params?.sessionId
       });
       logAPICall("ListContexts");
       logDebug(`Request: MaxResults=${maxResults}`, params?.nextToken ? `, NextToken=${params.nextToken}` : "");
@@ -10880,7 +10926,7 @@ var _Browser = class _Browser {
         logDebug(`VPC mode, endpoint_router_port: ${this.session.httpPort}`);
         this._endpointUrl = `ws://${this.session.networkInterfaceIp}:${this.session.httpPort}`;
       } else {
-        const { GetCdpLinkRequest: GetCdpLinkRequest2 } = await import("./model-5SHCBTKT.mjs");
+        const { GetCdpLinkRequest: GetCdpLinkRequest2 } = await import("./model-NXNY7XBJ.mjs");
         const request = new GetCdpLinkRequest2({
           authorization: `Bearer ${this.session.getAPIKey()}`,
           sessionId: this.session.sessionId
@@ -12932,13 +12978,14 @@ import fetch2 from "node-fetch";
 var _FileTransfer = class _FileTransfer {
   /**
    * Initialize FileTransfer with AgentBay client and session.
-   * 
+   *
    * @param agentBay - AgentBay instance for context service access
    * @param session - Created session object for context operations
    * @param httpTimeout - HTTP request timeout in seconds (default: 60.0)
    * @param followRedirects - Whether to follow HTTP redirects (default: true)
    */
   constructor(agentBay, session, httpTimeout = 60, followRedirects = true) {
+    this.contextPath = null;
     // Task completion states (for compatibility)
     this.finishedStates = /* @__PURE__ */ new Set(["success", "successful", "ok", "finished", "done", "completed", "complete"]);
     this.agentBay = agentBay;
@@ -12946,7 +12993,60 @@ var _FileTransfer = class _FileTransfer {
     this.session = session;
     this.httpTimeout = httpTimeout;
     this.followRedirects = followRedirects;
-    this.contextId = session.fileTransferContextId || "";
+    this.contextId = "";
+  }
+  /**
+   * Get the current context ID.
+   * @returns The context ID, or empty string if not yet loaded
+   */
+  getContextId() {
+    return this.contextId;
+  }
+  /**
+   * Ensure the file transfer context id is set by calling GetAndLoadInternalContext with SessionId and ContextTypes=["file_transfer"].
+   * Returns [success: boolean, errorMessage: string] tuple.
+   */
+  async ensureContextId() {
+    if (this.contextId) {
+      return [true, ""];
+    }
+    try {
+      const sid = this.session.getSessionId();
+      if (!sid) return [false, "No session ID"];
+      const client = this.agentBay.getClient();
+      let response;
+      try {
+        const req = {
+          authorization: `Bearer ${this.agentBay.getAPIKey()}`,
+          sessionId: sid,
+          contextTypes: ["file_transfer"]
+        };
+        response = await client.getAndLoadInternalContext(req);
+      } catch (e) {
+        return [false, e?.message || String(e)];
+      }
+      const body = response && response.body ? response.body : {};
+      if (body.Success === false && body.Code) {
+        return [false, body.Message || "Unknown error"];
+      }
+      const data = body.Data || body.data;
+      if (Array.isArray(data) && data.length > 0) {
+        for (const item of data) {
+          if (item && typeof item === "object") {
+            const context_id = item.ContextId || item.contextId;
+            const context_path = item.ContextPath || item.contextPath;
+            if (context_id && context_path) {
+              this.contextId = context_id;
+              this.contextPath = context_path;
+              return [true, ""];
+            }
+          }
+        }
+      }
+      return [false, "Response contains no data"];
+    } catch (e) {
+      return [false, e?.message || String(e)];
+    }
   }
   /**
    * Upload workflow:
@@ -12975,12 +13075,15 @@ var _FileTransfer = class _FileTransfer {
         };
       }
       if (!this.contextId) {
-        return {
-          success: false,
-          bytesSent: 0,
-          path: remotePath,
-          error: "No context ID"
-        };
+        const [success, errorMsg] = await this.ensureContextId();
+        if (!success) {
+          return {
+            success: false,
+            bytesSent: 0,
+            path: remotePath,
+            error: errorMsg || "No context ID"
+          };
+        }
       }
       const urlRes = await this.contextSvc.getFileUploadUrl(this.contextId, remotePath);
       if (!urlRes.success || !urlRes.url) {
@@ -13090,7 +13193,7 @@ var _FileTransfer = class _FileTransfer {
    * 1) Trigger session.context.sync(mode="upload") to sync cloud disk data to OSS
    * 2) Get pre-signed download URL via context.getFileDownloadUrl
    * 3) Download the file and save to local localPath
-   * 4) If wait=true, wait for download task to reach completion after step 1 
+   * 4) If wait=true, wait for download task to reach completion after step 1
    *    (ensuring backend has prepared the download object)
    *
    * Returns DownloadResult containing sync and download request_ids, HTTP status, byte count, etc.
@@ -13105,13 +13208,16 @@ var _FileTransfer = class _FileTransfer {
     } = options || {};
     try {
       if (!this.contextId) {
-        return {
-          success: false,
-          bytesReceived: 0,
-          path: remotePath,
-          localPath,
-          error: "No context ID"
-        };
+        const [success, errorMsg] = await this.ensureContextId();
+        if (!success) {
+          return {
+            success: false,
+            bytesReceived: 0,
+            path: remotePath,
+            localPath,
+            error: errorMsg || "No context ID"
+          };
+        }
       }
       let reqIdSync;
       try {
@@ -13470,7 +13576,7 @@ var _FileSystem = class _FileSystem {
   }
   /**
    * Ensure FileTransfer is initialized with the current session.
-   * 
+   *
    * @returns The FileTransfer instance
    */
   _ensureFileTransfer() {
@@ -13483,6 +13589,30 @@ var _FileSystem = class _FileSystem {
     }
     return this._fileTransfer;
   }
+  /**
+   * Get the context path for file transfer operations.
+   *
+   * This method ensures the context ID is loaded and returns the associated
+   * context path that was retrieved from GetAndLoadInternalContext API.
+   *
+   * @returns The context path if available, null otherwise.
+   *
+   * @example
+   * ```typescript
+   * const session = (await agentBay.create(params)).session;
+   * const contextPath = await session.fileSystem.getFileTransferContextPath();
+   * if (contextPath) {
+   *   console.log(`Context path: ${contextPath}`);
+   * }
+   * ```
+   */
+  async getFileTransferContextPath() {
+    const fileTransfer = this._ensureFileTransfer();
+    if (!fileTransfer.getContextId()) {
+      await fileTransfer.ensureContextId();
+    }
+    return fileTransfer.contextPath || null;
+  }
   /**
    * Creates a new directory at the specified path.
    * Corresponds to Python's create_directory() method
@@ -14382,8 +14512,8 @@ var _FileSystem = class _FileSystem {
     try {
       const fileTransfer = this._ensureFileTransfer();
       const result = await fileTransfer.upload(localPath, remotePath, options);
-      if (result.success && this.session.fileTransferContextId) {
-        const contextId = this.session.fileTransferContextId;
+      if (result.success) {
+        const contextId = fileTransfer.contextId;
         if (contextId) {
           try {
             const deleteResult = await this.session.agentBay.context.deleteFile(contextId, remotePath);
@@ -14430,8 +14560,8 @@ var _FileSystem = class _FileSystem {
     try {
       const fileTransfer = this._ensureFileTransfer();
       const result = await fileTransfer.download(remotePath, localPath, options);
-      if (result.success && this.session.fileTransferContextId) {
-        const contextId = this.session.fileTransferContextId;
+      if (result.success) {
+        const contextId = fileTransfer.contextId;
         if (contextId) {
           try {
             const deleteResult = await this.session.agentBay.context.deleteFile(contextId, remotePath);
@@ -14942,7 +15072,7 @@ var _Mobile = class _Mobile {
   async getAdbUrl(adbkeyPub) {
     try {
       const optionsJson = JSON.stringify({ adbkey_pub: adbkeyPub });
-      const { GetAdbLinkRequest: GetAdbLinkRequest2 } = await import("./model-5SHCBTKT.mjs");
+      const { GetAdbLinkRequest: GetAdbLinkRequest2 } = await import("./model-NXNY7XBJ.mjs");
       const request = new GetAdbLinkRequest2({
         authorization: `Bearer ${this.session.getAPIKey()}`,
         sessionId: this.session.sessionId,
@@ -15551,10 +15681,6 @@ var _Session = class _Session {
    * @param sessionId - The ID of this session.
    */
   constructor(agentBay, sessionId) {
-    // File transfer context ID
-    this.fileTransferContextId = null;
-    // Browser recording context ID
-    this.recordContextId = null;
     // VPC-related information
     this.isVpc = false;
     // Whether this session uses VPC resources
@@ -15721,32 +15847,15 @@ var _Session = class _Session {
    */
   async delete(syncContext = false) {
     try {
-      let shouldSync = false;
-      let syncContextId = null;
       if (syncContext) {
-        shouldSync = true;
-        logInfo("\u{1F504} User requested context synchronization");
-      } else if (this.enableBrowserReplay && this.recordContextId) {
-        shouldSync = true;
-        syncContextId = this.recordContextId;
-        logInfo(
-          `\u{1F3A5} Browser replay enabled - syncing recording context: ${syncContextId}`
-        );
-      }
-      if (shouldSync) {
         logDebug(
           "Triggering context synchronization before session deletion..."
         );
         const syncStartTime = Date.now();
         try {
           let syncResult;
-          if (syncContextId) {
-            syncResult = await this.context.sync(syncContextId, BROWSER_RECORD_PATH);
-            logInfo(`\u{1F3A5} Synced browser recording context: ${syncContextId}`);
-          } else {
-            syncResult = await this.context.sync();
-            logInfo("\u{1F504} Synced all contexts");
-          }
+          syncResult = await this.context.sync();
+          logInfo("\u{1F504} Synced all contexts");
           const syncDuration = Date.now() - syncStartTime;
           if (syncResult.success) {
             logInfo(`Context sync completed in ${syncDuration}ms`);
@@ -16819,16 +16928,6 @@ var Session = _Session;
 
 // src/agent-bay.ts
 var BROWSER_DATA_PATH2 = "/tmp/agentbay_browser";
-function generateRandomContextName(length = 8, includeTimestamp = true) {
-  const timestamp = (/* @__PURE__ */ new Date()).toISOString().replace(/[-T:.Z]/g, "").slice(0, 14);
-  const characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
-  let randomPart = "";
-  for (let i = 0; i < length; i++) {
-    randomPart += characters.charAt(Math.floor(Math.random() * characters.length));
-  }
-  return includeTimestamp ? `${timestamp}_${randomPart}` : randomPart;
-}
-__name(generateRandomContextName, "generateRandomContextName");
 var _AgentBay = class _AgentBay {
   /**
    * Initialize the AgentBay client.
@@ -16839,7 +16938,6 @@ var _AgentBay = class _AgentBay {
    * @param options.envFile - Custom path to .env file. If not provided, will search upward from current directory.
    */
   constructor(options = {}) {
-    this.fileTransferContext = null;
     loadDotEnvWithFallback(options.envFile);
     this.apiKey = options.apiKey || process.env.AGENTBAY_API_KEY || "";
     if (!this.apiKey) {
@@ -16873,7 +16971,7 @@ var _AgentBay = class _AgentBay {
   }
   /**
    * Wait for mobile simulate command to complete.
-   * 
+   *
    * @param session - The session to wait for mobile simulate
    * @param mobileSimPath - The dev info path to the mobile simulate
    * @param mobileSimMode - The mode of the mobile simulate. If not provided, will use the default mode.
@@ -16924,35 +17022,6 @@ var _AgentBay = class _AgentBay {
       logInfo(`Error executing mobile simulate command: ${error}`);
     }
   }
-  /**
-   * Update browser replay context with AppInstanceId from response data.
-   *
-   * @param responseData - Response data containing AppInstanceId
-   * @param recordContextId - The record context ID to update
-   */
-  async _updateBrowserReplayContext(responseData, recordContextId) {
-    if (!recordContextId) {
-      return;
-    }
-    try {
-      const appInstanceId = responseData?.appInstanceId;
-      if (!appInstanceId) {
-        logError("AppInstanceId not found in response data, skipping browser replay context update");
-        return;
-      }
-      const contextName = `browserreplay-${appInstanceId}`;
-      const contextObj = new Context(recordContextId, contextName);
-      logDebug(`Updating browser replay context: ${contextName} -> ${recordContextId}`);
-      const updateResult = await this.context.update(contextObj);
-      if (updateResult.success) {
-        logInfo(`\u2705 Successfully updated browser replay context: ${contextName}`);
-      } else {
-        logError(`\u26A0\uFE0F Failed to update browser replay context: ${updateResult.errorMessage}`);
-      }
-    } catch (error) {
-      logError(`\u274C Error updating browser replay context: ${error}`);
-    }
-  }
   /**
    * Create a new session in the AgentBay cloud environment.
    *
@@ -17018,20 +17087,6 @@ var _AgentBay = class _AgentBay {
           params.contextSync.push(mobileSimContextSync);
         }
       }
-      const contextName = `file-transfer-context-${Date.now()}`;
-      const contextResult = await this.context.get(contextName, true);
-      if (contextResult.success && contextResult.context) {
-        this.fileTransferContext = contextResult.context;
-        const fileTransferContextSync = new ContextSync(
-          contextResult.context.id,
-          "/tmp/file-transfer"
-        );
-        if (!params.contextSync) {
-          params.contextSync = [];
-        }
-        logDebug(`Adding context sync for file transfer operations: ${fileTransferContextSync}`);
-        params.contextSync.push(fileTransferContextSync);
-      }
       const request = new CreateMcpSessionRequest({
         authorization: "Bearer " + this.apiKey
       });
@@ -17090,21 +17145,6 @@ var _AgentBay = class _AgentBay {
         request.persistenceDataList.push(browserContextSync);
         needsContextSync = true;
       }
-      let recordContextId = "";
-      if (params.enableBrowserReplay) {
-        const recordPath = BROWSER_RECORD_PATH;
-        const recordContextName = generateRandomContextName();
-        const result = await this.context.get(recordContextName, true);
-        recordContextId = result.success ? result.contextId : "";
-        const recordPersistence = new CreateMcpSessionRequestPersistenceDataList({
-          contextId: recordContextId,
-          path: recordPath
-        });
-        if (!request.persistenceDataList) {
-          request.persistenceDataList = [];
-        }
-        request.persistenceDataList.push(recordPersistence);
-      }
       if (params.extraConfigs) {
         request.extraConfigs = JSON.stringify(params.extraConfigs);
         if (params.extraConfigs.mobile?.simulateConfig?.simulate) {
@@ -17230,8 +17270,6 @@ var _AgentBay = class _AgentBay {
       }
       session.resourceUrl = resourceUrl;
       session.enableBrowserReplay = params.enableBrowserReplay || false;
-      session.fileTransferContextId = this.fileTransferContext ? this.fileTransferContext.id : null;
-      session.recordContextId = recordContextId || null;
       session.imageId = params.imageId;
       if (params.extraConfigs && params.extraConfigs.mobile) {
         log("Applying mobile configuration...");
@@ -17246,9 +17284,6 @@ var _AgentBay = class _AgentBay {
           logError(`Warning: Failed to apply mobile configuration: ${error}`);
         }
       }
-      if (params.enableBrowserReplay) {
-        await this._updateBrowserReplayContext(data, recordContextId);
-      }
       if (params.isVpc) {
         logDebug("VPC session detected, automatically fetching MCP tools...");
         try {
@@ -17641,41 +17676,6 @@ var _AgentBay = class _AgentBay {
       session.token = getResult.data.token;
       session.resourceUrl = getResult.data.resourceUrl;
     }
-    if (getResult.data && getResult.data.contexts && getResult.data.contexts.length > 0) {
-      const contexts = getResult.data.contexts;
-      const fileTransferContexts = contexts.filter(
-        (ctx) => ctx.name && ctx.name.startsWith("file-transfer-context-")
-      );
-      if (fileTransferContexts.length === 0) {
-        const availableContexts = contexts.map((ctx) => ctx.name || "unknown");
-        logWarn(
-          `\u26A0\uFE0F  No file-transfer-context- found in contexts list for session ${sessionId}. Available contexts: ${availableContexts.join(", ")}`
-        );
-        session.fileTransferContextId = null;
-      } else if (fileTransferContexts.length === 1) {
-        const contextId = fileTransferContexts[0].id;
-        if (contextId) {
-          session.fileTransferContextId = contextId;
-          logInfo(`\u{1F4C1} Found file transfer context for recovered session: ${contextId}`);
-        } else {
-          logWarn(
-            `\u26A0\uFE0F  File transfer context found but missing 'id' field: ${JSON.stringify(fileTransferContexts[0])}`
-          );
-          session.fileTransferContextId = null;
-        }
-      } else {
-        const contextNames = fileTransferContexts.map((ctx) => ctx.name || "unknown");
-        logWarn(
-          `\u26A0\uFE0F  Multiple file-transfer-context- found in contexts list for session ${sessionId}. Found ${fileTransferContexts.length} contexts: ${contextNames.join(", ")}. Not setting fileTransferContextId.`
-        );
-        session.fileTransferContextId = null;
-      }
-    } else {
-      logWarn(
-        `\u26A0\uFE0F  No contexts list found in GetSession response for session ${sessionId}. fileTransferContextId will remain null.`
-      );
-      session.fileTransferContextId = null;
-    }
     return {
       requestId: getResult.requestId,
       success: true,
@@ -19042,6 +19042,10 @@ export {
   GetAdbLinkResponse,
   GetAdbLinkResponseBody,
   GetAdbLinkResponseBodyData,
+  GetAndLoadInternalContextRequest,
+  GetAndLoadInternalContextResponse,
+  GetAndLoadInternalContextResponseBody,
+  GetAndLoadInternalContextResponseBodyData,
   GetCdpLinkRequest,
   GetCdpLinkResponse,
   GetCdpLinkResponseBody,