pi-langfuse 1.5.19 → 1.6.1

package/src/utils.ts

@@ -334,6 +334,55 @@ export function getProviderPayload(event: Record<string, unknown>): unknown {
   return event.request ?? event.payload ?? event.body ?? event.providerPayload ?? event.messages ?? event;
 }
 
+export type ToolUpdateTransport =
+  | "anthropic-native"
+  | "openai-additional-tools"
+  | "openai-tool-search"
+  | "mid-conversation-system"
+  | "collapsed-leading-system";
+
+/** Classify the provider payload observed by this hook rather than guessing from the model name alone. */
+export function inferToolUpdateTransport(payload: unknown, hasPriorSystemState: boolean): ToolUpdateTransport | undefined {
+  let sawToolChangeBlock = false;
+  let sawAdditionalTools = false;
+  let sawToolSearch = false;
+  let sawTopLevelTools = false;
+  let sawMidConversationSystem = false;
+  let visited = 0;
+
+  const walk = (value: unknown, key?: string) => {
+    if (++visited > 2_000 || value === null || value === undefined) return;
+    if (key === "additional_tools") sawAdditionalTools = true;
+    if (key === "tools") sawTopLevelTools = true;
+    if (key === "tool_search" || key === "tool_search_output") sawToolSearch = true;
+    if (typeof value !== "object") return;
+    if (Array.isArray(value)) {
+      let sawNonSystem = false;
+      for (const item of value) {
+        if (item && typeof item === "object" && !Array.isArray(item)) {
+          const role = (item as Record<string, unknown>).role;
+          if (role === "system" && sawNonSystem) sawMidConversationSystem = true;
+          if (role && role !== "system") sawNonSystem = true;
+        }
+        walk(item);
+      }
+      return;
+    }
+    const record = value as Record<string, unknown>;
+    const type = record.type;
+    if (type === "tool_addition" || type === "tool_removal") sawToolChangeBlock = true;
+    for (const [childKey, child] of Object.entries(record)) walk(child, childKey);
+  };
+
+  walk(payload);
+  if (sawToolChangeBlock) return "anthropic-native";
+  if (sawAdditionalTools) return "openai-additional-tools";
+  if (sawToolSearch) return "openai-tool-search";
+  if (sawMidConversationSystem) return "mid-conversation-system";
+  if (hasPriorSystemState && sawTopLevelTools) return "collapsed-leading-system";
+  return undefined;
+}
+
 export function extractModelParameters(payload: unknown): Record<string, string | number> | undefined {
   if (!payload || typeof payload !== "object" || Array.isArray(payload)) {
     return undefined;
@@ -487,6 +536,32 @@ export function extractUsage(
   };
 }
 
+export function extractCacheMetrics(messageOrEvent: Record<string, unknown>): {
+  cacheReadTokens: number;
+  cacheWriteTokens: number;
+  uncachedInputTokens: number;
+  cacheHitRatio?: number;
+} | undefined {
+  const usage = (messageOrEvent.usage ??
+    (messageOrEvent.message && typeof messageOrEvent.message === "object"
+      ? (messageOrEvent.message as Record<string, unknown>).usage
+      : undefined)) as Record<string, unknown> | undefined;
+  if (!usage || typeof usage !== "object") return undefined;
+
+  const uncachedInputTokens = Number(
+    usage.input ?? usage.inputTokens ?? usage.prompt_tokens ?? usage.promptTokens ?? 0,
+  );
+  const cacheReadTokens = Number(usage.cacheRead ?? usage.cache_read ?? usage.cachedTokens ?? 0);
+  const cacheWriteTokens = Number(usage.cacheWrite ?? usage.cache_write ?? 0);
+  const denominator = uncachedInputTokens + cacheReadTokens;
+  return {
+    cacheReadTokens,
+    cacheWriteTokens,
+    uncachedInputTokens,
+    ...(denominator > 0 ? { cacheHitRatio: cacheReadTokens / denominator } : {}),
+  };
+}
+
 export function extractCostDetails(messageOrEvent: Record<string, unknown>): Record<string, number> | undefined {
   const usage = (messageOrEvent.usage ??
     (messageOrEvent.message && typeof messageOrEvent.message === "object"

package/types/langfuse-runtime-shims.d.ts

@@ -1,49 +0,0 @@
-declare module "@opentelemetry/sdk-trace-base" {
-  export class BasicTracerProvider {
-    constructor(options?: { resource?: unknown; spanProcessors?: unknown[] });
-    forceFlush?(): Promise<void>;
-    shutdown?(): Promise<void>;
-  }
-}
-
-declare module "@langfuse/otel" {
-  export class LangfuseSpanProcessor {
-    constructor(options: {
-      publicKey: string;
-      secretKey: string;
-      baseUrl: string;
-    });
-    forceFlush?(): Promise<void>;
-    shutdown?(): Promise<void>;
-  }
-}
-
-declare module "@langfuse/tracing" {
-  export function setLangfuseTracerProvider(provider: unknown): void;
-
-  export function startObservation(
-    name: string,
-    body?: Record<string, unknown>,
-    options?: { asType?: string },
-  ): unknown;
-
-  export function propagateAttributes<T>(
-    params: {
-      sessionId?: string;
-      traceName?: string;
-      metadata?: Record<string, string>;
-      tags?: string[];
-    },
-    fn: () => T,
-  ): T;
-}
-
-declare module "@langfuse/client" {
-  export class LangfuseClient {
-    constructor(options: {
-      publicKey: string;
-      secretKey: string;
-      baseUrl: string;
-    });
-  }
-}

package/types/node-shims.d.ts

@@ -1,29 +0,0 @@
-declare module "node:fs" {
-  export function mkdirSync(path: string, options?: { recursive?: boolean }): void;
-  export function readFileSync(path: string, encoding: string): string;
-  export function existsSync(path: string): boolean;
-  export function writeFileSync(path: string, data: string, encoding: string): void;
-}
-
-declare module "node:crypto" {
-  export function randomUUID(): string;
-}
-
-declare module "node:os" {
-  export function homedir(): string;
-}
-
-declare module "node:path" {
-  export function resolve(...paths: string[]): string;
-  export function dirname(path: string): string;
-  export function basename(path: string, suffix?: string): string;
-}
-
-declare module "node:url" {
-  export function fileURLToPath(url: string | URL): string;
-}
-
-declare const process: {
-  cwd(): string;
-  env: Record<string, string | undefined>;
-};

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

@@ -1,16 +0,0 @@
-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(
-      name: string,
-      options: { description?: string; handler: (args: string, ctx: any) => unknown },
-    ): void;
-  }
-}