taist 0.2.2 → 0.2.4

package/lib/transform.js

@@ -150,6 +150,15 @@ const __taist_debug = process.env.TAIST_DEBUG === 'true';
 if (process.env.TAIST_COLLECTOR_SOCKET && !__taist_reporter.isConnected()) {
   __taist_reporter.connectEager();
 }
+// Helper to find correlationId from GraphQL context argument (Apollo: parent, args, context, info)
+const __taist_findCorrelationIdInArgs = (args) => {
+  for (const arg of args) {
+    if (arg && typeof arg === 'object' && arg.taistCorrelationId) {
+      return arg.taistCorrelationId;
+    }
+  }
+  return null;
+};
 const __taist_wrap = (fn, name) => {
   if (typeof fn !== 'function') return fn;
   if (__taist_debug) console.log('[taist] wrapping:', name);
@@ -158,8 +167,10 @@ const __taist_wrap = (fn, name) => {
     const parentCtx = __taist_getContext();
     const id = __taist_generateId();
     const depth = parentCtx.depth;
-    // Get correlationId from parent context or fallback global
-    const correlationId = parentCtx.correlationId || __taist_getCorrelationId();
+    // Get correlationId: 1) parent context, 2) GraphQL context arg, 3) fallback global
+    const argsCorrelationId = __taist_findCorrelationIdInArgs(args);
+    const correlationId = parentCtx.correlationId || argsCorrelationId || __taist_getCorrelationId();
+    if (__taist_debug && argsCorrelationId) console.log('[taist] Found correlationId in args:', argsCorrelationId);
     const newCtx = {
       depth: depth + 1,
       traceId: parentCtx.traceId || id,
@@ -171,6 +182,7 @@ const __taist_wrap = (fn, name) => {
     const start = performance.now();
 
     const reportSuccess = (result) => {
+      if (__taist_debug) console.log('[taist] REPORT:', name, 'depth:', depth, 'correlationId:', correlationId);
       __taist_reporter.report({
         id, name, type: 'exit', args, result,
         duration: performance.now() - start,
@@ -184,6 +196,7 @@ const __taist_wrap = (fn, name) => {
     };
 
     const reportError = (err) => {
+      if (__taist_debug) console.log('[taist] REPORT ERROR:', name, 'depth:', depth, 'correlationId:', correlationId, 'error:', err.message);
       __taist_reporter.report({
         id, name, type: 'error', args,
         error: { name: err.name, message: err.message },

package/package.json

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