@wix/evalforge-evaluator 0.6.0 → 0.7.0

package/build/index.mjs

@@ -32,21 +32,38 @@ function loadConfig() {
       aiGatewayHeaders[key] = value;
     }
   }
+  const tracePushUrl = process.env.TRACE_PUSH_URL;
+  const routeHeader = process.env.EVAL_ROUTE_HEADER;
   return {
     serverUrl,
     apiPrefix,
     aiGatewayUrl,
     aiGatewayHeaders,
-    evaluationsDir
+    evaluationsDir,
+    tracePushUrl,
+    routeHeader
   };
 }
 
 // src/api-client.ts
-function createApiClient(serverUrl, apiPrefix = "") {
+function createApiClient(serverUrl, options = "") {
+  const opts = typeof options === "string" ? { apiPrefix: options } : options;
+  const apiPrefix = opts.apiPrefix ?? "";
+  const routeHeader = opts.routeHeader;
+  function buildHeaders(additionalHeaders) {
+    const headers = { ...additionalHeaders };
+    if (routeHeader) {
+      headers["x-wix-route"] = routeHeader;
+    }
+    return headers;
+  }
   async function fetchJson(path9) {
     const url = `${serverUrl}${apiPrefix}${path9}`;
     console.error(`[API] GET ${url}`);
-    const response = await fetch(url);
+    const headers = buildHeaders();
+    const response = await fetch(url, {
+      headers: Object.keys(headers).length > 0 ? headers : void 0
+    });
     if (!response.ok) {
       const errorText = await response.text();
       throw new Error(
@@ -60,7 +77,7 @@ function createApiClient(serverUrl, apiPrefix = "") {
     console.error(`[API] POST ${url}`);
     const response = await fetch(url, {
       method: "POST",
-      headers: { "Content-Type": "application/json" },
+      headers: buildHeaders({ "Content-Type": "application/json" }),
       body: JSON.stringify(body)
     });
     if (!response.ok) {
@@ -73,7 +90,11 @@ function createApiClient(serverUrl, apiPrefix = "") {
   async function deleteRequest(path9) {
     const url = `${serverUrl}${apiPrefix}${path9}`;
     console.error(`[API] DELETE ${url}`);
-    const response = await fetch(url, { method: "DELETE" });
+    const headers = buildHeaders();
+    const response = await fetch(url, {
+      method: "DELETE",
+      headers: Object.keys(headers).length > 0 ? headers : void 0
+    });
     if (!response.ok) {
       const errorText = await response.text();
       throw new Error(
@@ -86,7 +107,7 @@ function createApiClient(serverUrl, apiPrefix = "") {
     console.error(`[API] PUT ${url}`);
     const response = await fetch(url, {
       method: "PUT",
-      headers: { "Content-Type": "application/json" },
+      headers: buildHeaders({ "Content-Type": "application/json" }),
       body: JSON.stringify(body)
     });
     if (!response.ok) {
@@ -6059,8 +6080,36 @@ import { randomUUID } from "crypto";
 import { mkdir as mkdir2, writeFile } from "fs/promises";
 import { join as join2 } from "path";
 var DEFAULT_MODEL = "claude-sonnet-4-20250514";
-function emitTraceEvent(event) {
+function emitTraceEvent(event, tracePushUrl, routeHeader) {
   console.log(`${TRACE_EVENT_PREFIX}${JSON.stringify(event)}`);
+  if (tracePushUrl) {
+    pushTraceEvent(tracePushUrl, event, routeHeader).catch((err) => {
+      console.error("[Trace Push] Failed to push trace event:", err);
+    });
+  }
+}
+async function pushTraceEvent(url, event, routeHeader) {
+  try {
+    const headers = {
+      "Content-Type": "application/json"
+    };
+    if (routeHeader) {
+      headers["x-wix-route"] = routeHeader;
+    }
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(event)
+    });
+    if (!response.ok) {
+      const errorText = await response.text();
+      console.error(
+        `[Trace Push] HTTP ${response.status}: ${errorText.slice(0, 200)}`
+      );
+    }
+  } catch (err) {
+    console.error("[Trace Push] Network error:", err);
+  }
 }
 function createTraceEventFromMessage(message, context, stepNumber, isComplete) {
   let type = LiveTraceEventType.COMPLETION;
@@ -6218,7 +6267,11 @@ async function executeWithClaudeCode(skill, scenario, options) {
         false
         // Not complete yet
       );
-      emitTraceEvent(traceEvent);
+      emitTraceEvent(
+        traceEvent,
+        traceContext.tracePushUrl,
+        traceContext.routeHeader
+      );
     }
   }
   console.log(
@@ -6227,18 +6280,22 @@ async function executeWithClaudeCode(skill, scenario, options) {
     "messages"
   );
   if (traceContext) {
-    emitTraceEvent({
-      evalRunId: traceContext.evalRunId,
-      scenarioId: traceContext.scenarioId,
-      scenarioName: traceContext.scenarioName,
-      targetId: traceContext.targetId,
-      targetName: traceContext.targetName,
-      stepNumber: traceStepNumber + 1,
-      type: LiveTraceEventType.COMPLETION,
-      outputPreview: "Scenario execution completed",
-      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-      isComplete: true
-    });
+    emitTraceEvent(
+      {
+        evalRunId: traceContext.evalRunId,
+        scenarioId: traceContext.scenarioId,
+        scenarioName: traceContext.scenarioName,
+        targetId: traceContext.targetId,
+        targetName: traceContext.targetName,
+        stepNumber: traceStepNumber + 1,
+        type: LiveTraceEventType.COMPLETION,
+        outputPreview: "Scenario execution completed",
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        isComplete: true
+      },
+      traceContext.tracePushUrl,
+      traceContext.routeHeader
+    );
   }
   const endTime = /* @__PURE__ */ new Date();
   const totalDurationMs = endTime.getTime() - startTime.getTime();
@@ -6456,7 +6513,9 @@ async function callSkill(config, evalRunId2, scenario, skill, agent, workDir) {
       scenarioId: scenario.id,
       scenarioName: scenario.name,
       targetId: skill.id,
-      targetName: skill.name
+      targetName: skill.name,
+      tracePushUrl: config.tracePushUrl,
+      routeHeader: config.routeHeader
     }
   });
   const completedAt = (/* @__PURE__ */ new Date()).toISOString();
@@ -6569,9 +6628,13 @@ async function runEvaluation(projectId2, evalRunId2) {
     serverUrl: config.serverUrl,
     aiGatewayUrl: config.aiGatewayUrl,
     evaluationsDir: config.evaluationsDir,
-    hasAiGatewayHeaders: Object.keys(config.aiGatewayHeaders).length > 0
+    hasAiGatewayHeaders: Object.keys(config.aiGatewayHeaders).length > 0,
+    hasRouteHeader: !!config.routeHeader
+  });
+  const api = createApiClient(config.serverUrl, {
+    apiPrefix: config.apiPrefix,
+    routeHeader: config.routeHeader
   });
-  const api = createApiClient(config.serverUrl, config.apiPrefix);
   console.error(
     "[DEBUG-H2] fetchEvaluationData START",
     JSON.stringify({ serverUrl: config.serverUrl, timestamp: Date.now() })
@@ -6655,7 +6718,10 @@ runEvaluation(projectId, evalRunId).then(() => {
   console.error("[EVALUATOR-BOOT] runEvaluation FAILED:", err);
   try {
     const config = loadConfig();
-    const api = createApiClient(config.serverUrl, config.apiPrefix);
+    const api = createApiClient(config.serverUrl, {
+      apiPrefix: config.apiPrefix,
+      routeHeader: config.routeHeader
+    });
     await api.updateEvalRun(projectId, evalRunId, {
       status: EvalStatus.FAILED,
       completedAt: (/* @__PURE__ */ new Date()).toISOString()