pi-langfuse 1.5.8 → 1.5.10

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

@@ -8,7 +8,7 @@
  */
 
 import { basename } from "node:path";
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 import { state, resetRunState, runWithSession, setCurrentSession } from "./src/state.js";
 import { ensureConfig, promptForConfig, loadConfig } from "./src/config.js";
@@ -74,10 +74,18 @@ export default async function (pi: ExtensionAPI) {
     },
   });
 
-  const getSessionId = (ctx?: any) => {
+  const getSessionId = (ctx?: unknown): string | undefined => {
     try {
-      const sessionFile = ctx?.sessionManager?.getSessionFile?.();
-      return sessionFile ? basename(sessionFile, ".jsonl") : undefined;
+      const sessionManager = (ctx as ExtensionContext | undefined)?.sessionManager;
+      const sessionId = sessionManager?.getSessionId?.();
+      if (typeof sessionId === "string" && sessionId) {
+        return sessionId;
+      }
+
+      const sessionFile = sessionManager?.getSessionFile?.();
+      return typeof sessionFile === "string" && sessionFile
+        ? basename(sessionFile, ".jsonl")
+        : undefined;
     } catch {
       return undefined;
     }
@@ -159,11 +167,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.10",
 	"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) {
@@ -211,6 +219,9 @@ async function flushPendingScores(rt: LangfuseRuntime, signal: AbortSignal): Pro
 
   while (pendingScores.length > 0) {
     const scores = pendingScores.slice(0, MAX_SCORE_BATCH_SIZE);
+    for (const score of scores) {
+      score.id ??= randomUUID();
+    }
     try {
       const errors = await ingestBatch(
         rt,
@@ -642,6 +653,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 +667,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 +682,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;

package/src/types.ts

@@ -60,6 +60,7 @@ export interface LangfuseScoreClient {
 }
 
 export interface PendingScore {
+  id?: string;
   traceId?: string;
   sessionId?: string;
   observationId?: string;

package/types/pi-coding-agent.d.ts

@@ -1,4 +1,11 @@
 declare module "@earendil-works/pi-coding-agent" {
+  export interface ExtensionContext {
+    sessionManager?: {
+      getSessionId?: () => unknown;
+      getSessionFile?: () => unknown;
+    };
+  }
+
   export interface ExtensionAPI {
     on(event: string, handler: (event: any, ctx: any) => unknown): void;
     registerCommand(