@airtop/sdk 1.0.0-alpha2.6 → 1.0.0-alpha2.8

package/dist/index.cjs

@@ -1,4 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); 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; }var __getOwnPropNames = Object.getOwnPropertyNames;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); 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.6",
+      version: "1.0.0-alpha2.8",
       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.24",
+        "@airtop/core": "0.1.0-alpha.26",
         "date-fns": "4.1.0",
         loglayer: "6.3.3",
         "serialize-error": "12.0.0",
@@ -102,6 +102,11 @@ var AirtopBase = class {
    * @internal
    **/
   
+  /**
+   * The job id for the ongoing automation
+   * @internal
+   */
+  
   /**
    * 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
     };
   }
   /**
@@ -237,6 +244,99 @@ var processLogMessage = (log, logLevel, ...args) => {
 // src/window/AirtopWindowClient.ts
 
 
+// src/async-utils.ts
+
+var DEFAULT_TIMEOUT_SECONDS = 300;
+var DEFAULT_POLLING_INTERVAL_SECONDS = 2;
+async function waitForRequestCompletion(client, requestId, requestOptions) {
+  const startTime = Date.now();
+  const timeoutMs = _datefns.secondsToMilliseconds.call(void 0, _optionalChain([requestOptions, 'optionalAccess', _5 => _5.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, _datefns.secondsToMilliseconds.call(void 0, 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 AirtopNodeImpl = class {
+  /**
+   * The window client
+   */
+  
+  /**
+   * The node handle id to use for all requests
+   */
+  
+  /**
+   * Constructor
+   * @param client - The window client
+   * @param nodeHandleId - The node handle id to use for all requests
+   */
+  constructor(client, nodeHandleId) {
+    this.windowClient = client;
+    this.nodeHandleId = nodeHandleId;
+  }
+  /**
+   * 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);
+  }
+};
+
 // src/window/AirtopWindowScreenshot.ts
 function extractMimeAndBase64(dataUrl) {
   const match = dataUrl.match(/^data:(image\/\w+);base64,(.+)$/);
@@ -480,7 +580,7 @@ var AirtopWindowClient = class extends AirtopBase {
       prompt
     }).info("Performing a page query");
     const newConfig = config;
-    if (_optionalChain([config, 'optionalAccess', _5 => _5.configuration, 'access', _6 => _6.outputSchema])) {
+    if (_optionalChain([config, 'optionalAccess', _6 => _6.configuration, 'access', _7 => _7.outputSchema])) {
       newConfig.configuration.outputSchema = this.convertToJsonSchema(config.configuration.outputSchema);
     }
     return this.client.windows.pageQuery(
@@ -508,7 +608,7 @@ var AirtopWindowClient = class extends AirtopBase {
       prompt
     }).info("Performing a paginated extraction");
     const newConfig = config;
-    if (_optionalChain([config, 'optionalAccess', _7 => _7.configuration, 'access', _8 => _8.outputSchema])) {
+    if (_optionalChain([config, 'optionalAccess', _8 => _8.configuration, 'access', _9 => _9.outputSchema])) {
       newConfig.configuration.outputSchema = this.convertToJsonSchema(config.configuration.outputSchema);
     }
     return this.client.windows.paginatedExtraction(
@@ -609,6 +709,104 @@ var AirtopWindowClient = class extends AirtopBase {
       }
     );
   }
+  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: _datefns.secondsToMilliseconds.call(void 0, 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: _datefns.secondsToMilliseconds.call(void 0, 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: _datefns.secondsToMilliseconds.call(void 0, 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: _datefns.secondsToMilliseconds.call(void 0, 600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+    const nodeHandleId = apiResponse.data.modelResponse;
+    return new AirtopNodeImpl(this, nodeHandleId);
+  }
+  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: _datefns.secondsToMilliseconds.call(void 0, 600),
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+    const nodeHandleIds = apiResponse.data.modelResponse.split(",");
+    return nodeHandleIds.map((nodeHandleId) => new AirtopNodeImpl(this, nodeHandleId));
+  }
 };
 
 // src/window/AirtopWindow.ts
@@ -845,19 +1043,19 @@ var AirtopClient = class extends AirtopBase {
    */
   constructor(config) {
     super({
-      logLevel: _optionalChain([config, 'optionalAccess', _9 => _9.logLevel]),
+      logLevel: _optionalChain([config, 'optionalAccess', _10 => _10.logLevel]),
       client: new (0, _core.Airtop)({
         maxRetries: 0,
         timeout: _minutesToMilliseconds.minutesToMilliseconds.call(void 0, 1),
         apiKey: config.apiKey,
-        baseURL: _optionalChain([config, 'optionalAccess', _10 => _10.airtopUrl]),
-        logLevel: _optionalChain([config, 'optionalAccess', _11 => _11.logLevel]) || "off",
+        baseURL: _optionalChain([config, 'optionalAccess', _11 => _11.airtopUrl]),
+        logLevel: _optionalChain([config, 'optionalAccess', _12 => _12.logLevel]) || "off",
         defaultHeaders: {
           "x-airtop-sdk-source": "typescript",
           "x-airtop-sdk-version": version
         }
       }),
-      log: _optionalChain([config, 'optionalAccess', _12 => _12.logger]) || new (0, _loglayer.LogLayer)({
+      log: _optionalChain([config, 'optionalAccess', _13 => _13.logger]) || new (0, _loglayer.LogLayer)({
         errorSerializer: _serializeerror.serializeError,
         transport: new (0, _loglayer.ConsoleTransport)({
           logger: console,
@@ -868,7 +1066,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 +1093,8 @@ var AirtopClient = class extends AirtopBase {
    * @returns A new AirtopSession instance
    */
   async createSession(config, options = {}) {
-    const skipWaitSessionReady = _nullishCoalesce(config.skipWaitSessionReady, () => ( false));
-    delete config.skipWaitSessionReady;
+    const skipWaitSessionReady = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _14 => _14.skipWaitSessionReady]), () => ( false));
+     _optionalChainDelete([config, 'optionalAccess', _15 => delete _15.skipWaitSessionReady]);
     const sessionResponse = await this.client.sessions.create(
       {
         configuration: config