@neutrome/open-ai-router 0.1.17 → 0.1.18

package/package.json

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

package/src/observer.test.ts

@@ -13,15 +13,22 @@ describe("createTraceCollector", () => {
       data: { lilText: "MSG_START" },
     });
     collector.observe({
-      kind: "provider.started",
+      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:1\nSET_MODEL "gpt-4o"',
+      ';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

@@ -158,7 +158,20 @@ export function createTraceCollector(spanId = crypto.randomUUID()): {
 }
 
 function traceSegmentLabel(event: AuditEvent): string {
-  return event.kind.split(".", 1)[0] || "trace";
+  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";
+  }
 }
 
 export function renderProgramAsLil(program: Program): string {