@airtop/sdk 1.0.0-alpha2.0 → 1.0.0-alpha2.10

package/dist/index.js

@@ -9,7 +9,7 @@ var require_package = __commonJS({
     module.exports = {
       name: "@airtop/sdk",
       description: "Airtop SDK for TypeScript",
-      version: "1.0.0-alpha2.0",
+      version: "1.0.0-alpha2.10",
       type: "module",
       main: "./dist/index.cjs",
       module: "./dist/index.js",
@@ -47,7 +47,7 @@ var require_package = __commonJS({
       },
       dependencies: {
         "@airtop/json-schema-adapter": "workspace:*",
-        "@airtop/core": "0.1.0-alpha.17",
+        "@airtop/core": "0.1.0-alpha.27",
         "date-fns": "4.1.0",
         loglayer: "6.3.3",
         "serialize-error": "12.0.0",
@@ -102,6 +102,11 @@ var AirtopBase = class {
    * @internal
    **/
   outputJsonAdapter;
+  /**
+   * The job id for the ongoing automation
+   * @internal
+   */
+  jobId;
   /**
    * Creates a new instance of AirtopBase
    * @param config - Configuration options for the SDK
@@ -110,6 +115,7 @@ var AirtopBase = class {
     this.log = config.log.withPrefix("[Airtop]");
     this.client = config.client;
     this.outputJsonAdapter = config.outputSchemaAdapter;
+    this.jobId = config.jobId;
   }
   /**
    * Returns the API key used by the SDK
@@ -163,7 +169,8 @@ var AirtopBase = class {
   getCommonConfig() {
     return {
       client: this.client,
-      log: this.log
+      log: this.log,
+      jobId: this.jobId
     };
   }
   /**
@@ -235,7 +242,118 @@ var processLogMessage = (log, logLevel, ...args) => {
 import { NotFoundError } from "@airtop/core";
 
 // src/window/AirtopWindowClient.ts
+import { secondsToMilliseconds as secondsToMilliseconds3 } from "date-fns";
+
+// src/async-utils.ts
 import { secondsToMilliseconds as secondsToMilliseconds2 } from "date-fns";
+var DEFAULT_TIMEOUT_SECONDS = 300;
+var DEFAULT_POLLING_INTERVAL_SECONDS = 2;
+async function waitForRequestCompletion(client, requestId, requestOptions) {
+  const startTime = Date.now();
+  const timeoutMs = secondsToMilliseconds2(requestOptions?.timeoutInSeconds || DEFAULT_TIMEOUT_SECONDS);
+  while (Date.now() - startTime < timeoutMs) {
+    const apiResponse = await client.requests.getRequestStatus(requestId, requestOptions);
+    if (apiResponse.status === "completed") {
+      return apiResponse.response;
+    }
+    if (apiResponse.status === "error") {
+      throw new Error(apiResponse.error);
+    }
+    await new Promise((resolve) => setTimeout(resolve, secondsToMilliseconds2(DEFAULT_POLLING_INTERVAL_SECONDS)));
+  }
+  throw new Error("Waiting for request timed out");
+}
+async function withRequestCompletionPolling(client, fn, requestOptions) {
+  const response = await fn();
+  return waitForRequestCompletion(client, response.requestId, requestOptions);
+}
+
+// src/window/AirtopNode.ts
+var AirtopNode = class {
+  /**
+   * The window client
+   */
+  windowClient;
+  /**
+   * The node handle id to use for all requests
+   */
+  nodeHandleId;
+  /**
+   * The xpath selector of the node
+   */
+  selector;
+  /**
+   * Constructor
+   * @param client - The window client
+   * @param nodeData - The node data to use for all requests
+   */
+  constructor(client, nodeData) {
+    this.windowClient = client;
+    this.nodeHandleId = nodeData.id;
+    this.selector = nodeData.xpath;
+  }
+  /**
+   * Extract content from the node
+   * @param prompt - The prompt to use for the extraction
+   * @param config - The configuration to use for the extraction
+   * @param requestOptions - The request options to use for the extraction
+   */
+  async extract(prompt, config, requestOptions = {}) {
+    const augmentedConfig = {
+      ...config,
+      nodeHandleId: this.nodeHandleId
+    };
+    return this.windowClient.extract(prompt, augmentedConfig, requestOptions);
+  }
+  /**
+   * Act on the node
+   * @param prompt - The prompt to use for the action
+   * @param config - The configuration to use for the action
+   * @param requestOptions - The request options to use for the action
+   */
+  async act(prompt, config, requestOptions = {}) {
+    const augmentedConfig = {
+      ...config,
+      nodeHandleId: this.nodeHandleId
+    };
+    return this.windowClient.act(prompt, augmentedConfig, requestOptions);
+  }
+  /**
+   * Execute an LLM call on the node
+   * @param prompt - The prompt to use for the LLM call
+   * @param config - The configuration to use for the LLM call
+   * @param requestOptions - The request options to use for the LLM call
+   */
+  async llm(prompt, config, requestOptions = {}) {
+    return this.windowClient.llm(prompt, config, requestOptions);
+  }
+  /**
+   * Find one element in the node
+   * @param prompt - The prompt to use for the find one
+   * @param config - The configuration to use for the find one
+   * @param requestOptions - The request options to use for the find one
+   */
+  async findOne(prompt, config, requestOptions = {}) {
+    const augmentedConfig = {
+      ...config,
+      nodeHandleId: this.nodeHandleId
+    };
+    return this.windowClient.findOne(prompt, augmentedConfig, requestOptions);
+  }
+  /**
+   * Find many elements in the node
+   * @param prompt - The prompt to use for the find many
+   * @param config - The configuration to use for the find many
+   * @param requestOptions - The request options to use for the find many
+   */
+  async findMany(prompt, config, requestOptions = {}) {
+    const augmentedConfig = {
+      ...config,
+      nodeHandleId: this.nodeHandleId
+    };
+    return this.windowClient.findMany(prompt, augmentedConfig, requestOptions);
+  }
+};
 
 // src/window/AirtopWindowScreenshot.ts
 function extractMimeAndBase64(dataUrl) {
@@ -385,7 +503,7 @@ var AirtopWindowClient = class extends AirtopBase {
         sessionId: this.sessionId
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -417,7 +535,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -441,7 +559,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -463,7 +581,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -491,7 +609,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...newConfig || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -519,7 +637,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...newConfig || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -539,7 +657,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -559,7 +677,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -580,7 +698,7 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -604,11 +722,137 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config || {}
       },
       {
-        timeout: secondsToMilliseconds2(600),
+        timeout: secondsToMilliseconds3(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
   }
+  async extract(prompt, config, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Extracting content");
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.extract(
+        this.getWindowId(),
+        {
+          prompt,
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {}
+        },
+        {
+          timeout: secondsToMilliseconds3(600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  async act(prompt, config, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Acting on content");
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.act(
+        this.getWindowId(),
+        {
+          prompt,
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {}
+        },
+        {
+          timeout: secondsToMilliseconds3(600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  async llm(prompt, config, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Executing LLM call");
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.llm(
+        this.getWindowId(),
+        {
+          prompt,
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          timeout: secondsToMilliseconds3(600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  async findOne(prompt, config, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Executing LLM call");
+    const apiResponse = await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.findOne(
+        this.getWindowId(),
+        {
+          prompt,
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {}
+        },
+        {
+          timeout: secondsToMilliseconds3(600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+    try {
+      const nodeData = JSON.parse(apiResponse.data.modelResponse);
+      return new AirtopNode(this, nodeData);
+    } catch (error) {
+      this.log.withMetadata({ nodeDataStr: apiResponse.data.modelResponse }).withError(error).error("Error parsing node data");
+      throw error;
+    }
+  }
+  async findMany(prompt, config, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Executing LLM call");
+    const apiResponse = await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.findMany(
+        this.getWindowId(),
+        {
+          prompt,
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {}
+        },
+        {
+          timeout: secondsToMilliseconds3(600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+    const nodeHandles = apiResponse.data.modelResponse.split("___DELIM___");
+    return nodeHandles.map((nodeDataStr) => {
+      try {
+        const nodeData = JSON.parse(nodeDataStr);
+        return new AirtopNode(this, nodeData);
+      } catch (error) {
+        this.log.withMetadata({ nodeDataStr }).withError(error).error("Error parsing node data");
+      }
+    });
+  }
+  async waitForPage(config, requestOptions = {}) {
+    return await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.waitForPage(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          timeout: secondsToMilliseconds3(600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
 };
 
 // src/window/AirtopWindow.ts
@@ -782,7 +1026,7 @@ var AirtopSessionClient = class extends AirtopBase {
       {
         ...config,
         fileName,
-        sessionId: this.sessionId
+        sessionIds: [this.sessionId]
       },
       this.resolveRequestOptions(requestOptions)
     );
@@ -868,7 +1112,8 @@ var AirtopClient = class extends AirtopBase {
         contextFieldName: "context",
         metadataFieldName: "metadata"
       }),
-      outputSchemaAdapter: config.outputSchemaAdapter
+      outputSchemaAdapter: config.outputSchemaAdapter,
+      jobId: config.jobId
     });
     this.log.withPrefix("[Airtop SDK]");
     this.client.logLevel = config.logLevel;
@@ -894,8 +1139,8 @@ var AirtopClient = class extends AirtopBase {
    * @returns A new AirtopSession instance
    */
   async createSession(config, options = {}) {
-    const skipWaitSessionReady = config.skipWaitSessionReady ?? false;
-    delete config.skipWaitSessionReady;
+    const skipWaitSessionReady = config?.skipWaitSessionReady ?? false;
+    delete config?.skipWaitSessionReady;
     const sessionResponse = await this.client.sessions.create(
       {
         configuration: config
@@ -1003,23 +1248,6 @@ var AirtopClient = class extends AirtopBase {
   async getFile(fileId, requestOptions = {}) {
     return this.client.files.get(fileId, this.resolveRequestOptions(requestOptions));
   }
-  /**
-   * Lists files filtered by session IDs.
-   * @param sessionIds - Array of session IDs to retrieve files for
-   * @param config - Configuration options for the request
-   * @param requestOptions - Request options
-   * @returns Object containing pagination info and array of AirtopSession instances
-   */
-  async listFiles(sessionIds, config, requestOptions = {}) {
-    return this.client.files.list(
-      {
-        sessionIds,
-        limit: config.limit,
-        offset: config.offset
-      },
-      this.resolveRequestOptions(requestOptions)
-    );
-  }
   /**
    * Removes a file by ID.
    * @param fileId - ID of the file to remove
@@ -1028,6 +1256,14 @@ var AirtopClient = class extends AirtopBase {
   async removeFile(fileId, requestOptions = {}) {
     return this.client.files.delete(fileId, this.resolveRequestOptions(requestOptions));
   }
+  /**
+   * List files
+   * @param query
+   * @param requestOptions
+   */
+  async listFiles(query, requestOptions = {}) {
+    return this.client.files.list(query, this.resolveRequestOptions(requestOptions));
+  }
   /**
    * List all automations
    * @param requestOptions - Request options
@@ -1184,6 +1420,7 @@ export {
   AirtopBase,
   AirtopClient,
   AirtopMocks,
+  AirtopNode,
   AirtopPluginAugmentationType,
   AirtopSession,
   AirtopSessionClient,