@wix/evalforge-evaluator 0.28.0 → 0.29.0

package/build/index.mjs

@@ -6639,13 +6639,23 @@ function createTraceEventFromMessage(message, context, stepNumber, isComplete) {
   let toolName;
   let toolArgs;
   let outputPreview;
+  let filePath;
   for (const block of message.message.content) {
     if (block.type === "tool_use") {
       type = LiveTraceEventType.TOOL_USE;
       toolName = block.name;
-      toolArgs = JSON.stringify(block.input).slice(0, 200);
+      toolArgs = JSON.stringify(block.input).slice(0, 500);
+      const input = block.input;
+      if (input.file_path || input.path || input.target_file) {
+        filePath = String(input.file_path || input.path || input.target_file);
+        if (block.name === "Write" || block.name === "Edit" || block.name === "write" || block.name === "edit") {
+          type = LiveTraceEventType.FILE_WRITE;
+        } else if (block.name === "Read" || block.name === "read" || block.name === "View") {
+          type = LiveTraceEventType.FILE_READ;
+        }
+      }
     } else if (block.type === "text") {
-      outputPreview = block.text.slice(0, 200);
+      outputPreview = block.text.slice(0, 500);
     }
   }
   return {
@@ -6659,10 +6669,64 @@ function createTraceEventFromMessage(message, context, stepNumber, isComplete) {
     toolName,
     toolArgs,
     outputPreview,
+    filePath,
     timestamp: (/* @__PURE__ */ new Date()).toISOString(),
     isComplete
   };
 }
+function createTraceEventFromAnyMessage(message, context, stepNumber, isComplete) {
+  const baseEvent = {
+    evalRunId: context.evalRunId,
+    scenarioId: context.scenarioId,
+    scenarioName: context.scenarioName,
+    targetId: context.targetId,
+    targetName: context.targetName,
+    stepNumber,
+    timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+    isComplete
+  };
+  if (isAssistantMessage(message)) {
+    return createTraceEventFromMessage(
+      message,
+      context,
+      stepNumber,
+      isComplete
+    );
+  }
+  if (message.type === "user") {
+    const userMsg = message;
+    let outputPreview = "";
+    if (userMsg.message?.content) {
+      for (const block of userMsg.message.content) {
+        if ("text" in block && block.text) {
+          outputPreview = block.text.slice(0, 500);
+          break;
+        }
+      }
+    }
+    return {
+      ...baseEvent,
+      type: LiveTraceEventType.USER,
+      outputPreview: outputPreview || "(tool result)"
+    };
+  }
+  if (message.type === "system") {
+    const sysMsg = message;
+    return {
+      ...baseEvent,
+      type: LiveTraceEventType.SYSTEM,
+      outputPreview: sysMsg.message?.slice(0, 500) || sysMsg.subtype || "system"
+    };
+  }
+  if (message.type === "result") {
+    return null;
+  }
+  return {
+    ...baseEvent,
+    type: LiveTraceEventType.PROGRESS,
+    outputPreview: `Message type: ${message.type}`
+  };
+}
 async function executeWithClaudeCode(skill, scenario, options) {
   console.log("[executeWithClaudeCode] Starting execution", {
     skillId: skill.id,
@@ -6859,6 +6923,9 @@ async function executeWithClaudeCode(skill, scenario, options) {
   const SDK_TIMEOUT_MS = Math.max(3e5, maxTurns * 6e4);
   let timeoutHandle;
   let timedOut = false;
+  const HEARTBEAT_INTERVAL_MS = 1e4;
+  let heartbeatHandle;
+  const executionStartTime = Date.now();
   try {
     const timeoutPromise = new Promise((_, reject) => {
       timeoutHandle = setTimeout(() => {
@@ -6870,6 +6937,30 @@ async function executeWithClaudeCode(skill, scenario, options) {
         );
       }, SDK_TIMEOUT_MS);
     });
+    if (traceContext) {
+      heartbeatHandle = setInterval(() => {
+        const elapsedMs = Date.now() - executionStartTime;
+        const progressEvent = {
+          evalRunId: traceContext.evalRunId,
+          scenarioId: traceContext.scenarioId,
+          scenarioName: traceContext.scenarioName,
+          targetId: traceContext.targetId,
+          targetName: traceContext.targetName,
+          stepNumber: traceStepNumber,
+          type: LiveTraceEventType.PROGRESS,
+          outputPreview: `Executing... (${Math.round(elapsedMs / 1e3)}s elapsed, ${messageCount} messages)`,
+          elapsedMs,
+          timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+          isComplete: false
+        };
+        emitTraceEvent(
+          progressEvent,
+          traceContext.tracePushUrl,
+          traceContext.routeHeader,
+          traceContext.authToken
+        );
+      }, HEARTBEAT_INTERVAL_MS);
+    }
     const sdkPromise = (async () => {
       const evaluatorPromptSuffix = `
 
@@ -6892,21 +6983,23 @@ IMPORTANT: This is an automated evaluation run. Execute the requested changes im
             })
           );
         }
-        if (traceContext && isAssistantMessage(message)) {
+        if (traceContext) {
           traceStepNumber++;
-          const traceEvent = createTraceEventFromMessage(
+          const traceEvent = createTraceEventFromAnyMessage(
             message,
             traceContext,
             traceStepNumber,
             false
             // Not complete yet
           );
-          emitTraceEvent(
-            traceEvent,
-            traceContext.tracePushUrl,
-            traceContext.routeHeader,
-            traceContext.authToken
-          );
+          if (traceEvent) {
+            emitTraceEvent(
+              traceEvent,
+              traceContext.tracePushUrl,
+              traceContext.routeHeader,
+              traceContext.authToken
+            );
+          }
         }
       }
     })();
@@ -6914,6 +7007,9 @@ IMPORTANT: This is an automated evaluation run. Execute the requested changes im
     if (timeoutHandle) {
       clearTimeout(timeoutHandle);
     }
+    if (heartbeatHandle) {
+      clearInterval(heartbeatHandle);
+    }
     console.log(
       "[executeWithClaudeCode] Claude Agent SDK query completed, received",
       allMessages.length,
@@ -6923,6 +7019,9 @@ IMPORTANT: This is an automated evaluation run. Execute the requested changes im
     if (timeoutHandle) {
       clearTimeout(timeoutHandle);
     }
+    if (heartbeatHandle) {
+      clearInterval(heartbeatHandle);
+    }
     if (timedOut) {
       console.error("[SDK-TIMEOUT] Execution timed out:", sdkError);
     }
@@ -7466,6 +7565,32 @@ function diffSnapshots(before, after) {
   diffs.sort((a, b) => a.path.localeCompare(b.path));
   return diffs;
 }
+function extractTemplateFiles(before, after) {
+  const files = [];
+  const allPaths = /* @__PURE__ */ new Set([...Object.keys(before), ...Object.keys(after)]);
+  for (const path9 of allPaths) {
+    const beforeContent = before[path9];
+    const afterContent = after[path9];
+    if (afterContent === void 0) {
+      continue;
+    }
+    let status;
+    if (beforeContent === void 0) {
+      status = "new";
+    } else if (beforeContent !== afterContent) {
+      status = "modified";
+    } else {
+      status = "unchanged";
+    }
+    files.push({
+      path: path9,
+      content: afterContent,
+      status
+    });
+  }
+  files.sort((a, b) => a.path.localeCompare(b.path));
+  return files;
+}
 
 // src/run-scenario/callSkill.ts
 async function callSkill(config2, evalRunId2, scenario, skill, agent, workDir) {
@@ -7498,6 +7623,7 @@ async function callSkill(config2, evalRunId2, scenario, skill, agent, workDir) {
   const completedAt = (/* @__PURE__ */ new Date()).toISOString();
   const afterSnapshot = workDir ? snapshotDirectory(workDir) : {};
   const fileDiffs = diffSnapshots(beforeSnapshot, afterSnapshot);
+  const templateFiles = workDir ? extractTemplateFiles(beforeSnapshot, afterSnapshot) : void 0;
   return {
     id: randomUUID2(),
     targetId: skill.id,
@@ -7508,6 +7634,7 @@ async function callSkill(config2, evalRunId2, scenario, skill, agent, workDir) {
     duration: result.durationMs,
     outputText: result.outputText,
     fileDiffs: fileDiffs.length > 0 ? fileDiffs : void 0,
+    templateFiles: templateFiles && templateFiles.length > 0 ? templateFiles : void 0,
     startedAt,
     completedAt,
     llmTrace