@mastra/langfuse 0.0.0-vector-query-tool-provider-options-20250828222356
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/CHANGELOG.md +71 -0
- package/LICENSE.md +15 -0
- package/README.md +49 -0
- package/dist/ai-tracing.d.ts +34 -0
- package/dist/ai-tracing.d.ts.map +1 -0
- package/dist/index.cjs +135 -0
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.ts +8 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +133 -0
- package/dist/index.js.map +1 -0
- package/eslint.config.js +6 -0
- package/package.json +46 -0
- package/src/ai-tracing.test.ts +663 -0
- package/src/ai-tracing.ts +198 -0
- package/src/index.ts +9 -0
- package/tsconfig.build.json +9 -0
- package/tsconfig.json +5 -0
- package/tsup.config.ts +17 -0
- package/vitest.config.ts +11 -0
package/CHANGELOG.md
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
# @mastra/langfuse
|
|
2
|
+
|
|
3
|
+
## 0.0.0-vector-query-tool-provider-options-20250828222356
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [[`637f323`](https://github.com/mastra-ai/mastra/commit/637f32371d79a8f78c52c0d53411af0915fcec67), [`29de0e1`](https://github.com/mastra-ai/mastra/commit/29de0e1b0a7173317ae7d1ab0c0993167c659f2b), [`00c74e7`](https://github.com/mastra-ai/mastra/commit/00c74e73b1926be0d475693bb886fb67a22ff352), [`6ac697e`](https://github.com/mastra-ai/mastra/commit/6ac697edcc2435482c247cba615277ec4765dcc4), [`af90672`](https://github.com/mastra-ai/mastra/commit/af906722d8da28688882193b1e531026f9e2e81e), [`56041d0`](https://github.com/mastra-ai/mastra/commit/56041d018863a3da6b98c512e47348647c075fb3), [`5de1555`](https://github.com/mastra-ai/mastra/commit/5de15554d3d6695211945a36928f6657e76cddc9), [`1ed5a3e`](https://github.com/mastra-ai/mastra/commit/1ed5a3e19330374c4347a4237cd2f4b9ffb60376)]:
|
|
8
|
+
- @mastra/core@0.0.0-vector-query-tool-provider-options-20250828222356
|
|
9
|
+
|
|
10
|
+
## 0.0.4
|
|
11
|
+
|
|
12
|
+
### Patch Changes
|
|
13
|
+
|
|
14
|
+
- [`c6113ed`](https://github.com/mastra-ai/mastra/commit/c6113ed7f9df297e130d94436ceee310273d6430) Thanks [@wardpeet](https://github.com/wardpeet)! - Fix peerdpes for @mastra/core
|
|
15
|
+
|
|
16
|
+
- Updated dependencies []:
|
|
17
|
+
- @mastra/core@0.15.2
|
|
18
|
+
|
|
19
|
+
## 0.0.3
|
|
20
|
+
|
|
21
|
+
### Patch Changes
|
|
22
|
+
|
|
23
|
+
- [`95b2aa9`](https://github.com/mastra-ai/mastra/commit/95b2aa908230919e67efcac0d69005a2d5745298) Thanks [@wardpeet](https://github.com/wardpeet)! - Fix peerdeps @mastra/core
|
|
24
|
+
|
|
25
|
+
- Updated dependencies []:
|
|
26
|
+
- @mastra/core@0.15.1
|
|
27
|
+
|
|
28
|
+
## 0.0.2
|
|
29
|
+
|
|
30
|
+
### Patch Changes
|
|
31
|
+
|
|
32
|
+
- 03997ae: Update peerdeps
|
|
33
|
+
- e1a81da: initial release of LangFuse AI Observability exporter
|
|
34
|
+
- Updated dependencies [227c7e6]
|
|
35
|
+
- Updated dependencies [12cae67]
|
|
36
|
+
- Updated dependencies [fd3a3eb]
|
|
37
|
+
- Updated dependencies [6faaee5]
|
|
38
|
+
- Updated dependencies [4232b14]
|
|
39
|
+
- Updated dependencies [a89de7e]
|
|
40
|
+
- Updated dependencies [5a37d0c]
|
|
41
|
+
- Updated dependencies [4bde0cb]
|
|
42
|
+
- Updated dependencies [cf4f357]
|
|
43
|
+
- Updated dependencies [ad888a2]
|
|
44
|
+
- Updated dependencies [481751d]
|
|
45
|
+
- Updated dependencies [2454423]
|
|
46
|
+
- Updated dependencies [194e395]
|
|
47
|
+
- Updated dependencies [a722c0b]
|
|
48
|
+
- Updated dependencies [c30bca8]
|
|
49
|
+
- Updated dependencies [3b5fec7]
|
|
50
|
+
- Updated dependencies [a8f129d]
|
|
51
|
+
- @mastra/core@0.14.0
|
|
52
|
+
|
|
53
|
+
## 0.0.2-alpha.1
|
|
54
|
+
|
|
55
|
+
### Patch Changes
|
|
56
|
+
|
|
57
|
+
- 03997ae: Update peerdeps
|
|
58
|
+
- @mastra/core@0.14.0-alpha.7
|
|
59
|
+
|
|
60
|
+
## 0.0.2-alpha.0
|
|
61
|
+
|
|
62
|
+
### Patch Changes
|
|
63
|
+
|
|
64
|
+
- e1a81da: initial release of LangFuse AI Observability exporter
|
|
65
|
+
- Updated dependencies [6faaee5]
|
|
66
|
+
- Updated dependencies [4232b14]
|
|
67
|
+
- Updated dependencies [a89de7e]
|
|
68
|
+
- Updated dependencies [cf4f357]
|
|
69
|
+
- Updated dependencies [a722c0b]
|
|
70
|
+
- Updated dependencies [3b5fec7]
|
|
71
|
+
- @mastra/core@0.14.0-alpha.1
|
package/LICENSE.md
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
# Apache License 2.0
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2025 Kepler Software, Inc.
|
|
4
|
+
|
|
5
|
+
Licensed under the Apache License, Version 2.0 (the "License");
|
|
6
|
+
you may not use this file except in compliance with the License.
|
|
7
|
+
You may obtain a copy of the License at
|
|
8
|
+
|
|
9
|
+
http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
+
|
|
11
|
+
Unless required by applicable law or agreed to in writing, software
|
|
12
|
+
distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
+
See the License for the specific language governing permissions and
|
|
15
|
+
limitations under the License.
|
package/README.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
# @mastra/langfuse
|
|
2
|
+
|
|
3
|
+
Langfuse AI Observability exporter for Mastra applications.
|
|
4
|
+
|
|
5
|
+
## Installation
|
|
6
|
+
|
|
7
|
+
```bash
|
|
8
|
+
npm install @mastra/langfuse
|
|
9
|
+
```
|
|
10
|
+
|
|
11
|
+
## Usage
|
|
12
|
+
|
|
13
|
+
```typescript
|
|
14
|
+
import { LangfuseExporter } from '@mastra/langfuse';
|
|
15
|
+
|
|
16
|
+
// Use with Mastra
|
|
17
|
+
const mastra = new Mastra({
|
|
18
|
+
...,
|
|
19
|
+
observability: {
|
|
20
|
+
instances: {
|
|
21
|
+
langfuse: {
|
|
22
|
+
serviceName: 'service',
|
|
23
|
+
exporters: [
|
|
24
|
+
new LangfuseExporter({
|
|
25
|
+
publicKey: process.env.LANGFUSE_PUBLIC_KEY,
|
|
26
|
+
secretKey: process.env.LANGFUSE_SECRET_KEY,
|
|
27
|
+
baseUrl: process.env.LANGFUSE_BASE_URL,
|
|
28
|
+
realtime: true,
|
|
29
|
+
}),
|
|
30
|
+
],
|
|
31
|
+
},
|
|
32
|
+
},
|
|
33
|
+
},
|
|
34
|
+
});
|
|
35
|
+
```
|
|
36
|
+
|
|
37
|
+
## Features
|
|
38
|
+
|
|
39
|
+
### AI Tracing
|
|
40
|
+
|
|
41
|
+
- **Automatic span mapping**: Root spans become Langfuse traces
|
|
42
|
+
- **LLM generation support**: `LLM_GENERATION` spans become Langfuse generations with token usage
|
|
43
|
+
- **Type-specific metadata**: Extracts relevant metadata for each span type (agents, tools, workflows)
|
|
44
|
+
- **Error tracking**: Automatic error status and message tracking
|
|
45
|
+
- **Hierarchical traces**: Maintains parent-child relationships
|
|
46
|
+
|
|
47
|
+
## License
|
|
48
|
+
|
|
49
|
+
Apache 2.0
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Langfuse Exporter for Mastra AI Tracing
|
|
3
|
+
*
|
|
4
|
+
* This exporter sends tracing data to Langfuse for AI observability.
|
|
5
|
+
* Root spans start traces in Langfuse.
|
|
6
|
+
* LLM_GENERATION spans become Langfuse generations, all others become spans.
|
|
7
|
+
*/
|
|
8
|
+
import type { AITracingExporter, AITracingEvent } from '@mastra/core/ai-tracing';
|
|
9
|
+
export interface LangfuseExporterConfig {
|
|
10
|
+
/** Langfuse API key */
|
|
11
|
+
publicKey: string;
|
|
12
|
+
/** Langfuse secret key */
|
|
13
|
+
secretKey: string;
|
|
14
|
+
/** Langfuse host URL */
|
|
15
|
+
baseUrl: string;
|
|
16
|
+
/** Enable realtime mode - flushes after each event for immediate visibility */
|
|
17
|
+
realtime?: boolean;
|
|
18
|
+
/** Additional options to pass to the Langfuse client */
|
|
19
|
+
options?: any;
|
|
20
|
+
}
|
|
21
|
+
export declare class LangfuseExporter implements AITracingExporter {
|
|
22
|
+
name: string;
|
|
23
|
+
private client;
|
|
24
|
+
private realtime;
|
|
25
|
+
private traceMap;
|
|
26
|
+
constructor(config: LangfuseExporterConfig);
|
|
27
|
+
exportEvent(event: AITracingEvent): Promise<void>;
|
|
28
|
+
private handleSpanStarted;
|
|
29
|
+
private handleSpanUpdateOrEnd;
|
|
30
|
+
private buildTracePayload;
|
|
31
|
+
private buildSpanPayload;
|
|
32
|
+
shutdown(): Promise<void>;
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=ai-tracing.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ai-tracing.d.ts","sourceRoot":"","sources":["../src/ai-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAsC,MAAM,yBAAyB,CAAC;AAKrH,MAAM,WAAW,sBAAsB;IACrC,uBAAuB;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,0BAA0B;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,+EAA+E;IAC/E,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,wDAAwD;IACxD,OAAO,CAAC,EAAE,GAAG,CAAC;CACf;AASD,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,IAAI,SAAc;IAClB,OAAO,CAAC,MAAM,CAAW;IACzB,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,QAAQ,CAAgC;gBAEpC,MAAM,EAAE,sBAAsB;IAUpC,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;YAmBzC,iBAAiB;YA0BjB,qBAAqB;IA2BnC,OAAO,CAAC,iBAAiB;IAqBzB,OAAO,CAAC,gBAAgB;IAmDlB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAIhC"}
|
package/dist/index.cjs
ADDED
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var aiTracing = require('@mastra/core/ai-tracing');
|
|
4
|
+
var langfuse = require('langfuse');
|
|
5
|
+
|
|
6
|
+
// src/ai-tracing.ts
|
|
7
|
+
var LangfuseExporter = class {
|
|
8
|
+
name = "langfuse";
|
|
9
|
+
client;
|
|
10
|
+
realtime;
|
|
11
|
+
traceMap = /* @__PURE__ */ new Map();
|
|
12
|
+
constructor(config) {
|
|
13
|
+
this.realtime = config.realtime ?? false;
|
|
14
|
+
this.client = new langfuse.Langfuse({
|
|
15
|
+
publicKey: config.publicKey,
|
|
16
|
+
secretKey: config.secretKey,
|
|
17
|
+
baseUrl: config.baseUrl,
|
|
18
|
+
...config.options
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
async exportEvent(event) {
|
|
22
|
+
switch (event.type) {
|
|
23
|
+
case "span_started":
|
|
24
|
+
await this.handleSpanStarted(event.span);
|
|
25
|
+
break;
|
|
26
|
+
case "span_updated":
|
|
27
|
+
await this.handleSpanUpdateOrEnd(event.span, true);
|
|
28
|
+
break;
|
|
29
|
+
case "span_ended":
|
|
30
|
+
await this.handleSpanUpdateOrEnd(event.span, false);
|
|
31
|
+
break;
|
|
32
|
+
}
|
|
33
|
+
if (this.realtime) {
|
|
34
|
+
await this.client.flushAsync();
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
async handleSpanStarted(span) {
|
|
38
|
+
if (span.isRootSpan) {
|
|
39
|
+
const trace = this.client.trace(this.buildTracePayload(span));
|
|
40
|
+
this.traceMap.set(span.trace.id, { trace, spans: /* @__PURE__ */ new Map() });
|
|
41
|
+
}
|
|
42
|
+
const traceData = this.traceMap.get(span.trace.id);
|
|
43
|
+
if (!traceData) {
|
|
44
|
+
console.log("NO TRACE");
|
|
45
|
+
return;
|
|
46
|
+
}
|
|
47
|
+
const langfuseParent = span.parent && traceData.spans.has(span.parent.id) ? traceData.spans.get(span.parent.id) : traceData.trace;
|
|
48
|
+
const payload = this.buildSpanPayload(span, true);
|
|
49
|
+
const langfuseSpan = span.type === aiTracing.AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);
|
|
50
|
+
traceData.spans.set(span.id, langfuseSpan);
|
|
51
|
+
}
|
|
52
|
+
async handleSpanUpdateOrEnd(span, isUpdate) {
|
|
53
|
+
const traceData = this.traceMap.get(span.trace.id);
|
|
54
|
+
if (!traceData) {
|
|
55
|
+
console.log("NO TRACE");
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
const langfuseSpan = traceData.spans.get(span.id);
|
|
59
|
+
if (!langfuseSpan) {
|
|
60
|
+
console.log("NO SPAN");
|
|
61
|
+
return;
|
|
62
|
+
}
|
|
63
|
+
if (isUpdate) {
|
|
64
|
+
langfuseSpan.update(this.buildSpanPayload(span, false));
|
|
65
|
+
} else {
|
|
66
|
+
langfuseSpan.end(this.buildSpanPayload(span, false));
|
|
67
|
+
if (span.isRootSpan) {
|
|
68
|
+
traceData.trace.update({ output: span.output });
|
|
69
|
+
this.traceMap.delete(span.trace.id);
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
buildTracePayload(span) {
|
|
74
|
+
const payload = {
|
|
75
|
+
id: span.trace.id,
|
|
76
|
+
name: span.name
|
|
77
|
+
};
|
|
78
|
+
const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};
|
|
79
|
+
if (userId) payload.userId = userId;
|
|
80
|
+
if (sessionId) payload.sessionId = sessionId;
|
|
81
|
+
if (span.input) payload.input = span.input;
|
|
82
|
+
payload.metadata = {
|
|
83
|
+
spanType: span.type,
|
|
84
|
+
...aiTracing.sanitizeMetadata(span.attributes),
|
|
85
|
+
...aiTracing.sanitizeMetadata(remainingMetadata)
|
|
86
|
+
};
|
|
87
|
+
return payload;
|
|
88
|
+
}
|
|
89
|
+
buildSpanPayload(span, isCreate) {
|
|
90
|
+
const payload = {};
|
|
91
|
+
if (isCreate) {
|
|
92
|
+
payload.id = span.id;
|
|
93
|
+
payload.name = span.name;
|
|
94
|
+
payload.startTime = span.startTime;
|
|
95
|
+
if (span.input !== void 0) payload.input = span.input;
|
|
96
|
+
}
|
|
97
|
+
if (span.output !== void 0) payload.output = span.output;
|
|
98
|
+
if (span.endTime !== void 0) payload.endTime = span.endTime;
|
|
99
|
+
const attributes = span.attributes ?? {};
|
|
100
|
+
const attributesToOmit = [];
|
|
101
|
+
if (span.type === aiTracing.AISpanType.LLM_GENERATION) {
|
|
102
|
+
const llmAttr = attributes;
|
|
103
|
+
if (llmAttr.model !== void 0) {
|
|
104
|
+
payload.model = llmAttr.model;
|
|
105
|
+
attributesToOmit.push("model");
|
|
106
|
+
}
|
|
107
|
+
if (llmAttr.usage !== void 0) {
|
|
108
|
+
payload.usage = llmAttr.usage;
|
|
109
|
+
attributesToOmit.push("usage");
|
|
110
|
+
}
|
|
111
|
+
if (llmAttr.parameters !== void 0) {
|
|
112
|
+
payload.modelParameters = llmAttr.parameters;
|
|
113
|
+
attributesToOmit.push("parameters");
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
payload.metadata = {
|
|
117
|
+
spanType: span.type,
|
|
118
|
+
...aiTracing.sanitizeMetadata(aiTracing.omitKeys(attributes, attributesToOmit)),
|
|
119
|
+
...aiTracing.sanitizeMetadata(span.metadata)
|
|
120
|
+
};
|
|
121
|
+
if (span.errorInfo) {
|
|
122
|
+
payload.level = "ERROR";
|
|
123
|
+
payload.statusMessage = span.errorInfo.message;
|
|
124
|
+
}
|
|
125
|
+
return payload;
|
|
126
|
+
}
|
|
127
|
+
async shutdown() {
|
|
128
|
+
await this.client.shutdownAsync();
|
|
129
|
+
this.traceMap.clear();
|
|
130
|
+
}
|
|
131
|
+
};
|
|
132
|
+
|
|
133
|
+
exports.LangfuseExporter = LangfuseExporter;
|
|
134
|
+
//# sourceMappingURL=index.cjs.map
|
|
135
|
+
//# sourceMappingURL=index.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/ai-tracing.ts"],"names":["Langfuse","AISpanType","sanitizeMetadata","omitKeys"],"mappings":";;;;;;AAiCO,IAAM,mBAAN,MAAoD;AAAA,EACzD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EAE9C,YAAY,MAAA,EAAgC;AAC1C,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AACnC,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIA,iBAAA,CAAS;AAAA,MACzB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,SAAS,MAAA,CAAO,OAAA;AAAA,MAChB,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA;AAIJ,IAAA,IAAI,KAAK,QAAA,EAAU;AACjB,MAAA,MAAM,IAAA,CAAK,OAAO,UAAA,EAAW;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,QAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAC5D,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,KAAA,CAAM,EAAA,EAAI,EAAE,KAAA,EAAO,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBACJ,IAAA,CAAK,MAAA,IAAU,SAAA,CAAU,KAAA,CAAM,IAAI,IAAA,CAAK,MAAA,CAAO,EAAE,CAAA,GAC5C,UAAU,KAAA,CAAM,GAAA,CAAI,KAAK,MAAA,CAAO,EAAE,IACnC,SAAA,CAAU,KAAA;AAEhB,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAASC,oBAAA,CAAW,cAAA,GAAiB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,QAAA,EAAkC;AACrF,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAChD,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAA,OAAA,CAAQ,IAAI,SAAS,CAAA;AAErB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,YAAA,CAAa,MAAA,CAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAAA,IACxD,CAAA,MAAO;AACL,MAAA,YAAA,CAAa,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAEnD,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAC9C,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,EAAE,CAAA;AAAA,MACpC;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,kBAAkB,IAAA,EAAsC;AAC9D,IAAA,MAAM,OAAA,GAA+B;AAAA,MACnC,EAAA,EAAI,KAAK,KAAA,CAAM,EAAA;AAAA,MACf,MAAM,IAAA,CAAK;AAAA,KACb;AAEA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,mBAAkB,GAAI,IAAA,CAAK,YAAY,EAAC;AAEtE,IAAA,IAAI,MAAA,UAAgB,MAAA,GAAS,MAAA;AAC7B,IAAA,IAAI,SAAA,UAAmB,SAAA,GAAY,SAAA;AACnC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,OAAA,CAAQ,KAAA,GAAQ,IAAA,CAAK,KAAA;AAErC,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAGC,0BAAA,CAAiB,IAAA,CAAK,UAAU,CAAA;AAAA,MACnC,GAAGA,2BAAiB,iBAAiB;AAAA,KACvC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CAAiB,MAAiB,QAAA,EAAwC;AAChF,IAAA,MAAM,UAA+B,EAAC;AAEtC,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAA,CAAQ,KAAK,IAAA,CAAK,EAAA;AAClB,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AACpB,MAAA,OAAA,CAAQ,YAAY,IAAA,CAAK,SAAA;AACzB,MAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACrD;AAEA,IAAA,IAAI,IAAA,CAAK,MAAA,KAAW,MAAA,EAAW,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AACrD,IAAA,IAAI,IAAA,CAAK,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,UAAU,IAAA,CAAK,OAAA;AAEvD,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAGxC,IAAA,MAAM,mBAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,IAAA,KAASD,oBAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAEhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,kBAAkB,OAAA,CAAQ,UAAA;AAClC,QAAA,gBAAA,CAAiB,KAAK,YAAY,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAGC,0BAAA,CAAiBC,kBAAA,CAAS,UAAA,EAAY,gBAAgB,CAAC,CAAA;AAAA,MAC1D,GAAGD,0BAAA,CAAiB,IAAA,CAAK,QAAQ;AAAA,KACnC;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA;AAChB,MAAA,OAAA,CAAQ,aAAA,GAAgB,KAAK,SAAA,CAAU,OAAA;AAAA,IACzC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAChC,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.cjs","sourcesContent":["/**\n * Langfuse Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Langfuse for AI observability.\n * Root spans start traces in Langfuse.\n * LLM_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig {\n /** Langfuse API key */\n publicKey: string;\n /** Langfuse secret key */\n secretKey: string;\n /** Langfuse host URL */\n baseUrl: string;\n /** Enable realtime mode - flushes after each event for immediate visibility */\n realtime?: boolean;\n /** Additional options to pass to the Langfuse client */\n options?: any;\n}\n\ntype TraceData = {\n trace: LangfuseTraceClient; // Langfuse trace object\n spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation\n};\n\ntype LangfuseSpan = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient;\n\nexport class LangfuseExporter implements AITracingExporter {\n name = 'langfuse';\n private client: Langfuse;\n private realtime: boolean;\n private traceMap = new Map<string, TraceData>();\n\n constructor(config: LangfuseExporterConfig) {\n this.realtime = config.realtime ?? false;\n this.client = new Langfuse({\n publicKey: config.publicKey,\n secretKey: config.secretKey,\n baseUrl: config.baseUrl,\n ...config.options,\n });\n }\n\n async exportEvent(event: AITracingEvent): Promise<void> {\n switch (event.type) {\n case 'span_started':\n await this.handleSpanStarted(event.span);\n break;\n case 'span_updated':\n await this.handleSpanUpdateOrEnd(event.span, true);\n break;\n case 'span_ended':\n await this.handleSpanUpdateOrEnd(event.span, false);\n break;\n }\n\n // Flush immediately in realtime mode for instant visibility\n if (this.realtime) {\n await this.client.flushAsync();\n }\n }\n\n private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n if (span.isRootSpan) {\n const trace = this.client.trace(this.buildTracePayload(span));\n this.traceMap.set(span.trace.id, { trace, spans: new Map() });\n }\n\n const traceData = this.traceMap.get(span.trace.id);\n if (!traceData) {\n console.log('NO TRACE');\n // TODO: log warning\n return;\n }\n\n const langfuseParent =\n span.parent && traceData.spans.has(span.parent.id)\n ? (traceData.spans.get(span.parent.id) as LangfuseSpan)\n : traceData.trace;\n\n const payload = this.buildSpanPayload(span, true);\n\n const langfuseSpan =\n span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n traceData.spans.set(span.id, langfuseSpan);\n }\n\n private async handleSpanUpdateOrEnd(span: AnyAISpan, isUpdate: boolean): Promise<void> {\n const traceData = this.traceMap.get(span.trace.id);\n if (!traceData) {\n console.log('NO TRACE');\n // TODO: log warning\n return;\n }\n\n const langfuseSpan = traceData.spans.get(span.id);\n if (!langfuseSpan) {\n console.log('NO SPAN');\n // TODO: log warning\n return;\n }\n\n if (isUpdate) {\n langfuseSpan.update(this.buildSpanPayload(span, false));\n } else {\n langfuseSpan.end(this.buildSpanPayload(span, false));\n\n if (span.isRootSpan) {\n traceData.trace.update({ output: span.output });\n this.traceMap.delete(span.trace.id);\n }\n }\n }\n\n private buildTracePayload(span: AnyAISpan): Record<string, any> {\n const payload: Record<string, any> = {\n id: span.trace.id,\n name: span.name,\n };\n\n const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};\n\n if (userId) payload.userId = userId;\n if (sessionId) payload.sessionId = sessionId;\n if (span.input) payload.input = span.input;\n\n payload.metadata = {\n spanType: span.type,\n ...sanitizeMetadata(span.attributes),\n ...sanitizeMetadata(remainingMetadata),\n };\n\n return payload;\n }\n\n private buildSpanPayload(span: AnyAISpan, isCreate: boolean): Record<string, any> {\n const payload: Record<string, any> = {};\n\n if (isCreate) {\n payload.id = span.id;\n payload.name = span.name;\n payload.startTime = span.startTime;\n if (span.input !== undefined) payload.input = span.input;\n }\n\n if (span.output !== undefined) payload.output = span.output;\n if (span.endTime !== undefined) payload.endTime = span.endTime;\n\n const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n // Strip special fields from metadata if used in top-level keys\n const attributesToOmit: string[] = [];\n\n if (span.type === AISpanType.LLM_GENERATION) {\n const llmAttr = attributes as LLMGenerationAttributes;\n\n if (llmAttr.model !== undefined) {\n payload.model = llmAttr.model;\n attributesToOmit.push('model');\n }\n\n if (llmAttr.usage !== undefined) {\n payload.usage = llmAttr.usage;\n attributesToOmit.push('usage');\n }\n\n if (llmAttr.parameters !== undefined) {\n payload.modelParameters = llmAttr.parameters;\n attributesToOmit.push('parameters');\n }\n }\n\n payload.metadata = {\n spanType: span.type,\n ...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),\n ...sanitizeMetadata(span.metadata),\n };\n\n if (span.errorInfo) {\n payload.level = 'ERROR';\n payload.statusMessage = span.errorInfo.message;\n }\n\n return payload;\n }\n\n async shutdown(): Promise<void> {\n await this.client.shutdownAsync();\n this.traceMap.clear();\n }\n}\n"]}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Langfuse Observability Provider for Mastra
|
|
3
|
+
*
|
|
4
|
+
* This package provides Langfuse-specific observability features for Mastra applications.
|
|
5
|
+
* Currently includes AI tracing support with plans for additional observability features.
|
|
6
|
+
*/
|
|
7
|
+
export * from './ai-tracing.js';
|
|
8
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,cAAc,cAAc,CAAC"}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,133 @@
|
|
|
1
|
+
import { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';
|
|
2
|
+
import { Langfuse } from 'langfuse';
|
|
3
|
+
|
|
4
|
+
// src/ai-tracing.ts
|
|
5
|
+
var LangfuseExporter = class {
|
|
6
|
+
name = "langfuse";
|
|
7
|
+
client;
|
|
8
|
+
realtime;
|
|
9
|
+
traceMap = /* @__PURE__ */ new Map();
|
|
10
|
+
constructor(config) {
|
|
11
|
+
this.realtime = config.realtime ?? false;
|
|
12
|
+
this.client = new Langfuse({
|
|
13
|
+
publicKey: config.publicKey,
|
|
14
|
+
secretKey: config.secretKey,
|
|
15
|
+
baseUrl: config.baseUrl,
|
|
16
|
+
...config.options
|
|
17
|
+
});
|
|
18
|
+
}
|
|
19
|
+
async exportEvent(event) {
|
|
20
|
+
switch (event.type) {
|
|
21
|
+
case "span_started":
|
|
22
|
+
await this.handleSpanStarted(event.span);
|
|
23
|
+
break;
|
|
24
|
+
case "span_updated":
|
|
25
|
+
await this.handleSpanUpdateOrEnd(event.span, true);
|
|
26
|
+
break;
|
|
27
|
+
case "span_ended":
|
|
28
|
+
await this.handleSpanUpdateOrEnd(event.span, false);
|
|
29
|
+
break;
|
|
30
|
+
}
|
|
31
|
+
if (this.realtime) {
|
|
32
|
+
await this.client.flushAsync();
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
async handleSpanStarted(span) {
|
|
36
|
+
if (span.isRootSpan) {
|
|
37
|
+
const trace = this.client.trace(this.buildTracePayload(span));
|
|
38
|
+
this.traceMap.set(span.trace.id, { trace, spans: /* @__PURE__ */ new Map() });
|
|
39
|
+
}
|
|
40
|
+
const traceData = this.traceMap.get(span.trace.id);
|
|
41
|
+
if (!traceData) {
|
|
42
|
+
console.log("NO TRACE");
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
const langfuseParent = span.parent && traceData.spans.has(span.parent.id) ? traceData.spans.get(span.parent.id) : traceData.trace;
|
|
46
|
+
const payload = this.buildSpanPayload(span, true);
|
|
47
|
+
const langfuseSpan = span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);
|
|
48
|
+
traceData.spans.set(span.id, langfuseSpan);
|
|
49
|
+
}
|
|
50
|
+
async handleSpanUpdateOrEnd(span, isUpdate) {
|
|
51
|
+
const traceData = this.traceMap.get(span.trace.id);
|
|
52
|
+
if (!traceData) {
|
|
53
|
+
console.log("NO TRACE");
|
|
54
|
+
return;
|
|
55
|
+
}
|
|
56
|
+
const langfuseSpan = traceData.spans.get(span.id);
|
|
57
|
+
if (!langfuseSpan) {
|
|
58
|
+
console.log("NO SPAN");
|
|
59
|
+
return;
|
|
60
|
+
}
|
|
61
|
+
if (isUpdate) {
|
|
62
|
+
langfuseSpan.update(this.buildSpanPayload(span, false));
|
|
63
|
+
} else {
|
|
64
|
+
langfuseSpan.end(this.buildSpanPayload(span, false));
|
|
65
|
+
if (span.isRootSpan) {
|
|
66
|
+
traceData.trace.update({ output: span.output });
|
|
67
|
+
this.traceMap.delete(span.trace.id);
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
buildTracePayload(span) {
|
|
72
|
+
const payload = {
|
|
73
|
+
id: span.trace.id,
|
|
74
|
+
name: span.name
|
|
75
|
+
};
|
|
76
|
+
const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};
|
|
77
|
+
if (userId) payload.userId = userId;
|
|
78
|
+
if (sessionId) payload.sessionId = sessionId;
|
|
79
|
+
if (span.input) payload.input = span.input;
|
|
80
|
+
payload.metadata = {
|
|
81
|
+
spanType: span.type,
|
|
82
|
+
...sanitizeMetadata(span.attributes),
|
|
83
|
+
...sanitizeMetadata(remainingMetadata)
|
|
84
|
+
};
|
|
85
|
+
return payload;
|
|
86
|
+
}
|
|
87
|
+
buildSpanPayload(span, isCreate) {
|
|
88
|
+
const payload = {};
|
|
89
|
+
if (isCreate) {
|
|
90
|
+
payload.id = span.id;
|
|
91
|
+
payload.name = span.name;
|
|
92
|
+
payload.startTime = span.startTime;
|
|
93
|
+
if (span.input !== void 0) payload.input = span.input;
|
|
94
|
+
}
|
|
95
|
+
if (span.output !== void 0) payload.output = span.output;
|
|
96
|
+
if (span.endTime !== void 0) payload.endTime = span.endTime;
|
|
97
|
+
const attributes = span.attributes ?? {};
|
|
98
|
+
const attributesToOmit = [];
|
|
99
|
+
if (span.type === AISpanType.LLM_GENERATION) {
|
|
100
|
+
const llmAttr = attributes;
|
|
101
|
+
if (llmAttr.model !== void 0) {
|
|
102
|
+
payload.model = llmAttr.model;
|
|
103
|
+
attributesToOmit.push("model");
|
|
104
|
+
}
|
|
105
|
+
if (llmAttr.usage !== void 0) {
|
|
106
|
+
payload.usage = llmAttr.usage;
|
|
107
|
+
attributesToOmit.push("usage");
|
|
108
|
+
}
|
|
109
|
+
if (llmAttr.parameters !== void 0) {
|
|
110
|
+
payload.modelParameters = llmAttr.parameters;
|
|
111
|
+
attributesToOmit.push("parameters");
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
payload.metadata = {
|
|
115
|
+
spanType: span.type,
|
|
116
|
+
...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),
|
|
117
|
+
...sanitizeMetadata(span.metadata)
|
|
118
|
+
};
|
|
119
|
+
if (span.errorInfo) {
|
|
120
|
+
payload.level = "ERROR";
|
|
121
|
+
payload.statusMessage = span.errorInfo.message;
|
|
122
|
+
}
|
|
123
|
+
return payload;
|
|
124
|
+
}
|
|
125
|
+
async shutdown() {
|
|
126
|
+
await this.client.shutdownAsync();
|
|
127
|
+
this.traceMap.clear();
|
|
128
|
+
}
|
|
129
|
+
};
|
|
130
|
+
|
|
131
|
+
export { LangfuseExporter };
|
|
132
|
+
//# sourceMappingURL=index.js.map
|
|
133
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/ai-tracing.ts"],"names":[],"mappings":";;;;AAiCO,IAAM,mBAAN,MAAoD;AAAA,EACzD,IAAA,GAAO,UAAA;AAAA,EACC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAAuB;AAAA,EAE9C,YAAY,MAAA,EAAgC;AAC1C,IAAA,IAAA,CAAK,QAAA,GAAW,OAAO,QAAA,IAAY,KAAA;AACnC,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,QAAA,CAAS;AAAA,MACzB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB,SAAS,MAAA,CAAO,OAAA;AAAA,MAChB,GAAG,MAAA,CAAO;AAAA,KACX,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,KAAA,EAAsC;AACtD,IAAA,QAAQ,MAAM,IAAA;AAAM,MAClB,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,IAAI,CAAA;AACvC,QAAA;AAAA,MACF,KAAK,cAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AACjD,QAAA;AAAA,MACF,KAAK,YAAA;AACH,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,KAAA,CAAM,IAAA,EAAM,KAAK,CAAA;AAClD,QAAA;AAAA;AAIJ,IAAA,IAAI,KAAK,QAAA,EAAU;AACjB,MAAA,MAAM,IAAA,CAAK,OAAO,UAAA,EAAW;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAgC;AAC9D,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAM,QAAQ,IAAA,CAAK,MAAA,CAAO,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAC,CAAA;AAC5D,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,KAAA,CAAM,EAAA,EAAI,EAAE,KAAA,EAAO,KAAA,kBAAO,IAAI,GAAA,EAAI,EAAG,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,iBACJ,IAAA,CAAK,MAAA,IAAU,SAAA,CAAU,KAAA,CAAM,IAAI,IAAA,CAAK,MAAA,CAAO,EAAE,CAAA,GAC5C,UAAU,KAAA,CAAM,GAAA,CAAI,KAAK,MAAA,CAAO,EAAE,IACnC,SAAA,CAAU,KAAA;AAEhB,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,IAAI,CAAA;AAEhD,IAAA,MAAM,YAAA,GACJ,IAAA,CAAK,IAAA,KAAS,UAAA,CAAW,cAAA,GAAiB,cAAA,CAAe,UAAA,CAAW,OAAO,CAAA,GAAI,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAE5G,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,YAAY,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,qBAAA,CAAsB,IAAA,EAAiB,QAAA,EAAkC;AACrF,IAAA,MAAM,YAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,MAAM,EAAE,CAAA;AACjD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,OAAA,CAAQ,IAAI,UAAU,CAAA;AAEtB,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAChD,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAA,OAAA,CAAQ,IAAI,SAAS,CAAA;AAErB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,YAAA,CAAa,MAAA,CAAO,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAAA,IACxD,CAAA,MAAO;AACL,MAAA,YAAA,CAAa,GAAA,CAAI,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,KAAK,CAAC,CAAA;AAEnD,MAAA,IAAI,KAAK,UAAA,EAAY;AACnB,QAAA,SAAA,CAAU,MAAM,MAAA,CAAO,EAAE,MAAA,EAAQ,IAAA,CAAK,QAAQ,CAAA;AAC9C,QAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,EAAE,CAAA;AAAA,MACpC;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,kBAAkB,IAAA,EAAsC;AAC9D,IAAA,MAAM,OAAA,GAA+B;AAAA,MACnC,EAAA,EAAI,KAAK,KAAA,CAAM,EAAA;AAAA,MACf,MAAM,IAAA,CAAK;AAAA,KACb;AAEA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,mBAAkB,GAAI,IAAA,CAAK,YAAY,EAAC;AAEtE,IAAA,IAAI,MAAA,UAAgB,MAAA,GAAS,MAAA;AAC7B,IAAA,IAAI,SAAA,UAAmB,SAAA,GAAY,SAAA;AACnC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,OAAA,CAAQ,KAAA,GAAQ,IAAA,CAAK,KAAA;AAErC,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,gBAAA,CAAiB,IAAA,CAAK,UAAU,CAAA;AAAA,MACnC,GAAG,iBAAiB,iBAAiB;AAAA,KACvC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEQ,gBAAA,CAAiB,MAAiB,QAAA,EAAwC;AAChF,IAAA,MAAM,UAA+B,EAAC;AAEtC,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAA,CAAQ,KAAK,IAAA,CAAK,EAAA;AAClB,MAAA,OAAA,CAAQ,OAAO,IAAA,CAAK,IAAA;AACpB,MAAA,OAAA,CAAQ,YAAY,IAAA,CAAK,SAAA;AACzB,MAAA,IAAI,IAAA,CAAK,KAAA,KAAU,MAAA,EAAW,OAAA,CAAQ,QAAQ,IAAA,CAAK,KAAA;AAAA,IACrD;AAEA,IAAA,IAAI,IAAA,CAAK,MAAA,KAAW,MAAA,EAAW,OAAA,CAAQ,SAAS,IAAA,CAAK,MAAA;AACrD,IAAA,IAAI,IAAA,CAAK,OAAA,KAAY,MAAA,EAAW,OAAA,CAAQ,UAAU,IAAA,CAAK,OAAA;AAEvD,IAAA,MAAM,UAAA,GAAc,IAAA,CAAK,UAAA,IAAc,EAAC;AAGxC,IAAA,MAAM,mBAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,UAAA,CAAW,cAAA,EAAgB;AAC3C,MAAA,MAAM,OAAA,GAAU,UAAA;AAEhB,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,UAAU,MAAA,EAAW;AAC/B,QAAA,OAAA,CAAQ,QAAQ,OAAA,CAAQ,KAAA;AACxB,QAAA,gBAAA,CAAiB,KAAK,OAAO,CAAA;AAAA,MAC/B;AAEA,MAAA,IAAI,OAAA,CAAQ,eAAe,MAAA,EAAW;AACpC,QAAA,OAAA,CAAQ,kBAAkB,OAAA,CAAQ,UAAA;AAClC,QAAA,gBAAA,CAAiB,KAAK,YAAY,CAAA;AAAA,MACpC;AAAA,IACF;AAEA,IAAA,OAAA,CAAQ,QAAA,GAAW;AAAA,MACjB,UAAU,IAAA,CAAK,IAAA;AAAA,MACf,GAAG,gBAAA,CAAiB,QAAA,CAAS,UAAA,EAAY,gBAAgB,CAAC,CAAA;AAAA,MAC1D,GAAG,gBAAA,CAAiB,IAAA,CAAK,QAAQ;AAAA,KACnC;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,OAAA,CAAQ,KAAA,GAAQ,OAAA;AAChB,MAAA,OAAA,CAAQ,aAAA,GAAgB,KAAK,SAAA,CAAU,OAAA;AAAA,IACzC;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,MAAM,IAAA,CAAK,OAAO,aAAA,EAAc;AAChC,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AAAA,EACtB;AACF","file":"index.js","sourcesContent":["/**\n * Langfuse Exporter for Mastra AI Tracing\n *\n * This exporter sends tracing data to Langfuse for AI observability.\n * Root spans start traces in Langfuse.\n * LLM_GENERATION spans become Langfuse generations, all others become spans.\n */\n\nimport type { AITracingExporter, AITracingEvent, AnyAISpan, LLMGenerationAttributes } from '@mastra/core/ai-tracing';\nimport { AISpanType, sanitizeMetadata, omitKeys } from '@mastra/core/ai-tracing';\nimport { Langfuse } from 'langfuse';\nimport type { LangfuseTraceClient, LangfuseSpanClient, LangfuseGenerationClient } from 'langfuse';\n\nexport interface LangfuseExporterConfig {\n /** Langfuse API key */\n publicKey: string;\n /** Langfuse secret key */\n secretKey: string;\n /** Langfuse host URL */\n baseUrl: string;\n /** Enable realtime mode - flushes after each event for immediate visibility */\n realtime?: boolean;\n /** Additional options to pass to the Langfuse client */\n options?: any;\n}\n\ntype TraceData = {\n trace: LangfuseTraceClient; // Langfuse trace object\n spans: Map<string, LangfuseSpanClient | LangfuseGenerationClient>; // Maps span.id to Langfuse span/generation\n};\n\ntype LangfuseSpan = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient;\n\nexport class LangfuseExporter implements AITracingExporter {\n name = 'langfuse';\n private client: Langfuse;\n private realtime: boolean;\n private traceMap = new Map<string, TraceData>();\n\n constructor(config: LangfuseExporterConfig) {\n this.realtime = config.realtime ?? false;\n this.client = new Langfuse({\n publicKey: config.publicKey,\n secretKey: config.secretKey,\n baseUrl: config.baseUrl,\n ...config.options,\n });\n }\n\n async exportEvent(event: AITracingEvent): Promise<void> {\n switch (event.type) {\n case 'span_started':\n await this.handleSpanStarted(event.span);\n break;\n case 'span_updated':\n await this.handleSpanUpdateOrEnd(event.span, true);\n break;\n case 'span_ended':\n await this.handleSpanUpdateOrEnd(event.span, false);\n break;\n }\n\n // Flush immediately in realtime mode for instant visibility\n if (this.realtime) {\n await this.client.flushAsync();\n }\n }\n\n private async handleSpanStarted(span: AnyAISpan): Promise<void> {\n if (span.isRootSpan) {\n const trace = this.client.trace(this.buildTracePayload(span));\n this.traceMap.set(span.trace.id, { trace, spans: new Map() });\n }\n\n const traceData = this.traceMap.get(span.trace.id);\n if (!traceData) {\n console.log('NO TRACE');\n // TODO: log warning\n return;\n }\n\n const langfuseParent =\n span.parent && traceData.spans.has(span.parent.id)\n ? (traceData.spans.get(span.parent.id) as LangfuseSpan)\n : traceData.trace;\n\n const payload = this.buildSpanPayload(span, true);\n\n const langfuseSpan =\n span.type === AISpanType.LLM_GENERATION ? langfuseParent.generation(payload) : langfuseParent.span(payload);\n\n traceData.spans.set(span.id, langfuseSpan);\n }\n\n private async handleSpanUpdateOrEnd(span: AnyAISpan, isUpdate: boolean): Promise<void> {\n const traceData = this.traceMap.get(span.trace.id);\n if (!traceData) {\n console.log('NO TRACE');\n // TODO: log warning\n return;\n }\n\n const langfuseSpan = traceData.spans.get(span.id);\n if (!langfuseSpan) {\n console.log('NO SPAN');\n // TODO: log warning\n return;\n }\n\n if (isUpdate) {\n langfuseSpan.update(this.buildSpanPayload(span, false));\n } else {\n langfuseSpan.end(this.buildSpanPayload(span, false));\n\n if (span.isRootSpan) {\n traceData.trace.update({ output: span.output });\n this.traceMap.delete(span.trace.id);\n }\n }\n }\n\n private buildTracePayload(span: AnyAISpan): Record<string, any> {\n const payload: Record<string, any> = {\n id: span.trace.id,\n name: span.name,\n };\n\n const { userId, sessionId, ...remainingMetadata } = span.metadata ?? {};\n\n if (userId) payload.userId = userId;\n if (sessionId) payload.sessionId = sessionId;\n if (span.input) payload.input = span.input;\n\n payload.metadata = {\n spanType: span.type,\n ...sanitizeMetadata(span.attributes),\n ...sanitizeMetadata(remainingMetadata),\n };\n\n return payload;\n }\n\n private buildSpanPayload(span: AnyAISpan, isCreate: boolean): Record<string, any> {\n const payload: Record<string, any> = {};\n\n if (isCreate) {\n payload.id = span.id;\n payload.name = span.name;\n payload.startTime = span.startTime;\n if (span.input !== undefined) payload.input = span.input;\n }\n\n if (span.output !== undefined) payload.output = span.output;\n if (span.endTime !== undefined) payload.endTime = span.endTime;\n\n const attributes = (span.attributes ?? {}) as Record<string, any>;\n\n // Strip special fields from metadata if used in top-level keys\n const attributesToOmit: string[] = [];\n\n if (span.type === AISpanType.LLM_GENERATION) {\n const llmAttr = attributes as LLMGenerationAttributes;\n\n if (llmAttr.model !== undefined) {\n payload.model = llmAttr.model;\n attributesToOmit.push('model');\n }\n\n if (llmAttr.usage !== undefined) {\n payload.usage = llmAttr.usage;\n attributesToOmit.push('usage');\n }\n\n if (llmAttr.parameters !== undefined) {\n payload.modelParameters = llmAttr.parameters;\n attributesToOmit.push('parameters');\n }\n }\n\n payload.metadata = {\n spanType: span.type,\n ...sanitizeMetadata(omitKeys(attributes, attributesToOmit)),\n ...sanitizeMetadata(span.metadata),\n };\n\n if (span.errorInfo) {\n payload.level = 'ERROR';\n payload.statusMessage = span.errorInfo.message;\n }\n\n return payload;\n }\n\n async shutdown(): Promise<void> {\n await this.client.shutdownAsync();\n this.traceMap.clear();\n }\n}\n"]}
|
package/eslint.config.js
ADDED
package/package.json
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@mastra/langfuse",
|
|
3
|
+
"version": "0.0.0-vector-query-tool-provider-options-20250828222356",
|
|
4
|
+
"description": "Langfuse observability provider for Mastra - includes AI tracing and future observability features",
|
|
5
|
+
"type": "module",
|
|
6
|
+
"main": "dist/index.js",
|
|
7
|
+
"types": "dist/index.d.ts",
|
|
8
|
+
"exports": {
|
|
9
|
+
".": {
|
|
10
|
+
"import": {
|
|
11
|
+
"types": "./dist/index.d.ts",
|
|
12
|
+
"default": "./dist/index.js"
|
|
13
|
+
},
|
|
14
|
+
"require": {
|
|
15
|
+
"types": "./dist/index.d.ts",
|
|
16
|
+
"default": "./dist/index.cjs"
|
|
17
|
+
}
|
|
18
|
+
},
|
|
19
|
+
"./package.json": "./package.json"
|
|
20
|
+
},
|
|
21
|
+
"license": "Apache-2.0",
|
|
22
|
+
"dependencies": {
|
|
23
|
+
"langfuse": "^3.38.4"
|
|
24
|
+
},
|
|
25
|
+
"devDependencies": {
|
|
26
|
+
"@microsoft/api-extractor": "^7.52.8",
|
|
27
|
+
"@types/node": "^20.19.0",
|
|
28
|
+
"eslint": "^9.30.1",
|
|
29
|
+
"tsup": "^8.5.0",
|
|
30
|
+
"typescript": "^5.8.3",
|
|
31
|
+
"vitest": "^3.2.4",
|
|
32
|
+
"@internal/lint": "0.0.0-vector-query-tool-provider-options-20250828222356",
|
|
33
|
+
"@mastra/core": "0.0.0-vector-query-tool-provider-options-20250828222356",
|
|
34
|
+
"@internal/types-builder": "0.0.0-vector-query-tool-provider-options-20250828222356"
|
|
35
|
+
},
|
|
36
|
+
"peerDependencies": {
|
|
37
|
+
"@mastra/core": "0.0.0-vector-query-tool-provider-options-20250828222356"
|
|
38
|
+
},
|
|
39
|
+
"scripts": {
|
|
40
|
+
"build": "tsup --silent --config tsup.config.ts",
|
|
41
|
+
"build:watch": "pnpm build --watch",
|
|
42
|
+
"test": "vitest run",
|
|
43
|
+
"test:watch": "vitest watch",
|
|
44
|
+
"lint": "eslint ."
|
|
45
|
+
}
|
|
46
|
+
}
|