@paean-ai/adk 0.2.19 → 0.2.21

package/dist/cjs/agents/live_request_queue.js

@@ -0,0 +1,136 @@
+/**
+  * @license
+  * Copyright 2025 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var live_request_queue_exports = {};
+__export(live_request_queue_exports, {
+  LiveRequestQueue: () => LiveRequestQueue
+});
+module.exports = __toCommonJS(live_request_queue_exports);
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+class LiveRequestQueue {
+  constructor() {
+    // Keeps track of the data that are waiting to be sent.
+    this.queue = [];
+    // Keeps track of the promises that are waiting for data.
+    this.resolveFnFifoQueue = [];
+    this.isClosed = false;
+  }
+  /**
+   * Adds a request to the queue. If there is a pending `get()` call, it
+   * will be resolved with the given request.
+   * @param req The request to send.
+   */
+  send(req) {
+    if (this.isClosed) {
+      throw new Error("Cannot send to a closed queue.");
+    }
+    if (this.resolveFnFifoQueue.length > 0) {
+      const resolve = this.resolveFnFifoQueue.shift();
+      resolve(req);
+    } else {
+      this.queue.push(req);
+    }
+  }
+  /**
+   * Retrieves a request from the queue. If the queue is empty, it will
+   * wait until a request is available.
+   * @returns A promise that resolves with the next available request.
+   */
+  async get() {
+    if (this.queue.length > 0) {
+      return this.queue.shift();
+    }
+    if (this.isClosed) {
+      return { close: true };
+    }
+    return new Promise((resolve) => {
+      this.resolveFnFifoQueue.push(resolve);
+    });
+  }
+  /**
+   * Sends a close signal to the queue.
+   */
+  close() {
+    if (this.isClosed) {
+      return;
+    }
+    this.isClosed = true;
+    while (this.resolveFnFifoQueue.length > 0 && this.queue.length > 0) {
+      const resolve = this.resolveFnFifoQueue.shift();
+      const req = this.queue.shift();
+      resolve(req);
+    }
+    const closeRequest = { close: true };
+    while (this.resolveFnFifoQueue.length > 0) {
+      const resolve = this.resolveFnFifoQueue.shift();
+      resolve(closeRequest);
+    }
+  }
+  /**
+   * Sends a content object to the queue.
+   * @param content The content to send.
+   */
+  sendContent(content) {
+    this.send({ content });
+  }
+  /**
+   * Sends a blob to the model in realtime mode.
+   * @param blob The blob to send.
+   */
+  sendRealtime(blob) {
+    this.send({ blob });
+  }
+  /**
+   * Sends an activity start signal to mark the beginning of user input.
+   */
+  sendActivityStart() {
+    this.send({ activityStart: {} });
+  }
+  /**
+   * Sends an activity end signal to mark the end of user input.
+   */
+  sendActivityEnd() {
+    this.send({ activityEnd: {} });
+  }
+  /**
+   * Implements the async iterator protocol.
+   */
+  async *[Symbol.asyncIterator]() {
+    while (true) {
+      const request = await this.get();
+      yield request;
+      if (request.close) {
+        break;
+      }
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  LiveRequestQueue
+});