taist 0.2.13 → 0.2.15

package/lib/trace-reporter.js

@@ -542,9 +542,22 @@ export function report(trace) {
  * Flush the global reporter
  */
 export async function flush() {
-  if (globalReporter) {
-    await globalReporter.flush();
+  const reporter = globalThis[TAIST_REPORTER_KEY];
+  if (reporter) {
+    await reporter.flush();
   }
 }
 
+/**
+ * Reset the global reporter (for testing purposes).
+ * Closes the existing reporter and clears the singleton.
+ */
+export function resetGlobalReporter() {
+  const reporter = globalThis[TAIST_REPORTER_KEY];
+  if (reporter) {
+    reporter.close();
+  }
+  globalThis[TAIST_REPORTER_KEY] = null;
+}
+
 export default TraceReporter;

package/lib/vitest-reporter.js

@@ -33,6 +33,7 @@ import fs from 'fs';
  * @property {boolean} [silent=false] - Suppress output
  * @property {string | null} [outputFile=null] - Write to file instead of stdout
  * @property {number} [maxTraceGroups=10] - Max request groups to show in trace output
+ * @property {number} [traceGracePeriod=500] - Grace period in ms to wait for late-arriving traces before stopping collector
  */
 
 export class TaistReporter {
@@ -48,6 +49,7 @@ export class TaistReporter {
       silent: options.silent || false,
       outputFile: options.outputFile || null,
       maxTraceGroups: options.maxTraceGroups || 10,
+      traceGracePeriod: options.traceGracePeriod ?? 500, // Grace period for late-arriving traces
       ...options
     };
 
@@ -148,6 +150,12 @@ export class TaistReporter {
         await this.collectorReady;
       }
 
+      // Wait for late-arriving traces (e.g., from child processes where the
+      // outermost resolver trace completes after the HTTP response is sent)
+      if (this.options.traceGracePeriod > 0) {
+        await new Promise(resolve => setTimeout(resolve, this.options.traceGracePeriod));
+      }
+
       // Stop the collector gracefully - this sends shutdown signal to workers
       // and waits for them to flush their traces before closing (up to 2s timeout)
       await this.collector.stop();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "taist",
-  "version": "0.2.13",
+  "version": "0.2.15",
   "description": "Token-Optimized Testing Framework for AI-Assisted Development",
   "main": "index.js",
   "type": "module",