@narumitw/pi-langfuse 0.49.3 → 0.49.4

package/README.md

@@ -35,7 +35,7 @@ pi -e npm:@narumitw/pi-langfuse
 Try a local checkout:
 
 ```bash
-pi -e ./extensions/pi-langfuse
+pi -e ./packages/pi-langfuse
 ```
 
 The Langfuse v4 SDK requires Node.js 20 or newer.
@@ -173,7 +173,7 @@ The built-in mask specifically protects Langfuse credentials; it is not a genera
 ## 🗂️ Package layout
 
 ```txt
-extensions/pi-langfuse/
+packages/pi-langfuse/
 ├── src/
 │   ├── index.ts     # Pi package entrypoint
 │   ├── langfuse.ts  # Pi lifecycle integration and slash command

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@narumitw/pi-langfuse",
-	"version": "0.49.3",
+	"version": "0.49.4",
 	"description": "Pi extension that traces LLM generations and tool activity to Langfuse.",
 	"type": "module",
 	"license": "MIT",
@@ -23,6 +23,9 @@
 			"./src/index.ts"
 		]
 	},
+	"piExtension": {
+		"lifecycle": "stable"
+	},
 	"scripts": {
 		"check": "biome check . && npm run typecheck",
 		"format": "biome check --write .",
@@ -32,9 +35,9 @@
 		"@earendil-works/pi-coding-agent": "*"
 	},
 	"devDependencies": {
-		"@biomejs/biome": "2.5.7",
-		"@earendil-works/pi-coding-agent": "0.83.0",
-		"@types/node": "26.1.2",
+		"@biomejs/biome": "2.5.8",
+		"@earendil-works/pi-coding-agent": "0.84.2",
+		"@types/node": "26.2.0",
 		"typescript": "7.0.2"
 	},
 	"dependencies": {
@@ -49,6 +52,6 @@
 	"repository": {
 		"type": "git",
 		"url": "https://github.com/narumiruna/pi-extensions",
-		"directory": "extensions/pi-langfuse"
+		"directory": "packages/pi-langfuse"
 	}
 }

package/src/runtime.ts

@@ -39,7 +39,9 @@ class ProductionObservation implements Observation {
 	constructor(readonly native: LangfuseObservation) {}
 
 	update(attributes: ObservationAttributes): Observation {
-		this.native.updateOtelSpanAttributes(attributes as LangfuseObservationAttributes);
+		const { sessionId, ...observationAttributes } = attributes;
+		this.native.updateOtelSpanAttributes(observationAttributes as LangfuseObservationAttributes);
+		applySessionId(this.native, sessionId);
 		return this;
 	}
 
@@ -51,9 +53,7 @@ class ProductionObservation implements Observation {
 		if (name !== undefined) {
 			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_NAME, name);
 		}
-		if (sessionId !== undefined) {
-			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_SESSION_ID, sessionId);
-		}
+		applySessionId(this.native, sessionId);
 		if (tags !== undefined) {
 			this.native.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_TAGS, tags);
 		}
@@ -86,11 +86,14 @@ class ProductionTraceBackend implements TraceBackend {
 		attributes: ObservationAttributes,
 		options: { asType: ObservationType; parent?: Observation },
 	): Observation {
+		const { sessionId, ...observationAttributes } = attributes;
 		const parent = options.parent;
-		if (parent instanceof ProductionObservation) {
-			return new ProductionObservation(startChild(parent.native, name, attributes, options.asType));
-		}
-		return new ProductionObservation(startRoot(name, attributes, options.asType));
+		const native =
+			parent instanceof ProductionObservation
+				? startChild(parent.native, name, observationAttributes, options.asType)
+				: startRoot(name, observationAttributes, options.asType);
+		applySessionId(native, sessionId);
+		return new ProductionObservation(native);
 	}
 
 	async forceFlush(): Promise<void> {
@@ -177,6 +180,12 @@ const defaultFactories: RuntimeFactories = {
 	selectProvider: setLangfuseTracerProvider,
 };
 
+function applySessionId(observation: LangfuseObservation, sessionId: string | undefined): void {
+	if (sessionId !== undefined) {
+		observation.otelSpan.setAttribute(LangfuseOtelSpanAttributes.TRACE_SESSION_ID, sessionId);
+	}
+}
+
 function startRoot(
 	name: string,
 	attributes: ObservationAttributes,

package/src/tracing.ts

@@ -234,6 +234,14 @@ export class TraceRecorder {
 		private readonly context: RecorderContext,
 	) {}
 
+	private startObservation(
+		name: string,
+		attributes: ObservationAttributes,
+		options: { asType: ObservationType; parent?: Observation },
+	): Observation {
+		return this.backend.start(name, { ...attributes, sessionId: this.context.sessionId }, options);
+	}
+
 	hasActiveTrace(): boolean {
 		return this.root !== undefined;
 	}
@@ -275,7 +283,7 @@ export class TraceRecorder {
 				? "git:detached"
 				: undefined;
 
-		this.root = this.backend.start("pi.agent", attributes, { asType: "agent" });
+		this.root = this.startObservation("pi.agent", attributes, { asType: "agent" });
 		this.root.updateTrace?.({
 			name: "pi.trace",
 			sessionId: this.context.sessionId,
@@ -296,7 +304,7 @@ export class TraceRecorder {
 		this.lastAssistant = undefined;
 		this.unresolvedToolErrors = 0;
 		this.attemptIndex = index;
-		this.attempt = this.backend.start(
+		this.attempt = this.startObservation(
 			"pi.attempt",
 			{
 				metadata: {
@@ -325,7 +333,7 @@ export class TraceRecorder {
 		if (this.turn) this.closeTurn("Interrupted by the next Pi turn.", "ERROR");
 		this.counters.turns += 1;
 		this.turnIndex = turnIndex;
-		this.turn = this.backend.start(
+		this.turn = this.startObservation(
 			"pi.turn",
 			{ metadata: { "pi.turn.index": turnIndex }, version: TRACE_SCHEMA_VERSION },
 			{ asType: "span", parent: this.attempt ?? this.root },
@@ -372,7 +380,7 @@ export class TraceRecorder {
 			...(input.model?.api ? { "pi.request.api": input.model.api } : {}),
 			...(input.thinkingLevel ? { "pi.request.thinking_level": input.thinkingLevel } : {}),
 		};
-		const observation = this.backend.start(
+		const observation = this.startObservation(
 			"pi.llm",
 			{
 				...(input.payload !== undefined ? { input: this.capture(input.payload) } : {}),
@@ -559,7 +567,7 @@ export class TraceRecorder {
 		this.closeCompaction("Interrupted by another Pi compaction.");
 		this.counters.compactions += 1;
 		this.compaction = {
-			observation: this.backend.start(
+			observation: this.startObservation(
 				"pi.compaction",
 				{
 					metadata: {
@@ -711,7 +719,7 @@ export class TraceRecorder {
 	}
 
 	private startToolObservation(toolCallId: string, toolName: string, args?: unknown): Observation {
-		return this.backend.start(
+		return this.startObservation(
 			`pi.tool.${toolName}`,
 			{
 				...(args !== undefined ? { input: this.capture(args) } : {}),