@langwatch/scenario 0.2.6 → 0.2.9

package/dist/integrations/vitest/config.mjs

@@ -0,0 +1,35 @@
+import {
+  VitestReporter
+} from "../../chunk-K7KLHTDI.mjs";
+import "../../chunk-YPJZSK4J.mjs";
+import "../../chunk-7P6ASYW6.mjs";
+
+// src/integrations/vitest/config.ts
+import { defineConfig } from "vitest/config";
+function withScenario(config) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
+  const normalizedSetupFiles = ((_a = config.test) == null ? void 0 : _a.setupFiles) === void 0 ? [] : Array.isArray((_b = config.test) == null ? void 0 : _b.setupFiles) ? (_c = config.test) == null ? void 0 : _c.setupFiles : [(_d = config.test) == null ? void 0 : _d.setupFiles];
+  const normalizedGlobalSetup = ((_e = config.test) == null ? void 0 : _e.globalSetup) === void 0 ? [] : Array.isArray((_f = config.test) == null ? void 0 : _f.globalSetup) ? (_g = config.test) == null ? void 0 : _g.globalSetup : [(_h = config.test) == null ? void 0 : _h.globalSetup];
+  const normalizedReporters = ((_i = config.test) == null ? void 0 : _i.reporters) === void 0 ? [] : Array.isArray((_j = config.test) == null ? void 0 : _j.reporters) ? (_k = config.test) == null ? void 0 : _k.reporters : [(_l = config.test) == null ? void 0 : _l.reporters];
+  return defineConfig({
+    ...config,
+    test: {
+      ...config.test,
+      setupFiles: [
+        "@langwatch/scenario/integrations/vitest/setup",
+        ...normalizedSetupFiles
+      ],
+      globalSetup: [
+        "@langwatch/scenario/integrations/vitest/setup-global",
+        ...normalizedGlobalSetup
+      ],
+      reporters: [
+        ...normalizedReporters,
+        new VitestReporter()
+      ]
+    }
+  });
+}
+export {
+  withScenario
+};

package/dist/integrations/vitest/reporter.js

@@ -36,6 +36,126 @@ module.exports = __toCommonJS(reporter_exports);
 var import_fs = __toESM(require("fs"));
 var import_path = __toESM(require("path"));
 var import_chalk = __toESM(require("chalk"));
+
+// 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/utils/logger.ts
+var Logger = class _Logger {
+  constructor(context) {
+    this.context = context;
+  }
+  /**
+   * Creates a logger with context (e.g., class name)
+   */
+  static create(context) {
+    return new _Logger(context);
+  }
+  getLogLevel() {
+    return env.LOG_LEVEL ?? "INFO" /* INFO */;
+  }
+  getLogLevelIndex(level) {
+    return Object.values(LogLevel).indexOf(level);
+  }
+  /**
+   * Checks if logging should occur based on LOG_LEVEL env var
+   */
+  shouldLog(level) {
+    const currentLevelIndex = this.getLogLevelIndex(this.getLogLevel());
+    const requestedLevelIndex = this.getLogLevelIndex(level);
+    return currentLevelIndex >= 0 && requestedLevelIndex <= currentLevelIndex;
+  }
+  formatMessage(message) {
+    return this.context ? `[${this.context}] ${message}` : message;
+  }
+  error(message, data) {
+    if (this.shouldLog("ERROR" /* ERROR */)) {
+      const formattedMessage = this.formatMessage(message);
+      if (data) {
+        console.error(formattedMessage, data);
+      } else {
+        console.error(formattedMessage);
+      }
+    }
+  }
+  warn(message, data) {
+    if (this.shouldLog("WARN" /* WARN */)) {
+      const formattedMessage = this.formatMessage(message);
+      if (data) {
+        console.warn(formattedMessage, data);
+      } else {
+        console.warn(formattedMessage);
+      }
+    }
+  }
+  info(message, data) {
+    if (this.shouldLog("INFO" /* INFO */)) {
+      const formattedMessage = this.formatMessage(message);
+      if (data) {
+        console.info(formattedMessage, data);
+      } else {
+        console.info(formattedMessage);
+      }
+    }
+  }
+  debug(message, data) {
+    if (this.shouldLog("DEBUG" /* DEBUG */)) {
+      const formattedMessage = this.formatMessage(message);
+      if (data) {
+        console.log(formattedMessage, data);
+      } else {
+        console.log(formattedMessage);
+      }
+    }
+  }
+};
+
+// src/integrations/vitest/reporter.ts
+var logger = Logger.create("integrations:vitest:reporter");
 function getProjectRoot() {
   return process.cwd();
 }
@@ -61,7 +181,10 @@ var VitestReporter = class {
     const fullName = getFullTestName(test);
     const filePath = getLogFilePath(test.id);
     if (!import_fs.default.existsSync(filePath)) {
-      console.log(`No log file found ${filePath} for test ${fullName}`);
+      logger.warn(
+        `No log file found ${filePath} for test ${fullName}`,
+        test.id
+      );
       return;
     }
     const lines = import_fs.default.readFileSync(filePath, "utf-8").split("\n").filter(Boolean);

package/dist/integrations/vitest/reporter.mjs

@@ -1,139 +1,8 @@
+import {
+  VitestReporter
+} from "../../chunk-K7KLHTDI.mjs";
+import "../../chunk-YPJZSK4J.mjs";
 import "../../chunk-7P6ASYW6.mjs";
-
-// src/integrations/vitest/reporter.ts
-import fs from "fs";
-import path from "path";
-import chalk from "chalk";
-function getProjectRoot() {
-  return process.cwd();
-}
-var projectRoot = getProjectRoot();
-var logDir = path.join(projectRoot, ".scenario");
-if (!fs.existsSync(logDir)) fs.mkdirSync(logDir);
-function getLogFilePath(testId) {
-  return path.join(logDir, `${testId}.log`);
-}
-function getFullTestName(task) {
-  let name = task.name;
-  let parent = task.suite;
-  while (parent) {
-    name = `${parent.name} > ${name}`;
-    parent = parent.suite;
-  }
-  return name;
-}
-var VitestReporter = class {
-  results = [];
-  async onTestCaseResult(test) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
-    const fullName = getFullTestName(test);
-    const filePath = getLogFilePath(test.id);
-    if (!fs.existsSync(filePath)) {
-      console.log(`No log file found ${filePath} for test ${fullName}`);
-      return;
-    }
-    const lines = fs.readFileSync(filePath, "utf-8").split("\n").filter(Boolean);
-    const events = lines.map((line) => JSON.parse(line));
-    const runs = /* @__PURE__ */ new Map();
-    for (const event of events) {
-      const runId = event.scenarioRunId ?? "unknown";
-      if (!runs.has(runId)) runs.set(runId, []);
-      runs.get(runId).push(event);
-    }
-    for (const [runId, runEvents] of Array.from(runs.entries())) {
-      const started = runEvents.find(
-        (e) => e.type === "SCENARIO_RUN_STARTED"
-      );
-      const finished = runEvents.find(
-        (e) => e.type === "SCENARIO_RUN_FINISHED"
-      );
-      const messages = runEvents.filter(
-        (e) => e.type === "SCENARIO_MESSAGE_SNAPSHOT"
-      );
-      this.results.push({
-        name: ((_a = started == null ? void 0 : started.metadata) == null ? void 0 : _a.name) ?? fullName,
-        status: (finished == null ? void 0 : finished.status) ?? "UNKNOWN",
-        duration: started && finished ? finished.timestamp - started.timestamp : 0,
-        reasoning: (_b = finished == null ? void 0 : finished.results) == null ? void 0 : _b.reasoning,
-        criteria: (finished == null ? void 0 : finished.results) ? `Success Criteria: ${((_c = finished.results.metCriteria) == null ? void 0 : _c.length) ?? 0}/${(((_d = finished.results.metCriteria) == null ? void 0 : _d.length) ?? 0) + (((_e = finished.results.unmetCriteria) == null ? void 0 : _e.length) ?? 0)}` : void 0
-      });
-      console.log(
-        `
---- Scenario Run: ${((_f = started == null ? void 0 : started.metadata) == null ? void 0 : _f.name) ?? runId} ---`
-      );
-      if (started) {
-        console.log(`Description: ${((_g = started.metadata) == null ? void 0 : _g.description) ?? ""}`);
-      }
-      if (messages.length) {
-        console.log("Chat log:");
-        let lastMessageCount = 0;
-        for (const msg of messages) {
-          const allMessages = msg.messages ?? [];
-          for (const m of allMessages.slice(lastMessageCount)) {
-            const role = m.role;
-            let roleLabel = role;
-            if (role.toLowerCase() === "user") roleLabel = chalk.green("User");
-            else if (role.toLowerCase() === "agent")
-              roleLabel = chalk.cyan("Agent");
-            else if (role.toLowerCase() === "assistant")
-              roleLabel = chalk.cyan("Assistant");
-            else roleLabel = chalk.yellow(role);
-            console.log(`${roleLabel}: ${m.content}`);
-          }
-          lastMessageCount = allMessages.length;
-        }
-      }
-      if (finished) {
-        console.log("--- Verdict ---");
-        console.log(`Status: ${finished.status}`);
-        if (finished.results) {
-          console.log(`Verdict: ${finished.results.verdict}`);
-          if (finished.results.reasoning)
-            console.log(`Reasoning: ${finished.results.reasoning}`);
-          if ((_h = finished.results.metCriteria) == null ? void 0 : _h.length)
-            console.log(
-              `Met criteria: ${finished.results.metCriteria.join(", ")}`
-            );
-          if ((_i = finished.results.unmetCriteria) == null ? void 0 : _i.length)
-            console.log(
-              `Unmet criteria: ${finished.results.unmetCriteria.join(", ")}`
-            );
-          if (finished.results.error)
-            console.log(`Error: ${finished.results.error}`);
-        }
-      }
-      console.log("-----------------------------\n");
-    }
-    fs.unlinkSync(filePath);
-  }
-  async onTestRunEnd() {
-    if (this.results.length === 0) return;
-    const total = this.results.length;
-    const passed = this.results.filter((r) => r.status === "SUCCESS").length;
-    const failed = this.results.filter((r) => r.status !== "SUCCESS").length;
-    const successRate = (passed / total * 100).toFixed(1);
-    console.log();
-    console.log(chalk.bold.cyan("=== Scenario Test Report ==="));
-    console.log(`Total Scenarios: ${total}`);
-    console.log(chalk.green(`Passed: ${passed}`));
-    console.log(chalk.red(`Failed: ${failed}`));
-    console.log(`Success Rate: ${chalk.bold(`${successRate}%`)}`);
-    this.results.forEach((r, i) => {
-      const statusColor = r.status === "SUCCESS" ? chalk.green : chalk.red;
-      console.log();
-      console.log(
-        `${i + 1}. ${r.name} - ${statusColor(r.status)} in ${(r.duration / 1e3).toFixed(2)}s`
-      );
-      if (r.reasoning) {
-        console.log(chalk.greenBright("  Reasoning: ") + r.reasoning);
-      }
-      if (r.criteria) {
-        console.log(chalk.bold("  " + r.criteria));
-      }
-    });
-    console.log();
-  }
-};
 export {
   VitestReporter as default
 };

package/dist/integrations/vitest/setup-global.d.mts

@@ -0,0 +1,3 @@
+declare function setup(): void;
+
+export { setup as default };

package/dist/integrations/vitest/setup-global.d.ts

@@ -0,0 +1,3 @@
+declare function setup(): void;
+
+export { setup as default };

package/dist/integrations/vitest/setup-global.js

@@ -0,0 +1,30 @@
+"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);
+
+// src/integrations/vitest/setup-global.ts
+var setup_global_exports = {};
+__export(setup_global_exports, {
+  default: () => setup
+});
+module.exports = __toCommonJS(setup_global_exports);
+var import_xksuid = require("xksuid");
+function setup() {
+  const scenarioBatchRunId = `scenariobatchrun_${(0, import_xksuid.generate)()}`;
+  process.env.SCENARIO_BATCH_RUN_ID = scenarioBatchRunId;
+}

package/dist/integrations/vitest/setup-global.mjs

@@ -0,0 +1,11 @@
+import "../../chunk-7P6ASYW6.mjs";
+
+// src/integrations/vitest/setup-global.ts
+import { generate } from "xksuid";
+function setup() {
+  const scenarioBatchRunId = `scenariobatchrun_${generate()}`;
+  process.env.SCENARIO_BATCH_RUN_ID = scenarioBatchRunId;
+}
+export {
+  setup as default
+};

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-MOOKAYIE.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.6",
+  "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": {