@airtop/sdk 1.0.0-alpha2.45 → 1.0.0-alpha2.46

package/dist/index.cjs

@@ -1,4 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } function _optionalChainDelete(ops) { const result = _optionalChain(ops); return result == null ? true : result; }var __getOwnPropNames = Object.getOwnPropertyNames;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } function _optionalChainDelete(ops) { const result = _optionalChain(ops); return result == null ? true : result; }var __getOwnPropNames = Object.getOwnPropertyNames;
 var __commonJS = (cb, mod) => function __require() {
   return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
 };
@@ -9,7 +9,7 @@ var require_package = __commonJS({
     module.exports = {
       name: "@airtop/sdk",
       description: "Airtop SDK for TypeScript",
-      version: "1.0.0-alpha2.45",
+      version: "1.0.0-alpha2.46",
       type: "module",
       main: "./dist/index.cjs",
       module: "./dist/index.js",
@@ -198,7 +198,7 @@ var AirtopBase = class {
    * @internal
    */
   convertToJsonSchema(input) {
-    if (input && _optionalChain([this, 'access', _ => _.outputJsonAdapter, 'optionalAccess', _2 => _2.shouldRunConversion, 'call', _3 => _3(input)])) {
+    if (input && _optionalChain([this, 'access', _2 => _2.outputJsonAdapter, 'optionalAccess', _3 => _3.shouldRunConversion, 'call', _4 => _4(input)])) {
       return JSON.stringify(this.outputJsonAdapter.toJsonSchema(input));
     }
     if (typeof input === "object") {
@@ -350,7 +350,7 @@ var processLogMessage = (log, logLevel, ...args) => {
   }
   let finalMessage = messages.join(" ").trim();
   if (finalMessage === "\u2A2F" && error) {
-    finalMessage = _optionalChain([error, 'optionalAccess', _4 => _4.message]) || "";
+    finalMessage = _optionalChain([error, 'optionalAccess', _5 => _5.message]) || "";
   }
   if (error && hasData && messages.length > 0) {
     log.withError(error).withMetadata(data)[logLevel](finalMessage);
@@ -370,6 +370,8 @@ var processLogMessage = (log, logLevel, ...args) => {
 };
 
 // src/session/AirtopSessionClient.ts
+var _fs = require('fs'); var _fs2 = _interopRequireDefault(_fs);
+var _path = require('path'); var _path2 = _interopRequireDefault(_path);
 
 
 // src/async-utils.ts
@@ -392,7 +394,7 @@ ${JSON.stringify(metadata)}` : errorMessage;
 // src/async-utils.ts
 async function waitForRequestCompletion(client, requestId, requestOptions) {
   const startTime = Date.now();
-  const timeoutMs = _datefns.secondsToMilliseconds.call(void 0, _optionalChain([requestOptions, 'optionalAccess', _5 => _5.timeoutInSeconds]) || TIMEOUT_SECONDS_DEFAULT_VALUE);
+  const timeoutMs = _datefns.secondsToMilliseconds.call(void 0, _optionalChain([requestOptions, 'optionalAccess', _6 => _6.timeoutInSeconds]) || TIMEOUT_SECONDS_DEFAULT_VALUE);
   while (Date.now() - startTime < timeoutMs) {
     const apiResponse = await client.requests.getRequestStatusV2(requestId, requestOptions);
     if (apiResponse.status === "completed") {
@@ -410,7 +412,7 @@ async function withRequestCompletionPolling(client, fn, requestOptions) {
     const response = await fn();
     return waitForRequestCompletion(client, response.requestId, requestOptions);
   } catch (thrownError) {
-    if (_optionalChain([thrownError, 'access', _6 => _6.error, 'optionalAccess', _7 => _7.errors])) {
+    if (_optionalChain([thrownError, 'access', _7 => _7.error, 'optionalAccess', _8 => _8.errors])) {
       throw new AirtopError(thrownError.error.errors, {
         requestId: thrownError.error.requestId
       });
@@ -772,7 +774,7 @@ var AirtopWindowClient = class extends AirtopBase {
    */
   async monitor(condition, config = {}, requestOptions = {}) {
     this.log.withMetadata().info("Monitoring window");
-    if (!_optionalChain([config, 'optionalAccess', _8 => _8.configuration, 'optionalAccess', _9 => _9.interval, 'optionalAccess', _10 => _10.timeoutSeconds])) {
+    if (!_optionalChain([config, 'optionalAccess', _9 => _9.configuration, 'optionalAccess', _10 => _10.interval, 'optionalAccess', _11 => _11.timeoutSeconds])) {
       if (!config.configuration) {
         config.configuration = {};
       }
@@ -781,7 +783,7 @@ var AirtopWindowClient = class extends AirtopBase {
       }
       config.configuration.interval.timeoutSeconds = this.defaultTimeoutInSeconds;
     }
-    if (!_optionalChain([config, 'optionalAccess', _11 => _11.timeThresholdSeconds])) {
+    if (!_optionalChain([config, 'optionalAccess', _12 => _12.timeThresholdSeconds])) {
       config.timeThresholdSeconds = this.defaultTimeoutInSeconds;
     }
     return withRequestCompletionPolling(
@@ -812,7 +814,7 @@ var AirtopWindowClient = class extends AirtopBase {
       prompt
     }).info("Performing a page query");
     const newConfig = config;
-    if (_optionalChain([config, 'optionalAccess', _12 => _12.configuration, 'access', _13 => _13.outputSchema])) {
+    if (_optionalChain([config, 'optionalAccess', _13 => _13.configuration, 'access', _14 => _14.outputSchema])) {
       newConfig.configuration.outputSchema = this.convertToJsonSchema(config.configuration.outputSchema);
     }
     return withRequestCompletionPolling(
@@ -845,7 +847,7 @@ var AirtopWindowClient = class extends AirtopBase {
       prompt
     }).info("Performing a paginated extraction");
     const newConfig = config;
-    if (_optionalChain([config, 'optionalAccess', _14 => _14.configuration, 'access', _15 => _15.outputSchema])) {
+    if (_optionalChain([config, 'optionalAccess', _15 => _15.configuration, 'access', _16 => _16.outputSchema])) {
       newConfig.configuration.outputSchema = this.convertToJsonSchema(config.configuration.outputSchema);
     }
     return withRequestCompletionPolling(
@@ -1014,7 +1016,7 @@ var AirtopWindowClient = class extends AirtopBase {
           ...config || {},
           includeWebContext: true,
           // Always include web context for window.llm() calls
-          outputSchema: _optionalChain([config, 'optionalAccess', _16 => _16.outputSchema]) ? this.convertToJsonSchema(config.outputSchema) : void 0
+          outputSchema: _optionalChain([config, 'optionalAccess', _17 => _17.outputSchema]) ? this.convertToJsonSchema(config.outputSchema) : void 0
         },
         {
           ...this.resolveRequestOptions(requestOptions)
@@ -1137,7 +1139,7 @@ var AirtopWindowClient = class extends AirtopBase {
         {
           sessionId: this.sessionId,
           ...config || {},
-          automationId: _optionalChain([config, 'optionalAccess', _17 => _17.automationId]) || "auto",
+          automationId: _optionalChain([config, 'optionalAccess', _18 => _18.automationId]) || "auto",
           parameters: {
             customData: typeof formData === "string" ? formData : JSON.stringify(formData)
             // Will be interpreted by the LLM
@@ -1174,6 +1176,41 @@ var AirtopWindowClient = class extends AirtopBase {
       requestOptions
     );
   }
+  /**
+   * fileInput fills out a file input field in the window, using a file from the session
+   * @param fileId - The ID of the file to use
+   * @param config - Configuration options for the file input
+   * @param requestOptions - Request options
+   * @returns Promise resolving to the file input operation result
+   */
+  async fileInput(fileId, config, requestOptions = {}) {
+    return await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.fileInputAsync(
+        this.getWindowId(),
+        { sessionId: this.sessionId, ...config || {}, fileId },
+        { ...this.resolveRequestOptions(requestOptions) }
+      ),
+      requestOptions
+    );
+  }
+  /**
+   * uploadFileAndSelectInput uploads a file to the session, gets it ready to upload,
+   * and selects the file in the indicated input field
+   * @param filePath - The path to the file to upload
+   * @param config - Configuration options for the upload file and select input operation
+   * @param requestOptions - Request options
+   * @returns Promise resolving to the upload file and select input operation result
+   */
+  async uploadFileAndSelectInput({ filePath }, config, requestOptions = {}) {
+    const sessionClient = new AirtopSessionClient(this.getCommonConfig(), this.sessionId);
+    const file = await sessionClient.uploadFile(filePath, { pushToSession: true }, requestOptions);
+    return await this.fileInput(
+      file.id,
+      { ...config, fileId: file.id },
+      { ...this.resolveRequestOptions(requestOptions) }
+    );
+  }
 };
 
 // src/window/AirtopWindow.ts
@@ -1393,7 +1430,7 @@ var AirtopSessionClient = class extends AirtopBase {
           ...config || {},
           includeWebContext: false,
           // Do not include web context for session.llm() calls
-          outputSchema: _optionalChain([config, 'optionalAccess', _18 => _18.outputSchema]) ? this.convertToJsonSchema(config.outputSchema) : void 0
+          outputSchema: _optionalChain([config, 'optionalAccess', _19 => _19.outputSchema]) ? this.convertToJsonSchema(config.outputSchema) : void 0
         },
         {
           ...this.resolveRequestOptions(requestOptions)
@@ -1463,7 +1500,7 @@ var AirtopSessionClient = class extends AirtopBase {
     const { lookbackSeconds = 5, fileType } = configuration || {};
     this.log.info(`waiting for file to be available on session: ${this.sessionId}`);
     const startTime = /* @__PURE__ */ new Date();
-    const timeoutSeconds = _optionalChain([requestOptions, 'optionalAccess', _19 => _19.timeoutInSeconds]) || 120;
+    const timeoutSeconds = _optionalChain([requestOptions, 'optionalAccess', _20 => _20.timeoutInSeconds]) || 120;
     const timeoutPromise = new Promise((resolve) => {
       setTimeout(() => {
         this.log.info(`waiting for file timed out after ${timeoutSeconds} seconds`);
@@ -1532,6 +1569,102 @@ ${JSON.stringify(event, null, 2)}`);
     })();
     return Promise.race([timeoutPromise, processEventsPromise]);
   }
+  /**
+   * Uploads a file to the session and waits for it to be available.
+   * @param filePath - The path to the file to upload
+   * @param options - Additional configuration options for uploading the file,
+   * @param options.fileName - The name of the file to upload, defaults to the basename of the file path
+   * @param options.fileType - The type of the file to upload, defaults to "customer_upload"
+   * @param options.sessionIds - The IDs of the sessions to upload the file to, defaults to the current session id
+   * @param options.waitForUpload - Whether to wait for the upload to complete, defaults to true
+   * @param options.pushToSession - Whether to push the file to the session, defaults to true, waits for the upload to complete regardless of waitForUpload setting
+   * @param requestOptions - Request options
+   * @returns the uploaded file result
+   */
+  async uploadFile(filePath, options = {}, requestOptions = {}) {
+    this.log.info(`uploading file ${filePath}`);
+    const fileName = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _21 => _21.fileName]), () => ( _path2.default.basename(filePath)));
+    const fileType = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _22 => _22.fileType]), () => ( "customer_upload"));
+    const sessionIds = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _23 => _23.sessionIds]), () => ( [this.sessionId]));
+    const waitForUpload = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _24 => _24.waitForUpload]), () => ( true));
+    const pushToSession = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _25 => _25.pushToSession]), () => ( true));
+    const file = await this.client.files.createFile(
+      {
+        sessionIds,
+        fileName,
+        fileType
+      },
+      this.resolveRequestOptions(requestOptions)
+    );
+    const fileContent = _fs2.default.readFileSync(filePath);
+    if (waitForUpload || pushToSession) {
+      const uploadWaitPromise = this.waitForUpload(file.data.id, "uploaded");
+      fetch(file.data.uploadUrl, {
+        method: "PUT",
+        body: fileContent,
+        headers: {
+          "Content-Type": "application/octet-stream"
+        }
+      });
+      await uploadWaitPromise;
+    }
+    if (pushToSession) {
+      const pushWaitPromise = this.waitForUpload(file.data.id, "pushed");
+      this.client.files.push(file.data.id, {
+        sessionIds
+      });
+      await pushWaitPromise;
+    }
+    const fileDataResponse = await this.client.files.get(file.data.id);
+    const fileData = fileDataResponse.data;
+    return {
+      id: file.data.id,
+      downloadUrl: fileData.downloadUrl,
+      fileName: file.data.fileName,
+      fileType: file.data.fileType
+    };
+  }
+  /**
+   * Waits for a file to be uploaded, pushed to a session, then reach 'available' status. Throws an error if the file upload fails or times out.
+   * @param fileId - The ID of the file to wait for
+   * @param waitType - The type of wait to perform, "uploaded": just the file being uploaded to Airtop, "pushed": the file being pushed to a session
+   * @param timeoutSeconds - The timeout in seconds, defaults to 300
+   * @returns void
+   */
+  async waitForUpload(fileId, waitType = "uploaded", timeoutSeconds = 300) {
+    const waitEvent = waitType === "pushed" ? "file_upload_status" : "file_status";
+    const timeoutPromise = new Promise((_, reject) => {
+      setTimeout(() => {
+        reject(new Error(`waiting for file upload timed out after ${timeoutSeconds} seconds`));
+      }, timeoutSeconds * 1e3);
+    });
+    const processEventsPromise = (async () => {
+      const newEvents = await this.client.sessions.getEvents(
+        this.sessionId,
+        { all: false, reverse: false },
+        { timeout: timeoutSeconds * 1e3 }
+      );
+      for await (const event of newEvents) {
+        const e = event;
+        this.log.info(`waitForUpload processing event:
+${JSON.stringify(event, null, 2)}`);
+        if (e.event === waitEvent) {
+          if (e.fileId === fileId) {
+            if (e.status === "available") {
+              this.log.info(`uploaded file is now available on session: ${fileId}`);
+              return;
+            }
+            if (e.status === "upload_failed") {
+              this.log.info(`upload failed: ${this.sessionId} ${fileId}`);
+              throw new Error(`upload failed: ${e.eventData.error}`);
+            }
+          }
+        }
+      }
+      return null;
+    })();
+    return Promise.race([timeoutPromise, processEventsPromise]);
+  }
 };
 
 // src/session/AirtopSession.ts
@@ -1590,20 +1723,20 @@ var AirtopClient = class extends AirtopBase {
    */
   constructor(config) {
     super({
-      logLevel: _optionalChain([config, 'optionalAccess', _20 => _20.logLevel]),
+      logLevel: _optionalChain([config, 'optionalAccess', _26 => _26.logLevel]),
       client: new (0, _core.Airtop)({
         maxRetries: 0,
         timeout: _secondsToMilliseconds.secondsToMilliseconds.call(void 0, _nullishCoalesce(config.defaultTimeoutInSeconds, () => ( TIMEOUT_SECONDS_DEFAULT_VALUE))),
         apiKey: config.apiKey,
-        baseURL: _optionalChain([config, 'optionalAccess', _21 => _21.airtopUrl]),
-        logLevel: _optionalChain([config, 'optionalAccess', _22 => _22.logLevel]) || "off",
+        baseURL: _optionalChain([config, 'optionalAccess', _27 => _27.airtopUrl]),
+        logLevel: _optionalChain([config, 'optionalAccess', _28 => _28.logLevel]) || "off",
         defaultHeaders: {
           "x-airtop-sdk-source": "typescript",
           "x-airtop-sdk-version": version,
-          "x-airtop-sdk-environment": _nullishCoalesce(_optionalChain([config, 'optionalAccess', _23 => _23.environment]), () => ( "default"))
+          "x-airtop-sdk-environment": _nullishCoalesce(_optionalChain([config, 'optionalAccess', _29 => _29.environment]), () => ( "default"))
         }
       }),
-      log: _optionalChain([config, 'optionalAccess', _24 => _24.logger]) || new (0, _loglayer.LogLayer)({
+      log: _optionalChain([config, 'optionalAccess', _30 => _30.logger]) || new (0, _loglayer.LogLayer)({
         errorSerializer: _serializeerror.serializeError,
         transport: new (0, _loglayer.ConsoleTransport)({
           logger: console,
@@ -1643,8 +1776,8 @@ var AirtopClient = class extends AirtopBase {
    * @returns A new AirtopSession instance
    */
   async createSession(config, options = {}) {
-    const skipWaitSessionReady = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _25 => _25.skipWaitSessionReady]), () => ( false));
-     _optionalChainDelete([config, 'optionalAccess', _26 => delete _26.skipWaitSessionReady]);
+    const skipWaitSessionReady = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _31 => _31.skipWaitSessionReady]), () => ( false));
+     _optionalChainDelete([config, 'optionalAccess', _32 => delete _32.skipWaitSessionReady]);
     const sessionResponse = await this.client.sessions.create(
       {
         configuration: config
@@ -1900,19 +2033,19 @@ var AirtopAgentClient = class extends AirtopBase {
    */
   constructor(config) {
     super({
-      logLevel: _optionalChain([config, 'optionalAccess', _27 => _27.logLevel]),
+      logLevel: _optionalChain([config, 'optionalAccess', _33 => _33.logLevel]),
       client: new (0, _core.Airtop)({
         maxRetries: 0,
         timeout: _datefns.minutesToMilliseconds.call(void 0, 1),
         apiKey: config.apiKey,
-        baseURL: _optionalChain([config, 'optionalAccess', _28 => _28.airtopUrl]),
-        logLevel: _optionalChain([config, 'optionalAccess', _29 => _29.logLevel]) || "off",
+        baseURL: _optionalChain([config, 'optionalAccess', _34 => _34.airtopUrl]),
+        logLevel: _optionalChain([config, 'optionalAccess', _35 => _35.logLevel]) || "off",
         defaultHeaders: {
           "x-airtop-sdk-source": "typescript",
           "x-airtop-sdk-version": _process.version
         }
       }),
-      log: _optionalChain([config, 'optionalAccess', _30 => _30.logger]) || new (0, _loglayer.LogLayer)({
+      log: _optionalChain([config, 'optionalAccess', _36 => _36.logger]) || new (0, _loglayer.LogLayer)({
         errorSerializer: _serializeerror.serializeError,
         transport: new (0, _loglayer.ConsoleTransport)({
           logger: console,