@neutrome/open-ai-router 0.1.16 → 0.1.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neutrome/open-ai-router",
-  "version": "0.1.16",
+  "version": "0.1.17",
   "type": "module",
   "exports": {
     ".": "./src/index.ts"

package/src/observer.test.ts

@@ -0,0 +1,27 @@
+import { describe, expect, it } from "vitest";
+import { createTraceCollector } from "./observer.ts";
+
+describe("createTraceCollector", () => {
+  it("prefixes raw trace chunks with event comments", () => {
+    const collector = createTraceCollector("00000000-0000-4000-8000-000000000001");
+
+    collector.observe({
+      kind: "request.received",
+      requestId: "request-1",
+      executionId: "request-1",
+      timestamp: "2026-06-24T12:00:00.000Z",
+      data: { lilText: "MSG_START" },
+    });
+    collector.observe({
+      kind: "provider.started",
+      requestId: "request-1",
+      executionId: "provider-1",
+      timestamp: "2026-06-24T12:00:01.000Z",
+      data: { lilText: "SET_MODEL \"gpt-4o\"" },
+    });
+
+    expect(collector.getTrace().rawTrace).toBe(
+      ';request:0\nMSG_START\n\n;provider:1\nSET_MODEL "gpt-4o"',
+    );
+  });
+});

package/src/observer.ts

@@ -45,6 +45,7 @@ export function createTraceCollector(spanId = crypto.randomUUID()): {
   const executorSpans = new Map<string, TraceSpan>();
   const rawSegments: string[] = [];
   const startedAt = new Date().toISOString();
+  let rawSegmentIndex = 0;
   let finishedAt: string | undefined;
   let error: string | undefined;
 
@@ -152,10 +153,14 @@ export function createTraceCollector(spanId = crypto.randomUUID()): {
   function appendLilSegment(event: AuditEvent): void {
     const lilText = asString(event.data?.lilText)?.trim();
     if (!lilText) return;
-    rawSegments.push(lilText);
+    rawSegments.push(`;${traceSegmentLabel(event)}:${rawSegmentIndex++}\n${lilText}`);
   }
 }
 
+function traceSegmentLabel(event: AuditEvent): string {
+  return event.kind.split(".", 1)[0] || "trace";
+}
+
 export function renderProgramAsLil(program: Program): string {
   return program.code
     .map((instruction) => {