@llmtracer/sdk 2.1.0 → 2.5.0
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/README.md +84 -119
- package/dist/cjs/config.d.ts +8 -0
- package/dist/cjs/config.d.ts.map +1 -1
- package/dist/cjs/config.js +8 -0
- package/dist/cjs/config.js.map +1 -1
- package/dist/cjs/index.d.ts +11 -3
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/index.js +33 -1
- package/dist/cjs/index.js.map +1 -1
- package/dist/cjs/patcher.d.ts +1 -0
- package/dist/cjs/patcher.d.ts.map +1 -1
- package/dist/cjs/patcher.js +106 -1
- package/dist/cjs/patcher.js.map +1 -1
- package/dist/cjs/providers/anthropic.d.ts.map +1 -1
- package/dist/cjs/providers/anthropic.js +28 -3
- package/dist/cjs/providers/anthropic.js.map +1 -1
- package/dist/cjs/providers/google.d.ts +4 -0
- package/dist/cjs/providers/google.d.ts.map +1 -0
- package/dist/cjs/providers/google.js +138 -0
- package/dist/cjs/providers/google.js.map +1 -0
- package/dist/cjs/providers/index.d.ts +1 -0
- package/dist/cjs/providers/index.d.ts.map +1 -1
- package/dist/cjs/providers/index.js +4 -1
- package/dist/cjs/providers/index.js.map +1 -1
- package/dist/cjs/providers/openai.d.ts +6 -1
- package/dist/cjs/providers/openai.d.ts.map +1 -1
- package/dist/cjs/providers/openai.js +127 -5
- package/dist/cjs/providers/openai.js.map +1 -1
- package/dist/cjs/schema.d.ts +3 -0
- package/dist/cjs/schema.d.ts.map +1 -1
- package/dist/cjs/schema.js +20 -2
- package/dist/cjs/schema.js.map +1 -1
- package/dist/cjs/transport.d.ts +12 -0
- package/dist/cjs/transport.d.ts.map +1 -1
- package/dist/cjs/transport.js +115 -9
- package/dist/cjs/transport.js.map +1 -1
- package/dist/cjs/wrapper.d.ts +1 -1
- package/dist/cjs/wrapper.d.ts.map +1 -1
- package/dist/cjs/wrapper.js +13 -2
- package/dist/cjs/wrapper.js.map +1 -1
- package/dist/esm/config.d.ts +8 -0
- package/dist/esm/config.d.ts.map +1 -1
- package/dist/esm/config.js +8 -0
- package/dist/esm/config.js.map +1 -1
- package/dist/esm/index.d.ts +11 -3
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +33 -1
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/patcher.d.ts +1 -0
- package/dist/esm/patcher.d.ts.map +1 -1
- package/dist/esm/patcher.js +106 -2
- package/dist/esm/patcher.js.map +1 -1
- package/dist/esm/providers/anthropic.d.ts.map +1 -1
- package/dist/esm/providers/anthropic.js +28 -3
- package/dist/esm/providers/anthropic.js.map +1 -1
- package/dist/esm/providers/google.d.ts +4 -0
- package/dist/esm/providers/google.d.ts.map +1 -0
- package/dist/esm/providers/google.js +101 -0
- package/dist/esm/providers/google.js.map +1 -0
- package/dist/esm/providers/index.d.ts +1 -0
- package/dist/esm/providers/index.d.ts.map +1 -1
- package/dist/esm/providers/index.js +1 -0
- package/dist/esm/providers/index.js.map +1 -1
- package/dist/esm/providers/openai.d.ts +6 -1
- package/dist/esm/providers/openai.d.ts.map +1 -1
- package/dist/esm/providers/openai.js +122 -5
- package/dist/esm/providers/openai.js.map +1 -1
- package/dist/esm/schema.d.ts +3 -0
- package/dist/esm/schema.d.ts.map +1 -1
- package/dist/esm/schema.js +20 -2
- package/dist/esm/schema.js.map +1 -1
- package/dist/esm/transport.d.ts +12 -0
- package/dist/esm/transport.d.ts.map +1 -1
- package/dist/esm/transport.js +114 -9
- package/dist/esm/transport.js.map +1 -1
- package/dist/esm/wrapper.d.ts +1 -1
- package/dist/esm/wrapper.d.ts.map +1 -1
- package/dist/esm/wrapper.js +13 -2
- package/dist/esm/wrapper.js.map +1 -1
- package/package.json +10 -14
- package/dist/cjs/auto.d.ts +0 -2
- package/dist/cjs/auto.d.ts.map +0 -1
- package/dist/cjs/auto.js +0 -5
- package/dist/cjs/auto.js.map +0 -1
- package/dist/esm/auto.d.ts +0 -2
- package/dist/esm/auto.d.ts.map +0 -1
- package/dist/esm/auto.js +0 -3
- package/dist/esm/auto.js.map +0 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,UAAU,EAAE,YAAY,EAA6B,MAAM,WAAW,CAAC;AAChF,OAAO,KAAK,SAAS,MAAM,cAAc,CAAC;AAE1C,MAAM,UAAU,sBAAsB,CAAC,MAAW;IAChD,
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,UAAU,EAAE,YAAY,EAA6B,MAAM,WAAW,CAAC;AAChF,OAAO,KAAK,SAAS,MAAM,cAAc,CAAC;AAE1C,MAAM,UAAU,sBAAsB,CAAC,MAAW;IAChD,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;IAC5B,MAAM,SAAS,GAAG,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;IAC7C,MAAM,aAAa,GAAG,KAAK,EAAE,2BAA2B,IAAI,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,KAAK,EAAE,uBAAuB,IAAI,CAAC,CAAC;IAEtD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO,CAAC,GAAG,CACT,qCAAqC,SAAS,QAAQ,UAAU,iBAAiB,aAAa,eAAe,SAAS,EAAE,CACzH,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,CAAC,MAAM,EAAE,OAAO,IAAI,EAAE,CAAC;SACzC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;SACzC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SACvB,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO;QACL,QAAQ,EAAE,WAAW;QACrB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,SAAS;QACjC,YAAY,EAAE,SAAS,GAAG,aAAa,GAAG,SAAS;QACnD,aAAa,EAAE,UAAU;QACzB,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;QACjD,aAAa,EAAE,YAAY;KAC5B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAW,EAAE,OAAoB,EAAE,SAAiB,EAAE,IAAW;IACnG,IAAI,KAAK,GAAG,SAAS,CAAC;IACtB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,MAAM,YAAY,GAAa,EAAE,CAAC;IAElC,SAAS,MAAM;QACb,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CACT,qCAAqC,WAAW,QAAQ,YAAY,iBAAiB,aAAa,eAAe,SAAS,EAAE,CAC7H,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE;YAChC,QAAQ,EAAE,WAAW;YACrB,KAAK;YACL,YAAY,EAAE,WAAW,GAAG,aAAa,GAAG,SAAS;YACrD,aAAa,EAAE,YAAY;YAC3B,MAAM,EAAE,SAAS;YACjB,aAAa,EAAE,YAAY;SAC5B,EAAE,SAAS,CAAC,CAAC;QACd,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,MAAM,CAAC,KAAK;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;QACvB,GAAG,CAAC,MAAM,EAAE,IAAI;YACd,IAAI,IAAI,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;gBAClC,OAAO;oBACL,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;oBACxD,OAAO;wBACL,KAAK,CAAC,IAAI;4BACR,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;4BACjD,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gCACrB,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;gCAC/B,2BAA2B;gCAC3B,wDAAwD;gCACxD,+CAA+C;gCAC/C,8CAA8C;gCAC9C,IAAI,KAAK,EAAE,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oCACrD,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC;oCACrC,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,YAAY,IAAI,WAAW,CAAC;oCAC/D,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,2BAA2B,IAAI,aAAa,CAAC;oCAClF,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,uBAAuB,IAAI,SAAS,CAAC;gCACxE,CAAC;gCACD,IAAI,KAAK,EAAE,IAAI,KAAK,qBAAqB,IAAI,KAAK,CAAC,aAAa,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;oCACtF,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;gCAC9C,CAAC;gCACD,IAAI,KAAK,EAAE,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oCACnD,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,YAAY,CAAC;gCAC3D,CAAC;4BACH,CAAC;4BACD,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gCACpB,IAAI,CAAC;oCACH,MAAM,EAAE,CAAC;gCACX,CAAC;gCAAC,MAAM,CAAC;oCACP,UAAU;gCACZ,CAAC;4BACH,CAAC;4BACD,OAAO,UAAU,CAAC;wBACpB,CAAC;wBACD,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC;wBACvD,KAAK,EAAE,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC;qBACtD,CAAC;gBACJ,CAAC,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;gBAChC,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import { ContextData, ResponseData } from '../schema';
|
|
2
|
+
export declare function parseGoogleResponse(result: any): ResponseData;
|
|
3
|
+
export declare function wrapGoogleStream(result: any, context: ContextData, elapsedMs: number, args: any[]): any;
|
|
4
|
+
//# sourceMappingURL=google.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../../src/providers/google.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,WAAW,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAiChF,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,GAAG,YAAY,CAyB7D;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,GAAG,CAkEvG"}
|
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
import { config } from '../config';
|
|
2
|
+
import { buildEvent, printCapture } from '../schema';
|
|
3
|
+
import * as transport from '../transport';
|
|
4
|
+
/**
|
|
5
|
+
* Extract token counts from a Google GenAI response.
|
|
6
|
+
* The JS SDK may use camelCase while Python uses snake_case — we check both.
|
|
7
|
+
*/
|
|
8
|
+
function extractTokens(metadata) {
|
|
9
|
+
const inputTokens = metadata?.input_tokens ?? metadata?.inputTokens ??
|
|
10
|
+
metadata?.promptTokenCount ?? metadata?.prompt_token_count ?? 0;
|
|
11
|
+
const outputTokens = metadata?.output_tokens ?? metadata?.outputTokens ??
|
|
12
|
+
metadata?.candidatesTokenCount ?? metadata?.candidates_token_count ?? 0;
|
|
13
|
+
const cachedTokens = metadata?.cached_content_token_count ?? metadata?.cachedContentTokenCount ?? 0;
|
|
14
|
+
const thoughtsTokens = metadata?.thoughts_token_count ?? metadata?.thoughtsTokenCount ?? 0;
|
|
15
|
+
const toolUseTokens = metadata?.tool_use_prompt_token_count ?? metadata?.toolUsePromptTokenCount ?? 0;
|
|
16
|
+
const totalReported = metadata?.total_token_count ?? metadata?.totalTokenCount ?? 0;
|
|
17
|
+
return { inputTokens, outputTokens, cachedTokens, thoughtsTokens, toolUseTokens, totalReported };
|
|
18
|
+
}
|
|
19
|
+
export function parseGoogleResponse(result) {
|
|
20
|
+
// Usage metadata may be at result.usageMetadata or result.usage_metadata
|
|
21
|
+
const metadata = result?.usageMetadata ?? result?.usage_metadata;
|
|
22
|
+
const { inputTokens, outputTokens, cachedTokens, thoughtsTokens, toolUseTokens, totalReported } = extractTokens(metadata);
|
|
23
|
+
if (config.debug) {
|
|
24
|
+
console.log(`[llmtracer] Google tokens — base_in=${inputTokens} cached=${cachedTokens} tools=${toolUseTokens} base_out=${outputTokens} thoughts=${thoughtsTokens} total_reported=${totalReported}`);
|
|
25
|
+
}
|
|
26
|
+
const toolsInvoked = (result?.candidates?.[0]?.content?.parts || [])
|
|
27
|
+
.filter((p) => p.functionCall)
|
|
28
|
+
.map((p) => p.functionCall.name)
|
|
29
|
+
.filter(Boolean);
|
|
30
|
+
return {
|
|
31
|
+
provider: 'google',
|
|
32
|
+
model: result?.model ?? result?.modelVersion ?? 'unknown',
|
|
33
|
+
input_tokens: inputTokens + cachedTokens + toolUseTokens,
|
|
34
|
+
output_tokens: outputTokens + thoughtsTokens,
|
|
35
|
+
status: 'success',
|
|
36
|
+
tools_invoked: toolsInvoked,
|
|
37
|
+
};
|
|
38
|
+
}
|
|
39
|
+
export function wrapGoogleStream(result, context, elapsedMs, args) {
|
|
40
|
+
let model = 'unknown';
|
|
41
|
+
let lastMetadata = null;
|
|
42
|
+
function record() {
|
|
43
|
+
const { inputTokens, outputTokens, cachedTokens, thoughtsTokens, toolUseTokens, totalReported } = extractTokens(lastMetadata);
|
|
44
|
+
if (config.debug) {
|
|
45
|
+
console.log(`[llmtracer] Google tokens — base_in=${inputTokens} cached=${cachedTokens} tools=${toolUseTokens} base_out=${outputTokens} thoughts=${thoughtsTokens} total_reported=${totalReported}`);
|
|
46
|
+
}
|
|
47
|
+
const event = buildEvent(context, {
|
|
48
|
+
provider: 'google',
|
|
49
|
+
model,
|
|
50
|
+
input_tokens: inputTokens + cachedTokens + toolUseTokens,
|
|
51
|
+
output_tokens: outputTokens + thoughtsTokens,
|
|
52
|
+
status: 'success',
|
|
53
|
+
}, elapsedMs);
|
|
54
|
+
transport.enqueue(event);
|
|
55
|
+
if (config.debug)
|
|
56
|
+
printCapture(event);
|
|
57
|
+
}
|
|
58
|
+
return new Proxy(result, {
|
|
59
|
+
get(target, prop) {
|
|
60
|
+
if (prop === Symbol.asyncIterator) {
|
|
61
|
+
return function () {
|
|
62
|
+
const originalIterator = target[Symbol.asyncIterator]();
|
|
63
|
+
return {
|
|
64
|
+
async next() {
|
|
65
|
+
const iterResult = await originalIterator.next();
|
|
66
|
+
if (!iterResult.done) {
|
|
67
|
+
const chunk = iterResult.value;
|
|
68
|
+
// Extract model from chunk if available
|
|
69
|
+
if (chunk?.model || chunk?.modelVersion) {
|
|
70
|
+
model = chunk.model ?? chunk.modelVersion ?? model;
|
|
71
|
+
}
|
|
72
|
+
// Accumulate usage metadata — last chunk typically has the final totals
|
|
73
|
+
const metadata = chunk?.usageMetadata ?? chunk?.usage_metadata;
|
|
74
|
+
if (metadata) {
|
|
75
|
+
lastMetadata = metadata;
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
if (iterResult.done) {
|
|
79
|
+
try {
|
|
80
|
+
record();
|
|
81
|
+
}
|
|
82
|
+
catch {
|
|
83
|
+
// swallow
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
return iterResult;
|
|
87
|
+
},
|
|
88
|
+
return: originalIterator.return?.bind(originalIterator),
|
|
89
|
+
throw: originalIterator.throw?.bind(originalIterator),
|
|
90
|
+
};
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
const value = target[prop];
|
|
94
|
+
if (typeof value === 'function') {
|
|
95
|
+
return value.bind(target);
|
|
96
|
+
}
|
|
97
|
+
return value;
|
|
98
|
+
},
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
//# sourceMappingURL=google.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.js","sourceRoot":"","sources":["../../../src/providers/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,UAAU,EAAE,YAAY,EAA6B,MAAM,WAAW,CAAC;AAChF,OAAO,KAAK,SAAS,MAAM,cAAc,CAAC;AAE1C;;;GAGG;AACH,SAAS,aAAa,CAAC,QAAa;IAQlC,MAAM,WAAW,GACf,QAAQ,EAAE,YAAY,IAAI,QAAQ,EAAE,WAAW;QAC/C,QAAQ,EAAE,gBAAgB,IAAI,QAAQ,EAAE,kBAAkB,IAAI,CAAC,CAAC;IAClE,MAAM,YAAY,GAChB,QAAQ,EAAE,aAAa,IAAI,QAAQ,EAAE,YAAY;QACjD,QAAQ,EAAE,oBAAoB,IAAI,QAAQ,EAAE,sBAAsB,IAAI,CAAC,CAAC;IAC1E,MAAM,YAAY,GAChB,QAAQ,EAAE,0BAA0B,IAAI,QAAQ,EAAE,uBAAuB,IAAI,CAAC,CAAC;IACjF,MAAM,cAAc,GAClB,QAAQ,EAAE,oBAAoB,IAAI,QAAQ,EAAE,kBAAkB,IAAI,CAAC,CAAC;IACtE,MAAM,aAAa,GACjB,QAAQ,EAAE,2BAA2B,IAAI,QAAQ,EAAE,uBAAuB,IAAI,CAAC,CAAC;IAClF,MAAM,aAAa,GACjB,QAAQ,EAAE,iBAAiB,IAAI,QAAQ,EAAE,eAAe,IAAI,CAAC,CAAC;IAEhE,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC;AACnG,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAW;IAC7C,yEAAyE;IACzE,MAAM,QAAQ,GAAG,MAAM,EAAE,aAAa,IAAI,MAAM,EAAE,cAAc,CAAC;IACjE,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,GAC7F,aAAa,CAAC,QAAQ,CAAC,CAAC;IAE1B,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO,CAAC,GAAG,CACT,uCAAuC,WAAW,WAAW,YAAY,UAAU,aAAa,aAAa,YAAY,aAAa,cAAc,mBAAmB,aAAa,EAAE,CACvL,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;SACjE,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;SAClC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC;SACpC,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO;QACL,QAAQ,EAAE,QAAQ;QAClB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,MAAM,EAAE,YAAY,IAAI,SAAS;QACzD,YAAY,EAAE,WAAW,GAAG,YAAY,GAAG,aAAa;QACxD,aAAa,EAAE,YAAY,GAAG,cAAc;QAC5C,MAAM,EAAE,SAAS;QACjB,aAAa,EAAE,YAAY;KAC5B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAW,EAAE,OAAoB,EAAE,SAAiB,EAAE,IAAW;IAChG,IAAI,KAAK,GAAG,SAAS,CAAC;IACtB,IAAI,YAAY,GAAQ,IAAI,CAAC;IAE7B,SAAS,MAAM;QACb,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,GAC7F,aAAa,CAAC,YAAY,CAAC,CAAC;QAE9B,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CACT,uCAAuC,WAAW,WAAW,YAAY,UAAU,aAAa,aAAa,YAAY,aAAa,cAAc,mBAAmB,aAAa,EAAE,CACvL,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE;YAChC,QAAQ,EAAE,QAAQ;YAClB,KAAK;YACL,YAAY,EAAE,WAAW,GAAG,YAAY,GAAG,aAAa;YACxD,aAAa,EAAE,YAAY,GAAG,cAAc;YAC5C,MAAM,EAAE,SAAS;SAClB,EAAE,SAAS,CAAC,CAAC;QACd,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,MAAM,CAAC,KAAK;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;QACvB,GAAG,CAAC,MAAM,EAAE,IAAI;YACd,IAAI,IAAI,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;gBAClC,OAAO;oBACL,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;oBACxD,OAAO;wBACL,KAAK,CAAC,IAAI;4BACR,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;4BACjD,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gCACrB,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;gCAC/B,wCAAwC;gCACxC,IAAI,KAAK,EAAE,KAAK,IAAI,KAAK,EAAE,YAAY,EAAE,CAAC;oCACxC,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC;gCACrD,CAAC;gCACD,wEAAwE;gCACxE,MAAM,QAAQ,GAAG,KAAK,EAAE,aAAa,IAAI,KAAK,EAAE,cAAc,CAAC;gCAC/D,IAAI,QAAQ,EAAE,CAAC;oCACb,YAAY,GAAG,QAAQ,CAAC;gCAC1B,CAAC;4BACH,CAAC;4BACD,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gCACpB,IAAI,CAAC;oCACH,MAAM,EAAE,CAAC;gCACX,CAAC;gCAAC,MAAM,CAAC;oCACP,UAAU;gCACZ,CAAC;4BACH,CAAC;4BACD,OAAO,UAAU,CAAC;wBACpB,CAAC;wBACD,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC;wBACvD,KAAK,EAAE,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC;qBACtD,CAAC;gBACJ,CAAC,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;gBAChC,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC"}
|
|
@@ -1,4 +1,9 @@
|
|
|
1
1
|
import { ContextData, ResponseData } from '../schema';
|
|
2
|
-
export declare function parseOpenAIResponse(result: any): ResponseData;
|
|
2
|
+
export declare function parseOpenAIResponse(result: any, args?: any[]): ResponseData;
|
|
3
|
+
export declare function parseTranscriptionResponse(result: any, args?: any[]): ResponseData;
|
|
4
|
+
export declare function parseSpeechResponse(result: any, args?: any[]): ResponseData;
|
|
5
|
+
export declare function parseEmbeddingResponse(result: any, args?: any[]): ResponseData;
|
|
6
|
+
export declare function parseImageResponse(result: any, args?: any[]): ResponseData;
|
|
7
|
+
export declare function parseModerationResponse(result: any, args?: any[]): ResponseData;
|
|
3
8
|
export declare function wrapOpenAIStream(result: any, context: ContextData, elapsedMs: number, args: any[]): any;
|
|
4
9
|
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,WAAW,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGhF,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,GAAG,YAAY,
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,WAAW,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGhF,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,YAAY,CAyB3E;AAID,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,YAAY,CAelF;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,YAAY,CAe3E;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,YAAY,CAkB9E;AAED,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,YAAY,CAe1E;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,YAAY,CAc/E;AAGD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,GAAG,CA+EvG"}
|
|
@@ -1,15 +1,115 @@
|
|
|
1
1
|
import { config } from '../config';
|
|
2
2
|
import { buildEvent, printCapture } from '../schema';
|
|
3
3
|
import * as transport from '../transport';
|
|
4
|
-
export function parseOpenAIResponse(result) {
|
|
4
|
+
export function parseOpenAIResponse(result, args) {
|
|
5
|
+
const usage = result?.usage;
|
|
6
|
+
const inputTokens = usage?.prompt_tokens || 0;
|
|
7
|
+
const outputTokens = usage?.completion_tokens || 0;
|
|
8
|
+
if (config.debug) {
|
|
9
|
+
const reasoning = usage?.completion_tokens_details?.reasoning_tokens || 0;
|
|
10
|
+
const cached = usage?.prompt_tokens_details?.cached_tokens || 0;
|
|
11
|
+
console.log(`[llmtracer] OpenAI tokens — in=${inputTokens} out=${outputTokens} reasoning=${reasoning} cached=${cached}`);
|
|
12
|
+
}
|
|
13
|
+
const toolsInvoked = (result?.choices?.[0]?.message?.tool_calls || [])
|
|
14
|
+
.map((tc) => tc.function?.name)
|
|
15
|
+
.filter(Boolean);
|
|
5
16
|
return {
|
|
6
17
|
provider: 'openai',
|
|
7
18
|
model: result?.model || 'unknown',
|
|
8
|
-
input_tokens:
|
|
9
|
-
output_tokens:
|
|
19
|
+
input_tokens: inputTokens,
|
|
20
|
+
output_tokens: outputTokens,
|
|
10
21
|
status: 'success',
|
|
22
|
+
tools_invoked: toolsInvoked,
|
|
11
23
|
};
|
|
12
24
|
}
|
|
25
|
+
// ── Multi-modal response parsers ─────────────────────────────────────────
|
|
26
|
+
export function parseTranscriptionResponse(result, args) {
|
|
27
|
+
try {
|
|
28
|
+
const kwargs = args?.[0] || {};
|
|
29
|
+
return {
|
|
30
|
+
provider: 'openai',
|
|
31
|
+
model: kwargs.model || 'whisper-1',
|
|
32
|
+
input_tokens: 0,
|
|
33
|
+
output_tokens: 0,
|
|
34
|
+
status: 'success',
|
|
35
|
+
event_type: 'speech_to_text',
|
|
36
|
+
usage: { audio_duration_s: result?.duration ?? null },
|
|
37
|
+
};
|
|
38
|
+
}
|
|
39
|
+
catch {
|
|
40
|
+
return { provider: 'openai', model: 'whisper-1', input_tokens: 0, output_tokens: 0, status: 'success', event_type: 'speech_to_text' };
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
export function parseSpeechResponse(result, args) {
|
|
44
|
+
try {
|
|
45
|
+
const kwargs = args?.[0] || {};
|
|
46
|
+
return {
|
|
47
|
+
provider: 'openai',
|
|
48
|
+
model: kwargs.model || 'tts-1',
|
|
49
|
+
input_tokens: 0,
|
|
50
|
+
output_tokens: 0,
|
|
51
|
+
status: 'success',
|
|
52
|
+
event_type: 'text_to_speech',
|
|
53
|
+
usage: { character_count: (kwargs.input || '').length, voice: kwargs.voice || 'unknown' },
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
catch {
|
|
57
|
+
return { provider: 'openai', model: 'tts-1', input_tokens: 0, output_tokens: 0, status: 'success', event_type: 'text_to_speech' };
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
export function parseEmbeddingResponse(result, args) {
|
|
61
|
+
try {
|
|
62
|
+
const kwargs = args?.[0] || {};
|
|
63
|
+
const model = result?.model || kwargs.model || 'unknown';
|
|
64
|
+
const inputTokens = result?.usage?.prompt_tokens || 0;
|
|
65
|
+
const dimensions = result?.data?.[0]?.embedding?.length ?? null;
|
|
66
|
+
return {
|
|
67
|
+
provider: 'openai',
|
|
68
|
+
model,
|
|
69
|
+
input_tokens: inputTokens,
|
|
70
|
+
output_tokens: 0,
|
|
71
|
+
status: 'success',
|
|
72
|
+
event_type: 'embedding',
|
|
73
|
+
usage: { dimensions },
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
catch {
|
|
77
|
+
return { provider: 'openai', model: 'unknown', input_tokens: 0, output_tokens: 0, status: 'success', event_type: 'embedding' };
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
export function parseImageResponse(result, args) {
|
|
81
|
+
try {
|
|
82
|
+
const kwargs = args?.[0] || {};
|
|
83
|
+
return {
|
|
84
|
+
provider: 'openai',
|
|
85
|
+
model: kwargs.model || 'dall-e-3',
|
|
86
|
+
input_tokens: 0,
|
|
87
|
+
output_tokens: 0,
|
|
88
|
+
status: 'success',
|
|
89
|
+
event_type: 'image_generation',
|
|
90
|
+
usage: { image_count: result?.data?.length || 0, size: kwargs.size ?? null, quality: kwargs.quality ?? null },
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
catch {
|
|
94
|
+
return { provider: 'openai', model: 'dall-e-3', input_tokens: 0, output_tokens: 0, status: 'success', event_type: 'image_generation' };
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
export function parseModerationResponse(result, args) {
|
|
98
|
+
try {
|
|
99
|
+
const kwargs = args?.[0] || {};
|
|
100
|
+
return {
|
|
101
|
+
provider: 'openai',
|
|
102
|
+
model: result?.model || kwargs.model || 'text-moderation-latest',
|
|
103
|
+
input_tokens: 0,
|
|
104
|
+
output_tokens: 0,
|
|
105
|
+
status: 'success',
|
|
106
|
+
event_type: 'moderation',
|
|
107
|
+
};
|
|
108
|
+
}
|
|
109
|
+
catch {
|
|
110
|
+
return { provider: 'openai', model: 'text-moderation-latest', input_tokens: 0, output_tokens: 0, status: 'success', event_type: 'moderation' };
|
|
111
|
+
}
|
|
112
|
+
}
|
|
13
113
|
export function wrapOpenAIStream(result, context, elapsedMs, args) {
|
|
14
114
|
const chunks = [];
|
|
15
115
|
function record() {
|
|
@@ -22,12 +122,29 @@ export function wrapOpenAIStream(result, context, elapsedMs, args) {
|
|
|
22
122
|
}
|
|
23
123
|
}
|
|
24
124
|
const model = chunks[0]?.model || 'unknown';
|
|
125
|
+
const inputTokens = usage?.prompt_tokens || 0;
|
|
126
|
+
const outputTokens = usage?.completion_tokens || 0;
|
|
127
|
+
if (config.debug) {
|
|
128
|
+
const reasoning = usage?.completion_tokens_details?.reasoning_tokens || 0;
|
|
129
|
+
const cached = usage?.prompt_tokens_details?.cached_tokens || 0;
|
|
130
|
+
console.log(`[llmtracer] OpenAI tokens — in=${inputTokens} out=${outputTokens} reasoning=${reasoning} cached=${cached}`);
|
|
131
|
+
}
|
|
132
|
+
// Extract tool names from accumulated stream chunks
|
|
133
|
+
const toolsInvoked = [];
|
|
134
|
+
for (const chunk of chunks) {
|
|
135
|
+
const toolCalls = chunk?.choices?.[0]?.delta?.tool_calls || [];
|
|
136
|
+
for (const tc of toolCalls) {
|
|
137
|
+
if (tc.function?.name)
|
|
138
|
+
toolsInvoked.push(tc.function.name);
|
|
139
|
+
}
|
|
140
|
+
}
|
|
25
141
|
const event = buildEvent(context, {
|
|
26
142
|
provider: 'openai',
|
|
27
143
|
model,
|
|
28
|
-
input_tokens:
|
|
29
|
-
output_tokens:
|
|
144
|
+
input_tokens: inputTokens,
|
|
145
|
+
output_tokens: outputTokens,
|
|
30
146
|
status: 'success',
|
|
147
|
+
tools_invoked: toolsInvoked,
|
|
31
148
|
}, elapsedMs);
|
|
32
149
|
transport.enqueue(event);
|
|
33
150
|
if (config.debug)
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,UAAU,EAAE,YAAY,EAA6B,MAAM,WAAW,CAAC;AAChF,OAAO,KAAK,SAAS,MAAM,cAAc,CAAC;AAE1C,MAAM,UAAU,mBAAmB,CAAC,MAAW;
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,UAAU,EAAE,YAAY,EAA6B,MAAM,WAAW,CAAC;AAChF,OAAO,KAAK,SAAS,MAAM,cAAc,CAAC;AAE1C,MAAM,UAAU,mBAAmB,CAAC,MAAW,EAAE,IAAY;IAC3D,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;IAC5B,MAAM,WAAW,GAAG,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;IAC9C,MAAM,YAAY,GAAG,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;IAEnD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,MAAM,SAAS,GAAG,KAAK,EAAE,yBAAyB,EAAE,gBAAgB,IAAI,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG,KAAK,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC,CAAC;QAChE,OAAO,CAAC,GAAG,CACT,kCAAkC,WAAW,QAAQ,YAAY,cAAc,SAAS,WAAW,MAAM,EAAE,CAC5G,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;SACnE,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC;SACnC,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO;QACL,QAAQ,EAAE,QAAQ;QAClB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,SAAS;QACjC,YAAY,EAAE,WAAW;QACzB,aAAa,EAAE,YAAY;QAC3B,MAAM,EAAE,SAAS;QACjB,aAAa,EAAE,YAAY;KAC5B,CAAC;AACJ,CAAC;AAED,4EAA4E;AAE5E,MAAM,UAAU,0BAA0B,CAAC,MAAW,EAAE,IAAY;IAClE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,WAAW;YAClC,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,gBAAgB;YAC5B,KAAK,EAAE,EAAE,gBAAgB,EAAE,MAAM,EAAE,QAAQ,IAAI,IAAI,EAAE;SACtD,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IACxI,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAW,EAAE,IAAY;IAC3D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,OAAO;YAC9B,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,gBAAgB;YAC5B,KAAK,EAAE,EAAE,eAAe,EAAE,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,SAAS,EAAE;SAC1F,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IACpI,CAAC;AACH,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAW,EAAE,IAAY;IAC9D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,SAAS,CAAC;QACzD,MAAM,WAAW,GAAG,MAAM,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC;QAChE,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK;YACL,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,CAAC;YAChB,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,WAAW;YACvB,KAAK,EAAE,EAAE,UAAU,EAAE;SACtB,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC;IACjI,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAW,EAAE,IAAY;IAC1D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,UAAU;YACjC,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,kBAAkB;YAC9B,KAAK,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI,EAAE;SAC9G,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,kBAAkB,EAAE,CAAC;IACzI,CAAC;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAAW,EAAE,IAAY;IAC/D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK,IAAI,wBAAwB;YAChE,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,YAAY;SACzB,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,wBAAwB,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC;IACjJ,CAAC;AACH,CAAC;AAGD,MAAM,UAAU,gBAAgB,CAAC,MAAW,EAAE,OAAoB,EAAE,SAAiB,EAAE,IAAW;IAChG,MAAM,MAAM,GAAU,EAAE,CAAC;IAEzB,SAAS,MAAM;QACb,+FAA+F;QAC/F,IAAI,KAAK,GAAQ,IAAI,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;gBACrB,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBACxB,MAAM;YACR,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,SAAS,CAAC;QAC5C,MAAM,WAAW,GAAG,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;QAC9C,MAAM,YAAY,GAAG,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;QAEnD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,KAAK,EAAE,yBAAyB,EAAE,gBAAgB,IAAI,CAAC,CAAC;YAC1E,MAAM,MAAM,GAAG,KAAK,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC,CAAC;YAChE,OAAO,CAAC,GAAG,CACT,kCAAkC,WAAW,QAAQ,YAAY,cAAc,SAAS,WAAW,MAAM,EAAE,CAC5G,CAAC;QACJ,CAAC;QAED,oDAAoD;QACpD,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,IAAI,EAAE,CAAC;YAC/D,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;gBAC3B,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;oBAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE;YAChC,QAAQ,EAAE,QAAQ;YAClB,KAAK;YACL,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,YAAY;YAC3B,MAAM,EAAE,SAAS;YACjB,aAAa,EAAE,YAAY;SAC5B,EAAE,SAAS,CAAC,CAAC;QACd,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,MAAM,CAAC,KAAK;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,gFAAgF;IAChF,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;QACvB,GAAG,CAAC,MAAM,EAAE,IAAI;YACd,IAAI,IAAI,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;gBAClC,OAAO;oBACL,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;oBACxD,OAAO;wBACL,KAAK,CAAC,IAAI;4BACR,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,CAAC;4BACjD,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gCACrB,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;4BAChC,CAAC;4BACD,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gCACpB,IAAI,CAAC;oCACH,MAAM,EAAE,CAAC;gCACX,CAAC;gCAAC,MAAM,CAAC;oCACP,UAAU;gCACZ,CAAC;4BACH,CAAC;4BACD,OAAO,UAAU,CAAC;wBACpB,CAAC;wBACD,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC;wBACvD,KAAK,EAAE,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC;qBACtD,CAAC;gBACJ,CAAC,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;gBAChC,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
package/dist/esm/schema.d.ts
CHANGED
|
@@ -11,6 +11,9 @@ export interface ResponseData {
|
|
|
11
11
|
input_tokens: number;
|
|
12
12
|
output_tokens: number;
|
|
13
13
|
status: string;
|
|
14
|
+
tools_invoked?: string[];
|
|
15
|
+
event_type?: string;
|
|
16
|
+
usage?: Record<string, any>;
|
|
14
17
|
}
|
|
15
18
|
export declare function buildContext(provider: string, args: any[]): ContextData;
|
|
16
19
|
export declare function buildEvent(contextData: ContextData, responseData: ResponseData, latencyMs: number): Record<string, any>;
|
package/dist/esm/schema.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/schema.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAc,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;
|
|
1
|
+
{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/schema.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAc,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAMzC,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,YAAY,GAAG,SAAS,CAAC;IAClC,MAAM,EAAE,UAAU,CAAC,OAAO,aAAa,CAAC,CAAC;IACzC,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC7B;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,WAAW,CAMvE;AAED,wBAAgB,UAAU,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAiCvH;AAwCD,wBAAgB,YAAY,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CA8B7D"}
|
package/dist/esm/schema.js
CHANGED
|
@@ -1,7 +1,8 @@
|
|
|
1
1
|
import { randomUUID } from 'crypto';
|
|
2
2
|
import { getCurrent } from './context';
|
|
3
3
|
import { getCallerInfo } from './caller';
|
|
4
|
-
|
|
4
|
+
import { config } from './config';
|
|
5
|
+
const SDK_VERSION = '2.4.0';
|
|
5
6
|
const SDK_LANGUAGE = 'typescript';
|
|
6
7
|
export function buildContext(provider, args) {
|
|
7
8
|
return {
|
|
@@ -19,7 +20,15 @@ export function buildEvent(contextData, responseData, latencyMs) {
|
|
|
19
20
|
sdk_language: SDK_LANGUAGE,
|
|
20
21
|
...responseData,
|
|
21
22
|
...contextData.caller,
|
|
23
|
+
event_type: responseData.event_type || 'chat_completion',
|
|
24
|
+
usage: responseData.usage || null,
|
|
22
25
|
};
|
|
26
|
+
if (config.sessionId) {
|
|
27
|
+
event.session_id = config.sessionId;
|
|
28
|
+
}
|
|
29
|
+
if (config.appName) {
|
|
30
|
+
event.app_name = config.appName;
|
|
31
|
+
}
|
|
23
32
|
if (contextData.context) {
|
|
24
33
|
event.trace_id = contextData.context.traceId;
|
|
25
34
|
event.span_id = contextData.context.spanId;
|
|
@@ -48,6 +57,11 @@ const COST_PER_M = {
|
|
|
48
57
|
'claude-sonnet-4-5': [3, 15],
|
|
49
58
|
'claude-haiku-4-5': [0.8, 4],
|
|
50
59
|
'claude-opus-4-6': [15, 75],
|
|
60
|
+
'gemini-2.5-pro': [1.25, 10],
|
|
61
|
+
'gemini-2.5-flash': [0.15, 0.6],
|
|
62
|
+
'gemini-2.0-flash': [0.1, 0.4],
|
|
63
|
+
'gemini-1.5-pro': [1.25, 5],
|
|
64
|
+
'gemini-1.5-flash': [0.075, 0.3],
|
|
51
65
|
};
|
|
52
66
|
function estimateCost(model, inputTokens, outputTokens) {
|
|
53
67
|
// Try exact match first, then prefix match
|
|
@@ -83,6 +97,10 @@ export function printCapture(event) {
|
|
|
83
97
|
traceInfo = ` trace:${firstValue}`;
|
|
84
98
|
}
|
|
85
99
|
const costStr = cost > 0 ? ` $${cost.toFixed(4)}` : '';
|
|
86
|
-
|
|
100
|
+
let toolsInfo = '';
|
|
101
|
+
if (event.tools_invoked && event.tools_invoked.length > 0) {
|
|
102
|
+
toolsInfo = ` tools:${event.tools_invoked.join(',')}`;
|
|
103
|
+
}
|
|
104
|
+
console.log(`[llmtracer] ${event.provider}/${model} ${input}\u2192${output} tokens${costStr} ${latency}s (${file})${traceInfo}${toolsInfo}`);
|
|
87
105
|
}
|
|
88
106
|
//# sourceMappingURL=schema.js.map
|
package/dist/esm/schema.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAgB,UAAU,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;
|
|
1
|
+
{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAgB,UAAU,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,MAAM,WAAW,GAAG,OAAO,CAAC;AAC5B,MAAM,YAAY,GAAG,YAAY,CAAC;AAmBlC,MAAM,UAAU,YAAY,CAAC,QAAgB,EAAE,IAAW;IACxD,OAAO;QACL,OAAO,EAAE,UAAU,EAAE;QACrB,MAAM,EAAE,aAAa,EAAE;QACvB,QAAQ;KACT,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,WAAwB,EAAE,YAA0B,EAAE,SAAiB;IAChG,MAAM,KAAK,GAAwB;QACjC,QAAQ,EAAE,UAAU,EAAE;QACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,UAAU,EAAE,SAAS;QACrB,WAAW,EAAE,WAAW;QACxB,YAAY,EAAE,YAAY;QAC1B,GAAG,YAAY;QACf,GAAG,WAAW,CAAC,MAAM;QACrB,UAAU,EAAE,YAAY,CAAC,UAAU,IAAI,iBAAiB;QACxD,KAAK,EAAE,YAAY,CAAC,KAAK,IAAI,IAAI;KAClC,CAAC;IAEF,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACrB,KAAK,CAAC,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC;IACtC,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC;IAClC,CAAC;IAED,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;QACxB,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC;QAC7C,KAAK,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;QAC3C,IAAI,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YACrC,KAAK,CAAC,cAAc,GAAG,WAAW,CAAC,OAAO,CAAC,YAAY,CAAC;QAC1D,CAAC;QACD,IAAI,WAAW,CAAC,OAAO,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjF,KAAK,CAAC,IAAI,GAAG,EAAE,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,GAAqC;IACnD,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC;IACnB,aAAa,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC;IAC1B,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;IACvB,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;IACjB,eAAe,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IAC3B,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;IACd,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;IAClB,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IACrB,mBAAmB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;IAC5B,kBAAkB,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;IAC5B,eAAe,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;IACzB,mBAAmB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;IAC5B,kBAAkB,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;IAC5B,iBAAiB,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;IAC3B,gBAAgB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;IAC5B,kBAAkB,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC;IAC/B,kBAAkB,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IAC9B,gBAAgB,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3B,kBAAkB,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC;CACjC,CAAC;AAEF,SAAS,YAAY,CAAC,KAAa,EAAE,WAAmB,EAAE,YAAoB;IAC5E,2CAA2C;IAC3C,IAAI,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1C,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC1B,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,CAAC;IACrB,OAAO,CAAC,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,OAAS,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAA0B;IACrD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,SAAS,CAAC;IACvC,MAAM,KAAK,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;IACzD,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;IAC3D,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC;IACpF,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAE5D,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACtB,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC/D,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACzC,IAAI,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,QAAQ,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC3E,CAAC;IAED,IAAI,SAAS,GAAG,EAAE,CAAC;IACnB,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAChD,IAAI,UAAU;YAAE,SAAS,GAAG,WAAW,UAAU,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAExD,IAAI,SAAS,GAAG,EAAE,CAAC;IACnB,IAAI,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1D,SAAS,GAAG,WAAW,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;IACzD,CAAC;IAED,OAAO,CAAC,GAAG,CACT,eAAe,KAAK,CAAC,QAAQ,IAAI,KAAK,KAAK,KAAK,SAAS,MAAM,UAAU,OAAO,KAAK,OAAO,OAAO,IAAI,IAAI,SAAS,GAAG,SAAS,EAAE,CACnI,CAAC;AACJ,CAAC"}
|
package/dist/esm/transport.d.ts
CHANGED
|
@@ -3,4 +3,16 @@ export declare function enqueue(event: Record<string, any>): void;
|
|
|
3
3
|
export declare function flush(): Promise<void>;
|
|
4
4
|
export declare function stop(): void;
|
|
5
5
|
export declare function getQueueLength(): number;
|
|
6
|
+
declare function exitFlush(): void;
|
|
7
|
+
declare function registerExitHandlers(): void;
|
|
8
|
+
export declare const _testExports: {
|
|
9
|
+
exitFlush: typeof exitFlush;
|
|
10
|
+
registerExitHandlers: typeof registerExitHandlers;
|
|
11
|
+
readonly exitHandlersRegistered: boolean;
|
|
12
|
+
resetExitFlag(): void;
|
|
13
|
+
readonly retryAttempt: number;
|
|
14
|
+
readonly backoffUntil: number;
|
|
15
|
+
resetRetryState(): void;
|
|
16
|
+
};
|
|
17
|
+
export {};
|
|
6
18
|
//# sourceMappingURL=transport.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/transport.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/transport.ts"],"names":[],"mappings":"AAWA,wBAAgB,KAAK,IAAI,IAAI,CAQ5B;AAED,wBAAgB,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAiBxD;AAED,wBAAsB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAE3C;AAED,wBAAgB,IAAI,IAAI,IAAI,CAS3B;AAED,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AA8DD,iBAAS,SAAS,IAAI,IAAI,CAoBzB;AAED,iBAAS,oBAAoB,IAAI,IAAI,CAqBpC;AAGD,eAAO,MAAM,YAAY;;;;;;;;CAQxB,CAAC"}
|
package/dist/esm/transport.js
CHANGED
|
@@ -2,18 +2,34 @@ import { config } from './config';
|
|
|
2
2
|
let eventQueue = [];
|
|
3
3
|
let flushTimer = null;
|
|
4
4
|
let firstBatch = true;
|
|
5
|
+
let exitHandlersRegistered = false;
|
|
6
|
+
// Retry state: tracks consecutive failures and next retry time
|
|
7
|
+
let retryAttempt = 0;
|
|
8
|
+
let backoffUntil = 0;
|
|
5
9
|
export function start() {
|
|
6
10
|
// Reset state for fresh start
|
|
7
11
|
firstBatch = true;
|
|
12
|
+
retryAttempt = 0;
|
|
13
|
+
backoffUntil = 0;
|
|
14
|
+
if (!config.skipExitHandlers) {
|
|
15
|
+
registerExitHandlers();
|
|
16
|
+
}
|
|
8
17
|
}
|
|
9
18
|
export function enqueue(event) {
|
|
19
|
+
// sampleRate check: skip event if it doesn't pass the sampling threshold
|
|
20
|
+
if (config.sampleRate < 1.0) {
|
|
21
|
+
if (Math.random() >= config.sampleRate)
|
|
22
|
+
return;
|
|
23
|
+
}
|
|
24
|
+
if (eventQueue.length >= config.maxQueueSize)
|
|
25
|
+
eventQueue.shift();
|
|
10
26
|
eventQueue.push(event);
|
|
11
27
|
if (firstBatch) {
|
|
12
28
|
firstBatch = false;
|
|
13
29
|
// Flush immediately on first event
|
|
14
30
|
setImmediate(() => doFlush());
|
|
15
|
-
// Start regular
|
|
16
|
-
flushTimer = setInterval(() => doFlush(),
|
|
31
|
+
// Start regular interval using configured flushIntervalMs
|
|
32
|
+
flushTimer = setInterval(() => doFlush(), config.flushIntervalMs);
|
|
17
33
|
flushTimer.unref();
|
|
18
34
|
}
|
|
19
35
|
}
|
|
@@ -27,16 +43,23 @@ export function stop() {
|
|
|
27
43
|
}
|
|
28
44
|
eventQueue = [];
|
|
29
45
|
firstBatch = true;
|
|
46
|
+
retryAttempt = 0;
|
|
47
|
+
backoffUntil = 0;
|
|
30
48
|
}
|
|
31
49
|
export function getQueueLength() {
|
|
32
50
|
return eventQueue.length;
|
|
33
51
|
}
|
|
34
|
-
async function doFlush() {
|
|
52
|
+
async function doFlush(timeoutMs = 10000) {
|
|
35
53
|
if (eventQueue.length === 0)
|
|
36
54
|
return;
|
|
37
|
-
|
|
55
|
+
// If we're in backoff, skip this flush cycle
|
|
56
|
+
if (backoffUntil > 0 && Date.now() < backoffUntil)
|
|
57
|
+
return;
|
|
58
|
+
// Take up to maxBatchSize events from the queue
|
|
59
|
+
const batchSize = Math.min(eventQueue.length, config.maxBatchSize);
|
|
60
|
+
const events = eventQueue.splice(0, batchSize);
|
|
38
61
|
try {
|
|
39
|
-
const payload = JSON.stringify({ events });
|
|
62
|
+
const payload = JSON.stringify({ schema_version: '1.0', events });
|
|
40
63
|
const response = await fetch(config.endpoint, {
|
|
41
64
|
method: 'POST',
|
|
42
65
|
headers: {
|
|
@@ -44,19 +67,101 @@ async function doFlush() {
|
|
|
44
67
|
'Authorization': `Bearer ${config.apiKey}`,
|
|
45
68
|
},
|
|
46
69
|
body: payload,
|
|
47
|
-
signal: AbortSignal.timeout(
|
|
70
|
+
signal: AbortSignal.timeout(timeoutMs),
|
|
48
71
|
});
|
|
49
72
|
if (config.debug) {
|
|
50
73
|
console.log(`[llmtracer] Flushed ${events.length} events \u2192 ${response.status}`);
|
|
51
74
|
}
|
|
75
|
+
// Success — reset retry state
|
|
76
|
+
retryAttempt = 0;
|
|
77
|
+
backoffUntil = 0;
|
|
52
78
|
}
|
|
53
79
|
catch (e) {
|
|
54
80
|
if (config.debug) {
|
|
55
81
|
console.log(`[llmtracer] Flush failed: ${e}`);
|
|
56
82
|
}
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
83
|
+
if (retryAttempt < config.maxRetries) {
|
|
84
|
+
// Re-queue events for retry (prepend so they go out first)
|
|
85
|
+
const requeue = events.slice(0, config.maxBatchSize);
|
|
86
|
+
eventQueue.unshift(...requeue);
|
|
87
|
+
// Set exponential backoff: delay = min(1000 * 2^attempt, 30000) + jitter(0-1000ms)
|
|
88
|
+
retryAttempt++;
|
|
89
|
+
const baseDelay = Math.min(1000 * Math.pow(2, retryAttempt), 30000);
|
|
90
|
+
const jitter = Math.floor(Math.random() * 1000);
|
|
91
|
+
backoffUntil = Date.now() + baseDelay + jitter;
|
|
92
|
+
if (config.debug) {
|
|
93
|
+
console.log(`[llmtracer] Retry ${retryAttempt}/${config.maxRetries} in ${baseDelay + jitter}ms`);
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
else {
|
|
97
|
+
// Max retries exceeded — drop the batch
|
|
98
|
+
if (config.debug) {
|
|
99
|
+
console.log(`[llmtracer] Dropping ${events.length} events after ${config.maxRetries} retries`);
|
|
100
|
+
}
|
|
101
|
+
retryAttempt = 0;
|
|
102
|
+
backoffUntil = 0;
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
function exitFlush() {
|
|
107
|
+
try {
|
|
108
|
+
if (eventQueue.length === 0)
|
|
109
|
+
return;
|
|
110
|
+
const events = eventQueue.splice(0);
|
|
111
|
+
const payload = JSON.stringify({ schema_version: '1.0', events });
|
|
112
|
+
// Best-effort flush with short timeout — never log, never throw
|
|
113
|
+
fetch(config.endpoint, {
|
|
114
|
+
method: 'POST',
|
|
115
|
+
headers: {
|
|
116
|
+
'Content-Type': 'application/json',
|
|
117
|
+
'Authorization': `Bearer ${config.apiKey}`,
|
|
118
|
+
},
|
|
119
|
+
body: payload,
|
|
120
|
+
signal: AbortSignal.timeout(3000),
|
|
121
|
+
}).catch(() => { });
|
|
122
|
+
}
|
|
123
|
+
catch {
|
|
124
|
+
// crash-proof: swallow everything
|
|
125
|
+
}
|
|
126
|
+
}
|
|
127
|
+
function registerExitHandlers() {
|
|
128
|
+
if (exitHandlersRegistered)
|
|
129
|
+
return;
|
|
130
|
+
exitHandlersRegistered = true;
|
|
131
|
+
try {
|
|
132
|
+
process.on('beforeExit', () => {
|
|
133
|
+
try {
|
|
134
|
+
exitFlush();
|
|
135
|
+
}
|
|
136
|
+
catch { /* swallow */ }
|
|
137
|
+
});
|
|
138
|
+
process.on('SIGINT', () => {
|
|
139
|
+
try {
|
|
140
|
+
exitFlush();
|
|
141
|
+
}
|
|
142
|
+
catch { /* swallow */ }
|
|
143
|
+
process.exit(0);
|
|
144
|
+
});
|
|
145
|
+
process.on('SIGTERM', () => {
|
|
146
|
+
try {
|
|
147
|
+
exitFlush();
|
|
148
|
+
}
|
|
149
|
+
catch { /* swallow */ }
|
|
150
|
+
process.exit(0);
|
|
151
|
+
});
|
|
152
|
+
}
|
|
153
|
+
catch {
|
|
154
|
+
// crash-proof: if process.on fails, just skip
|
|
60
155
|
}
|
|
61
156
|
}
|
|
157
|
+
// Export for testing only
|
|
158
|
+
export const _testExports = {
|
|
159
|
+
exitFlush,
|
|
160
|
+
registerExitHandlers,
|
|
161
|
+
get exitHandlersRegistered() { return exitHandlersRegistered; },
|
|
162
|
+
resetExitFlag() { exitHandlersRegistered = false; },
|
|
163
|
+
get retryAttempt() { return retryAttempt; },
|
|
164
|
+
get backoffUntil() { return backoffUntil; },
|
|
165
|
+
resetRetryState() { retryAttempt = 0; backoffUntil = 0; },
|
|
166
|
+
};
|
|
62
167
|
//# sourceMappingURL=transport.js.map
|