@langwatch/scenario 0.4.5 → 0.4.6

package/dist/index.js

@@ -3897,6 +3897,7 @@ var ScenarioExecution = class {
     });
     this.currentTurnSpan = this.tracer.startSpan("Scenario Turn", {
       attributes: {
+        "langwatch.origin": "simulation",
         "scenario.name": this.config.name,
         "scenario.id": this.config.id,
         [import_observability3.attributes.ATTR_LANGWATCH_THREAD_ID]: this.state.threadId,
@@ -4412,12 +4413,17 @@ var EventBus = class _EventBus {
   }
   /**
    * Stops accepting new events and drains the processing queue.
+   * Times out after the specified duration to prevent blocking indefinitely
+   * when the events endpoint is slow or unavailable.
    */
-  async drain() {
+  async drain(timeoutMs = 3e5) {
     this.logger.debug("Draining event stream");
     this.events$.complete();
     if (this.processingPromise) {
-      await this.processingPromise;
+      await Promise.race([
+        this.processingPromise,
+        new Promise((resolve) => setTimeout(resolve, timeoutMs))
+      ]);
     }
   }
   /**

package/dist/index.mjs

@@ -3837,6 +3837,7 @@ var ScenarioExecution = class {
     });
     this.currentTurnSpan = this.tracer.startSpan("Scenario Turn", {
       attributes: {
+        "langwatch.origin": "simulation",
         "scenario.name": this.config.name,
         "scenario.id": this.config.id,
         [attributes3.ATTR_LANGWATCH_THREAD_ID]: this.state.threadId,
@@ -4359,12 +4360,17 @@ var EventBus = class _EventBus {
   }
   /**
    * Stops accepting new events and drains the processing queue.
+   * Times out after the specified duration to prevent blocking indefinitely
+   * when the events endpoint is slow or unavailable.
    */
-  async drain() {
+  async drain(timeoutMs = 3e5) {
     this.logger.debug("Draining event stream");
     this.events$.complete();
     if (this.processingPromise) {
-      await this.processingPromise;
+      await Promise.race([
+        this.processingPromise,
+        new Promise((resolve) => setTimeout(resolve, timeoutMs))
+      ]);
     }
   }
   /**

package/dist/integrations/vitest/setup.js

@@ -606,12 +606,17 @@ var EventBus = class _EventBus {
   }
   /**
    * Stops accepting new events and drains the processing queue.
+   * Times out after the specified duration to prevent blocking indefinitely
+   * when the events endpoint is slow or unavailable.
    */
-  async drain() {
+  async drain(timeoutMs = 3e5) {
     this.logger.debug("Draining event stream");
     this.events$.complete();
     if (this.processingPromise) {
-      await this.processingPromise;
+      await Promise.race([
+        this.processingPromise,
+        new Promise((resolve) => setTimeout(resolve, timeoutMs))
+      ]);
     }
   }
   /**

package/dist/integrations/vitest/setup.mjs

@@ -589,12 +589,17 @@ var EventBus = class _EventBus {
   }
   /**
    * Stops accepting new events and drains the processing queue.
+   * Times out after the specified duration to prevent blocking indefinitely
+   * when the events endpoint is slow or unavailable.
    */
-  async drain() {
+  async drain(timeoutMs = 3e5) {
     this.logger.debug("Draining event stream");
     this.events$.complete();
     if (this.processingPromise) {
-      await this.processingPromise;
+      await Promise.race([
+        this.processingPromise,
+        new Promise((resolve) => setTimeout(resolve, timeoutMs))
+      ]);
     }
   }
   /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langwatch/scenario",
-  "version": "0.4.5",
+  "version": "0.4.6",
   "description": "A TypeScript library for testing AI agents using scenarios",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -22,7 +22,7 @@
   },
   "homepage": "https://github.com/langwatch/scenario#readme",
   "author": "LangWatch",
-  "license": "MIT",
+  "license": "AGPL-3.0",
   "engines": {
     "node": ">=20",
     "pnpm": ">=8"