@neutrome/open-ai-router 0.8.0 → 0.8.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/package.json +3 -3
- package/src/telemetry/router.ts +24 -6
- package/src/telemetry.test.ts +78 -0
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@neutrome/open-ai-router",
|
|
3
|
-
"version": "0.8.
|
|
3
|
+
"version": "0.8.1",
|
|
4
4
|
"type": "module",
|
|
5
5
|
"files": [
|
|
6
6
|
"src",
|
|
@@ -16,8 +16,8 @@
|
|
|
16
16
|
"hono": "^4.12.18",
|
|
17
17
|
"posthog-node": "^5.41.0",
|
|
18
18
|
"zod": "^4.2.1",
|
|
19
|
-
"@neutrome/lil-engine": "0.6.
|
|
20
|
-
"@neutrome/lilsdk": "0.6.
|
|
19
|
+
"@neutrome/lil-engine": "0.6.1",
|
|
20
|
+
"@neutrome/lilsdk": "0.6.1"
|
|
21
21
|
},
|
|
22
22
|
"devDependencies": {
|
|
23
23
|
"@types/node": "^25.9.3",
|
package/src/telemetry/router.ts
CHANGED
|
@@ -55,6 +55,7 @@ export function createRouterTelemetry(
|
|
|
55
55
|
const spans: ExecutionSummarySpan[] = [];
|
|
56
56
|
const active = new Map<string, ActiveSpan>();
|
|
57
57
|
const generations: ActiveSpan[] = [];
|
|
58
|
+
const generationReports: Promise<void>[] = [];
|
|
58
59
|
const errors = new Map<string, TelemetryError>();
|
|
59
60
|
let traceError: string | undefined;
|
|
60
61
|
let finalizedAt: string | undefined;
|
|
@@ -93,11 +94,12 @@ export function createRouterTelemetry(
|
|
|
93
94
|
requestId: event.requestId,
|
|
94
95
|
executionId: event.executionId,
|
|
95
96
|
});
|
|
97
|
+
completeSpan(event.executionId, span);
|
|
98
|
+
return;
|
|
96
99
|
}
|
|
97
100
|
if (!isInvocationFinished(event.kind)) return;
|
|
98
101
|
finishSpan(span.summary, data.finishedAt ?? event.timestamp);
|
|
99
|
-
|
|
100
|
-
if (span.summary.kind === "provider") generations.push(span);
|
|
102
|
+
completeSpan(event.executionId, span);
|
|
101
103
|
}
|
|
102
104
|
|
|
103
105
|
function observeDetached(
|
|
@@ -135,13 +137,11 @@ export function createRouterTelemetry(
|
|
|
135
137
|
finalizedAt = finishedAt;
|
|
136
138
|
for (const span of active.values()) {
|
|
137
139
|
finishSpan(span.summary, finishedAt, "incomplete");
|
|
138
|
-
|
|
140
|
+
queueGeneration(span);
|
|
139
141
|
}
|
|
140
142
|
active.clear();
|
|
141
143
|
await reportAll([
|
|
142
|
-
...
|
|
143
|
-
traceDriver?.recordGeneration(toGeneration(span)),
|
|
144
|
-
),
|
|
144
|
+
...generationReports,
|
|
145
145
|
...[...errors.values()].map((error) => errorDriver?.recordError(error)),
|
|
146
146
|
]);
|
|
147
147
|
}
|
|
@@ -208,6 +208,24 @@ export function createRouterTelemetry(
|
|
|
208
208
|
: { timeToFirstTokenMs: activeSpan.timeToFirstTokenMs }),
|
|
209
209
|
};
|
|
210
210
|
}
|
|
211
|
+
|
|
212
|
+
function completeSpan(executionId: string, span: ActiveSpan): void {
|
|
213
|
+
active.delete(executionId);
|
|
214
|
+
queueGeneration(span);
|
|
215
|
+
}
|
|
216
|
+
|
|
217
|
+
function queueGeneration(span: ActiveSpan): void {
|
|
218
|
+
if (span.summary.kind !== "provider") return;
|
|
219
|
+
generations.push(span);
|
|
220
|
+
const report = traceDriver?.recordGeneration(toGeneration(span));
|
|
221
|
+
if (report) {
|
|
222
|
+
generationReports.push(
|
|
223
|
+
report.catch((error) => {
|
|
224
|
+
console.error("[telemetry] report failed", error);
|
|
225
|
+
}),
|
|
226
|
+
);
|
|
227
|
+
}
|
|
228
|
+
}
|
|
211
229
|
}
|
|
212
230
|
|
|
213
231
|
function applyFailure(
|
package/src/telemetry.test.ts
CHANGED
|
@@ -147,6 +147,7 @@ describe("createRouterTelemetry", () => {
|
|
|
147
147
|
requestId: "request-1",
|
|
148
148
|
executionId: "provider-1",
|
|
149
149
|
errorKind: "provider",
|
|
150
|
+
timestamp: "2026-07-13T12:00:01.000Z",
|
|
150
151
|
data: {
|
|
151
152
|
message: "Provider returned error",
|
|
152
153
|
stage: "provider_invoke",
|
|
@@ -160,6 +161,13 @@ describe("createRouterTelemetry", () => {
|
|
|
160
161
|
}),
|
|
161
162
|
);
|
|
162
163
|
|
|
164
|
+
expect(traceDriver.recordGeneration).toHaveBeenCalledWith(
|
|
165
|
+
expect.objectContaining({
|
|
166
|
+
startedAt: "2026-07-13T12:00:00.000Z",
|
|
167
|
+
finishedAt: "2026-07-13T12:00:01.000Z",
|
|
168
|
+
}),
|
|
169
|
+
);
|
|
170
|
+
|
|
163
171
|
await telemetry.flush();
|
|
164
172
|
|
|
165
173
|
expect(traceDriver.recordGeneration).toHaveBeenCalledWith(
|
|
@@ -178,6 +186,76 @@ describe("createRouterTelemetry", () => {
|
|
|
178
186
|
status: 400,
|
|
179
187
|
}),
|
|
180
188
|
);
|
|
189
|
+
expect(traceDriver.recordGeneration).toHaveBeenCalledTimes(1);
|
|
190
|
+
});
|
|
191
|
+
|
|
192
|
+
it("reports consecutive provider attempts at their own terminal times", async () => {
|
|
193
|
+
const traceDriver = { recordGeneration: vi.fn(async () => {}) };
|
|
194
|
+
const telemetry = createRouterTelemetry({ traceDriver });
|
|
195
|
+
|
|
196
|
+
telemetry.observe(
|
|
197
|
+
createExecutionEvent({
|
|
198
|
+
kind: "provider.started",
|
|
199
|
+
requestId: "request-1",
|
|
200
|
+
executionId: "provider-1",
|
|
201
|
+
data: {
|
|
202
|
+
startedAt: "2026-07-13T12:00:00.000Z",
|
|
203
|
+
provider: "cloudflare",
|
|
204
|
+
model: "@cf/first",
|
|
205
|
+
},
|
|
206
|
+
}),
|
|
207
|
+
);
|
|
208
|
+
telemetry.observe(
|
|
209
|
+
createExecutionEvent({
|
|
210
|
+
kind: "execution.failed",
|
|
211
|
+
requestId: "request-1",
|
|
212
|
+
executionId: "provider-1",
|
|
213
|
+
errorKind: "provider",
|
|
214
|
+
timestamp: "2026-07-13T12:00:01.000Z",
|
|
215
|
+
data: { message: "first failed" },
|
|
216
|
+
}),
|
|
217
|
+
);
|
|
218
|
+
telemetry.observe(
|
|
219
|
+
createExecutionEvent({
|
|
220
|
+
kind: "provider.started",
|
|
221
|
+
requestId: "request-1",
|
|
222
|
+
executionId: "provider-2",
|
|
223
|
+
data: {
|
|
224
|
+
startedAt: "2026-07-13T12:00:02.000Z",
|
|
225
|
+
provider: "openrouter",
|
|
226
|
+
model: "second",
|
|
227
|
+
},
|
|
228
|
+
}),
|
|
229
|
+
);
|
|
230
|
+
telemetry.observe(
|
|
231
|
+
createExecutionEvent({
|
|
232
|
+
kind: "provider.finished",
|
|
233
|
+
requestId: "request-1",
|
|
234
|
+
executionId: "provider-2",
|
|
235
|
+
timestamp: "2026-07-13T12:00:03.000Z",
|
|
236
|
+
data: { finishedAt: "2026-07-13T12:00:03.000Z" },
|
|
237
|
+
}),
|
|
238
|
+
);
|
|
239
|
+
|
|
240
|
+
expect(traceDriver.recordGeneration.mock.calls).toEqual([
|
|
241
|
+
[
|
|
242
|
+
expect.objectContaining({
|
|
243
|
+
model: "@cf/first",
|
|
244
|
+
startedAt: "2026-07-13T12:00:00.000Z",
|
|
245
|
+
finishedAt: "2026-07-13T12:00:01.000Z",
|
|
246
|
+
}),
|
|
247
|
+
],
|
|
248
|
+
[
|
|
249
|
+
expect.objectContaining({
|
|
250
|
+
model: "second",
|
|
251
|
+
startedAt: "2026-07-13T12:00:02.000Z",
|
|
252
|
+
finishedAt: "2026-07-13T12:00:03.000Z",
|
|
253
|
+
}),
|
|
254
|
+
],
|
|
255
|
+
]);
|
|
256
|
+
|
|
257
|
+
await telemetry.flush();
|
|
258
|
+
expect(traceDriver.recordGeneration).toHaveBeenCalledTimes(2);
|
|
181
259
|
});
|
|
182
260
|
});
|
|
183
261
|
|