@exellix/ai-tasks 10.0.13 → 10.1.0

package/dist/node-execution/traceLifecycle.js

@@ -0,0 +1,312 @@
+/**
+ * GTF-FR-001 lifecycle helpers for v2 execution trace index dual-write.
+ * Uses @x12i/graphenix-trace-format appendTraceEvent until upstream exports these APIs.
+ */
+import { randomUUID } from "crypto";
+import { appendTraceEvent, deriveNodeStatusFromUnits } from "@x12i/graphenix-trace-format";
+import { traceError } from "./mapObservationToTrace.js";
+function appendUnique(ids, id) {
+    if (!ids.includes(id))
+        ids.push(id);
+}
+function nextSequence(trace) {
+    return (trace.events?.length ?? 0) + 1;
+}
+function appendLifecycleEvent(trace, event) {
+    appendTraceEvent(trace, {
+        ...event,
+        sequence: event.sequence ?? nextSequence(trace),
+    }, { mutate: true });
+}
+function ensureNodeRunning(trace, nodeId, ts) {
+    const node = trace.nodeExecutions[nodeId];
+    if (!node)
+        return;
+    if (node.status === "pending") {
+        trace.nodeExecutions[nodeId] = { ...node, status: "running", startedAt: ts };
+        appendUnique(trace.graphExecution.startedNodeIds, nodeId);
+        appendLifecycleEvent(trace, {
+            id: `evt:${randomUUID()}`,
+            ts,
+            level: "info",
+            type: "node.started",
+            nodeId,
+        });
+    }
+}
+export function markUnitStarted(trace, unit, at) {
+    const ts = at ?? new Date().toISOString();
+    const unitId = unit.unitId;
+    const existing = trace.unitExecutions[unitId];
+    if (!existing)
+        return;
+    const strategyKey = unit.strategyKey ?? unit.actionKey;
+    trace.unitExecutions[unitId] = {
+        ...existing,
+        status: "running",
+        startedAt: ts,
+        ...(typeof strategyKey === "string" ? { strategyKey } : {}),
+    };
+    appendUnique(trace.graphExecution.startedUnitIds, unitId);
+    ensureNodeRunning(trace, unit.nodeId, ts);
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "info",
+        type: "unit.started",
+        nodeId: unit.nodeId,
+        unitId,
+        unitKind: unit.unitKind,
+        evidence: {
+            unitKind: unit.unitKind,
+            invokeMode: unit.invokeMode,
+            strategyKey: unit.strategyKey,
+            skillKey: unit.skillKey,
+            modelSlot: unit.modelSlot,
+        },
+    });
+}
+export function markUnitCompleted(trace, unitId, outcome, unit) {
+    const ts = outcome.completedAt ?? new Date().toISOString();
+    const existing = trace.unitExecutions[unitId];
+    if (!existing)
+        return;
+    trace.unitExecutions[unitId] = {
+        ...existing,
+        status: "completed",
+        completedAt: ts,
+        durationMs: outcome.durationMs,
+        ...(outcome.modelInvocationIds !== undefined
+            ? { modelInvocationIds: outcome.modelInvocationIds }
+            : {}),
+    };
+    appendUnique(trace.graphExecution.completedUnitIds, unitId);
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "info",
+        type: "unit.completed",
+        nodeId: existing.nodeId,
+        unitId,
+        unitKind: existing.unitKind,
+        evidence: {
+            durationMs: outcome.durationMs,
+            modelInvocationIds: outcome.modelInvocationIds,
+            unitKind: unit?.unitKind ?? existing.unitKind,
+        },
+    });
+}
+export function markUnitFailed(trace, unitId, error, outcome, unit) {
+    const ts = outcome?.completedAt ?? new Date().toISOString();
+    const durationMs = outcome?.durationMs ?? 0;
+    const errorMessage = outcome?.errorMessage ??
+        (error instanceof Error ? error.message : typeof error === "string" ? error : String(error));
+    const existing = trace.unitExecutions[unitId];
+    if (!existing)
+        return;
+    trace.unitExecutions[unitId] = {
+        ...existing,
+        status: "failed",
+        completedAt: ts,
+        durationMs,
+        error: traceError(errorMessage, "UNIT_FAILED"),
+    };
+    appendUnique(trace.graphExecution.failedUnitIds, unitId);
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "error",
+        type: "unit.failed",
+        nodeId: existing.nodeId,
+        unitId,
+        unitKind: existing.unitKind,
+        evidence: {
+            errorMessage,
+            durationMs,
+            unitKind: unit?.unitKind ?? existing.unitKind,
+        },
+    });
+}
+function plannedResolvedInvocation(unit, trace) {
+    const selection = unit.modelSelection ?? { kind: "profileChoice", key: "vol/default" };
+    return {
+        provider: "pending",
+        modelId: "pending",
+        profileChoiceKey: selection.kind === "profileChoice" ? selection.key : undefined,
+        registryVersion: trace.profileRegistry?.version,
+        registryHash: trace.profileRegistry?.registryHash,
+        resolutionSource: "profileRegistry",
+    };
+}
+export function recordModelInvocationStarted(trace, args) {
+    const ts = args.startedAt ?? new Date().toISOString();
+    const { unit, invocationId, nodeId } = args;
+    const selection = unit.modelSelection ?? { kind: "profileChoice", key: "vol/default" };
+    trace.modelInvocations[invocationId] = {
+        invocationId,
+        nodeId,
+        unitId: unit.unitId,
+        slot: unit.modelSlot ?? "skillModel",
+        attempt: args.attempt ?? 1,
+        status: "started",
+        startedAt: ts,
+        plannedSelection: selection,
+        resolvedInvocation: plannedResolvedInvocation(unit, trace),
+    };
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "info",
+        type: "model.invocation.started",
+        nodeId,
+        unitId: unit.unitId,
+        modelSlot: unit.modelSlot,
+        evidence: { invocationId, modelSlot: unit.modelSlot },
+    });
+}
+function applyDataPolicyPreviews(trace, outcome) {
+    const policy = trace.diagnostics?.dataPolicy;
+    if (!policy?.storePrompts) {
+        return {};
+    }
+    const max = policy.maxPreviewChars ?? 512;
+    const clip = (s) => s === undefined ? undefined : s.length > max ? s.slice(0, max) : s;
+    return {
+        promptPreview: clip(outcome.promptPreview),
+        completionPreview: clip(outcome.completionPreview),
+    };
+}
+export function recordModelInvocationCompleted(trace, invocationId, outcome) {
+    const ts = outcome.completedAt ?? new Date().toISOString();
+    const existing = trace.modelInvocations[invocationId];
+    if (!existing)
+        return;
+    const previews = applyDataPolicyPreviews(trace, outcome);
+    trace.modelInvocations[invocationId] = {
+        ...existing,
+        status: "completed",
+        completedAt: ts,
+        durationMs: outcome.durationMs,
+        ...(outcome.usage ? { usage: outcome.usage } : {}),
+        ...(outcome.cost
+            ? {
+                cost: {
+                    costUsd: outcome.cost.costUsd,
+                    costStatus: outcome.cost.costStatus ?? "unknown",
+                },
+            }
+            : {}),
+        ...(outcome.resolvedInvocation ? { resolvedInvocation: outcome.resolvedInvocation } : {}),
+        ...previews,
+    };
+    const unitId = existing.unitId;
+    const unitExec = trace.unitExecutions[unitId];
+    if (unitExec) {
+        const ids = [...(unitExec.modelInvocationIds ?? [])];
+        appendUnique(ids, invocationId);
+        trace.unitExecutions[unitId] = { ...unitExec, modelInvocationIds: ids };
+    }
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "info",
+        type: "model.invocation.completed",
+        nodeId: existing.nodeId,
+        unitId: existing.unitId,
+        modelSlot: existing.slot,
+        evidence: { invocationId, durationMs: outcome.durationMs },
+    });
+}
+export function recordModelInvocationFailed(trace, invocationId, error, outcome) {
+    const ts = outcome?.completedAt ?? new Date().toISOString();
+    const durationMs = outcome?.durationMs ?? 0;
+    const errorMessage = outcome?.errorMessage ??
+        (error instanceof Error ? error.message : typeof error === "string" ? error : String(error));
+    const existing = trace.modelInvocations[invocationId];
+    if (!existing)
+        return;
+    trace.modelInvocations[invocationId] = {
+        ...existing,
+        status: "failed",
+        completedAt: ts,
+        durationMs,
+        error: traceError(errorMessage, "MODEL_INVOCATION_FAILED"),
+    };
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "error",
+        type: "model.invocation.failed",
+        nodeId: existing.nodeId,
+        unitId: existing.unitId,
+        modelSlot: existing.slot,
+        evidence: { invocationId, errorMessage },
+    });
+}
+export function appendFallbackAppliedEvent(trace, unit, extra) {
+    const ts = new Date().toISOString();
+    appendLifecycleEvent(trace, {
+        id: `evt:${randomUUID()}`,
+        ts,
+        level: "info",
+        type: "fallback.applied",
+        nodeId: unit.nodeId,
+        unitId: unit.unitId,
+        evidence: {
+            unitKind: unit.unitKind,
+            ...extra,
+        },
+    });
+}
+export function rollupNodeExecution(trace, nodeId) {
+    const nodeTrace = trace.nodeExecutions[nodeId];
+    if (!nodeTrace)
+        return;
+    const status = deriveNodeStatusFromUnits(trace, nodeId);
+    const units = nodeTrace.unitIds
+        .map((id) => trace.unitExecutions[id])
+        .filter((u) => !!u);
+    const startedTimes = units
+        .map((u) => u.startedAt)
+        .filter((t) => typeof t === "string")
+        .sort();
+    const completedTimes = units
+        .map((u) => u.completedAt)
+        .filter((t) => typeof t === "string")
+        .sort();
+    const startedAt = nodeTrace.startedAt ?? startedTimes[0];
+    const completedAt = status === "completed" || status === "failed" || status === "skipped"
+        ? (completedTimes[completedTimes.length - 1] ?? new Date().toISOString())
+        : undefined;
+    let durationMs = units.reduce((sum, u) => sum + (u.durationMs ?? 0), 0);
+    if (startedAt && completedAt) {
+        durationMs = Math.max(0, new Date(completedAt).getTime() - new Date(startedAt).getTime());
+    }
+    trace.nodeExecutions[nodeId] = {
+        ...nodeTrace,
+        status,
+        ...(startedAt ? { startedAt } : {}),
+        ...(completedAt ? { completedAt, durationMs } : {}),
+    };
+    if (status === "completed") {
+        appendUnique(trace.graphExecution.completedNodeIds, nodeId);
+        appendLifecycleEvent(trace, {
+            id: `evt:${randomUUID()}`,
+            ts: completedAt ?? new Date().toISOString(),
+            level: "info",
+            type: "node.completed",
+            nodeId,
+        });
+    }
+    else if (status === "failed") {
+        appendUnique(trace.graphExecution.failedNodeIds, nodeId);
+        appendLifecycleEvent(trace, {
+            id: `evt:${randomUUID()}`,
+            ts: completedAt ?? new Date().toISOString(),
+            level: "error",
+            type: "node.failed",
+            nodeId,
+        });
+    }
+}
+//# sourceMappingURL=traceLifecycle.js.map

package/dist/node-execution/traceLifecycle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceLifecycle.js","sourceRoot":"","sources":["../../src/node-execution/traceLifecycle.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,yBAAyB,EAAE,MAAM,8BAA8B,CAAC;AAS3F,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,SAAS,YAAY,CAAC,GAAa,EAAE,EAAU;IAC7C,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,YAAY,CAAC,KAA4B;IAChD,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,oBAAoB,CAC3B,KAA4B,EAC5B,KAA2D;IAE3D,gBAAgB,CACd,KAAK,EACL;QACE,GAAG,KAAK;QACR,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,YAAY,CAAC,KAAK,CAAC;KAClC,EACf,EAAE,MAAM,EAAE,IAAI,EAAE,CACjB,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,KAA4B,EAAE,MAAc,EAAE,EAAU;IACjF,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC1C,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC9B,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;QAC7E,YAAY,CAAC,KAAK,CAAC,cAAc,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC1D,oBAAoB,CAAC,KAAK,EAAE;YAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;YACzB,EAAE;YACF,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,cAAc;YACpB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAsCD,MAAM,UAAU,eAAe,CAC7B,KAA4B,EAC5B,IAAyB,EACzB,EAAW;IAEX,MAAM,EAAE,GAAG,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3B,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC9C,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC;IACvD,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG;QAC7B,GAAG,QAAQ;QACX,MAAM,EAAE,SAAS;QACjB,SAAS,EAAE,EAAE;QACb,GAAG,CAAC,OAAO,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5D,CAAC;IACF,YAAY,CAAC,KAAK,CAAC,cAAc,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IAC1D,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAE1C,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,MAAM;QACb,IAAI,EAAE,cAAc;QACpB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,MAAM;QACN,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,QAAQ,EAAE;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,KAA4B,EAC5B,MAAc,EACd,OAA8B,EAC9B,IAA0B;IAE1B,MAAM,EAAE,GAAG,OAAO,CAAC,WAAW,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC3D,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC9C,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG;QAC7B,GAAG,QAAQ;QACX,MAAM,EAAE,WAAW;QACnB,WAAW,EAAE,EAAE;QACf,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,GAAG,CAAC,OAAO,CAAC,kBAAkB,KAAK,SAAS;YAC1C,CAAC,CAAC,EAAE,kBAAkB,EAAE,OAAO,CAAC,kBAAkB,EAAE;YACpD,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;IACF,YAAY,CAAC,KAAK,CAAC,cAAc,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;IAE5D,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,MAAM;QACb,IAAI,EAAE,gBAAgB;QACtB,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,MAAM;QACN,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,QAAQ,EAAE;YACR,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;YAC9C,QAAQ,EAAE,IAAI,EAAE,QAAQ,IAAI,QAAQ,CAAC,QAAQ;SAC9C;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,KAA4B,EAC5B,MAAc,EACd,KAAc,EACd,OAA4B,EAC5B,IAA0B;IAE1B,MAAM,EAAE,GAAG,OAAO,EAAE,WAAW,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,CAAC,CAAC;IAC5C,MAAM,YAAY,GAChB,OAAO,EAAE,YAAY;QACrB,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC/F,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC9C,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG;QAC7B,GAAG,QAAQ;QACX,MAAM,EAAE,QAAQ;QAChB,WAAW,EAAE,EAAE;QACf,UAAU;QACV,KAAK,EAAE,UAAU,CAAC,YAAY,EAAE,aAAa,CAAC;KAC/C,CAAC;IACF,YAAY,CAAC,KAAK,CAAC,cAAc,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAEzD,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,OAAO;QACd,IAAI,EAAE,aAAa;QACnB,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,MAAM;QACN,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,QAAQ,EAAE;YACR,YAAY;YACZ,UAAU;YACV,QAAQ,EAAE,IAAI,EAAE,QAAQ,IAAI,QAAQ,CAAC,QAAQ;SAC9C;KACF,CAAC,CAAC;AACL,CAAC;AAED,SAAS,yBAAyB,CAChC,IAAyB,EACzB,KAA4B;IAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,CAAC;IACvF,OAAO;QACL,QAAQ,EAAE,SAAS;QACnB,OAAO,EAAE,SAAS;QAClB,gBAAgB,EAAE,SAAS,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;QAChF,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE,OAAO;QAC/C,YAAY,EAAE,KAAK,CAAC,eAAe,EAAE,YAAY;QACjD,gBAAgB,EAAE,iBAAiB;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,KAA4B,EAC5B,IAA8B;IAE9B,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACtD,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,EAAE,aAAa,EAAE,CAAC;IAEvF,KAAK,CAAC,gBAAgB,CAAC,YAAY,CAAC,GAAG;QACrC,YAAY;QACZ,MAAM;QACN,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,IAAI,EAAE,IAAI,CAAC,SAAS,IAAI,YAAY;QACpC,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,CAAC;QAC1B,MAAM,EAAE,SAAS;QACjB,SAAS,EAAE,EAAE;QACb,gBAAgB,EAAE,SAAS;QAC3B,kBAAkB,EAAE,yBAAyB,CAAC,IAAI,EAAE,KAAK,CAAC;KAC5B,CAAC;IAEjC,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,MAAM;QACb,IAAI,EAAE,0BAA0B;QAChC,MAAM;QACN,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,QAAQ,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE;KACtD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,uBAAuB,CAC9B,KAA4B,EAC5B,OAA+B;IAE/B,MAAM,MAAM,GAAG,KAAK,CAAC,WAAW,EAAE,UAAU,CAAC;IAC7C,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,CAAC,eAAe,IAAI,GAAG,CAAC;IAC1C,MAAM,IAAI,GAAG,CAAC,CAAqB,EAAE,EAAE,CACrC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrE,OAAO;QACL,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;QAC1C,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC;KACnD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,KAA4B,EAC5B,YAAoB,EACpB,OAA+B;IAE/B,MAAM,EAAE,GAAG,OAAO,CAAC,WAAW,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC3D,MAAM,QAAQ,GAAG,KAAK,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;IACtD,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,MAAM,QAAQ,GAAG,uBAAuB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACzD,KAAK,CAAC,gBAAgB,CAAC,YAAY,CAAC,GAAG;QACrC,GAAG,QAAQ;QACX,MAAM,EAAE,WAAW;QACnB,WAAW,EAAE,EAAE;QACf,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAClD,GAAG,CAAC,OAAO,CAAC,IAAI;YACd,CAAC,CAAC;gBACE,IAAI,EAAE;oBACJ,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO;oBAC7B,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU,IAAI,SAAS;iBAC7B;aACtB;YACH,CAAC,CAAC,EAAE,CAAC;QACP,GAAG,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzF,GAAG,QAAQ;KACZ,CAAC;IAEF,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC/B,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC9C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAC,CAAC;QACrD,YAAY,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QAChC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,kBAAkB,EAAE,GAAG,EAAE,CAAC;IAC1E,CAAC;IAED,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,MAAM;QACb,IAAI,EAAE,4BAA4B;QAClC,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,SAAS,EAAE,QAAQ,CAAC,IAAI;QACxB,QAAQ,EAAE,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE;KAC3D,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,KAA4B,EAC5B,YAAoB,EACpB,KAAc,EACd,OAAgC;IAEhC,MAAM,EAAE,GAAG,OAAO,EAAE,WAAW,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,CAAC,CAAC;IAC5C,MAAM,YAAY,GAChB,OAAO,EAAE,YAAY;QACrB,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC/F,MAAM,QAAQ,GAAG,KAAK,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;IACtD,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,KAAK,CAAC,gBAAgB,CAAC,YAAY,CAAC,GAAG;QACrC,GAAG,QAAQ;QACX,MAAM,EAAE,QAAQ;QAChB,WAAW,EAAE,EAAE;QACf,UAAU;QACV,KAAK,EAAE,UAAU,CAAC,YAAY,EAAE,yBAAyB,CAAC;KAC3D,CAAC;IAEF,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,OAAO;QACd,IAAI,EAAE,yBAAyB;QAC/B,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,SAAS,EAAE,QAAQ,CAAC,IAAI;QACxB,QAAQ,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;KACzC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,KAA4B,EAC5B,IAAyB,EACzB,KAA+B;IAE/B,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,oBAAoB,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE;QACF,KAAK,EAAE,MAAM;QACb,IAAI,EAAE,kBAAkB;QACxB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,QAAQ,EAAE;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,GAAG,KAAK;SACT;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAA4B,EAAE,MAAc;IAC9E,MAAM,SAAS,GAAG,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAC/C,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,MAAM,MAAM,GAAG,yBAAyB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO;SAC5B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;SACrC,MAAM,CAAC,CAAC,CAAC,EAA8B,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAElD,MAAM,YAAY,GAAG,KAAK;SACvB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;SACvB,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;SACjD,IAAI,EAAE,CAAC;IACV,MAAM,cAAc,GAAG,KAAK;SACzB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;SACzB,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;SACjD,IAAI,EAAE,CAAC;IAEV,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IACzD,MAAM,WAAW,GACf,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,SAAS;QACnE,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACzE,CAAC,CAAC,SAAS,CAAC;IAEhB,IAAI,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACxE,IAAI,SAAS,IAAI,WAAW,EAAE,CAAC;QAC7B,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5F,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG;QAC7B,GAAG,SAAS;QACZ,MAAM;QACN,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACpD,CAAC;IAEF,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;QAC3B,YAAY,CAAC,KAAK,CAAC,cAAc,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;QAC5D,oBAAoB,CAAC,KAAK,EAAE;YAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;YACzB,EAAE,EAAE,WAAW,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC3C,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,gBAAgB;YACtB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;SAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,YAAY,CAAC,KAAK,CAAC,cAAc,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QACzD,oBAAoB,CAAC,KAAK,EAAE;YAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;YACzB,EAAE,EAAE,WAAW,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC3C,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,aAAa;YACnB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}

package/dist/node-execution/unitTrace.d.ts

@@ -1,12 +1,23 @@
 import type { ExecutionUnitPlanV2, GraphExecutionTraceV2 } from "@x12i/graphenix-executable-contracts";
+import type { LlmCallObservation } from "../types/llmCall.js";
 export type UnitTraceContext = {
     trace?: GraphExecutionTraceV2;
     nodeId: string;
 };
-export declare function appendUnitTraceEvent(ctx: UnitTraceContext, type: string, unit: ExecutionUnitPlanV2, extra?: Record<string, unknown>): void;
 export declare function unitUsesModelInvocation(unit: ExecutionUnitPlanV2): boolean;
+export declare function beginUnitTrace(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2): {
+    invocationId?: string;
+    startedAtMs: number;
+};
+export declare function completeUnitTrace(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, startedAtMs: number, invocationId: string | undefined, observations: LlmCallObservation[]): void;
+export declare function failUnitTrace(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, startedAtMs: number, invocationId: string | undefined, error: unknown): void;
+export declare function appendFallbackApplied(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, extra?: Record<string, unknown>): void;
+/** @deprecated Use beginUnitTrace — kept for incremental migration. */
+export declare function appendUnitTraceEvent(ctx: UnitTraceContext, type: string, unit: ExecutionUnitPlanV2, extra?: Record<string, unknown>): void;
+/** @deprecated Use beginUnitTrace / completeUnitTrace. */
 export declare function appendModelInvocationStarted(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, invocationId?: string): string;
+/** @deprecated Use completeUnitTrace. */
 export declare function appendModelInvocationCompleted(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, invocationId: string, extra?: Record<string, unknown>): void;
+/** @deprecated Use failUnitTrace. */
 export declare function appendModelInvocationFailed(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, invocationId: string, error: unknown): void;
-export declare function appendFallbackApplied(ctx: UnitTraceContext, unit: ExecutionUnitPlanV2, extra?: Record<string, unknown>): void;
 //# sourceMappingURL=unitTrace.d.ts.map

package/dist/node-execution/unitTrace.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"unitTrace.d.ts","sourceRoot":"","sources":["../../src/node-execution/unitTrace.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAEvG,MAAM,MAAM,gBAAgB,GAAG;IAC7B,KAAK,CAAC,EAAE,qBAAqB,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,mBAAmB,EACzB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,IAAI,CAoBN;AAED,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,mBAAmB,GAAG,OAAO,CAE1E;AAED,wBAAgB,4BAA4B,CAC1C,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,YAAY,CAAC,EAAE,MAAM,GACpB,MAAM,CAOR;AAED,wBAAgB,8BAA8B,CAC5C,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,YAAY,EAAE,MAAM,EACpB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,IAAI,CAKN;AAED,wBAAgB,2BAA2B,CACzC,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,OAAO,GACb,IAAI,CAKN;AAED,wBAAgB,qBAAqB,CACnC,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,IAAI,CAEN"}
+{"version":3,"file":"unitTrace.d.ts","sourceRoot":"","sources":["../../src/node-execution/unitTrace.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AACvG,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAa9D,MAAM,MAAM,gBAAgB,GAAG;IAC7B,KAAK,CAAC,EAAE,qBAAqB,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,mBAAmB,GAAG,OAAO,CAU1E;AAED,wBAAgB,cAAc,CAC5B,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,GACxB;IAAE,YAAY,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAAE,CAShD;AAYD,wBAAgB,iBAAiB,CAC/B,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,GAAG,SAAS,EAChC,YAAY,EAAE,kBAAkB,EAAE,GACjC,IAAI,CA8CN;AAED,wBAAgB,aAAa,CAC3B,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,GAAG,SAAS,EAChC,KAAK,EAAE,OAAO,GACb,IAAI,CAiBN;AAED,wBAAgB,qBAAqB,CACnC,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,IAAI,CAGN;AAED,uEAAuE;AACvE,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,mBAAmB,EACzB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,IAAI,CAKN;AAED,0DAA0D;AAC1D,wBAAgB,4BAA4B,CAC1C,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,YAAY,CAAC,EAAE,MAAM,GACpB,MAAM,CAKR;AAED,yCAAyC;AACzC,wBAAgB,8BAA8B,CAC5C,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,YAAY,EAAE,MAAM,EACpB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,IAAI,CAMN;AAED,qCAAqC;AACrC,wBAAgB,2BAA2B,CACzC,GAAG,EAAE,gBAAgB,EACrB,IAAI,EAAE,mBAAmB,EACzB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,OAAO,GACb,IAAI,CAIN"}

package/dist/node-execution/unitTrace.js

@@ -1,52 +1,123 @@
 import { randomUUID } from "crypto";
-import { appendTraceEvent } from "@x12i/graphenix-trace-format";
 import { AI_UNIT_KINDS } from "@x12i/graphenix-plan-format";
-export function appendUnitTraceEvent(ctx, type, unit, extra) {
+import { mapObservationToModelInvocationOutcome } from "./mapObservationToTrace.js";
+import { appendFallbackAppliedEvent, markUnitCompleted, markUnitFailed, markUnitStarted, recordModelInvocationCompleted, recordModelInvocationFailed, recordModelInvocationStarted, } from "./traceLifecycle.js";
+export function unitUsesModelInvocation(unit) {
+    if (!AI_UNIT_KINDS.has(unit.unitKind) || unit.unitKind === "finalizer") {
+        return false;
+    }
+    if (unit.unitKind === "externalPostUtility") {
+        const key = unit.strategyKey ?? unit.actionKey;
+        if (key === "merge" || key === "pick-best")
+            return false;
+        return !!unit.modelSlot;
+    }
+    return true;
+}
+export function beginUnitTrace(ctx, unit) {
+    const startedAtMs = performance.now();
     if (!ctx.trace)
-        return;
-    const sequence = (ctx.trace.events?.length ?? 0) + 1;
-    appendTraceEvent(ctx.trace, {
-        id: `evt:${randomUUID()}`,
-        ts: new Date().toISOString(),
-        sequence,
-        level: type.includes("failed") ? "error" : "info",
-        type,
+        return { startedAtMs };
+    markUnitStarted(ctx.trace, unit);
+    const invocationId = unitUsesModelInvocation(unit)
+        ? startModelInvocation(ctx, unit)
+        : undefined;
+    return { invocationId, startedAtMs };
+}
+function startModelInvocation(ctx, unit) {
+    const invocationId = `inv:${randomUUID()}`;
+    recordModelInvocationStarted(ctx.trace, {
+        invocationId,
+        unit,
         nodeId: ctx.nodeId,
-        unitId: unit.unitId,
-        evidence: {
-            unitKind: unit.unitKind,
-            invokeMode: unit.invokeMode,
-            strategyKey: unit.strategyKey,
-            skillKey: unit.skillKey,
-            modelSlot: unit.modelSlot,
-            ...extra,
-        },
     });
+    return invocationId;
 }
-export function unitUsesModelInvocation(unit) {
-    return AI_UNIT_KINDS.has(unit.unitKind) && unit.unitKind !== "finalizer";
+export function completeUnitTrace(ctx, unit, startedAtMs, invocationId, observations) {
+    if (!ctx.trace)
+        return;
+    const durationMs = Math.max(1, Math.round(performance.now() - startedAtMs));
+    const completedAt = new Date().toISOString();
+    const modelInvocationIds = [];
+    if (invocationId) {
+        modelInvocationIds.push(invocationId);
+        const outcome = observations[0]
+            ? mapObservationToModelInvocationOutcome(observations[0], unit, ctx.trace)
+            : { durationMs };
+        recordModelInvocationCompleted(ctx.trace, invocationId, {
+            ...outcome,
+            durationMs: outcome.durationMs || durationMs,
+            completedAt,
+        });
+        for (let i = 1; i < observations.length; i++) {
+            const extraId = `inv:${randomUUID()}`;
+            recordModelInvocationStarted(ctx.trace, {
+                invocationId: extraId,
+                unit,
+                nodeId: ctx.nodeId,
+                startedAt: completedAt,
+                attempt: i + 1,
+            });
+            const extraOutcome = mapObservationToModelInvocationOutcome(observations[i], unit, ctx.trace);
+            recordModelInvocationCompleted(ctx.trace, extraId, {
+                ...extraOutcome,
+                completedAt,
+            });
+            modelInvocationIds.push(extraId);
+        }
+    }
+    markUnitCompleted(ctx.trace, unit.unitId, {
+        durationMs,
+        completedAt,
+        ...(modelInvocationIds.length > 0 ? { modelInvocationIds } : {}),
+    }, unit);
+}
+export function failUnitTrace(ctx, unit, startedAtMs, invocationId, error) {
+    if (!ctx.trace)
+        return;
+    const durationMs = Math.max(0, Math.round(performance.now() - startedAtMs));
+    const completedAt = new Date().toISOString();
+    if (invocationId) {
+        recordModelInvocationFailed(ctx.trace, invocationId, error, { durationMs, completedAt });
+    }
+    markUnitFailed(ctx.trace, unit.unitId, error, { durationMs, completedAt }, unit);
+}
+export function appendFallbackApplied(ctx, unit, extra) {
+    if (!ctx.trace)
+        return;
+    appendFallbackAppliedEvent(ctx.trace, unit, extra);
 }
+/** @deprecated Use beginUnitTrace — kept for incremental migration. */
+export function appendUnitTraceEvent(ctx, type, unit, extra) {
+    void type;
+    void extra;
+    if (!ctx.trace)
+        return;
+    if (type === "unit.started")
+        markUnitStarted(ctx.trace, unit);
+}
+/** @deprecated Use beginUnitTrace / completeUnitTrace. */
 export function appendModelInvocationStarted(ctx, unit, invocationId) {
+    if (!ctx.trace)
+        return invocationId ?? `inv:${randomUUID()}`;
     const id = invocationId ?? `inv:${randomUUID()}`;
-    appendUnitTraceEvent(ctx, "model.invocation.started", unit, {
-        invocationId: id,
-        modelSlot: unit.modelSlot,
-    });
+    recordModelInvocationStarted(ctx.trace, { invocationId: id, unit, nodeId: ctx.nodeId });
     return id;
 }
+/** @deprecated Use completeUnitTrace. */
 export function appendModelInvocationCompleted(ctx, unit, invocationId, extra) {
-    appendUnitTraceEvent(ctx, "model.invocation.completed", unit, {
-        invocationId,
-        ...extra,
+    void unit;
+    if (!ctx.trace)
+        return;
+    recordModelInvocationCompleted(ctx.trace, invocationId, {
+        durationMs: typeof extra?.durationMs === "number" ? extra.durationMs : 0,
     });
 }
+/** @deprecated Use failUnitTrace. */
 export function appendModelInvocationFailed(ctx, unit, invocationId, error) {
-    appendUnitTraceEvent(ctx, "model.invocation.failed", unit, {
-        invocationId,
-        errorMessage: error instanceof Error ? error.message : String(error),
-    });
-}
-export function appendFallbackApplied(ctx, unit, extra) {
-    appendUnitTraceEvent(ctx, "fallback.applied", unit, extra);
+    void unit;
+    if (!ctx.trace)
+        return;
+    recordModelInvocationFailed(ctx.trace, invocationId, error);
 }
 //# sourceMappingURL=unitTrace.js.map

package/dist/node-execution/unitTrace.js.map

@@ -1 +1 @@
-{"version":3,"file":"unitTrace.js","sourceRoot":"","sources":["../../src/node-execution/unitTrace.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAQ5D,MAAM,UAAU,oBAAoB,CAClC,GAAqB,EACrB,IAAY,EACZ,IAAyB,EACzB,KAA+B;IAE/B,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IACvB,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACrD,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE;QAC1B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;QACzB,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAC5B,QAAQ;QACR,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;QACjD,IAAI;QACJ,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,QAAQ,EAAE;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG,KAAK;SACT;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,IAAyB;IAC/D,OAAO,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,KAAK,WAAW,CAAC;AAC3E,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,GAAqB,EACrB,IAAyB,EACzB,YAAqB;IAErB,MAAM,EAAE,GAAG,YAAY,IAAI,OAAO,UAAU,EAAE,EAAE,CAAC;IACjD,oBAAoB,CAAC,GAAG,EAAE,0BAA0B,EAAE,IAAI,EAAE;QAC1D,YAAY,EAAE,EAAE;QAChB,SAAS,EAAE,IAAI,CAAC,SAAS;KAC1B,CAAC,CAAC;IACH,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,GAAqB,EACrB,IAAyB,EACzB,YAAoB,EACpB,KAA+B;IAE/B,oBAAoB,CAAC,GAAG,EAAE,4BAA4B,EAAE,IAAI,EAAE;QAC5D,YAAY;QACZ,GAAG,KAAK;KACT,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,GAAqB,EACrB,IAAyB,EACzB,YAAoB,EACpB,KAAc;IAEd,oBAAoB,CAAC,GAAG,EAAE,yBAAyB,EAAE,IAAI,EAAE;QACzD,YAAY;QACZ,YAAY,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;KACrE,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,GAAqB,EACrB,IAAyB,EACzB,KAA+B;IAE/B,oBAAoB,CAAC,GAAG,EAAE,kBAAkB,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;AAC7D,CAAC"}
+{"version":3,"file":"unitTrace.js","sourceRoot":"","sources":["../../src/node-execution/unitTrace.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAG5D,OAAO,EAAE,sCAAsC,EAAE,MAAM,4BAA4B,CAAC;AACpF,OAAO,EACL,0BAA0B,EAC1B,iBAAiB,EACjB,cAAc,EACd,eAAe,EACf,8BAA8B,EAC9B,2BAA2B,EAC3B,4BAA4B,GAE7B,MAAM,qBAAqB,CAAC;AAO7B,MAAM,UAAU,uBAAuB,CAAC,IAAyB;IAC/D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;QACvE,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,IAAI,CAAC,QAAQ,KAAK,qBAAqB,EAAE,CAAC;QAC5C,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC;QAC/C,IAAI,GAAG,KAAK,OAAO,IAAI,GAAG,KAAK,WAAW;YAAE,OAAO,KAAK,CAAC;QACzD,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;IAC1B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,GAAqB,EACrB,IAAyB;IAEzB,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IACtC,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO,EAAE,WAAW,EAAE,CAAC;IAEvC,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACjC,MAAM,YAAY,GAAG,uBAAuB,CAAC,IAAI,CAAC;QAChD,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE,IAAI,CAAC;QACjC,CAAC,CAAC,SAAS,CAAC;IACd,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAqB,EAAE,IAAyB;IAC5E,MAAM,YAAY,GAAG,OAAO,UAAU,EAAE,EAAE,CAAC;IAC3C,4BAA4B,CAAC,GAAG,CAAC,KAAM,EAAE;QACvC,YAAY;QACZ,IAAI;QACJ,MAAM,EAAE,GAAG,CAAC,MAAM;KACnB,CAAC,CAAC;IACH,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,GAAqB,EACrB,IAAyB,EACzB,WAAmB,EACnB,YAAgC,EAChC,YAAkC;IAElC,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IAEvB,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC;IAC5E,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC7C,MAAM,kBAAkB,GAAa,EAAE,CAAC;IAExC,IAAI,YAAY,EAAE,CAAC;QACjB,kBAAkB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG,YAAY,CAAC,CAAC,CAAC;YAC7B,CAAC,CAAC,sCAAsC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC;YAC1E,CAAC,CAAE,EAAE,UAAU,EAAoC,CAAC;QACtD,8BAA8B,CAAC,GAAG,CAAC,KAAK,EAAE,YAAY,EAAE;YACtD,GAAG,OAAO;YACV,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,UAAU;YAC5C,WAAW;SACZ,CAAC,CAAC;QAEH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,OAAO,GAAG,OAAO,UAAU,EAAE,EAAE,CAAC;YACtC,4BAA4B,CAAC,GAAG,CAAC,KAAK,EAAE;gBACtC,YAAY,EAAE,OAAO;gBACrB,IAAI;gBACJ,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,SAAS,EAAE,WAAW;gBACtB,OAAO,EAAE,CAAC,GAAG,CAAC;aACf,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,sCAAsC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YAC9F,8BAA8B,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE;gBACjD,GAAG,YAAY;gBACf,WAAW;aACZ,CAAC,CAAC;YACH,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,iBAAiB,CACf,GAAG,CAAC,KAAK,EACT,IAAI,CAAC,MAAM,EACX;QACE,UAAU;QACV,WAAW;QACX,GAAG,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACjE,EACD,IAAI,CACL,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,GAAqB,EACrB,IAAyB,EACzB,WAAmB,EACnB,YAAgC,EAChC,KAAc;IAEd,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IAEvB,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC;IAC5E,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAE7C,IAAI,YAAY,EAAE,CAAC;QACjB,2BAA2B,CAAC,GAAG,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,CAAC;IAC3F,CAAC;IAED,cAAc,CACZ,GAAG,CAAC,KAAK,EACT,IAAI,CAAC,MAAM,EACX,KAAK,EACL,EAAE,UAAU,EAAE,WAAW,EAAE,EAC3B,IAAI,CACL,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,GAAqB,EACrB,IAAyB,EACzB,KAA+B;IAE/B,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IACvB,0BAA0B,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;AACrD,CAAC;AAED,uEAAuE;AACvE,MAAM,UAAU,oBAAoB,CAClC,GAAqB,EACrB,IAAY,EACZ,IAAyB,EACzB,KAA+B;IAE/B,KAAK,IAAI,CAAC;IACV,KAAK,KAAK,CAAC;IACX,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IACvB,IAAI,IAAI,KAAK,cAAc;QAAE,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;AAChE,CAAC;AAED,0DAA0D;AAC1D,MAAM,UAAU,4BAA4B,CAC1C,GAAqB,EACrB,IAAyB,EACzB,YAAqB;IAErB,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO,YAAY,IAAI,OAAO,UAAU,EAAE,EAAE,CAAC;IAC7D,MAAM,EAAE,GAAG,YAAY,IAAI,OAAO,UAAU,EAAE,EAAE,CAAC;IACjD,4BAA4B,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;IACxF,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,yCAAyC;AACzC,MAAM,UAAU,8BAA8B,CAC5C,GAAqB,EACrB,IAAyB,EACzB,YAAoB,EACpB,KAA+B;IAE/B,KAAK,IAAI,CAAC;IACV,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IACvB,8BAA8B,CAAC,GAAG,CAAC,KAAK,EAAE,YAAY,EAAE;QACtD,UAAU,EAAE,OAAO,KAAK,EAAE,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;KACzE,CAAC,CAAC;AACL,CAAC;AAED,qCAAqC;AACrC,MAAM,UAAU,2BAA2B,CACzC,GAAqB,EACrB,IAAyB,EACzB,YAAoB,EACpB,KAAc;IAEd,KAAK,IAAI,CAAC;IACV,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IACvB,2BAA2B,CAAC,GAAG,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;AAC9D,CAAC"}

package/dist/post-steps/audit/runAudit.d.ts

@@ -4,6 +4,7 @@
 import type { RunTaskRequest, RunTaskResponse } from "../../types/index.js";
 import type { AuditConfig, AuditPostStepMetadata, IntermediateStep } from "../../types/task-types.js";
 import type { DebugTraceCollector } from "../../observability/debugTrace.js";
+import type { LlmCallObservation } from "../../types/llmCall.js";
 export type RunAuditPostStepContext<TParsed = any> = {
     request: RunTaskRequest;
     mainResult: RunTaskResponse<TParsed>;
@@ -12,6 +13,8 @@ export type RunAuditPostStepContext<TParsed = any> = {
     reRunMain: (overrideContext: string) => Promise<RunTaskResponse<TParsed>>;
     /** Optional runTask trace collector (executionMode: "trace"). */
     traceCollector?: DebugTraceCollector | null;
+    /** Optional executionTrace v2 observation sink (nodePlan POST units). */
+    observationSink?: (obs: LlmCallObservation) => void;
 };
 export type RunAuditPostStepResult = {
     outputText: string;

package/dist/post-steps/audit/runAudit.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runAudit.d.ts","sourceRoot":"","sources":["../../../src/post-steps/audit/runAudit.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC5E,OAAO,KAAK,EACV,WAAW,EAEX,qBAAqB,EAErB,gBAAgB,EACjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAgB7E,MAAM,MAAM,uBAAuB,CAAC,OAAO,GAAG,GAAG,IAAI;IACnD,OAAO,EAAE,cAAc,CAAC;IACxB,UAAU,EAAE,eAAe,CAAC,OAAO,CAAC,CAAC;IACrC,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,EAAE,CAAC,eAAe,EAAE,MAAM,KAAK,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,iEAAiE;IACjE,cAAc,CAAC,EAAE,mBAAmB,GAAG,IAAI,CAAC;CAC7C,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,qBAAqB,CAAC;IAChC,iBAAiB,EAAE,gBAAgB,EAAE,CAAC;CACvC,CAAC;AA2CF,wBAAsB,gBAAgB,CAAC,OAAO,GAAG,GAAG,EAClD,GAAG,EAAE,uBAAuB,CAAC,OAAO,CAAC,GACpC,OAAO,CAAC,sBAAsB,CAAC,CA6XjC"}
+{"version":3,"file":"runAudit.d.ts","sourceRoot":"","sources":["../../../src/post-steps/audit/runAudit.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC5E,OAAO,KAAK,EACV,WAAW,EAEX,qBAAqB,EAErB,gBAAgB,EACjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAU7E,OAAO,KAAK,EAAiB,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAMhF,MAAM,MAAM,uBAAuB,CAAC,OAAO,GAAG,GAAG,IAAI;IACnD,OAAO,EAAE,cAAc,CAAC;IACxB,UAAU,EAAE,eAAe,CAAC,OAAO,CAAC,CAAC;IACrC,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,EAAE,CAAC,eAAe,EAAE,MAAM,KAAK,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,iEAAiE;IACjE,cAAc,CAAC,EAAE,mBAAmB,GAAG,IAAI,CAAC;IAC5C,yEAAyE;IACzE,eAAe,CAAC,EAAE,CAAC,GAAG,EAAE,kBAAkB,KAAK,IAAI,CAAC;CACrD,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,qBAAqB,CAAC;IAChC,iBAAiB,EAAE,gBAAgB,EAAE,CAAC;CACvC,CAAC;AA2CF,wBAAsB,gBAAgB,CAAC,OAAO,GAAG,GAAG,EAClD,GAAG,EAAE,uBAAuB,CAAC,OAAO,CAAC,GACpC,OAAO,CAAC,sBAAsB,CAAC,CAiajC"}