@airtop/sdk 1.0.0-alpha2.2 → 1.0.0-alpha2.21

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.2",
+      version: "1.0.0-alpha2.21",
       type: "module",
       main: "./dist/index.cjs",
       module: "./dist/index.js",
@@ -38,34 +38,35 @@ var require_package = __commonJS({
         "agentic"
       ],
       scripts: {
-        "build:second-stage": "tsup src/index.ts",
         "test:e2e": "vitest --run",
         "build:dev": "node_modules/.bin/hash-runner",
+        build: "tsup src/index.ts",
         clean: "rm -rf .turbo node_modules dist",
-        lint: "biome check --write --unsafe src && biome format src --write && biome lint src --fix",
-        "verify-types": "tsc --noEmit"
+        lint: "biome check --no-errors-on-unmatched --write --unsafe src",
+        "lint:staged": "biome check --no-errors-on-unmatched --write --unsafe --staged src",
+        "verify-types": "tsc --noEmit && tsc --noEmit --project tsconfig.e2e.json"
       },
       dependencies: {
         "@airtop/json-schema-adapter": "workspace:*",
-        "@airtop/core": "0.1.0-alpha.17",
+        "@airtop/core": "0.1.0-alpha.37",
         "date-fns": "4.1.0",
-        loglayer: "6.3.3",
+        loglayer: "6.6.0",
         "serialize-error": "12.0.0",
         uuid: "11.1.0"
       },
       devDependencies: {
-        "@biomejs/biome": "1.9.4",
-        "@dotenvx/dotenvx": "1.39.0",
+        "@biomejs/biome": "2.1.3",
+        "@dotenvx/dotenvx": "1.48.4",
         "@internal/tsconfig": "workspace:*",
         "deep-utility-types": "1.3.1",
         "env-var": "7.5.0",
-        "hash-runner": "2.0.1",
+        "hash-runner": "3.1.0",
         nanoid: "5.1.5",
-        tsup: "8.4.0",
-        typescript: "5.8.2",
+        tsup: "8.5.0",
+        typescript: "5.9.2",
         "utility-types": "3.11.0",
         uuid: "11.1.0",
-        vitest: "3.1.1",
+        vitest: "3.2.4",
         zod: "*"
       },
       bugs: "https://github.com/airtop-ai/airtop-sdk/issues",
@@ -79,13 +80,19 @@ var require_package = __commonJS({
       optionalDependencies: {
         "@airtop/json-schema-adapter-zod": "workspace:*"
       },
-      packageManager: "pnpm@10.5.0"
+      packageManager: "pnpm@10.12.4"
     };
   }
 });
 
 // src/AirtopBase.ts
 import { secondsToMilliseconds } from "date-fns";
+
+// src/constants.ts
+var TIMEOUT_SECONDS_DEFAULT_VALUE = 300;
+var DEFAULT_POLLING_INTERVAL_MS = 500;
+
+// src/AirtopBase.ts
 var AirtopBase = class {
   /**
    * Logger instance for the SDK
@@ -102,6 +109,21 @@ var AirtopBase = class {
    * @internal
    **/
   outputJsonAdapter;
+  /**
+   * The job id for the ongoing automation
+   * @internal
+   */
+  jobId;
+  /**
+   * Instance for sending agent events
+   * @internal
+   */
+  agentEventPublisher;
+  /**
+   * The default timeout in seconds for API requests.
+   * @internal
+   */
+  defaultTimeoutInSeconds;
   /**
    * Creates a new instance of AirtopBase
    * @param config - Configuration options for the SDK
@@ -110,6 +132,29 @@ var AirtopBase = class {
     this.log = config.log.withPrefix("[Airtop]");
     this.client = config.client;
     this.outputJsonAdapter = config.outputSchemaAdapter;
+    this.jobId = config.jobId;
+    this.defaultTimeoutInSeconds = config.defaultTimeoutInSeconds ?? TIMEOUT_SECONDS_DEFAULT_VALUE;
+    if (config.agentEventPublisher) {
+      this.agentEventPublisher = config.agentEventPublisher;
+    }
+  }
+  /**
+   * Sets the publisher for sending agent events. Internal use only.
+   * @internal
+   */
+  _setAgentEventPublisher(logger) {
+    this.agentEventPublisher = logger;
+  }
+  /**
+   * Sends a payload to the agent with the specified event name.
+   * @param eventName The name of the event
+   * @param payload The payload to send to the agent
+   * @internal
+   */
+  _sendAgentPayload(eventName, payload) {
+    if (this.agentEventPublisher) {
+      this.agentEventPublisher.withMetadata(payload).info(eventName);
+    }
   }
   /**
    * Returns the API key used by the SDK
@@ -163,7 +208,10 @@ var AirtopBase = class {
   getCommonConfig() {
     return {
       client: this.client,
-      log: this.log
+      log: this.log,
+      jobId: this.jobId,
+      outputSchemaAdapter: this.outputJsonAdapter,
+      agentEventPublisher: this.agentEventPublisher
     };
   }
   /**
@@ -174,7 +222,7 @@ var AirtopBase = class {
    */
   resolveRequestOptions(options = {}) {
     return {
-      timeout: secondsToMilliseconds(options.timeoutInSeconds || 60),
+      timeout: secondsToMilliseconds(options.timeoutInSeconds || this.defaultTimeoutInSeconds),
       maxRetries: options.maxRetries || 0,
       signal: options.abortSignal
     };
@@ -183,7 +231,7 @@ var AirtopBase = class {
 
 // src/AirtopClient.ts
 import { Airtop as AirtopCore } from "@airtop/core";
-import { minutesToMilliseconds } from "date-fns/minutesToMilliseconds";
+import { secondsToMilliseconds as secondsToMilliseconds3 } from "date-fns/secondsToMilliseconds";
 import { ConsoleTransport, LogLayer } from "loglayer";
 import { serializeError } from "serialize-error";
 
@@ -234,9 +282,166 @@ var processLogMessage = (log, logLevel, ...args) => {
 // src/session/AirtopSessionClient.ts
 import { NotFoundError } from "@airtop/core";
 
-// src/window/AirtopWindowClient.ts
+// src/async-utils.ts
 import { secondsToMilliseconds as secondsToMilliseconds2 } from "date-fns";
 
+// src/types.ts
+var AirtopError = class extends Error {
+  issues;
+  metadata;
+  constructor(issues, metadata) {
+    const errorMessage = issues.map((issue) => issue.message).join("\n");
+    const errorMessageWithMetadata = metadata ? `${errorMessage}
+${JSON.stringify(metadata)}` : errorMessage;
+    super(errorMessageWithMetadata);
+    this.issues = issues;
+    this.metadata = metadata;
+  }
+};
+
+// src/async-utils.ts
+async function waitForRequestCompletion(client, requestId, requestOptions) {
+  const startTime = Date.now();
+  const timeoutMs = secondsToMilliseconds2(requestOptions?.timeoutInSeconds || TIMEOUT_SECONDS_DEFAULT_VALUE);
+  while (Date.now() - startTime < timeoutMs) {
+    const apiResponse = await client.requests.getRequestStatusV2(requestId, requestOptions);
+    if (apiResponse.status === "completed") {
+      return apiResponse.response;
+    }
+    if (apiResponse.status === "error") {
+      throw new Error(apiResponse.error);
+    }
+    await new Promise((resolve) => setTimeout(resolve, DEFAULT_POLLING_INTERVAL_MS));
+  }
+  throw new Error("Waiting for request timed out");
+}
+async function withRequestCompletionPolling(client, fn, requestOptions) {
+  try {
+    const response = await fn();
+    return waitForRequestCompletion(client, response.requestId, requestOptions);
+  } catch (thrownError) {
+    if (thrownError.error?.errors) {
+      throw new AirtopError(thrownError.error.errors, {
+        requestId: thrownError.error.requestId
+      });
+    }
+    throw thrownError;
+  }
+}
+
+// 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, nodeHandleId: this.nodeHandleId }, 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 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 findOneOptional(prompt, config, requestOptions = {}) {
+    const augmentedConfig = {
+      ...config,
+      nodeHandleId: this.nodeHandleId
+    };
+    return this.windowClient.findOneOptional(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);
+  }
+  /**
+   * 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 findManyOptional(prompt, config, requestOptions = {}) {
+    const augmentedConfig = {
+      ...config,
+      nodeHandleId: this.nodeHandleId
+    };
+    return this.windowClient.findManyOptional(prompt, augmentedConfig, requestOptions);
+  }
+};
+
 // src/window/AirtopWindowScreenshot.ts
 function extractMimeAndBase64(dataUrl) {
   const match = dataUrl.match(/^data:(image\/\w+);base64,(.+)$/);
@@ -336,6 +541,19 @@ var AirtopWindowClient = class extends AirtopBase {
     });
     this.windowId = windowId;
     this.sessionId = sessionId;
+    this.client.windows.get(
+      this.windowId,
+      {
+        sessionId: this.sessionId
+      },
+      this.resolveRequestOptions()
+    ).then((results) => {
+      this._sendAgentPayload("LiveViewUrl", {
+        liveViewUrl: results.data.liveViewUrl,
+        windowId: this.windowId,
+        sessionId: this.sessionId
+      });
+    });
   }
   /**
    * Gets the window ID.
@@ -377,17 +595,19 @@ var AirtopWindowClient = class extends AirtopBase {
     this.log.withMetadata({
       elementDescription
     }).info("Clicking on element");
-    return this.client.windows.click(
-      this.getWindowId(),
-      {
-        ...config,
-        elementDescription,
-        sessionId: this.sessionId
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.clickAsync(
+        this.getWindowId(),
+        {
+          ...config,
+          elementDescription,
+          sessionId: this.sessionId
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -409,17 +629,19 @@ var AirtopWindowClient = class extends AirtopBase {
    */
   async hover(elementDescription, config, requestOptions = {}) {
     this.log.withMetadata(config).info("Hovering over window");
-    return this.client.windows.hover(
-      this.getWindowId(),
-      {
-        elementDescription,
-        sessionId: this.sessionId,
-        ...config || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.hoverAsync(
+        this.getWindowId(),
+        {
+          elementDescription,
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -441,7 +663,6 @@ var AirtopWindowClient = class extends AirtopBase {
         ...config
       },
       {
-        timeout: secondsToMilliseconds2(600),
         ...this.resolveRequestOptions(requestOptions)
       }
     );
@@ -453,19 +674,33 @@ var AirtopWindowClient = class extends AirtopBase {
    * @param requestOptions - Request options
    * @returns Promise resolving to the monitoring operation result
    */
-  async monitor(condition, config, requestOptions = {}) {
+  async monitor(condition, config = {}, requestOptions = {}) {
     this.log.withMetadata().info("Monitoring window");
-    return this.client.windows.monitor(
-      this.getWindowId(),
-      {
-        condition,
-        sessionId: this.sessionId,
-        ...config || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
+    if (!config?.configuration?.interval?.timeoutSeconds) {
+      if (!config.configuration) {
+        config.configuration = {};
+      }
+      if (!config.configuration.interval) {
+        config.configuration.interval = {};
       }
+      config.configuration.interval.timeoutSeconds = this.defaultTimeoutInSeconds;
+    }
+    if (!config?.timeThresholdSeconds) {
+      config.timeThresholdSeconds = this.defaultTimeoutInSeconds;
+    }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.monitorAsync(
+        this.getWindowId(),
+        {
+          condition,
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -483,17 +718,19 @@ var AirtopWindowClient = class extends AirtopBase {
     if (config?.configuration.outputSchema) {
       newConfig.configuration.outputSchema = this.convertToJsonSchema(config.configuration.outputSchema);
     }
-    return this.client.windows.pageQuery(
-      this.getWindowId(),
-      {
-        sessionId: this.sessionId,
-        prompt,
-        ...newConfig || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.pageQueryAsync(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          prompt,
+          ...newConfig || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -511,17 +748,19 @@ var AirtopWindowClient = class extends AirtopBase {
     if (config?.configuration.outputSchema) {
       newConfig.configuration.outputSchema = this.convertToJsonSchema(config.configuration.outputSchema);
     }
-    return this.client.windows.paginatedExtraction(
-      this.getWindowId(),
-      {
-        prompt,
-        sessionId: this.sessionId,
-        ...newConfig || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.paginatedExtractionAsync(
+        this.getWindowId(),
+        {
+          prompt,
+          sessionId: this.sessionId,
+          ...newConfig || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -532,16 +771,18 @@ var AirtopWindowClient = class extends AirtopBase {
    */
   async scrape(config, requestOptions = {}) {
     this.log.info("Scraping window");
-    return this.client.windows.scrape(
-      this.getWindowId(),
-      {
-        sessionId: this.sessionId,
-        ...config || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.scrapeAsync(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -552,16 +793,18 @@ var AirtopWindowClient = class extends AirtopBase {
    */
   async screenshot(config, requestOptions = {}) {
     this.log.info("Screenshotting window");
-    const resp = await this.client.windows.screenshot(
-      this.getWindowId(),
-      {
-        sessionId: this.sessionId,
-        ...config || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    const resp = await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.screenshotAsync(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
     return new AirtopWindowScreenshot(resp);
   }
@@ -573,16 +816,18 @@ var AirtopWindowClient = class extends AirtopBase {
    */
   async scroll(config, requestOptions = {}) {
     this.log.info("Scrolling window");
-    return this.client.windows.scroll(
-      this.getWindowId(),
-      {
-        sessionId: this.sessionId,
-        ...config || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
-      }
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.scrollAsync(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
   /**
@@ -596,17 +841,198 @@ var AirtopWindowClient = class extends AirtopBase {
     this.log.withMetadata({
       text
     }).info("Typing text");
-    return this.client.windows.type(
-      this.getWindowId(),
-      {
-        sessionId: this.sessionId,
-        text,
-        ...config || {}
-      },
-      {
-        timeout: secondsToMilliseconds2(600),
-        ...this.resolveRequestOptions(requestOptions)
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.typeAsync(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          text,
+          ...config || {}
+        },
+        {
+          ...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 || {}
+        },
+        {
+          ...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 || {}
+        },
+        {
+          ...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,
+          jobId: this.jobId,
+          ...config || {},
+          includeWebContext: true,
+          // Always include web context for window.llm() calls
+          outputSchema: config?.outputSchema ? this.convertToJsonSchema(config.outputSchema) : void 0
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  async findOnePrivate(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 || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+    if (apiResponse.errors.length > 0) {
+      throw new AirtopError(apiResponse.errors);
+    }
+    try {
+      if (!apiResponse.data.nodeHandle) {
+        return null;
       }
+      return new AirtopNode(this, apiResponse.data.nodeHandle);
+    } catch (error) {
+      this.log.withMetadata({ nodeDataStr: apiResponse.data.nodeHandle }).withError(error).error("Error parsing node data");
+      throw error;
+    }
+  }
+  async findOne(prompt, config, requestOptions = {}) {
+    const configOverride = { ...config, optional: false };
+    return await this.findOnePrivate(prompt, configOverride, requestOptions);
+  }
+  async findOneOptional(prompt, config, requestOptions = {}) {
+    const configOverride = { ...config, optional: true };
+    return await this.findOnePrivate(prompt, configOverride, requestOptions);
+  }
+  async findManyPrivate(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 || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+    if (apiResponse.errors.length > 0) {
+      throw new AirtopError(apiResponse.errors);
+    }
+    const nodeHandleData = apiResponse.data.nodeHandles;
+    return nodeHandleData.map((nodeHandle) => new AirtopNode(this, nodeHandle));
+  }
+  async findMany(prompt, config, requestOptions = {}) {
+    const configOverride = { ...config, optional: false };
+    return await this.findManyPrivate(prompt, configOverride, requestOptions);
+  }
+  async findManyOptional(prompt, config, requestOptions = {}) {
+    const configOverride = { ...config, optional: true };
+    return await this.findManyPrivate(prompt, configOverride, requestOptions);
+  }
+  async waitForPage(config, requestOptions = {}) {
+    return await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.waitForPage(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  async navigate(direction, config, requestOptions = {}) {
+    return await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.navigate(
+        this.getWindowId(),
+        {
+          direction,
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {}
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  async fillForm(formData, config, requestOptions = {}) {
+    return await withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.executeAutomation(
+        this.getWindowId(),
+        {
+          sessionId: this.sessionId,
+          ...config || {},
+          automationId: config?.automationId || "auto",
+          parameters: {
+            customData: typeof formData === "string" ? formData : JSON.stringify(formData)
+            // Will be interpreted by the LLM
+          }
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
     );
   }
 };
@@ -682,6 +1108,9 @@ var AirtopSessionClient = class extends AirtopBase {
   getSessionId() {
     return this.sessionId;
   }
+  async listWindows(requestOptions = {}) {
+    return this.client.sessions.listWindows(this.sessionId, this.resolveRequestOptions(requestOptions));
+  }
   /**
    * Gets the state of the session using the attached session id.
    * @param requestOptions - Request options
@@ -760,6 +1189,19 @@ var AirtopSessionClient = class extends AirtopBase {
       },
       this.resolveRequestOptions(requestOptions)
     );
+    this.client.windows.get(
+      results.data.windowId,
+      {
+        sessionId: this.sessionId
+      },
+      this.resolveRequestOptions()
+    ).then((windowResults) => {
+      this._sendAgentPayload("LiveViewUrl", {
+        liveViewUrl: windowResults.data.liveViewUrl,
+        windowId: results.data.windowId,
+        sessionId: this.sessionId
+      });
+    });
     return new AirtopWindow(this.getCommonConfig(), this.sessionId, results);
   }
   /**
@@ -782,11 +1224,67 @@ var AirtopSessionClient = class extends AirtopBase {
       {
         ...config,
         fileName,
-        sessionId: this.sessionId
+        sessionIds: [this.sessionId]
       },
       this.resolveRequestOptions(requestOptions)
     );
   }
+  async llm(prompt, config, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Executing LLM call");
+    const currentWindows = await this.listWindows();
+    if (currentWindows.data.windows.length === 0) {
+      throw new AirtopError([
+        {
+          message: "No windows found in the session. Please create a window before calling the llm() method."
+        }
+      ]);
+    }
+    const firstWindow = currentWindows.data.windows[0];
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.windows.llm(
+        firstWindow.windowId,
+        {
+          prompt,
+          sessionId: this.sessionId,
+          jobId: this.jobId,
+          ...config || {},
+          includeWebContext: false,
+          // Do not include web context for session.llm() calls
+          outputSchema: config?.outputSchema ? this.convertToJsonSchema(config.outputSchema) : void 0
+        },
+        {
+          ...this.resolveRequestOptions(requestOptions)
+        }
+      )
+    );
+  }
+  /**
+   * Calls the service endpoint.
+   * @param prompt - The prompt to send to the service, describing the actions to take
+   * @param service - The service to call, if not provided, the service will be inferred from the prompt
+   * @param requestOptions - Request options
+   */
+  async service(prompt, service, requestOptions = {}) {
+    this.log.withMetadata({ prompt }).info("Service");
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.sessions.service(this.sessionId, { prompt, service }),
+      requestOptions
+    );
+  }
+  /**
+   * Retrieves the list of connected services available for the current session.
+   * @param requestOptions - Request options
+   * @returns A promise that resolves to the connected services response containing service details
+   */
+  async getConnectedServices(requestOptions = {}) {
+    return withRequestCompletionPolling(
+      this.client,
+      () => this.client.sessions.getConnectedServices(this.sessionId),
+      requestOptions
+    );
+  }
 };
 
 // src/session/AirtopSession.ts
@@ -848,7 +1346,7 @@ var AirtopClient = class extends AirtopBase {
       logLevel: config?.logLevel,
       client: new AirtopCore({
         maxRetries: 0,
-        timeout: minutesToMilliseconds(1),
+        timeout: secondsToMilliseconds3(config.defaultTimeoutInSeconds ?? TIMEOUT_SECONDS_DEFAULT_VALUE),
         apiKey: config.apiKey,
         baseURL: config?.airtopUrl,
         logLevel: config?.logLevel || "off",
@@ -868,7 +1366,10 @@ var AirtopClient = class extends AirtopBase {
         contextFieldName: "context",
         metadataFieldName: "metadata"
       }),
-      outputSchemaAdapter: config.outputSchemaAdapter
+      outputSchemaAdapter: config.outputSchemaAdapter,
+      jobId: config.jobId,
+      agentEventPublisher: config.agentEventPublisher,
+      defaultTimeoutInSeconds: config.defaultTimeoutInSeconds ?? TIMEOUT_SECONDS_DEFAULT_VALUE
     });
     this.log.withPrefix("[Airtop SDK]");
     this.client.logLevel = config.logLevel;
@@ -894,8 +1395,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 +1504,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 +1512,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
@@ -1053,43 +1545,6 @@ var AirtopClient = class extends AirtopBase {
   }
 };
 
-// src/plugin/AirtopPlugin.types.ts
-var AirtopPluginAugmentationType = /* @__PURE__ */ ((AirtopPluginAugmentationType2) => {
-  AirtopPluginAugmentationType2["AirtopClient"] = "AirtopClient";
-  AirtopPluginAugmentationType2["AirtopWindowClient"] = "AirtopWindowClient";
-  AirtopPluginAugmentationType2["AirtopWindow"] = "AirtopWindow";
-  AirtopPluginAugmentationType2["AirtopSessionClient"] = "AirtopSessionClient";
-  AirtopPluginAugmentationType2["AirtopSession"] = "AirtopSession";
-  AirtopPluginAugmentationType2["AirtopWindowScreenshot"] = "AirtopWindowScreenshot";
-  return AirtopPluginAugmentationType2;
-})(AirtopPluginAugmentationType || {});
-
-// src/plugin/AirtopPlugin.ts
-function registerAirtopPlugin(plugin) {
-  for (const pluginToAdd of plugin.pluginsToAdd) {
-    switch (pluginToAdd.augmentationType) {
-      case "AirtopClient" /* AirtopClient */:
-        pluginToAdd.augment(AirtopClient.prototype);
-        break;
-      case "AirtopSessionClient" /* AirtopSessionClient */:
-        pluginToAdd.augment(AirtopSessionClient.prototype);
-        break;
-      case "AirtopSession" /* AirtopSession */:
-        pluginToAdd.augment(AirtopSession.prototype);
-        break;
-      case "AirtopWindowClient" /* AirtopWindowClient */:
-        pluginToAdd.augment(AirtopWindowClient.prototype);
-        break;
-      case "AirtopWindow" /* AirtopWindow */:
-        pluginToAdd.augment(AirtopWindow.prototype);
-        break;
-      case "AirtopWindowScreenshot" /* AirtopWindowScreenshot */:
-        pluginToAdd.augment(AirtopWindowScreenshot.prototype);
-        break;
-    }
-  }
-}
-
 // src/AirtopMocks.ts
 import { Airtop as AirtopCore2 } from "@airtop/core";
 import { MockLogLayer } from "loglayer";
@@ -1180,16 +1635,246 @@ var AirtopMocks = class {
     });
   }
 };
+
+// src/agent/AirtopAgentClient.ts
+import { version as version2 } from "process";
+import { Airtop as AirtopCore3 } from "@airtop/core";
+import { minutesToMilliseconds } from "date-fns";
+import { ConsoleTransport as ConsoleTransport2, LogLayer as LogLayer2 } from "loglayer";
+import { serializeError as serializeError2 } from "serialize-error";
+var AirtopAgentClient = class extends AirtopBase {
+  /**
+   * Creates a new instance of the Airtop SDK.
+   * @param config - Configuration options for the Airtop SDK
+   */
+  constructor(config) {
+    super({
+      logLevel: config?.logLevel,
+      client: new AirtopCore3({
+        maxRetries: 0,
+        timeout: minutesToMilliseconds(1),
+        apiKey: config.apiKey,
+        baseURL: config?.airtopUrl,
+        logLevel: config?.logLevel || "off",
+        defaultHeaders: {
+          "x-airtop-sdk-source": "typescript",
+          "x-airtop-sdk-version": version2
+        }
+      }),
+      log: config?.logger || new LogLayer2({
+        errorSerializer: serializeError2,
+        transport: new ConsoleTransport2({
+          logger: console,
+          messageField: "message",
+          enabled: config.logLevel !== "off",
+          level: config.logLevel === "off" ? "error" : config.logLevel || "error"
+        }),
+        contextFieldName: "context",
+        metadataFieldName: "metadata"
+      }),
+      outputSchemaAdapter: config.outputSchemaAdapter,
+      jobId: config.jobId
+    });
+    this.log.withPrefix("[Airtop SDK]");
+    this.client.logLevel = config.logLevel;
+    this.client.logger = {
+      debug: (message, ...rest) => {
+        processLogMessage(this.log, "debug", message, rest);
+      },
+      error: (message, ...rest) => {
+        processLogMessage(this.log, "error", message, rest);
+      },
+      info: (message, ...rest) => {
+        processLogMessage(this.log, "info", message, rest);
+      },
+      warn: (message, ...rest) => {
+        processLogMessage(this.log, "warn", message, rest);
+      }
+    };
+  }
+  /**
+   * Creates a new agent.
+   * @param params - Parameters for creating the agent. Corresponds to `AirtopAgentCreateAgentParams`.
+   * @param requestOptions - Request options.
+   * @returns The created agent data, `AirtopCreateAgentResponse`.
+   */
+  async createAgent(params, requestOptions = {}) {
+    this.log.info("Creating agent");
+    return this.client.agents.createAgent(params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Retrieves a specific agent by its ID.
+   * @param agentId - The ID of the agent to retrieve.
+   * @param requestOptions - Request options.
+   * @returns The agent data, `AirtopGetAgentResponse`.
+   */
+  async getAgent(agentId, requestOptions = {}) {
+    this.log.withMetadata({ agentId }).info("Retrieving agent");
+    return this.client.agents.getAgent(agentId, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Lists agents.
+   * @param params - Optional parameters for listing agents (e.g., pagination). Corresponds to `AirtopAgentGetAgentsParams`.
+   * @param requestOptions - Request options.
+   * @returns A list of agents, `AirtopGetAgentsResponse`.
+   */
+  async listAgents(params, requestOptions = {}) {
+    this.log.info("Listing agents");
+    return this.client.agents.getAgents(params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Updates an existing agent.
+   * @param agentId - The ID of the agent to update.
+   * @param params - The new data for the agent. Corresponds to `AirtopAgentUpdateAgentParams`.
+   * @param requestOptions - Request options.
+   * @returns The updated agent data, `AirtopUpdateAgentResponse`.
+   */
+  async updateAgent(agentId, params, requestOptions = {}) {
+    this.log.withMetadata({ agentId }).info("Updating agent");
+    return this.client.agents.updateAgent(agentId, params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Deletes one or more agents.
+   * @param params - Parameters for deleting agents. Corresponds to `AirtopAgentDeleteAgentsParams`.
+   * @param requestOptions - Request options.
+   * @returns A response confirming the deletion, `AirtopDeleteAgentsResponse`.
+   */
+  async deleteAgents(params, requestOptions = {}) {
+    this.log.info("Deleting agent(s)");
+    return this.client.agents.deleteAgents(params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Duplicates an agent.
+   * @param agentId - The ID of the agent to duplicate.
+   * @param params - Parameters for duplicating the agent (e.g., new name). Corresponds to `AirtopAgentDuplicateAgentParams`.
+   * @param requestOptions - Request options.
+   * @returns The duplicated agent data, `AirtopDuplicateAgentResponse`.
+   */
+  async duplicateAgent(agentId, params, requestOptions = {}) {
+    this.log.withMetadata({ agentId }).info("Duplicating agent");
+    return this.client.agents.duplicateAgent(agentId, params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Creates a new version for an agent.
+   * @param agentId - The ID of the agent for which to create a version.
+   * @param params - Parameters for creating the agent version. Corresponds to `AirtopAgentCreateVersionParams`.
+   * @param requestOptions - Request options.
+   * @returns The created agent version data, `AirtopCreateAgentVersionResponse`.
+   */
+  async createAgentVersion(agentId, params, requestOptions = {}) {
+    this.log.withMetadata({ agentId }).info("Creating agent version");
+    return this.client.agents.createVersion(agentId, params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Lists versions of an agent.
+   * @param agentId - The ID of the agent whose versions to list.
+   * @param params - Optional parameters for listing agent versions. Corresponds to `AirtopAgentGetVersionsParams`.
+   * @param requestOptions - Request options.
+   * @returns A list of agent versions, `AirtopGetAgentVersionsResponse`.
+   */
+  async listAgentVersions(agentId, params, requestOptions = {}) {
+    this.log.withMetadata({ agentId }).info("Listing agent versions");
+    return this.client.agents.getVersions(agentId, params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Creates an invocation for an agent.
+   * @param agentId - The ID of the agent for which to create an invocation.
+   * @param params - Parameters for creating the agent invocation. Corresponds to `AirtopAgentCreateInvocationParams`.
+   * @param requestOptions - Request options.
+   * @returns The created agent invocation data, `AirtopCreateAgentInvocationResponse`.
+   */
+  async createAgentInvocation(agentId, params, requestOptions = {}) {
+    this.log.withMetadata({ agentId }).info("Creating agent invocation");
+    return this.client.agents.createInvocation(agentId, params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Lists invocations of an agent.
+   * @param agentIds - The ID of the agent whose invocations to list.
+   * @param params - Optional parameters for listing agent invocations. Corresponds to `AirtopAgentGetInvocationsParams`.
+   * @param requestOptions - Request options.
+   * @returns A list of agent invocations, `AirtopGetAgentInvocationsResponse`.
+   */
+  async listAgentInvocations(agentIds, params, requestOptions = {}) {
+    this.log.withMetadata({ agentId: agentIds }).info("Listing agent invocations");
+    return this.client.agents.getInvocations(agentIds, params, this.resolveRequestOptions(requestOptions));
+  }
+  /**
+   * Cancels a specific invocation of an agent.
+   * @param agentId - The ID of the agent.
+   * @param invocationId - The ID of the invocation to cancel.
+   * @param paramsBody - Optional body parameters for cancelling the invocation. Corresponds to `AirtopAgentCancelInvocationParams`.
+   * @param requestOptions - Request options.
+   * @returns A promise that resolves when the operation is complete.
+   */
+  async cancelAgentInvocation(agentId, invocationId, paramsBody, requestOptions = {}) {
+    this.log.withMetadata({ agentId, invocationId }).info("Cancelling agent invocation");
+    const resolvedOptions = this.resolveRequestOptions(requestOptions);
+    const finalOptions = { ...resolvedOptions };
+    const params = { id: agentId, ...paramsBody };
+    if (paramsBody) {
+      finalOptions.body = paramsBody;
+    }
+    await this.client.agents.cancelInvocation(invocationId, params, finalOptions);
+  }
+};
+
+// src/window/AirtopWindowClient.types.ts
+var WindowNavigateDirection = /* @__PURE__ */ ((WindowNavigateDirection2) => {
+  WindowNavigateDirection2["BACK"] = "backward";
+  WindowNavigateDirection2["FORWARD"] = "forward";
+  return WindowNavigateDirection2;
+})(WindowNavigateDirection || {});
+
+// src/plugin/AirtopPlugin.types.ts
+var AirtopPluginAugmentationType = /* @__PURE__ */ ((AirtopPluginAugmentationType2) => {
+  AirtopPluginAugmentationType2["AirtopClient"] = "AirtopClient";
+  AirtopPluginAugmentationType2["AirtopWindowClient"] = "AirtopWindowClient";
+  AirtopPluginAugmentationType2["AirtopWindow"] = "AirtopWindow";
+  AirtopPluginAugmentationType2["AirtopSessionClient"] = "AirtopSessionClient";
+  AirtopPluginAugmentationType2["AirtopSession"] = "AirtopSession";
+  AirtopPluginAugmentationType2["AirtopWindowScreenshot"] = "AirtopWindowScreenshot";
+  return AirtopPluginAugmentationType2;
+})(AirtopPluginAugmentationType || {});
+
+// src/plugin/AirtopPlugin.ts
+function registerAirtopPlugin(plugin) {
+  for (const pluginToAdd of plugin.pluginsToAdd) {
+    switch (pluginToAdd.augmentationType) {
+      case "AirtopClient" /* AirtopClient */:
+        pluginToAdd.augment(AirtopClient.prototype);
+        break;
+      case "AirtopSessionClient" /* AirtopSessionClient */:
+        pluginToAdd.augment(AirtopSessionClient.prototype);
+        break;
+      case "AirtopSession" /* AirtopSession */:
+        pluginToAdd.augment(AirtopSession.prototype);
+        break;
+      case "AirtopWindowClient" /* AirtopWindowClient */:
+        pluginToAdd.augment(AirtopWindowClient.prototype);
+        break;
+      case "AirtopWindow" /* AirtopWindow */:
+        pluginToAdd.augment(AirtopWindow.prototype);
+        break;
+      case "AirtopWindowScreenshot" /* AirtopWindowScreenshot */:
+        pluginToAdd.augment(AirtopWindowScreenshot.prototype);
+        break;
+    }
+  }
+}
 export {
+  AirtopAgentClient,
   AirtopBase,
   AirtopClient,
+  AirtopError,
   AirtopMocks,
+  AirtopNode,
   AirtopPluginAugmentationType,
   AirtopSession,
   AirtopSessionClient,
   AirtopWindow,
   AirtopWindowClient,
   AirtopWindowScreenshot,
+  WindowNavigateDirection,
   registerAirtopPlugin
 };
 //# sourceMappingURL=index.js.map