@narumitw/pi-langfuse 0.26.0 → 0.31.0

package/README.md

@@ -168,6 +168,7 @@ The built-in mask specifically protects Langfuse credentials; it is not a genera
 ```txt
 extensions/pi-langfuse/
 ├── src/
+│   ├── index.ts     # Pi package entrypoint
 │   ├── langfuse.ts  # Pi lifecycle integration and slash command
 │   ├── tracing.ts   # Observation lifecycle, outcomes, and bounded metadata
 │   ├── sanitizer.ts # Content bounding and opaque-signature removal
@@ -180,7 +181,7 @@ extensions/pi-langfuse/
 └── package.json
 ```
 
-Only `langfuse.ts` is a Pi entrypoint; the other source modules are internal.
+`index.ts` is the Pi entrypoint and forwards to `langfuse.ts`; the other source modules are internal.
 
 ## 🔎 Keywords
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@narumitw/pi-langfuse",
-	"version": "0.26.0",
+	"version": "0.31.0",
 	"description": "Pi extension that traces LLM generations and tool activity to Langfuse.",
 	"type": "module",
 	"license": "MIT",
@@ -20,7 +20,7 @@
 	],
 	"pi": {
 		"extensions": [
-			"./src/langfuse.ts"
+			"./src/index.ts"
 		]
 	},
 	"scripts": {
@@ -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/index.ts

@@ -0,0 +1 @@
+export { default } from "./langfuse.js";

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;