@narumitw/pi-langfuse 0.28.0 → 0.31.0

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@narumitw/pi-langfuse",
-	"version": "0.28.0",
+	"version": "0.31.0",
 	"description": "Pi extension that traces LLM generations and tool activity to Langfuse.",
 	"type": "module",
 	"license": "MIT",
@@ -30,15 +30,15 @@
 	},
 	"dependencies": {
 		"@langfuse/otel": "^5.9.1",
-		"@langfuse/tracing": "^4.0.0",
+		"@langfuse/tracing": "^5.9.1",
 		"@opentelemetry/api": "^1.9.1",
-		"@opentelemetry/exporter-trace-otlp-http": "^0.220.0",
-		"@opentelemetry/sdk-trace-base": "^2.9.0",
-		"@opentelemetry/sdk-trace-node": "^2.9.0"
+		"@opentelemetry/exporter-trace-otlp-http": "^0.221.0",
+		"@opentelemetry/sdk-trace-base": "^2.10.0",
+		"@opentelemetry/sdk-trace-node": "^2.10.0"
 	},
 	"devDependencies": {
-		"@biomejs/biome": "2.5.3",
-		"@earendil-works/pi-coding-agent": "0.80.10",
+		"@biomejs/biome": "2.5.5",
+		"@earendil-works/pi-coding-agent": "0.82.1",
 		"@types/node": "26.1.1",
 		"typescript": "7.0.2"
 	},

package/src/runtime.ts

@@ -3,7 +3,7 @@ import { LangfuseSpanProcessor } from "@langfuse/otel";
 import {
 	type LangfuseObservation,
 	type LangfuseObservationAttributes,
-	type LangfuseTraceAttributes,
+	LangfuseOtelSpanAttributes,
 	setLangfuseTracerProvider,
 	startObservation,
 } from "@langfuse/tracing";
@@ -44,7 +44,28 @@ class ProductionObservation implements Observation {
 	}
 
 	updateTrace(attributes: ObservationAttributes): Observation {
-		this.native.updateTrace(attributes as LangfuseTraceAttributes);
+		const { input, output, metadata, name, sessionId, tags, version } = attributes;
+		if (input !== undefined || output !== undefined) {
+			this.native.setTraceIO({ input, output });
+		}
+		if (name !== undefined) {
+			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_NAME, name);
+		}
+		if (sessionId !== undefined) {
+			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_SESSION_ID, sessionId);
+		}
+		if (tags !== undefined) {
+			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_TAGS, tags);
+		}
+		if (version !== undefined) {
+			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.VERSION, version);
+		}
+		for (const [key, value] of Object.entries(metadata ?? {})) {
+			this.native.otelSpan.setAttribute(
+				`${LangfuseOtelSpanAttributes.TRACE_METADATA}.${key}`,
+				String(value),
+			);
+		}
 		return this;
 	}
 
@@ -67,9 +88,7 @@ class ProductionTraceBackend implements TraceBackend {
 	): Observation {
 		const parent = options.parent;
 		if (parent instanceof ProductionObservation) {
-			return new ProductionObservation(
-				startChild(parent.native, name, attributes, options.asType),
-			);
+			return new ProductionObservation(startChild(parent.native, name, attributes, options.asType));
 		}
 		return new ProductionObservation(startRoot(name, attributes, options.asType));
 	}
@@ -236,10 +255,7 @@ function maskSecretValue(
 			return data.map((item) => maskSecretValue(item, secrets, active));
 		}
 		return Object.fromEntries(
-			Object.entries(data).map(([key, value]) => [
-				key,
-				maskSecretValue(value, secrets, active),
-			]),
+			Object.entries(data).map(([key, value]) => [key, maskSecretValue(value, secrets, active)]),
 		);
 	} finally {
 		active.delete(data);

package/src/tracing.ts

@@ -243,8 +243,7 @@ export class TraceRecorder {
 	}
 
 	beginAgent(input: BeginAgentInput): void {
-		if (this.root)
-			this.closeActiveTrace("Interrupted by a new Pi run.", undefined, "interrupted");
+		if (this.root) this.closeActiveTrace("Interrupted by a new Pi run.", undefined, "interrupted");
 
 		this.lastOutput = undefined;
 		this.lastAssistant = undefined;