@airtop/sdk 1.0.0-alpha2.13 → 1.0.0-alpha2.15

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.13",
+      version: "1.0.0-alpha2.15",
       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.28",
+        "@airtop/core": "0.1.0-alpha.29",
         "date-fns": "4.1.0",
         loglayer: "6.3.3",
         "serialize-error": "12.0.0",
@@ -121,6 +121,9 @@ var AirtopBase = class {
     this.client = config.client;
     this.outputJsonAdapter = config.outputSchemaAdapter;
     this.jobId = config.jobId;
+    if (config.agentEventPublisher) {
+      this.agentEventPublisher = config.agentEventPublisher;
+    }
   }
   /**
    * Sets the publisher for sending agent events. Internal use only.
@@ -193,7 +196,9 @@ var AirtopBase = class {
     return {
       client: this.client,
       log: this.log,
-      jobId: this.jobId
+      jobId: this.jobId,
+      outputSchemaAdapter: this.outputJsonAdapter,
+      agentEventPublisher: this.agentEventPublisher
     };
   }
   /**
@@ -513,6 +518,20 @@ var AirtopWindowClient = class extends AirtopBase {
     });
     this.windowId = windowId;
     this.sessionId = sessionId;
+    this.client.windows.get(
+      this.windowId,
+      {
+        sessionId: this.sessionId,
+        ...config
+      },
+      this.resolveRequestOptions()
+    ).then((results) => {
+      this._sendAgentPayload("LiveViewUrl", {
+        liveViewUrl: results.data.liveViewUrl,
+        windowId: this.windowId,
+        sessionId: this.sessionId
+      });
+    });
   }
   /**
    * Gets the window ID.
@@ -541,11 +560,6 @@ var AirtopWindowClient = class extends AirtopBase {
       },
       this.resolveRequestOptions(requestOptions)
     );
-    this._sendAgentPayload("LiveViewUrl", {
-      liveViewUrl: results.data.liveViewUrl,
-      windowId: this.windowId,
-      sessionId: this.sessionId
-    });
     return new AirtopWindow(this.getCommonConfig(), this.sessionId, results);
   }
   /**
@@ -1119,6 +1133,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);
   }
   /**
@@ -1228,7 +1255,8 @@ var AirtopClient = class extends AirtopBase {
         metadataFieldName: "metadata"
       }),
       outputSchemaAdapter: config.outputSchemaAdapter,
-      jobId: config.jobId
+      jobId: config.jobId,
+      agentEventPublisher: config.agentEventPublisher
     });
     this.log.withPrefix("[Airtop SDK]");
     this.client.logLevel = config.logLevel;
@@ -1448,8 +1476,191 @@ function registerAirtopPlugin(plugin) {
   }
 }
 
-// src/AirtopMocks.ts
+// src/agent/AirtopAgentClient.ts
+import { version as version2 } from "node:process";
 import { Airtop as AirtopCore2 } from "@airtop/core";
+import { minutesToMilliseconds as minutesToMilliseconds2 } from "date-fns";
+import { ConsoleTransport as ConsoleTransport2 } from "loglayer";
+import { 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 AirtopCore2({
+        maxRetries: 0,
+        timeout: minutesToMilliseconds2(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/AirtopMocks.ts
+import { Airtop as AirtopCore3 } from "@airtop/core";
 import { MockLogLayer } from "loglayer";
 import { v4 as uuidv4 } from "uuid";
 function createNoOpClient(client) {
@@ -1472,7 +1683,7 @@ function createNoOpClient(client) {
   });
 }
 function generateCommonAirtopConfigMock() {
-  const client = new AirtopCore2({
+  const client = new AirtopCore3({
     apiKey: "test-api-key"
   });
   return {
@@ -1539,6 +1750,7 @@ var AirtopMocks = class {
   }
 };
 export {
+  AirtopAgentClient,
   AirtopBase,
   AirtopClient,
   AirtopError,