pi-langfuse 1.3.1 → 1.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-langfuse",
-	"version": "1.3.1",
+	"version": "1.3.2",
 	"description": "Langfuse extension for Pi coding agent",
 	"repository": {
 		"type": "git",
@@ -34,7 +34,8 @@
 		"@langfuse/client": "^5.3.0",
 		"@langfuse/otel": "^5.3.0",
 		"@langfuse/tracing": "^5.3.0",
-		"@opentelemetry/sdk-node": "^0.218.0"
+		"@opentelemetry/sdk-node": "^0.218.0",
+		"@opentelemetry/sdk-trace-base": "^2.0.1"
 	},
 	"peerDependencies": {
 		"@earendil-works/pi-coding-agent": "*"

package/src/config.ts

@@ -42,23 +42,10 @@ export function saveConfig(config: Config) {
   writeFileSync(CONFIG_PATH, `${JSON.stringify(config, null, 2)}\n`, "utf-8");
 }
 
-export async function ensureConfig(ctx: any): Promise<boolean> {
-  if (!state.config) {
-    state.config = loadConfigFromEnv() || loadConfigFromFile();
-  }
-
-  if (state.config) {
-    return true;
-  }
-
-  if (state.setupAttemptedThisSession) {
-    return false;
-  }
-  state.setupAttemptedThisSession = true;
-
+async function collectConfigFromUI(ctx: any, reason: string): Promise<Config | null> {
   if (!ctx.hasUI) {
-    console.log("📊 Langfuse: Missing config. Run this extension in Pi UI to complete setup, or set LANGFUSE_PUBLIC_KEY / LANGFUSE_SECRET_KEY / LANGFUSE_BASE_URL.");
-    return false;
+    console.log(`📊 Langfuse: ${reason}. Run this extension in Pi UI to complete setup, or set LANGFUSE_PUBLIC_KEY / LANGFUSE_SECRET_KEY / LANGFUSE_BASE_URL.`);
+    return null;
   }
 
   ctx.ui.notify("Langfuse setup required. Enter your API keys to enable tracing.", "info");
@@ -66,21 +53,25 @@ export async function ensureConfig(ctx: any): Promise<boolean> {
   const publicKey = (await ctx.ui.input("Langfuse public key:", "pk-lf-..."))?.trim();
   if (!publicKey) {
     ctx.ui.notify("Langfuse setup cancelled.", "warning");
-    return false;
+    return null;
   }
 
   const secretKey = (await ctx.ui.input("Langfuse secret key:", "sk-lf-..."))?.trim();
   if (!secretKey) {
     ctx.ui.notify("Langfuse setup cancelled.", "warning");
-    return false;
+    return null;
   }
 
   const hostInput = (await ctx.ui.input("Langfuse host:", DEFAULT_LANGFUSE_HOST))?.trim();
-  state.config = {
+  return {
     publicKey,
     secretKey,
     host: hostInput || DEFAULT_LANGFUSE_HOST,
   };
+}
+
+async function saveConfigFromUI(ctx: any, config: Config): Promise<boolean> {
+  state.config = config;
 
   try {
     saveConfig(state.config);
@@ -94,9 +85,42 @@ export async function ensureConfig(ctx: any): Promise<boolean> {
   }
 }
 
+export async function ensureConfig(ctx: any): Promise<boolean> {
+  if (!state.config) {
+    state.config = loadConfigFromEnv() || loadConfigFromFile();
+  }
+
+  if (state.config) {
+    return true;
+  }
+
+  if (state.setupAttemptedThisSession) {
+    return false;
+  }
+  state.setupAttemptedThisSession = true;
+
+  const config = await collectConfigFromUI(ctx, "Missing config");
+  if (!config) {
+    return false;
+  }
+
+  return saveConfigFromUI(ctx, config);
+}
+
 export async function promptForConfig(ctx: any): Promise<boolean> {
   state.setupAttemptedThisSession = false;
   state.config = null;
   await shutdownRuntime();
-  return ensureConfig(ctx);
+
+  const config = await collectConfigFromUI(ctx, "Manual setup requested");
+  if (!config) {
+    state.config = loadConfigFromEnv() || loadConfigFromFile();
+    return false;
+  }
+
+  const saved = await saveConfigFromUI(ctx, config);
+  if (saved) {
+    ctx.ui.notify("Langfuse tracing enabled for future agent runs.", "info");
+  }
+  return saved;
 }

package/src/langfuse.ts

@@ -9,8 +9,8 @@ export async function getRuntime(): Promise<LangfuseRuntime> {
   }
 
   if (!runtime) {
-    const [{ NodeSDK }, { LangfuseSpanProcessor }, tracing, { LangfuseClient }] = await Promise.all([
-      import("@opentelemetry/sdk-node"),
+    const [{ BasicTracerProvider }, { LangfuseSpanProcessor }, tracing, { LangfuseClient }] = await Promise.all([
+      import("@opentelemetry/sdk-trace-base"),
       import("@langfuse/otel"),
       import("@langfuse/tracing"),
       import("@langfuse/client"),
@@ -21,8 +21,8 @@ export async function getRuntime(): Promise<LangfuseRuntime> {
       secretKey: state.config.secretKey,
       baseUrl: state.config.host,
     });
-    const sdk = new NodeSDK({ spanProcessors: [spanProcessor] });
-    sdk.start();
+    const tracerProvider = new BasicTracerProvider({ spanProcessors: [spanProcessor] });
+    tracing.setLangfuseTracerProvider(tracerProvider);
 
     runtime = {
       startObservation: tracing.startObservation as unknown as LangfuseRuntime["startObservation"],
@@ -33,7 +33,8 @@ export async function getRuntime(): Promise<LangfuseRuntime> {
         baseUrl: state.config.host,
       }) as LangfuseScoreClient,
       spanProcessor,
-      sdk,
+      tracerProvider,
+      clearTracerProvider: () => tracing.setLangfuseTracerProvider(null),
     };
   }
 
@@ -48,12 +49,12 @@ export async function shutdownRuntime(): Promise<void> {
   try {
     await runtime.scoreClient.flush?.();
     await runtime.scoreClient.shutdown?.();
-    await runtime.spanProcessor?.forceFlush?.();
-    await runtime.spanProcessor?.shutdown?.();
-    await runtime.sdk?.shutdown?.();
+    await runtime.tracerProvider?.forceFlush?.();
+    await runtime.tracerProvider?.shutdown?.();
   } catch (e) {
     console.warn("📊 Langfuse: Failed to flush/shutdown cleanly", e);
   } finally {
+    runtime.clearTracerProvider?.();
     runtime = null;
   }
 }

package/src/types.ts

@@ -62,7 +62,8 @@ export interface LangfuseRuntime {
   ) => LangfuseObservation;
   scoreClient: LangfuseScoreClient;
   spanProcessor?: { forceFlush?: () => Promise<void>; shutdown?: () => Promise<void> };
-  sdk?: { start?: () => void; shutdown?: () => Promise<void> };
+  tracerProvider?: { forceFlush?: () => Promise<void>; shutdown?: () => Promise<void> };
+  clearTracerProvider?: () => void;
 }
 
 export interface GenerationState {