@workbench-ai/agent-driver 0.0.44

package/dist/trace-builder.d.ts

@@ -0,0 +1,88 @@
+import type { JsonValue, TraceEvent, TraceSpan, TraceSummary } from "./types.js";
+export interface TraceBundle {
+    spans: TraceSpan[];
+    events: TraceEvent[];
+    summaries: TraceSummary[];
+}
+export declare class HarnessTraceBuilder {
+    private readonly args;
+    private readonly spans;
+    private readonly events;
+    private readonly activeToolSpans;
+    private readonly dirtySpanIds;
+    private turnSpanId;
+    private turnStartedAt;
+    private outputSpanId;
+    private outputStartedAt;
+    private usageEventCount;
+    private flushedEventCount;
+    constructor(args: {
+        attemptNumber: number;
+        stageId: string;
+        stageRunIndex: number;
+        stageSpanId: string;
+    });
+    startTurn(args: {
+        at?: string;
+        provider?: string | null;
+        model?: string | null;
+        sessionId?: string | null;
+        operationId?: string | null;
+        promptText?: string | null;
+        attributes?: Record<string, JsonValue>;
+    }): void;
+    completeTurn(args: {
+        at?: string;
+        status?: string | null;
+        provider?: string | null;
+        model?: string | null;
+        sessionId?: string | null;
+        operationId?: string | null;
+        errorMessage?: string | null;
+        attributes?: Record<string, JsonValue>;
+    }): void;
+    startToolCall(args: {
+        at?: string;
+        toolId?: string | null;
+        toolName?: string | null;
+        attributes?: Record<string, JsonValue>;
+    }): void;
+    completeToolCall(args: {
+        at?: string;
+        toolId?: string | null;
+        toolName?: string | null;
+        attributes?: Record<string, JsonValue>;
+    }): void;
+    startAssistantOutput(args: {
+        at?: string;
+        phase?: string | null;
+        itemId?: string | null;
+    }): void;
+    appendOutputDelta(args: {
+        at?: string;
+        delta: string;
+        phase?: string | null;
+        itemId?: string | null;
+    }): void;
+    completeAssistantOutput(args: {
+        at?: string;
+        text: string;
+        phase?: string | null;
+        itemId?: string | null;
+    }): void;
+    recordUsage(args: {
+        at?: string;
+        inputTokens?: number | null;
+        outputTokens?: number | null;
+        attributes?: Record<string, JsonValue>;
+    }): void;
+    recordNote(message: string, at?: string, attributes?: Record<string, JsonValue>): void;
+    recordError(message: string, at?: string, attributes?: Record<string, JsonValue>): void;
+    flushBundle(): TraceBundle;
+    buildBundle(finalOutput: string, endedAt?: string): TraceBundle;
+    private ensureOutputSpan;
+    private completeOutputSpan;
+    private upsertSpan;
+    private appendEvent;
+}
+//# sourceMappingURL=trace-builder.d.ts.map

package/dist/trace-builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace-builder.d.ts","sourceRoot":"","sources":["../src/trace-builder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,SAAS,EACT,UAAU,EAEV,SAAS,EAET,YAAY,EACb,MAAM,YAAY,CAAC;AAIpB,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,MAAM,EAAE,UAAU,EAAE,CAAC;IACrB,SAAS,EAAE,YAAY,EAAE,CAAC;CAC3B;AASD,qBAAa,mBAAmB;IAa5B,OAAO,CAAC,QAAQ,CAAC,IAAI;IAZvB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAmB;IACzC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoB;IAC3C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAqC;IACrE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAqB;IAClD,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,aAAa,CAAuB;IAC5C,OAAO,CAAC,YAAY,CAAuB;IAC3C,OAAO,CAAC,eAAe,CAAuB;IAC9C,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,iBAAiB,CAAK;gBAGX,IAAI,EAAE;QACrB,aAAa,EAAE,MAAM,CAAC;QACtB,OAAO,EAAE,MAAM,CAAC;QAChB,aAAa,EAAE,MAAM,CAAC;QACtB,WAAW,EAAE,MAAM,CAAC;KACrB;IAGH,SAAS,CAAC,IAAI,EAAE;QACd,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC5B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC3B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KACxC,GAAG,IAAI;IA4CR,YAAY,CAAC,IAAI,EAAE;QACjB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC5B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KACxC,GAAG,IAAI;IAkDR,aAAa,CAAC,IAAI,EAAE;QAClB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KACxC,GAAG,IAAI;IA+CR,gBAAgB,CAAC,IAAI,EAAE;QACrB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KACxC,GAAG,IAAI;IAiDR,oBAAoB,CAAC,IAAI,EAAE;QACzB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACxB,GAAG,IAAI;IAOR,iBAAiB,CAAC,IAAI,EAAE;QACtB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACxB,GAAG,IAAI;IA0BR,uBAAuB,CAAC,IAAI,EAAE;QAC5B,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KACxB,GAAG,IAAI;IAOR,WAAW,CAAC,IAAI,EAAE;QAChB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC5B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;KACxC,GAAG,IAAI;IA4CR,UAAU,CACR,OAAO,EAAE,MAAM,EACf,EAAE,SAAW,EACb,UAAU,GAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAM,GACzC,IAAI;IAeP,WAAW,CACT,OAAO,EAAE,MAAM,EACf,EAAE,SAAW,EACb,UAAU,GAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAM,GACzC,IAAI;IAiCP,WAAW,IAAI,WAAW;IAe1B,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,SAAW,GAAG,WAAW;IAgDjE,OAAO,CAAC,gBAAgB;IAoCxB,OAAO,CAAC,kBAAkB;IA4C1B,OAAO,CAAC,UAAU;IAWlB,OAAO,CAAC,WAAW;CAGpB"}

package/dist/trace-builder.js

@@ -0,0 +1,493 @@
+import { createId, nowIso } from "./internal-utils.js";
+export class HarnessTraceBuilder {
+    args;
+    spans = [];
+    events = [];
+    activeToolSpans = new Map();
+    dirtySpanIds = new Set();
+    turnSpanId = null;
+    turnStartedAt = null;
+    outputSpanId = null;
+    outputStartedAt = null;
+    usageEventCount = 0;
+    flushedEventCount = 0;
+    constructor(args) {
+        this.args = args;
+    }
+    startTurn(args) {
+        const at = args.at ?? nowIso();
+        this.turnSpanId = this.turnSpanId ?? createId("trace_span");
+        this.turnStartedAt = this.turnStartedAt ?? at;
+        const existing = this.turnSpanId != null
+            ? this.spans.find((span) => span.id === this.turnSpanId) ?? null
+            : null;
+        this.upsertSpan(this.turnSpanId, {
+            id: this.turnSpanId,
+            parent_id: this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "turn",
+            title: "Model turn",
+            status: "running",
+            started_at: this.turnStartedAt,
+            ended_at: null,
+            attributes: {
+                ...(existing?.attributes ?? {}),
+                provider: args.provider ?? readTraceString(existing?.attributes, [["provider"]]) ?? null,
+                model: args.model ?? readTraceString(existing?.attributes, [["model"]]) ?? null,
+                session_id: args.sessionId ?? readTraceString(existing?.attributes, [["session_id"]]) ?? null,
+                operation_id: args.operationId ?? readTraceString(existing?.attributes, [["operation_id"]]) ?? null,
+                prompt_text: args.promptText ?? readTraceString(existing?.attributes, [["prompt_text"]]) ?? null,
+                prompt_format: args.promptText ? "text" : readTraceString(existing?.attributes, [["prompt_format"]]) ?? null,
+                prompt_source: args.promptText ? "rendered_stage_prompt" : readTraceString(existing?.attributes, [["prompt_source"]]) ?? null,
+                ...(args.attributes ?? {}),
+            },
+        });
+        this.appendEvent(createTraceEvent({
+            spanId: this.turnSpanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "status",
+            at,
+            message: "Model turn started",
+        }));
+    }
+    completeTurn(args) {
+        const at = args.at ?? nowIso();
+        const status = normalizeTraceStatus(args.status ?? null);
+        const existing = this.turnSpanId != null
+            ? this.spans.find((span) => span.id === this.turnSpanId) ?? null
+            : null;
+        if (!this.turnSpanId) {
+            this.turnSpanId = createId("trace_span");
+            this.turnStartedAt = at;
+        }
+        this.upsertSpan(this.turnSpanId, {
+            id: this.turnSpanId,
+            parent_id: this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "turn",
+            title: "Model turn",
+            status,
+            started_at: this.turnStartedAt ?? at,
+            ended_at: at,
+            attributes: {
+                ...(existing?.attributes ?? {}),
+                provider: args.provider ?? readTraceString(existing?.attributes, [["provider"]]) ?? null,
+                model: args.model ?? readTraceString(existing?.attributes, [["model"]]) ?? null,
+                session_id: args.sessionId ?? readTraceString(existing?.attributes, [["session_id"]]) ?? null,
+                operation_id: args.operationId ?? readTraceString(existing?.attributes, [["operation_id"]]) ?? null,
+                error_message: args.errorMessage ?? null,
+                ...(args.attributes ?? {}),
+            },
+        });
+        this.appendEvent(createTraceEvent({
+            spanId: this.turnSpanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: status === "failed" || status === "canceled" ? "error" : "status",
+            at,
+            message: args.errorMessage ??
+                (status === "completed"
+                    ? "Model turn completed"
+                    : `Model turn ${status}`),
+        }));
+    }
+    startToolCall(args) {
+        const at = args.at ?? nowIso();
+        const toolId = args.toolId?.trim() || createId("tool");
+        if (this.activeToolSpans.has(toolId)) {
+            return;
+        }
+        const spanId = createId("trace_span");
+        const title = `Tool call: ${args.toolName?.trim() || "tool"}`;
+        const attributes = {
+            tool_name: args.toolName ?? null,
+            tool_id: toolId,
+            ...(args.attributes ?? {}),
+        };
+        this.activeToolSpans.set(toolId, {
+            spanId,
+            startedAt: at,
+            title,
+            attributes,
+        });
+        this.spans.push({
+            id: spanId,
+            parent_id: this.turnSpanId ?? this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "tool_call",
+            title,
+            status: "running",
+            started_at: at,
+            ended_at: null,
+            attributes,
+        });
+        this.dirtySpanIds.add(spanId);
+        this.appendEvent(createTraceEvent({
+            spanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "status",
+            at,
+            message: `${title} started`,
+            attributes,
+        }));
+    }
+    completeToolCall(args) {
+        const at = args.at ?? nowIso();
+        const toolId = args.toolId?.trim() || createId("tool");
+        if (!this.activeToolSpans.has(toolId)) {
+            this.startToolCall({
+                at,
+                toolId,
+                toolName: args.toolName,
+                attributes: args.attributes,
+            });
+        }
+        const tool = this.activeToolSpans.get(toolId);
+        if (!tool) {
+            return;
+        }
+        this.upsertSpan(tool.spanId, {
+            id: tool.spanId,
+            parent_id: this.turnSpanId ?? this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "tool_call",
+            title: tool.title,
+            status: "completed",
+            started_at: tool.startedAt,
+            ended_at: at,
+            attributes: {
+                ...tool.attributes,
+                ...(args.attributes ?? {}),
+            },
+        });
+        this.appendEvent(createTraceEvent({
+            spanId: tool.spanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "message",
+            at,
+            message: `${tool.title} completed`,
+            attributes: {
+                ...tool.attributes,
+                ...(args.attributes ?? {}),
+            },
+        }));
+        this.activeToolSpans.delete(toolId);
+    }
+    startAssistantOutput(args) {
+        this.ensureOutputSpan(args.at ?? nowIso(), {
+            phase: args.phase ?? null,
+            item_id: args.itemId ?? null,
+        });
+    }
+    appendOutputDelta(args) {
+        const at = args.at ?? nowIso();
+        this.ensureOutputSpan(at, {
+            phase: args.phase ?? null,
+            item_id: args.itemId ?? null,
+        });
+        if (!this.outputSpanId || !args.delta) {
+            return;
+        }
+        this.appendEvent(createTraceEvent({
+            spanId: this.outputSpanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "output",
+            at,
+            message: "Assistant streamed output",
+            attributes: {
+                delta_text: args.delta,
+                delta_length: args.delta.length,
+            },
+        }));
+    }
+    completeAssistantOutput(args) {
+        this.completeOutputSpan(args.at ?? nowIso(), args.text, {
+            phase: args.phase ?? null,
+            item_id: args.itemId ?? null,
+        });
+    }
+    recordUsage(args) {
+        const at = args.at ?? nowIso();
+        const spanId = createId("trace_span");
+        const usage = {
+            input_tokens: args.inputTokens ?? null,
+            output_tokens: args.outputTokens ?? null,
+            ...(args.attributes ?? {}),
+        };
+        this.spans.push({
+            id: spanId,
+            parent_id: this.turnSpanId ?? this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "usage",
+            title: this.usageEventCount === 0 ? "Usage snapshot" : "Usage update",
+            status: "completed",
+            started_at: at,
+            ended_at: at,
+            attributes: usage,
+        });
+        this.dirtySpanIds.add(spanId);
+        this.appendEvent(createTraceEvent({
+            spanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "usage",
+            at,
+            message: `Usage updated${args.inputTokens != null || args.outputTokens != null
+                ? ` · in ${args.inputTokens ?? 0} / out ${args.outputTokens ?? 0}`
+                : ""}`,
+            attributes: usage,
+        }));
+        this.usageEventCount += 1;
+    }
+    recordNote(message, at = nowIso(), attributes = {}) {
+        this.appendEvent(createTraceEvent({
+            spanId: this.turnSpanId ?? this.args.stageSpanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "note",
+            at,
+            message,
+            attributes,
+        }));
+    }
+    recordError(message, at = nowIso(), attributes = {}) {
+        const spanId = createId("trace_span");
+        this.spans.push({
+            id: spanId,
+            parent_id: this.turnSpanId ?? this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "error",
+            title: "Runtime error",
+            status: "failed",
+            started_at: at,
+            ended_at: at,
+            attributes: {
+                message,
+                ...attributes,
+            },
+        });
+        this.dirtySpanIds.add(spanId);
+        this.appendEvent(createTraceEvent({
+            spanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "error",
+            at,
+            message,
+            attributes,
+        }));
+    }
+    flushBundle() {
+        const dirtySpanIds = [...this.dirtySpanIds];
+        const spans = this.spans
+            .filter((span) => dirtySpanIds.includes(span.id))
+            .sort((left, right) => left.started_at.localeCompare(right.started_at));
+        const events = this.events.slice(this.flushedEventCount);
+        this.dirtySpanIds.clear();
+        this.flushedEventCount = this.events.length;
+        return {
+            spans,
+            events,
+            summaries: [],
+        };
+    }
+    buildBundle(finalOutput, endedAt = nowIso()) {
+        if (this.outputSpanId) {
+            this.completeOutputSpan(endedAt, finalOutput, {});
+        }
+        if (this.turnSpanId) {
+            const existing = this.spans.find((span) => span.id === this.turnSpanId);
+            if (existing && !existing.ended_at) {
+                this.upsertSpan(this.turnSpanId, {
+                    ...existing,
+                    status: existing.status === "running" ? "completed" : existing.status,
+                    ended_at: endedAt,
+                });
+            }
+        }
+        for (const [toolId, tool] of this.activeToolSpans.entries()) {
+            this.upsertSpan(tool.spanId, {
+                id: tool.spanId,
+                parent_id: this.turnSpanId ?? this.args.stageSpanId,
+                attempt_number: this.args.attemptNumber,
+                stage_id: this.args.stageId,
+                stage_run_index: this.args.stageRunIndex,
+                kind: "tool_call",
+                title: tool.title,
+                status: "warning",
+                started_at: tool.startedAt,
+                ended_at: endedAt,
+                attributes: {
+                    ...tool.attributes,
+                    note: "tool span completed implicitly during bundle finalization",
+                },
+            });
+            this.activeToolSpans.delete(toolId);
+        }
+        return {
+            spans: [...this.spans].sort((left, right) => left.started_at.localeCompare(right.started_at)),
+            events: [...this.events].sort((left, right) => left.at.localeCompare(right.at)),
+            summaries: [],
+        };
+    }
+    ensureOutputSpan(at, attributes) {
+        if (this.outputSpanId) {
+            return;
+        }
+        this.outputSpanId = createId("trace_span");
+        this.outputStartedAt = at;
+        this.spans.push({
+            id: this.outputSpanId,
+            parent_id: this.turnSpanId ?? this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "assistant_output",
+            title: "Assistant output",
+            status: "running",
+            started_at: at,
+            ended_at: null,
+            attributes,
+        });
+        this.dirtySpanIds.add(this.outputSpanId);
+        this.appendEvent(createTraceEvent({
+            spanId: this.outputSpanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "status",
+            at,
+            message: "Assistant output started",
+        }));
+    }
+    completeOutputSpan(at, messageText, attributes) {
+        this.ensureOutputSpan(at, attributes);
+        if (!this.outputSpanId) {
+            return;
+        }
+        this.upsertSpan(this.outputSpanId, {
+            id: this.outputSpanId,
+            parent_id: this.turnSpanId ?? this.args.stageSpanId,
+            attempt_number: this.args.attemptNumber,
+            stage_id: this.args.stageId,
+            stage_run_index: this.args.stageRunIndex,
+            kind: "assistant_output",
+            title: "Assistant output",
+            status: "completed",
+            started_at: this.outputStartedAt ?? at,
+            ended_at: at,
+            attributes: {
+                ...attributes,
+                output_length: messageText.length,
+            },
+        });
+        this.appendEvent(createTraceEvent({
+            spanId: this.outputSpanId,
+            attemptNumber: this.args.attemptNumber,
+            stageId: this.args.stageId,
+            stageRunIndex: this.args.stageRunIndex,
+            kind: "output",
+            at,
+            message: messageText
+                ? truncateValue(messageText, 140)
+                : "Assistant output completed",
+            attributes: {
+                output_length: messageText.length,
+                output_text: messageText,
+            },
+        }));
+    }
+    upsertSpan(spanId, next) {
+        const index = this.spans.findIndex((span) => span.id === spanId);
+        if (index === -1) {
+            this.spans.push(next);
+            this.dirtySpanIds.add(spanId);
+            return;
+        }
+        this.spans[index] = next;
+        this.dirtySpanIds.add(spanId);
+    }
+    appendEvent(event) {
+        this.events.push(event);
+    }
+}
+function createTraceEvent(args) {
+    return {
+        id: createId("trace_event"),
+        span_id: args.spanId,
+        attempt_number: args.attemptNumber,
+        stage_id: args.stageId,
+        stage_run_index: args.stageRunIndex,
+        kind: args.kind,
+        at: args.at,
+        message: args.message,
+        attributes: args.attributes ?? {},
+    };
+}
+function normalizeTraceStatus(rawStatus) {
+    if (!rawStatus) {
+        return "completed";
+    }
+    if (["failed", "error"].includes(rawStatus)) {
+        return "failed";
+    }
+    if (["interrupted", "canceled", "cancelled"].includes(rawStatus)) {
+        return "canceled";
+    }
+    if (["running", "inprogress", "in_progress", "started"].includes(rawStatus.toLowerCase())) {
+        return "running";
+    }
+    return "completed";
+}
+function readTraceString(value, paths) {
+    if (!value) {
+        return null;
+    }
+    for (const path of paths) {
+        const current = readNestedValue(value, path);
+        if (typeof current === "string" && current.length > 0) {
+            return current;
+        }
+    }
+    return null;
+}
+function readNestedValue(value, path) {
+    let current = value;
+    for (const segment of path) {
+        if (!current || Array.isArray(current) || typeof current !== "object") {
+            return undefined;
+        }
+        current = current[segment];
+    }
+    return current;
+}
+function truncateValue(value, maxLength) {
+    if (value.length <= maxLength) {
+        return value;
+    }
+    return `${value.slice(0, maxLength - 1)}…`;
+}

package/dist/trace-replay.d.ts

@@ -0,0 +1,33 @@
+import type { JsonValue, TraceSpan } from "./types.js";
+import type { TraceBundle } from "./trace-builder.js";
+export interface HarnessTraceReplayEntry {
+    at: string;
+}
+export interface HarnessTraceReplaySource<TEntry extends HarnessTraceReplayEntry = HarnessTraceReplayEntry> {
+    entries: readonly TEntry[];
+}
+export interface HarnessTraceReplayerBuildArgs<TEntry extends HarnessTraceReplayEntry = HarnessTraceReplayEntry> {
+    artifact: {
+        attempt_number: number;
+        stage_id: string;
+        run_index: number;
+        output_file: string;
+        events_file: string;
+        raw_events_file?: string | null;
+        final_output?: string | null;
+    };
+    source: HarnessTraceReplaySource<TEntry>;
+    oldTurnSpan: TraceSpan | null;
+    stageSpanId: string;
+    stageStartedAt: string;
+    endedAt: string;
+    readFinalOutput: () => Promise<string>;
+}
+export interface HarnessTraceReplayer<TEntry extends HarnessTraceReplayEntry = HarnessTraceReplayEntry> {
+    harnessId: string;
+    parseRawReplayEntries(entries: Array<Record<string, unknown>>): HarnessTraceReplaySource<TEntry> | null;
+    parseHarnessReplayEntries(entries: Array<Record<string, unknown>>): HarnessTraceReplaySource<TEntry> | null;
+    buildTraceBundle(args: HarnessTraceReplayerBuildArgs<TEntry>): Promise<TraceBundle>;
+}
+export declare function readTraceString(value: Record<string, JsonValue> | undefined, key: string): string | null;
+//# sourceMappingURL=trace-replay.d.ts.map

package/dist/trace-replay.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace-replay.d.ts","sourceRoot":"","sources":["../src/trace-replay.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,SAAS,EACT,SAAS,EACV,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,WAAW,uBAAuB;IACtC,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,wBAAwB,CACvC,MAAM,SAAS,uBAAuB,GAAG,uBAAuB;IAEhE,OAAO,EAAE,SAAS,MAAM,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,6BAA6B,CAC5C,MAAM,SAAS,uBAAuB,GAAG,uBAAuB;IAEhE,QAAQ,EAAE;QACR,cAAc,EAAE,MAAM,CAAC;QACvB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAChC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;KAC9B,CAAC;IACF,MAAM,EAAE,wBAAwB,CAAC,MAAM,CAAC,CAAC;IACzC,WAAW,EAAE,SAAS,GAAG,IAAI,CAAC;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;CACxC;AAED,MAAM,WAAW,oBAAoB,CACnC,MAAM,SAAS,uBAAuB,GAAG,uBAAuB;IAEhE,SAAS,EAAE,MAAM,CAAC;IAClB,qBAAqB,CACnB,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,GACtC,wBAAwB,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;IAC3C,yBAAyB,CACvB,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,GACtC,wBAAwB,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;IAC3C,gBAAgB,CACd,IAAI,EAAE,6BAA6B,CAAC,MAAM,CAAC,GAC1C,OAAO,CAAC,WAAW,CAAC,CAAC;CACzB;AAED,wBAAgB,eAAe,CAC7B,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,SAAS,EAC5C,GAAG,EAAE,MAAM,GACV,MAAM,GAAG,IAAI,CAGf"}

package/dist/trace-replay.js

@@ -0,0 +1,4 @@
+export function readTraceString(value, key) {
+    const current = value?.[key];
+    return typeof current === "string" && current.length > 0 ? current : null;
+}