@neutrome/open-ai-router 0.1.16 → 0.1.18

package/package.json

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

package/src/observer.test.ts

@@ -0,0 +1,34 @@
+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.request",
+      requestId: "request-1",
+      executionId: "provider-1",
+      timestamp: "2026-06-24T12:00:01.000Z",
+      data: { lilText: "SET_MODEL \"gpt-4o\"" },
+    });
+    collector.observe({
+      kind: "provider.response_chunk",
+      requestId: "request-1",
+      executionId: "provider-1",
+      timestamp: "2026-06-24T12:00:02.000Z",
+      data: { lilText: "STREAM_DELTA \"Hi\"" },
+    });
+
+    expect(collector.getTrace().rawTrace).toBe(
+      ';request:0\nMSG_START\n\n;provider-request:1\nSET_MODEL "gpt-4o"\n\n;provider-response-chunk:2\nSTREAM_DELTA "Hi"',
+    );
+  });
+});

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,7 +153,24 @@ 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 {
+  switch (event.kind) {
+    case "request.received":
+      return "request";
+    case "provider.request":
+      return "provider-request";
+    case "provider.response":
+      return "provider-response";
+    case "provider.response_chunk":
+      return "provider-response-chunk";
+    case "tool.executed":
+      return "tool";
+    default:
+      return event.kind.replace(/\./g, "-") || "trace";
   }
 }