pi-langfuse 1.5.8 → 1.5.9

package/README.md

@@ -77,6 +77,14 @@ export LANGFUSE_BASE_URL="https://cloud.langfuse.com"  # optional; LANGFUSE_HOST
 
 Saved config takes precedence. Environment variables are only used when `~/.pi/agent/pi-langfuse/config.json` is missing or incomplete.
 
+For short-lived SDK hosts, set the bounded final score-delivery attempt during shutdown:
+
+```bash
+export PI_LANGFUSE_SCORE_SHUTDOWN_TIMEOUT=2  # seconds; defaults to 2 seconds
+```
+
+The extension attempts queued trace-level scores before other shutdown telemetry work. This value cannot extend the overall shutdown deadline.
+
 Privacy controls can also be set through environment variables:
 
 ```bash

package/README_CN.md

@@ -77,6 +77,14 @@ export LANGFUSE_BASE_URL="https://cloud.langfuse.com"  # 可选；也支持 LANG
 
 保存的配置优先级更高。只有当 `~/.pi/agent/pi-langfuse/config.json` 缺失或不完整时，扩展才会使用环境变量。
 
+对于短生命周期的 SDK 宿主，可设置关闭时最终分数发送尝试的上限：
+
+```bash
+export PI_LANGFUSE_SCORE_SHUTDOWN_TIMEOUT=2  # 单位为秒；默认 2 秒
+```
+
+扩展会在其他关闭遥测工作之前尝试发送已排队的 trace 级分数。该值不会延长总关闭超时。
+
 隐私采集策略也可以通过环境变量设置：
 
 ```bash

package/index.ts

@@ -159,11 +159,11 @@ export default async function (pi: ExtensionAPI) {
   pi.on("agent_end", async (event, ctx) => withSession(ctx, async () => {
     await finishAgentRun(event);
     const sessionId = state.currentSessionId;
-    setTimeout(() => {
-      shutdownRuntime(sessionId).catch((error) => {
-        console.warn("📊 Langfuse: Deferred shutdown failed", error);
-      });
-    }, 0);
+    try {
+      await shutdownRuntime(sessionId);
+    } catch (error) {
+      console.warn("📊 Langfuse: Shutdown failed", error);
+    }
   }));
 
   const handleSessionInterruption = (reason: string) => {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-langfuse",
-	"version": "1.5.8",
+	"version": "1.5.9",
 	"description": "Langfuse extension for Pi coding agent",
 	"repository": {
 		"type": "git",

package/src/langfuse.ts

@@ -60,6 +60,7 @@ interface RestFallbackStore {
 const OTEL_VISIBILITY_TIMEOUT_MS = 1_500;
 const OTEL_VISIBILITY_POLL_INTERVAL_MS = 200;
 const DEFAULT_SHUTDOWN_STEP_TIMEOUT_MS = 2_000;
+const DEFAULT_SCORE_SHUTDOWN_TIMEOUT_MS = 2_000;
 const DEFAULT_LANGFUSE_REQUEST_TIMEOUT_SECONDS = 5;
 const DEFAULT_SCORE_FLUSH_AT = 10;
 const DEFAULT_SCORE_FLUSH_INTERVAL_MS = 1_000;
@@ -80,6 +81,13 @@ function resolvePositiveEnvNumber(name: string, fallback: number, integer = fals
   return integer ? Math.floor(parsed) : parsed;
 }
 
+function getScoreShutdownTimeoutMs(): number {
+  return resolvePositiveEnvNumber(
+    "PI_LANGFUSE_SCORE_SHUTDOWN_TIMEOUT",
+    DEFAULT_SCORE_SHUTDOWN_TIMEOUT_MS / 1_000,
+  ) * 1_000;
+}
+
 function delay(ms: number, signal?: AbortSignal) {
   return new Promise<void>((resolve, reject) => {
     if (signal?.aborted) {
@@ -642,6 +650,12 @@ function doShutdownRuntime(): Promise<void> {
     const deadline = Date.now() + shutdownStepTimeoutMs;
     const controller = new AbortController();
     const abortTimeout = setTimeout(() => controller.abort(), shutdownStepTimeoutMs);
+    const scoreController = new AbortController();
+    const scoreAbortTimeout = setTimeout(
+      () => scoreController.abort(),
+      Math.min(getScoreShutdownTimeoutMs(), shutdownStepTimeoutMs),
+    );
+    scoreAbortTimeout.unref?.();
     stopScoreFlush(rt);
 
     try {
@@ -650,13 +664,13 @@ function doShutdownRuntime(): Promise<void> {
         () => rt.scoreFlushPromise,
         deadline,
       );
+      await flushPendingScores(rt, scoreController.signal);
       await withShutdownDeadline("OTel force flush", () => rt.tracerProvider?.forceFlush?.(), deadline);
       await withShutdownDeadline(
         "REST fallback ingestion",
         () => fallbackToRestIngestion(rt, controller.signal),
         deadline,
       );
-      await flushPendingScores(rt, controller.signal);
       await withShutdownDeadline("Langfuse score flush", () => rt.scoreClient.flush?.(), deadline);
       await withShutdownDeadline("Langfuse client shutdown", () => rt.scoreClient.shutdown?.(), deadline);
       await withShutdownDeadline("OTel tracer shutdown", () => rt.tracerProvider?.shutdown?.(), deadline);
@@ -665,6 +679,8 @@ function doShutdownRuntime(): Promise<void> {
       console.warn("📊 Langfuse: Failed to flush/shutdown cleanly", e);
     } finally {
       clearTimeout(abortTimeout);
+      clearTimeout(scoreAbortTimeout);
+      scoreController.abort();
       clearScoreFlushTimer(rt);
       rt.scoreFlushController?.abort();
       rt.scoreFlushController = undefined;