@traccia2/sdk 0.0.7 → 0.0.8
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/dist/exporter/http-exporter.d.ts +8 -2
- package/dist/exporter/http-exporter.d.ts.map +1 -1
- package/dist/exporter/http-exporter.js +51 -12
- package/dist/exporter/http-exporter.js.map +1 -1
- package/dist/integrations/index.d.ts +1 -1
- package/dist/integrations/index.js +3 -3
- package/dist/integrations/langchain-callback.d.ts +57 -91
- package/dist/integrations/langchain-callback.d.ts.map +1 -1
- package/dist/integrations/langchain-callback.js +464 -316
- package/dist/integrations/langchain-callback.js.map +1 -1
- package/dist/integrations/langchain-callback.old.d.ts +96 -0
- package/dist/integrations/langchain-callback.old.d.ts.map +1 -0
- package/dist/integrations/langchain-callback.old.js +371 -0
- package/dist/integrations/langchain-callback.old.js.map +1 -0
- package/dist/tracer/provider.d.ts +3 -3
- package/dist/tracer/provider.d.ts.map +1 -1
- package/dist/tracer/provider.js +9 -0
- package/dist/tracer/provider.js.map +1 -1
- package/dist/types.d.ts +4 -0
- package/dist/types.d.ts.map +1 -1
- package/package.json +1 -1
- package/src/__tests__/integrations-langchain.test.ts +354 -340
- package/src/exporter/http-exporter.ts +57 -13
- package/src/integrations/index.ts +1 -1
- package/src/integrations/langchain-callback.old.ts +438 -0
- package/src/integrations/langchain-callback.ts +723 -351
- package/src/tracer/provider.ts +9 -4
- package/src/types.ts +4 -0
- package/dist/integrations/langchain-callback.new.d.ts +0 -62
- package/dist/integrations/langchain-callback.new.d.ts.map +0 -1
- package/dist/integrations/langchain-callback.new.js +0 -519
- package/dist/integrations/langchain-callback.new.js.map +0 -1
- package/src/integrations/langchain-callback.new.ts +0 -810
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* HTTP Exporter for sending spans to a backend service.
|
|
3
3
|
*/
|
|
4
|
-
import { ISpan, ISpanExporter } from '../types';
|
|
4
|
+
import { ISpan, ISpanExporter, Resource } from '../types';
|
|
5
5
|
export declare const DEFAULT_ENDPOINT = "https://api.dashboard.com/api/v1/traces";
|
|
6
6
|
/**
|
|
7
7
|
* HTTP Exporter configuration.
|
|
@@ -13,6 +13,7 @@ export interface HttpExporterOptions {
|
|
|
13
13
|
maxRetries?: number;
|
|
14
14
|
backoffBase?: number;
|
|
15
15
|
backoffJitter?: number;
|
|
16
|
+
resource?: Resource;
|
|
16
17
|
}
|
|
17
18
|
/**
|
|
18
19
|
* HTTP Exporter for sending spans to a backend.
|
|
@@ -24,6 +25,7 @@ export declare class HttpExporter implements ISpanExporter {
|
|
|
24
25
|
private maxRetries;
|
|
25
26
|
private backoffBase;
|
|
26
27
|
private backoffJitter;
|
|
28
|
+
private resource;
|
|
27
29
|
constructor(options?: HttpExporterOptions);
|
|
28
30
|
/**
|
|
29
31
|
* Export spans to the backend.
|
|
@@ -34,9 +36,13 @@ export declare class HttpExporter implements ISpanExporter {
|
|
|
34
36
|
*/
|
|
35
37
|
shutdown(): Promise<void>;
|
|
36
38
|
/**
|
|
37
|
-
* Serialize spans to JSON bytes.
|
|
39
|
+
* Serialize spans to JSON bytes in OpenTelemetry-like format.
|
|
38
40
|
*/
|
|
39
41
|
private serializeSpans;
|
|
42
|
+
/**
|
|
43
|
+
* Convert numeric status to string code for OTLP compatibility.
|
|
44
|
+
*/
|
|
45
|
+
private statusToString;
|
|
40
46
|
/**
|
|
41
47
|
* Get request headers.
|
|
42
48
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"http-exporter.d.ts","sourceRoot":"","sources":["../../src/exporter/http-exporter.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;
|
|
1
|
+
{"version":3,"file":"http-exporter.d.ts","sourceRoot":"","sources":["../../src/exporter/http-exporter.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAE1D,eAAO,MAAM,gBAAgB,4CAA4C,CAAC;AAI1E;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED;;GAEG;AACH,qBAAa,YAAa,YAAW,aAAa;IAChD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,QAAQ,CAAW;gBAEf,OAAO,GAAE,mBAAwB;IAe7C;;OAEG;IACG,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAkC9C;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B;;OAEG;IACH,OAAO,CAAC,cAAc;IAwCtB;;OAEG;IACH,OAAO,CAAC,cAAc;IAatB;;OAEG;IACH,OAAO,CAAC,UAAU;IAalB;;OAEG;IACH,OAAO,CAAC,WAAW;IA0CnB;;OAEG;IACH,OAAO,CAAC,cAAc;IAMtB;;OAEG;IACH,OAAO,CAAC,KAAK;CAGd"}
|
|
@@ -52,6 +52,12 @@ class HttpExporter {
|
|
|
52
52
|
this.maxRetries = options.maxRetries || 5;
|
|
53
53
|
this.backoffBase = options.backoffBase || 1;
|
|
54
54
|
this.backoffJitter = options.backoffJitter || 0.5;
|
|
55
|
+
this.resource = options.resource || {
|
|
56
|
+
'sdk.name': 'traccia-sdk-ts',
|
|
57
|
+
'sdk.version': '1.0.0',
|
|
58
|
+
'service.name': 'unknown-service',
|
|
59
|
+
'service.version': 'unknown',
|
|
60
|
+
};
|
|
55
61
|
}
|
|
56
62
|
/**
|
|
57
63
|
* Export spans to the backend.
|
|
@@ -91,25 +97,58 @@ class HttpExporter {
|
|
|
91
97
|
// No-op for HTTP exporter
|
|
92
98
|
}
|
|
93
99
|
/**
|
|
94
|
-
* Serialize spans to JSON bytes.
|
|
100
|
+
* Serialize spans to JSON bytes in OpenTelemetry-like format.
|
|
95
101
|
*/
|
|
96
102
|
serializeSpans(spans) {
|
|
97
|
-
|
|
103
|
+
// Group all spans under a single scope for now
|
|
104
|
+
const scope = {
|
|
105
|
+
name: this.resource['sdk.name'] || 'agent-tracing-sdk',
|
|
106
|
+
version: this.resource['sdk.version'] || '0.1.0',
|
|
107
|
+
};
|
|
108
|
+
const otelSpans = spans.map((span) => ({
|
|
98
109
|
traceId: span.context.traceId,
|
|
99
110
|
spanId: span.context.spanId,
|
|
100
|
-
parentSpanId: span.parentSpanId,
|
|
111
|
+
parentSpanId: span.parentSpanId ?? null,
|
|
101
112
|
name: span.name,
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
durationNs: span.durationNs,
|
|
113
|
+
startTimeUnixNano: span.startTimeNs,
|
|
114
|
+
endTimeUnixNano: span.endTimeNs,
|
|
105
115
|
attributes: span.attributes,
|
|
106
|
-
events: span.events,
|
|
107
|
-
status: span.status,
|
|
108
|
-
statusDescription: span.statusDescription,
|
|
109
|
-
traceFlags: span.context.traceFlags,
|
|
110
|
-
traceState: span.context.traceState,
|
|
116
|
+
events: span.events || [],
|
|
117
|
+
status: typeof span.status === 'object' && 'code' in span.status ? span.status : { code: this.statusToString(span.status), message: span.statusDescription || '' },
|
|
111
118
|
}));
|
|
112
|
-
|
|
119
|
+
const payload = {
|
|
120
|
+
resource: {
|
|
121
|
+
'service.name': this.resource['service.name'],
|
|
122
|
+
'service.version': this.resource['service.version'],
|
|
123
|
+
...this.resource,
|
|
124
|
+
},
|
|
125
|
+
items: [
|
|
126
|
+
{
|
|
127
|
+
scopeSpans: [
|
|
128
|
+
{
|
|
129
|
+
scope,
|
|
130
|
+
spans: otelSpans,
|
|
131
|
+
},
|
|
132
|
+
],
|
|
133
|
+
},
|
|
134
|
+
],
|
|
135
|
+
};
|
|
136
|
+
return JSON.stringify(payload);
|
|
137
|
+
}
|
|
138
|
+
/**
|
|
139
|
+
* Convert numeric status to string code for OTLP compatibility.
|
|
140
|
+
*/
|
|
141
|
+
statusToString(status) {
|
|
142
|
+
switch (status) {
|
|
143
|
+
case 0:
|
|
144
|
+
return 'UNSET';
|
|
145
|
+
case 1:
|
|
146
|
+
return 'OK';
|
|
147
|
+
case 2:
|
|
148
|
+
return 'ERROR';
|
|
149
|
+
default:
|
|
150
|
+
return 'UNSET';
|
|
151
|
+
}
|
|
113
152
|
}
|
|
114
153
|
/**
|
|
115
154
|
* Get request headers.
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"http-exporter.js","sourceRoot":"","sources":["../../src/exporter/http-exporter.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,6CAA+B;AAC/B,2CAA6B;AAGhB,QAAA,gBAAgB,GAAG,yCAAyC,CAAC;AAE1E,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"http-exporter.js","sourceRoot":"","sources":["../../src/exporter/http-exporter.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,6CAA+B;AAC/B,2CAA6B;AAGhB,QAAA,gBAAgB,GAAG,yCAAyC,CAAC;AAE1E,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;AAexD;;GAEG;AACH,MAAa,YAAY;IASvB,YAAY,UAA+B,EAAE;QAC3C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,wBAAgB,CAAC;QACrD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,GAAG,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI;YAClC,UAAU,EAAE,gBAAgB;YAC5B,aAAa,EAAE,OAAO;YACtB,cAAc,EAAE,iBAAiB;YACjC,iBAAiB,EAAE,SAAS;SAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,KAAc;QACzB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAElC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAExD,IAAI,MAAM,IAAI,GAAG,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC;oBAClC,OAAO,IAAI,CAAC;gBACd,CAAC;gBAED,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;oBACxC,OAAO,KAAK,CAAC;gBACf,CAAC;gBAED,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;gBAC7C,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;YAAC,MAAM,CAAC;gBACP,sCAAsC;gBACtC,IAAI,OAAO,GAAG,IAAI,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC;oBAClC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;oBAC7C,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,0BAA0B;IAC5B,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,KAAc;QACnC,+CAA+C;QAC/C,MAAM,KAAK,GAAG;YACZ,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,mBAAmB;YACtD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,OAAO;SACjD,CAAC;QAEF,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC3B,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;YACvC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,iBAAiB,EAAE,IAAI,CAAC,WAAW;YACnC,eAAe,EAAE,IAAI,CAAC,SAAS;YAC/B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;YACzB,MAAM,EAAE,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,IAAI,EAAE,EAAE;SACnK,CAAC,CAAC,CAAC;QAEJ,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE;gBACR,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC;gBAC7C,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC;gBACnD,GAAG,IAAI,CAAC,QAAQ;aACjB;YACD,KAAK,EAAE;gBACL;oBACE,UAAU,EAAE;wBACV;4BACE,KAAK;4BACL,KAAK,EAAE,SAAS;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;QAEF,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,MAAW;QAChC,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,CAAC;gBACJ,OAAO,OAAO,CAAC;YACjB,KAAK,CAAC;gBACJ,OAAO,IAAI,CAAC;YACd,KAAK,CAAC;gBACJ,OAAO,OAAO,CAAC;YACjB;gBACE,OAAO,OAAO,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,UAAU;QAChB,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;YAClC,YAAY,EAAE,sBAAsB;SACrC,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC;QACrD,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,OAAe,EAAE,OAA+B;QAClE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,MAAM,OAAO,GAAG,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC;YAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAEtC,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,IAAI,EAAE,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC,MAAM;gBAC/B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,OAAO;oBACV,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC;iBAC7C;gBACD,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC;YAEF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE;gBACnD,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;oBACvB,+BAA+B;gBACjC,CAAC,CAAC,CAAC;gBAEH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACtB,OAAO,CAAC,QAAQ,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC;gBACtC,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC5B,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBACzB,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClB,MAAM,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACvB,OAAO,CAAC,GAAG,EAAE,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAe;QACpC,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC;QAClD,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC,GAAG,IAAI,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;CACF;AA3MD,oCA2MC"}
|
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
* @module integrations
|
|
6
6
|
*/
|
|
7
7
|
export { TracciaCallbackHandler } from './langchain-callback';
|
|
8
|
-
export {
|
|
8
|
+
export { TracciaCallbackHandlerOld } from './langchain-callback.old';
|
|
9
9
|
export { getTraciaHandler, withTracing, createTracedOpenAI, createTracedAgentExecutor, createTracedLLMChain, traced, setupLangChainWithTracing, } from './auto-langchain';
|
|
10
10
|
export { instrumentLangGraph, createTracedNode, createTracedConditional, } from './langgraph-instrumentation';
|
|
11
11
|
export { wrapGraphWithTracing, tracedNode, tracedConditional, createSimpleTracedGraph, traceableFunction, createAgentWorkflow, } from './auto-langgraph';
|
|
@@ -6,12 +6,12 @@
|
|
|
6
6
|
* @module integrations
|
|
7
7
|
*/
|
|
8
8
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
-
exports.POPULAR_OLLAMA_MODELS = exports.getOllamaSetupInstructions = exports.createOllamaStreamingChatbot = exports.createOllamaChatbot = exports.setupOllamaWithTracing = exports.createOllamaWithTracing = exports.createAgentWorkflow = exports.traceableFunction = exports.createSimpleTracedGraph = exports.tracedConditional = exports.tracedNode = exports.wrapGraphWithTracing = exports.createTracedConditional = exports.createTracedNode = exports.instrumentLangGraph = exports.setupLangChainWithTracing = exports.traced = exports.createTracedLLMChain = exports.createTracedAgentExecutor = exports.createTracedOpenAI = exports.withTracing = exports.getTraciaHandler = exports.
|
|
9
|
+
exports.POPULAR_OLLAMA_MODELS = exports.getOllamaSetupInstructions = exports.createOllamaStreamingChatbot = exports.createOllamaChatbot = exports.setupOllamaWithTracing = exports.createOllamaWithTracing = exports.createAgentWorkflow = exports.traceableFunction = exports.createSimpleTracedGraph = exports.tracedConditional = exports.tracedNode = exports.wrapGraphWithTracing = exports.createTracedConditional = exports.createTracedNode = exports.instrumentLangGraph = exports.setupLangChainWithTracing = exports.traced = exports.createTracedLLMChain = exports.createTracedAgentExecutor = exports.createTracedOpenAI = exports.withTracing = exports.getTraciaHandler = exports.TracciaCallbackHandlerOld = exports.TracciaCallbackHandler = void 0;
|
|
10
10
|
// LangChain integrations
|
|
11
11
|
var langchain_callback_1 = require("./langchain-callback");
|
|
12
12
|
Object.defineProperty(exports, "TracciaCallbackHandler", { enumerable: true, get: function () { return langchain_callback_1.TracciaCallbackHandler; } });
|
|
13
|
-
var
|
|
14
|
-
Object.defineProperty(exports, "
|
|
13
|
+
var langchain_callback_old_1 = require("./langchain-callback.old");
|
|
14
|
+
Object.defineProperty(exports, "TracciaCallbackHandlerOld", { enumerable: true, get: function () { return langchain_callback_old_1.TracciaCallbackHandlerOld; } });
|
|
15
15
|
var auto_langchain_1 = require("./auto-langchain");
|
|
16
16
|
Object.defineProperty(exports, "getTraciaHandler", { enumerable: true, get: function () { return auto_langchain_1.getTraciaHandler; } });
|
|
17
17
|
Object.defineProperty(exports, "withTracing", { enumerable: true, get: function () { return auto_langchain_1.withTracing; } });
|
|
@@ -1,96 +1,62 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
import {
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
1
|
+
import type { AgentAction, AgentFinish } from "@langchain/core/agents";
|
|
2
|
+
import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
|
|
3
|
+
import type { Document } from "@langchain/core/documents";
|
|
4
|
+
import type { Serialized } from "@langchain/core/load/serializable";
|
|
5
|
+
import { BaseMessage, type BaseMessageFields, type MessageContent } from "@langchain/core/messages";
|
|
6
|
+
import type { LLMResult } from "@langchain/core/outputs";
|
|
7
|
+
import type { ChainValues } from "@langchain/core/utils/types";
|
|
8
|
+
export type LlmMessage = {
|
|
9
|
+
role: string;
|
|
10
|
+
content: BaseMessageFields["content"];
|
|
11
|
+
additional_kwargs?: BaseMessageFields["additional_kwargs"];
|
|
12
|
+
};
|
|
13
|
+
export type AnonymousLlmMessage = {
|
|
14
|
+
content: BaseMessageFields["content"];
|
|
15
|
+
additional_kwargs?: BaseMessageFields["additional_kwargs"];
|
|
16
|
+
};
|
|
17
|
+
type ConstructorParams = {
|
|
18
|
+
userId?: string;
|
|
19
|
+
sessionId?: string;
|
|
20
|
+
tags?: string[];
|
|
21
|
+
version?: string;
|
|
22
|
+
traceMetadata?: Record<string, unknown>;
|
|
23
|
+
};
|
|
24
24
|
export declare class TracciaCallbackHandler extends BaseCallbackHandler {
|
|
25
25
|
name: string;
|
|
26
26
|
private tracer;
|
|
27
|
-
private
|
|
28
|
-
private
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
* Handle chain error.
|
|
61
|
-
*/
|
|
62
|
-
handleChainError(error: Error, runId: string): Promise<void>;
|
|
63
|
-
/**
|
|
64
|
-
* Handle tool start - called when a tool is invoked.
|
|
65
|
-
*/
|
|
66
|
-
handleToolStart(tool: any, input: string, runId: string, _parentRunId?: string): Promise<void>;
|
|
67
|
-
/**
|
|
68
|
-
* Handle tool end - called when a tool finishes execution.
|
|
69
|
-
*/
|
|
70
|
-
handleToolEnd(output: string, runId: string): Promise<void>;
|
|
71
|
-
/**
|
|
72
|
-
* Handle tool error.
|
|
73
|
-
*/
|
|
74
|
-
handleToolError(error: Error, runId: string): Promise<void>;
|
|
75
|
-
/**
|
|
76
|
-
* Handle agent action.
|
|
77
|
-
*/
|
|
78
|
-
handleAgentAction(action: any, runId: string): Promise<void>;
|
|
79
|
-
/**
|
|
80
|
-
* Handle agent finish.
|
|
81
|
-
*/
|
|
82
|
-
handleAgentFinish(finish: any, runId: string): Promise<void>;
|
|
83
|
-
onLLMStart(llm: any, prompts: string[], runId: string, parentRunId?: string): Promise<void>;
|
|
84
|
-
onLLMEnd(output: any, runId: string): Promise<void>;
|
|
85
|
-
onLLMError(error: Error, runId: string): Promise<void>;
|
|
86
|
-
onLLMNewToken(_token: string, idx?: any, runId?: string): Promise<void>;
|
|
87
|
-
onChainStart(chain: any, inputs: any, runId: string, parentRunId?: string): Promise<void>;
|
|
88
|
-
onChainEnd(output: any, runId: string): Promise<void>;
|
|
89
|
-
onChainError(error: Error, runId: string): Promise<void>;
|
|
90
|
-
onToolStart(tool: any, input: string, runId: string, parentRunId?: string): Promise<void>;
|
|
91
|
-
onToolEnd(output: string, runId: string): Promise<void>;
|
|
92
|
-
onToolError(error: Error, runId: string): Promise<void>;
|
|
93
|
-
onAgentAction(action: any, runId: string): Promise<void>;
|
|
94
|
-
onAgentFinish(finish: any, runId: string): Promise<void>;
|
|
27
|
+
private tags;
|
|
28
|
+
private completionStartTimes;
|
|
29
|
+
private promptToParentRunMap;
|
|
30
|
+
private runMap;
|
|
31
|
+
last_trace_id: string | null;
|
|
32
|
+
constructor(params?: ConstructorParams);
|
|
33
|
+
handleLLMNewToken(_token: string, _idx: any, runId: string, _parentRunId?: string, _tags?: string[], _fields?: any): Promise<void>;
|
|
34
|
+
handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, _runType?: string, name?: string): Promise<void>;
|
|
35
|
+
handleAgentAction(action: AgentAction, runId: string, parentRunId?: string): Promise<void>;
|
|
36
|
+
handleAgentEnd?(action: AgentFinish, runId: string, _parentRunId?: string): Promise<void>;
|
|
37
|
+
handleChainError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
38
|
+
handleGenerationStart(llm: Serialized, messages: (LlmMessage | MessageContent | AnonymousLlmMessage)[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
|
|
39
|
+
handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
|
|
40
|
+
handleChainEnd(outputs: ChainValues, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
41
|
+
handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
|
|
42
|
+
handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
|
|
43
|
+
handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
|
|
44
|
+
handleRetrieverEnd(documents: Document<Record<string, any>>[], runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
45
|
+
handleRetrieverError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
46
|
+
handleToolEnd(output: string, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
47
|
+
handleToolError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
48
|
+
handleLLMEnd(output: LLMResult, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
49
|
+
handleLLMError(err: any, runId: string, _parentRunId?: string | undefined): Promise<void>;
|
|
50
|
+
private registerLangfusePrompt;
|
|
51
|
+
private deregisterLangfusePrompt;
|
|
52
|
+
private startAndRegisterSpan;
|
|
53
|
+
private handleSpanEnd;
|
|
54
|
+
private parseAzureRefusalError;
|
|
55
|
+
private joinTagsAndMetaData;
|
|
56
|
+
private stripLangfuseKeysFromMetadata;
|
|
57
|
+
private extractUsageMetadata;
|
|
58
|
+
private extractModelNameFromMetadata;
|
|
59
|
+
private extractChatMessageContent;
|
|
95
60
|
}
|
|
61
|
+
export {};
|
|
96
62
|
//# sourceMappingURL=langchain-callback.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"langchain-callback.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":"AAAA
|
|
1
|
+
{"version":3,"file":"langchain-callback.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain-callback.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AACrE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAGL,WAAW,EAEX,KAAK,iBAAiB,EACtB,KAAK,cAAc,EACpB,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAc,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACrE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAU/D,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACtC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;CAC5D,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACtC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;CAC5D,CAAC;AAEF,KAAK,iBAAiB,GAAG;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzC,CAAC;AAEF,qBAAa,sBAAuB,SAAQ,mBAAmB;IAC7D,IAAI,SAA4B;IAChC,OAAO,CAAC,MAAM,CAA0B;IAExC,OAAO,CAAC,IAAI,CAA4B;IAExC,OAAO,CAAC,oBAAoB,CAA4B;IACxD,OAAO,CAAC,oBAAoB,CAAC;IAC7B,OAAO,CAAC,MAAM,CAAiC;IAExC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAQ;gBAE/B,MAAM,CAAC,EAAE,iBAAiB;IAQhC,iBAAiB,CACrB,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,GAAG,EACT,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,EAAE,EAChB,OAAO,CAAC,EAAE,GAAG,GACZ,OAAO,CAAC,IAAI,CAAC;IAQV,gBAAgB,CACpB,KAAK,EAAE,UAAU,EACjB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,QAAQ,CAAC,EAAE,MAAM,EACjB,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAwDV,iBAAiB,CACrB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAgBV,cAAc,CAAC,CACnB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAaV,gBAAgB,CACpB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAkBV,qBAAqB,CACzB,GAAG,EAAE,UAAU,EACf,QAAQ,EAAE,CAAC,UAAU,GAAG,cAAc,GAAG,mBAAmB,CAAC,EAAE,EAC/D,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACjD,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAiEV,oBAAoB,CACxB,GAAG,EAAE,UAAU,EACf,QAAQ,EAAE,WAAW,EAAE,EAAE,EACzB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACjD,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAuBV,cAAc,CAClB,OAAO,EAAE,WAAW,EACpB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAoCV,cAAc,CAClB,GAAG,EAAE,UAAU,EACf,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACjD,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAmBV,eAAe,CACnB,IAAI,EAAE,UAAU,EAChB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAmBV,oBAAoB,CACxB,SAAS,EAAE,UAAU,EACrB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,EAChC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,EAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC9C,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;IAmBV,kBAAkB,CACtB,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,EAC1C,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAeV,oBAAoB,CACxB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAcV,aAAa,CACjB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAaV,eAAe,CACnB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAgBV,YAAY,CAChB,MAAM,EAAE,SAAS,EACjB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IA+EV,cAAc,CAClB,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,GAChC,OAAO,CAAC,IAAI,CAAC;IAkBhB,OAAO,CAAC,sBAAsB;IAY9B,OAAO,CAAC,wBAAwB;IAIhC,OAAO,CAAC,oBAAoB;IAsB5B,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,sBAAsB;IAY9B,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,6BAA6B;IAoBrC,OAAO,CAAC,oBAAoB;IAmB5B,OAAO,CAAC,4BAA4B;IAYpC,OAAO,CAAC,yBAAyB;CA8DlC"}
|