@langwatch/scenario 0.2.2 → 0.2.9

package/dist/integrations/vitest/setup.js

@@ -30,19 +30,66 @@ var import_vitest = require("vitest");
 // src/events/event-bus.ts
 var import_rxjs = require("rxjs");
 
+// src/config/env.ts
+var import_zod = require("zod");
+
+// src/config/log-levels.ts
+var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
+  LogLevel2["ERROR"] = "ERROR";
+  LogLevel2["WARN"] = "WARN";
+  LogLevel2["INFO"] = "INFO";
+  LogLevel2["DEBUG"] = "DEBUG";
+  return LogLevel2;
+})(LogLevel || {});
+
+// src/config/env.ts
+var envSchema = import_zod.z.object({
+  /**
+   * LangWatch API key for event reporting.
+   * If not provided, events will not be sent to LangWatch.
+   */
+  LANGWATCH_API_KEY: import_zod.z.string().optional(),
+  /**
+   * LangWatch endpoint URL for event reporting.
+   * Defaults to the production LangWatch endpoint.
+   */
+  LANGWATCH_ENDPOINT: import_zod.z.string().url().default("https://app.langwatch.ai"),
+  /**
+   * Disables simulation report info messages when set to any truthy value.
+   * Useful for CI/CD environments or when you want cleaner output.
+   */
+  SCENARIO_DISABLE_SIMULATION_REPORT_INFO: import_zod.z.string().optional().transform((val) => Boolean(val)),
+  /**
+   * Node environment - affects logging and behavior.
+   * Defaults to 'development' if not specified.
+   */
+  NODE_ENV: import_zod.z.enum(["development", "production", "test"]).default("development"),
+  /**
+   * Log level for the scenario package.
+   * Defaults to 'info' if not specified.
+   */
+  LOG_LEVEL: import_zod.z.nativeEnum(LogLevel).optional(),
+  /**
+   * Scenario batch run ID.
+   * If not provided, a random ID will be generated.
+   */
+  SCENARIO_BATCH_RUN_ID: import_zod.z.string().optional()
+});
+var env = envSchema.parse(process.env);
+
 // src/config/load.ts
 var import_promises = __toESM(require("fs/promises"));
 var import_node_path = __toESM(require("path"));
 var import_node_url = require("url");
 
 // src/domain/core/config.ts
-var import_zod = require("zod");
+var import_zod2 = require("zod");
 var DEFAULT_TEMPERATURE = 0;
-var scenarioProjectConfigSchema = import_zod.z.object({
-  defaultModel: import_zod.z.object({
-    model: import_zod.z.custom(),
-    temperature: import_zod.z.number().min(0).max(1).optional().default(DEFAULT_TEMPERATURE),
-    maxTokens: import_zod.z.number().optional()
+var scenarioProjectConfigSchema = import_zod2.z.object({
+  defaultModel: import_zod2.z.object({
+    model: import_zod2.z.custom(),
+    temperature: import_zod2.z.number().min(0).max(1).optional().default(DEFAULT_TEMPERATURE),
+    maxTokens: import_zod2.z.number().optional()
   }).optional()
 }).strict();
 
@@ -58,7 +105,7 @@ var Logger = class _Logger {
     return new _Logger(context);
   }
   getLogLevel() {
-    return env.SCENARIO_LOG_LEVEL ?? "INFO" /* INFO */;
+    return env.LOG_LEVEL ?? "INFO" /* INFO */;
   }
   getLogLevelIndex(level) {
     return Object.values(LogLevel).indexOf(level);
@@ -116,63 +163,40 @@ var Logger = class _Logger {
   }
 };
 
-// src/config/env.ts
-var import_zod2 = require("zod");
-
-// src/config/log-levels.ts
-var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
-  LogLevel2["ERROR"] = "ERROR";
-  LogLevel2["WARN"] = "WARN";
-  LogLevel2["INFO"] = "INFO";
-  LogLevel2["DEBUG"] = "DEBUG";
-  return LogLevel2;
-})(LogLevel || {});
-
-// src/config/env.ts
-var envSchema = import_zod2.z.object({
-  /**
-   * LangWatch API key for event reporting.
-   * If not provided, events will not be sent to LangWatch.
-   */
-  LANGWATCH_API_KEY: import_zod2.z.string().optional(),
-  /**
-   * LangWatch endpoint URL for event reporting.
-   * Defaults to the production LangWatch endpoint.
-   */
-  LANGWATCH_ENDPOINT: import_zod2.z.string().url().default("https://app.langwatch.ai"),
-  /**
-   * Disables simulation report info messages when set to any truthy value.
-   * Useful for CI/CD environments or when you want cleaner output.
-   */
-  SCENARIO_DISABLE_SIMULATION_REPORT_INFO: import_zod2.z.string().optional().transform((val) => Boolean(val)),
-  /**
-   * Node environment - affects logging and behavior.
-   * Defaults to 'development' if not specified.
-   */
-  NODE_ENV: import_zod2.z.enum(["development", "production", "test"]).default("development"),
-  /**
-   * Log level for the scenario package.
-   * Defaults to 'info' if not specified.
-   */
-  SCENARIO_LOG_LEVEL: import_zod2.z.nativeEnum(LogLevel).optional(),
-  /**
-   * Scenario batch run ID.
-   * If not provided, a random ID will be generated.
-   */
-  SCENARIO_BATCH_RUN_ID: import_zod2.z.string().optional()
-});
-var env = envSchema.parse(process.env);
-
-// src/config/index.ts
+// src/config/get-project-config.ts
 var logger = new Logger("scenario.config");
 
 // src/utils/ids.ts
+var import_node_crypto = __toESM(require("crypto"));
+var import_node_process = __toESM(require("process"));
 var import_xksuid = require("xksuid");
+var batchRunId;
 function getBatchRunId() {
-  if (!env.SCENARIO_BATCH_RUN_ID) {
-    env.SCENARIO_BATCH_RUN_ID = `scenariobatchrun_${(0, import_xksuid.generate)()}`;
+  if (batchRunId) {
+    return batchRunId;
   }
-  return env.SCENARIO_BATCH_RUN_ID;
+  if (import_node_process.default.env.SCENARIO_BATCH_RUN_ID) {
+    console.log("process.env.SCENARIO_BATCH_RUN_ID", import_node_process.default.env.SCENARIO_BATCH_RUN_ID);
+    return batchRunId = import_node_process.default.env.SCENARIO_BATCH_RUN_ID;
+  }
+  if (import_node_process.default.env.VITEST_WORKER_ID || import_node_process.default.env.JEST_WORKER_ID) {
+    const parentProcessId = import_node_process.default.ppid;
+    const now = /* @__PURE__ */ new Date();
+    const year = now.getUTCFullYear();
+    const week = String(getISOWeekNumber(now)).padStart(2, "0");
+    const raw = `${parentProcessId}_${year}_w${week}`;
+    const hash = import_node_crypto.default.createHash("sha256").update(raw).digest("hex").slice(0, 12);
+    return batchRunId = `scenariobatchrun_${hash}`;
+  }
+  return batchRunId = `scenariobatchrun_${(0, import_xksuid.generate)()}`;
+}
+function getISOWeekNumber(date) {
+  const tmp = new Date(Date.UTC(date.getUTCFullYear(), date.getUTCMonth(), date.getUTCDate()));
+  const dayNum = tmp.getUTCDay() || 7;
+  tmp.setUTCDate(tmp.getUTCDate() + 4 - dayNum);
+  const yearStart = new Date(Date.UTC(tmp.getUTCFullYear(), 0, 1));
+  const weekNo = Math.ceil(((tmp.getTime() - yearStart.getTime()) / 864e5 + 1) / 7);
+  return weekNo;
 }
 
 // src/events/event-alert-message-logger.ts
@@ -186,12 +210,11 @@ var EventAlertMessageLogger = class _EventAlertMessageLogger {
     if (this.isGreetingDisabled()) {
       return;
     }
-    const batchRunId = getBatchRunId();
-    if (_EventAlertMessageLogger.shownBatchIds.has(batchRunId)) {
+    if (_EventAlertMessageLogger.shownBatchIds.has(getBatchRunId())) {
       return;
     }
-    _EventAlertMessageLogger.shownBatchIds.add(batchRunId);
-    this.displayGreeting(batchRunId);
+    _EventAlertMessageLogger.shownBatchIds.add(getBatchRunId());
+    this.displayGreeting();
   }
   /**
    * Shows a fancy message about how to watch the simulation.
@@ -206,7 +229,7 @@ var EventAlertMessageLogger = class _EventAlertMessageLogger {
   isGreetingDisabled() {
     return env.SCENARIO_DISABLE_SIMULATION_REPORT_INFO === true;
   }
-  displayGreeting(batchRunId) {
+  displayGreeting() {
     const separator = "\u2500".repeat(60);
     if (!env.LANGWATCH_API_KEY) {
       console.log(`
@@ -220,7 +243,10 @@ ${separator}`);
       console.log("   \u2022 Set LANGWATCH_API_KEY environment variable");
       console.log("   \u2022 Or configure apiKey in scenario.config.js");
       console.log("");
-      console.log(`\u{1F4E6} Batch Run ID: ${batchRunId}`);
+      console.log(`\u{1F4E6} Batch Run ID: ${getBatchRunId()}`);
+      console.log("");
+      console.log("\u{1F507} To disable these messages:");
+      console.log("   \u2022 Set SCENARIO_DISABLE_SIMULATION_REPORT_INFO=true");
       console.log(`${separator}
 `);
     } else {
@@ -234,7 +260,10 @@ ${separator}`);
         `   API Key: ${env.LANGWATCH_API_KEY.length > 0 ? "Configured" : "Not configured"}`
       );
       console.log("");
-      console.log(`\u{1F4E6} Batch Run ID: ${batchRunId}`);
+      console.log(`\u{1F4E6} Batch Run ID: ${getBatchRunId()}`);
+      console.log("");
+      console.log("\u{1F507} To disable these messages:");
+      console.log("   \u2022 Set SCENARIO_DISABLE_SIMULATION_REPORT_INFO=true");
       console.log(`${separator}
 `);
     }
@@ -510,6 +539,7 @@ var EventBus = class _EventBus {
 };
 
 // src/integrations/vitest/setup.ts
+var logger2 = Logger.create("integrations:vitest:setup");
 function getProjectRoot() {
   return process.cwd();
 }
@@ -530,7 +560,7 @@ var subs = [];
       try {
         import_fs.default.appendFileSync(filePath, JSON.stringify(event) + "\n");
       } catch (error) {
-        console.error("Error writing to log file:", error);
+        logger2.error("Error writing to log file:", error);
       }
     })
   );
@@ -542,7 +572,7 @@ EventBus.addGlobalListener((bus) => {
       try {
         import_fs.default.appendFileSync(filePath, JSON.stringify(event) + "\n");
       } catch (error) {
-        console.error("Error writing to log file:", error);
+        logger2.error("Error writing to log file:", error);
       }
     })
   );

package/dist/integrations/vitest/setup.mjs

@@ -1,12 +1,16 @@
 import {
   EventBus
-} from "../../chunk-NUZZAQV2.mjs";
+} from "../../chunk-7H6OGEQ5.mjs";
+import {
+  Logger
+} from "../../chunk-YPJZSK4J.mjs";
 import "../../chunk-7P6ASYW6.mjs";
 
 // src/integrations/vitest/setup.ts
 import fs from "fs";
 import path from "path";
 import { beforeEach, afterEach } from "vitest";
+var logger = Logger.create("integrations:vitest:setup");
 function getProjectRoot() {
   return process.cwd();
 }
@@ -27,7 +31,7 @@ beforeEach((ctx) => {
       try {
         fs.appendFileSync(filePath, JSON.stringify(event) + "\n");
       } catch (error) {
-        console.error("Error writing to log file:", error);
+        logger.error("Error writing to log file:", error);
       }
     })
   );
@@ -39,7 +43,7 @@ EventBus.addGlobalListener((bus) => {
       try {
         fs.appendFileSync(filePath, JSON.stringify(event) + "\n");
       } catch (error) {
-        console.error("Error writing to log file:", error);
+        logger.error("Error writing to log file:", error);
       }
     })
   );

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langwatch/scenario",
-  "version": "0.2.2",
+  "version": "0.2.9",
   "description": "A TypeScript library for testing AI agents using scenarios",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -72,6 +72,16 @@
       "types": "./dist/integrations/vitest/setup.d.ts",
       "require": "./dist/integrations/vitest/setup.js",
       "import": "./dist/integrations/vitest/setup.mjs"
+    },
+    "./integrations/vitest/setup-global": {
+      "types": "./dist/integrations/vitest/setup-global.d.ts",
+      "require": "./dist/integrations/vitest/setup-global.js",
+      "import": "./dist/integrations/vitest/setup-global.mjs"
+    },
+    "./integrations/vitest/config": {
+      "types": "./dist/integrations/vitest/config.d.ts",
+      "require": "./dist/integrations/vitest/config.js",
+      "import": "./dist/integrations/vitest/config.mjs"
     }
   },
   "peerDependencies": {
@@ -84,8 +94,7 @@
     "test": "vitest",
     "test:ci": "vitest run",
     "lint": "eslint .",
-    "examples:vitest:run": "export SCENARIO_BATCH_ID=scenariobatch_$(uuidgen) && pnpm run buildpack && (cd examples/vitest && pnpm install) && pnpm -F vitest-example run test",
-    "hash-source": "find src -name \"*.ts\" -type f | sort | xargs cat | sha256sum | cut -d' ' -f1",
-    "generate:api-reference": "npx typedoc src --out api-reference-docs && rm -rf ../docs/docs/public/reference/javascript/scenario && mv api-reference-docs ../docs/docs/public/reference/javascript/scenario && pnpm run hash-source > ../docs/docs/public/reference/javascript/.docs-source-hash"
+    "examples:vitest:run": "export SCENARIO_BATCH_ID=scenariobatch_$(uuidgen) && (cd examples/vitest && pnpm install) && pnpm -F vitest-example run test",
+    "generate:api-reference": "npx typedoc src --out api-reference-docs && rm -rf ../docs/docs/public/reference/javascript/scenario && mv api-reference-docs ../docs/docs/public/reference/javascript/scenario"
   }
 }