@elizaos/core 1.5.15 → 1.6.0-alpha.4

package/dist/node/index.node.js

@@ -25755,7 +25755,7 @@ var getDefaultProjectName = () => {
 };
 
 // ../../node_modules/langsmith/dist/index.js
-var __version__ = "0.3.69";
+var __version__ = "0.3.71";
 
 // ../../node_modules/langsmith/dist/utils/env.js
 var globalEnv;
@@ -26816,13 +26816,13 @@ class AutoBatchQueue {
     this.sizeBytes += size;
     return itemPromise;
   }
-  pop(upToSizeBytes) {
+  pop({ upToSizeBytes, upToSize }) {
     if (upToSizeBytes < 1) {
       throw new Error("Number of bytes to pop off may not be less than 1.");
     }
     const popped = [];
     let poppedSizeBytes = 0;
-    while (poppedSizeBytes + (this.peek()?.size ?? 0) < upToSizeBytes && this.items.length > 0) {
+    while (poppedSizeBytes + (this.peek()?.size ?? 0) < upToSizeBytes && this.items.length > 0 && popped.length < upToSize) {
       const item = this.items.shift();
       if (item) {
         popped.push(item);
@@ -26850,6 +26850,7 @@ class AutoBatchQueue {
 }
 var DEFAULT_UNCOMPRESSED_BATCH_SIZE_LIMIT_BYTES = 24 * 1024 * 1024;
 var SERVER_INFO_REQUEST_TIMEOUT_MS = 1e4;
+var DEFAULT_BATCH_SIZE_LIMIT = 100;
 var DEFAULT_API_URL = "https://api.smith.langchain.com";
 
 class Client {
@@ -26959,6 +26960,12 @@ class Client {
       writable: true,
       value: undefined
     });
+    Object.defineProperty(this, "batchSizeLimit", {
+      enumerable: true,
+      configurable: true,
+      writable: true,
+      value: undefined
+    });
     Object.defineProperty(this, "fetchOptions", {
       enumerable: true,
       configurable: true,
@@ -27061,6 +27068,7 @@ class Client {
     this.autoBatchTracing = config.autoBatchTracing ?? this.autoBatchTracing;
     this.blockOnRootRunFinalization = config.blockOnRootRunFinalization ?? this.blockOnRootRunFinalization;
     this.batchSizeBytesLimit = config.batchSizeBytesLimit;
+    this.batchSizeLimit = config.batchSizeLimit;
     this.fetchOptions = config.fetchOptions || {};
     this.manualFlushMode = config.manualFlushMode ?? this.manualFlushMode;
     if (getOtelEnabled()) {
@@ -27280,14 +27288,21 @@ class Client {
     const serverInfo = await this._ensureServerInfo();
     return this.batchSizeBytesLimit ?? serverInfo.batch_ingest_config?.size_limit_bytes ?? DEFAULT_UNCOMPRESSED_BATCH_SIZE_LIMIT_BYTES;
   }
+  async _getBatchSizeLimit() {
+    const serverInfo = await this._ensureServerInfo();
+    return this.batchSizeLimit ?? serverInfo.batch_ingest_config?.size_limit ?? DEFAULT_BATCH_SIZE_LIMIT;
+  }
   async _getDatasetExamplesMultiPartSupport() {
     const serverInfo = await this._ensureServerInfo();
     return serverInfo.instance_flags?.dataset_examples_multipart_enabled ?? false;
   }
-  drainAutoBatchQueue(batchSizeLimit) {
+  drainAutoBatchQueue({ batchSizeLimitBytes, batchSizeLimit }) {
     const promises = [];
     while (this.autoBatchQueue.items.length > 0) {
-      const [batch, done] = this.autoBatchQueue.pop(batchSizeLimit);
+      const [batch, done] = this.autoBatchQueue.pop({
+        upToSizeBytes: batchSizeLimitBytes,
+        upToSize: batchSizeLimit
+      });
       if (!batch.length) {
         done();
         break;
@@ -27376,13 +27391,20 @@ class Client {
       return itemPromise;
     }
     const sizeLimitBytes = await this._getBatchSizeLimitBytes();
-    if (this.autoBatchQueue.sizeBytes > sizeLimitBytes) {
-      this.drainAutoBatchQueue(sizeLimitBytes);
+    const sizeLimit = await this._getBatchSizeLimit();
+    if (this.autoBatchQueue.sizeBytes > sizeLimitBytes || this.autoBatchQueue.items.length > sizeLimit) {
+      this.drainAutoBatchQueue({
+        batchSizeLimitBytes: sizeLimitBytes,
+        batchSizeLimit: sizeLimit
+      });
     }
     if (this.autoBatchQueue.items.length > 0) {
       this.autoBatchTimeout = setTimeout(() => {
         this.autoBatchTimeout = undefined;
-        this.drainAutoBatchQueue(sizeLimitBytes);
+        this.drainAutoBatchQueue({
+          batchSizeLimitBytes: sizeLimitBytes,
+          batchSizeLimit: sizeLimit
+        });
       }, this.autoBatchAggregationDelayMs);
     }
     return itemPromise;
@@ -27435,7 +27457,11 @@ class Client {
   }
   async flush() {
     const sizeLimitBytes = await this._getBatchSizeLimitBytes();
-    await this.drainAutoBatchQueue(sizeLimitBytes);
+    const sizeLimit = await this._getBatchSizeLimit();
+    await this.drainAutoBatchQueue({
+      batchSizeLimitBytes: sizeLimitBytes,
+      batchSizeLimit: sizeLimit
+    });
   }
   _cloneCurrentOTELContext() {
     const otel_trace = getOTELTrace();
@@ -46350,5 +46376,5 @@ export {
   AgentRuntime
 };
 
-//# debugId=7AD07AFE80EE1A4264756E2164756E21
+//# debugId=714E2C90B3E960C464756E2164756E21
 //# sourceMappingURL=index.node.js.map