@librechat/agents 3.2.64 → 3.2.66

package/dist/cjs/instrumentation.cjs

@@ -41,25 +41,37 @@ function ensureOpenTelemetryContextManager() {
 	contextManager.enable();
 	if (!_opentelemetry_api.context.setGlobalContextManager(contextManager)) contextManager.disable();
 }
+function resolveLangfuseEnvironment(langfuse) {
+	const candidates = [
+		langfuse?.environment,
+		process.env.LANGFUSE_TRACING_ENVIRONMENT,
+		process.env.NODE_ENV
+	];
+	for (const candidate of candidates) if (candidate != null && candidate.trim() !== "") return candidate.trim();
+}
 function getLangfuseSpanProcessorParams(langfuse) {
 	if (langfuse?.enabled === false) return;
+	const environment = resolveLangfuseEnvironment(langfuse);
 	if (require_langfuse.hasLangfuseConfigCredentials(langfuse)) return {
 		publicKey: langfuse.publicKey,
 		secretKey: langfuse.secretKey,
-		...require_misc.isPresent(langfuse.baseUrl) ? { baseUrl: langfuse.baseUrl } : {}
+		...require_misc.isPresent(langfuse.baseUrl) ? { baseUrl: langfuse.baseUrl } : {},
+		...require_misc.isPresent(environment) ? { environment } : {}
 	};
 	if (require_langfuse.hasLangfuseEnvConfig()) {
 		const baseUrl = langfuse?.baseUrl ?? process.env.LANGFUSE_BASE_URL ?? process.env.LANGFUSE_BASEURL;
 		return {
 			publicKey: process.env.LANGFUSE_PUBLIC_KEY,
 			secretKey: process.env.LANGFUSE_SECRET_KEY,
-			...require_misc.isPresent(baseUrl) ? { baseUrl } : {}
+			...require_misc.isPresent(baseUrl) ? { baseUrl } : {},
+			...require_misc.isPresent(environment) ? { environment } : {}
 		};
 	}
 	if (require_misc.isPresent(langfuse?.baseUrl) && require_langfuse.hasLangfuseEnvCredentials()) return {
 		publicKey: process.env.LANGFUSE_PUBLIC_KEY,
 		secretKey: process.env.LANGFUSE_SECRET_KEY,
-		baseUrl: langfuse.baseUrl
+		baseUrl: langfuse.baseUrl,
+		...require_misc.isPresent(environment) ? { environment } : {}
 	};
 }
 function hashCacheKeyValue(value) {

package/dist/cjs/instrumentation.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"instrumentation.cjs","names":["resolveTraceIdSeedForSpan","context","isPresent","traceIdFromSeed","ROOT_CONTEXT","AsyncLocalStorageContextManager","hasLangfuseConfigCredentials","hasLangfuseEnvConfig","hasLangfuseEnvCredentials","createLangfuseSpanProcessor","resolveLangfuseConfigForSpan","createLibreChatTraceAttributes","BasicTracerProvider"],"sources":["../../src/instrumentation.ts"],"sourcesContent":["import { createHash, randomBytes } from 'node:crypto';\nimport { setLangfuseTracerProvider } from '@langfuse/tracing';\nimport { BasicTracerProvider } from '@opentelemetry/sdk-trace-base';\nimport { context, ROOT_CONTEXT, createContextKey } from '@opentelemetry/api';\nimport { AsyncLocalStorageContextManager } from '@opentelemetry/context-async-hooks';\nimport type {\n  IdGenerator,\n  ReadableSpan,\n  Span,\n  SpanProcessor,\n} from '@opentelemetry/sdk-trace-base';\nimport type { LangfuseSpanProcessorParams } from '@langfuse/otel';\nimport type { Context } from '@opentelemetry/api';\nimport type * as t from '@/types';\nimport {\n  createLibreChatTraceAttributes,\n  hasLangfuseConfigCredentials,\n  hasLangfuseEnvCredentials,\n  hasLangfuseEnvConfig,\n} from '@/langfuse';\nimport {\n  resolveLangfuseConfigForSpan,\n  resolveTraceIdSeedForSpan,\n} from '@/langfuseRuntimeScope';\nimport { createLangfuseSpanProcessor } from '@/langfuseToolOutputTracing';\nimport { traceIdFromSeed } from '@/langfuseRuntimeContext';\nimport { isPresent } from '@/utils/misc';\n\n/**\n * Per-run seed for deterministic Langfuse trace ids. When a run opts in\n * (`LangfuseConfig.deterministicTraceId`), it executes its stream inside\n * `runWithTraceIdSeed(runId, ...)` from `./langfuseRuntimeContext`, and the\n * IdGenerator below derives the root trace id from that seed instead of a\n * random one. This lets external systems (e.g. a host app recording user\n * feedback after the fact) attach scores or observations to the trace by\n * regenerating the same id from the run/message id; no trace lookup required.\n * With no active seed it falls back to random ids, so default behavior is\n * unchanged.\n */\nclass SeededTraceIdGenerator implements IdGenerator {\n  generateTraceId(): string {\n    const seed = resolveTraceIdSeedForSpan(context.active());\n    return isPresent(seed)\n      ? traceIdFromSeed(seed)\n      : randomBytes(16).toString('hex');\n  }\n\n  generateSpanId(): string {\n    return randomBytes(8).toString('hex');\n  }\n}\n\nlet langfuseTracerProvider: BasicTracerProvider | undefined;\nlet langfuseRoutingSpanProcessor: RoutingLangfuseSpanProcessor | undefined;\nconst contextManagerProbeKey = createContextKey(\n  'langfuse-context-manager-probe'\n);\n\nfunction hasActiveContextManager(): boolean {\n  return context.with(\n    ROOT_CONTEXT.setValue(contextManagerProbeKey, true),\n    () => context.active().getValue(contextManagerProbeKey) === true\n  );\n}\n\nexport function ensureOpenTelemetryContextManager(): void {\n  if (hasActiveContextManager()) {\n    return;\n  }\n\n  const contextManager = new AsyncLocalStorageContextManager();\n  contextManager.enable();\n  if (!context.setGlobalContextManager(contextManager)) {\n    contextManager.disable();\n  }\n}\n\nfunction getLangfuseSpanProcessorParams(\n  langfuse?: t.LangfuseConfig\n): LangfuseSpanProcessorParams | undefined {\n  if (langfuse?.enabled === false) {\n    return undefined;\n  }\n  if (hasLangfuseConfigCredentials(langfuse)) {\n    return {\n      publicKey: langfuse.publicKey,\n      secretKey: langfuse.secretKey,\n      ...(isPresent(langfuse.baseUrl) ? { baseUrl: langfuse.baseUrl } : {}),\n    };\n  }\n  if (hasLangfuseEnvConfig()) {\n    const baseUrl =\n      langfuse?.baseUrl ??\n      process.env.LANGFUSE_BASE_URL ??\n      process.env.LANGFUSE_BASEURL;\n    return {\n      publicKey: process.env.LANGFUSE_PUBLIC_KEY as string,\n      secretKey: process.env.LANGFUSE_SECRET_KEY as string,\n      ...(isPresent(baseUrl) ? { baseUrl } : {}),\n    };\n  }\n  if (isPresent(langfuse?.baseUrl) && hasLangfuseEnvCredentials()) {\n    return {\n      publicKey: process.env.LANGFUSE_PUBLIC_KEY as string,\n      secretKey: process.env.LANGFUSE_SECRET_KEY as string,\n      baseUrl: langfuse.baseUrl,\n    };\n  }\n  return undefined;\n}\n\nfunction hashCacheKeyValue(value: string | undefined): string | undefined {\n  return isPresent(value)\n    ? createHash('sha256').update(value, 'utf8').digest('hex')\n    : undefined;\n}\n\nfunction getLangfuseTracerProviderKey(\n  params: LangfuseSpanProcessorParams,\n  langfuse?: t.LangfuseConfig\n): string {\n  return JSON.stringify({\n    publicKey: params.publicKey,\n    secretKeyHash: hashCacheKeyValue(params.secretKey),\n    baseUrl: params.baseUrl,\n    environment: params.environment,\n    toolOutputTracing: langfuse?.toolOutputTracing,\n  });\n}\n\nclass RoutingLangfuseSpanProcessor implements SpanProcessor {\n  // Processors live for the process lifetime. LibreChat tenant Langfuse\n  // destinations are expected to be a bounded admin-managed set, and shutdown\n  // drains every cached processor when the provider is disposed.\n  private readonly processors = new Map<string, SpanProcessor>();\n  private readonly spanProcessors = new WeakMap<object, SpanProcessor>();\n\n  ensureProcessor(langfuse?: t.LangfuseConfig): SpanProcessor | undefined {\n    const params = getLangfuseSpanProcessorParams(langfuse);\n    if (params == null) {\n      return undefined;\n    }\n\n    const processorKey = getLangfuseTracerProviderKey(params, langfuse);\n    const existing = this.processors.get(processorKey);\n    if (existing != null) {\n      return existing;\n    }\n\n    const processor = createLangfuseSpanProcessor(params, langfuse);\n    this.processors.set(processorKey, processor);\n    return processor;\n  }\n\n  onStart(span: Span, parentContext: Context): void {\n    const langfuse = resolveLangfuseConfigForSpan(parentContext);\n    const processor = this.ensureProcessor(langfuse);\n    if (processor == null) {\n      return;\n    }\n\n    const librechatTraceAttributes = createLibreChatTraceAttributes(\n      langfuse?.librechatTraceAttributes ?? {}\n    );\n    if (Object.keys(librechatTraceAttributes).length > 0) {\n      span.setAttributes(librechatTraceAttributes);\n    }\n\n    this.spanProcessors.set(span, processor);\n    processor.onStart(span, parentContext);\n  }\n\n  onEnd(span: ReadableSpan): void {\n    this.spanProcessors.get(span)?.onEnd(span);\n  }\n\n  async forceFlush(): Promise<void> {\n    await Promise.all(\n      Array.from(this.processors.values(), (processor) =>\n        processor.forceFlush()\n      )\n    );\n  }\n\n  async shutdown(): Promise<void> {\n    await Promise.all(\n      Array.from(this.processors.values(), (processor) => processor.shutdown())\n    );\n  }\n}\n\nexport function initializeLangfuseTracing(\n  langfuse?: t.LangfuseConfig\n): BasicTracerProvider | undefined {\n  const params = getLangfuseSpanProcessorParams(langfuse);\n  if (params == null) {\n    return undefined;\n  }\n\n  if (langfuseTracerProvider != null) {\n    langfuseRoutingSpanProcessor?.ensureProcessor(langfuse);\n    return langfuseTracerProvider;\n  }\n\n  ensureOpenTelemetryContextManager();\n  langfuseRoutingSpanProcessor = new RoutingLangfuseSpanProcessor();\n  langfuseRoutingSpanProcessor.ensureProcessor(langfuse);\n  langfuseTracerProvider = new BasicTracerProvider({\n    spanProcessors: [langfuseRoutingSpanProcessor],\n    idGenerator: new SeededTraceIdGenerator(),\n  });\n\n  setLangfuseTracerProvider(langfuseTracerProvider);\n  return langfuseTracerProvider;\n}\n\nexport function initializeLangfuseTracingFromEnv():\n  | BasicTracerProvider\n  | undefined {\n  return initializeLangfuseTracing();\n}\n\ninitializeLangfuseTracingFromEnv();\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAM,yBAAN,MAAoD;CAClD,kBAA0B;EACxB,MAAM,OAAOA,6BAAAA,0BAA0BC,mBAAAA,QAAQ,OAAO,CAAC;EACvD,OAAOC,aAAAA,UAAU,IAAI,IACjBC,+BAAAA,gBAAgB,IAAI,KAAA,GAAA,YAAA,YAAA,CACR,EAAE,CAAC,CAAC,SAAS,KAAK;CACpC;CAEA,iBAAyB;EACvB,QAAA,GAAA,YAAA,YAAA,CAAmB,CAAC,CAAC,CAAC,SAAS,KAAK;CACtC;AACF;AAEA,IAAI;AACJ,IAAI;AACJ,MAAM,0BAAA,GAAA,mBAAA,iBAAA,CACJ,gCACF;AAEA,SAAS,0BAAmC;CAC1C,OAAOF,mBAAAA,QAAQ,KACbG,mBAAAA,aAAa,SAAS,wBAAwB,IAAI,SAC5CH,mBAAAA,QAAQ,OAAO,CAAC,CAAC,SAAS,sBAAsB,MAAM,IAC9D;AACF;AAEA,SAAgB,oCAA0C;CACxD,IAAI,wBAAwB,GAC1B;CAGF,MAAM,iBAAiB,IAAII,mCAAAA,gCAAgC;CAC3D,eAAe,OAAO;CACtB,IAAI,CAACJ,mBAAAA,QAAQ,wBAAwB,cAAc,GACjD,eAAe,QAAQ;AAE3B;AAEA,SAAS,+BACP,UACyC;CACzC,IAAI,UAAU,YAAY,OACxB;CAEF,IAAIK,iBAAAA,6BAA6B,QAAQ,GACvC,OAAO;EACL,WAAW,SAAS;EACpB,WAAW,SAAS;EACpB,GAAIJ,aAAAA,UAAU,SAAS,OAAO,IAAI,EAAE,SAAS,SAAS,QAAQ,IAAI,CAAC;CACrE;CAEF,IAAIK,iBAAAA,qBAAqB,GAAG;EAC1B,MAAM,UACJ,UAAU,WACV,QAAQ,IAAI,qBACZ,QAAQ,IAAI;EACd,OAAO;GACL,WAAW,QAAQ,IAAI;GACvB,WAAW,QAAQ,IAAI;GACvB,GAAIL,aAAAA,UAAU,OAAO,IAAI,EAAE,QAAQ,IAAI,CAAC;EAC1C;CACF;CACA,IAAIA,aAAAA,UAAU,UAAU,OAAO,KAAKM,iBAAAA,0BAA0B,GAC5D,OAAO;EACL,WAAW,QAAQ,IAAI;EACvB,WAAW,QAAQ,IAAI;EACvB,SAAS,SAAS;CACpB;AAGJ;AAEA,SAAS,kBAAkB,OAA+C;CACxE,OAAON,aAAAA,UAAU,KAAK,KAAA,GAAA,YAAA,WAAA,CACP,QAAQ,CAAC,CAAC,OAAO,OAAO,MAAM,CAAC,CAAC,OAAO,KAAK,IACvD,KAAA;AACN;AAEA,SAAS,6BACP,QACA,UACQ;CACR,OAAO,KAAK,UAAU;EACpB,WAAW,OAAO;EAClB,eAAe,kBAAkB,OAAO,SAAS;EACjD,SAAS,OAAO;EAChB,aAAa,OAAO;EACpB,mBAAmB,UAAU;CAC/B,CAAC;AACH;AAEA,IAAM,+BAAN,MAA4D;CAI1D,6BAA8B,IAAI,IAA2B;CAC7D,iCAAkC,IAAI,QAA+B;CAErE,gBAAgB,UAAwD;EACtE,MAAM,SAAS,+BAA+B,QAAQ;EACtD,IAAI,UAAU,MACZ;EAGF,MAAM,eAAe,6BAA6B,QAAQ,QAAQ;EAClE,MAAM,WAAW,KAAK,WAAW,IAAI,YAAY;EACjD,IAAI,YAAY,MACd,OAAO;EAGT,MAAM,YAAYO,kCAAAA,4BAA4B,QAAQ,QAAQ;EAC9D,KAAK,WAAW,IAAI,cAAc,SAAS;EAC3C,OAAO;CACT;CAEA,QAAQ,MAAY,eAA8B;EAChD,MAAM,WAAWC,6BAAAA,6BAA6B,aAAa;EAC3D,MAAM,YAAY,KAAK,gBAAgB,QAAQ;EAC/C,IAAI,aAAa,MACf;EAGF,MAAM,2BAA2BC,iBAAAA,+BAC/B,UAAU,4BAA4B,CAAC,CACzC;EACA,IAAI,OAAO,KAAK,wBAAwB,CAAC,CAAC,SAAS,GACjD,KAAK,cAAc,wBAAwB;EAG7C,KAAK,eAAe,IAAI,MAAM,SAAS;EACvC,UAAU,QAAQ,MAAM,aAAa;CACvC;CAEA,MAAM,MAA0B;EAC9B,KAAK,eAAe,IAAI,IAAI,CAAC,EAAE,MAAM,IAAI;CAC3C;CAEA,MAAM,aAA4B;EAChC,MAAM,QAAQ,IACZ,MAAM,KAAK,KAAK,WAAW,OAAO,IAAI,cACpC,UAAU,WAAW,CACvB,CACF;CACF;CAEA,MAAM,WAA0B;EAC9B,MAAM,QAAQ,IACZ,MAAM,KAAK,KAAK,WAAW,OAAO,IAAI,cAAc,UAAU,SAAS,CAAC,CAC1E;CACF;AACF;AAEA,SAAgB,0BACd,UACiC;CAEjC,IADe,+BAA+B,QACrC,KAAK,MACZ;CAGF,IAAI,0BAA0B,MAAM;EAClC,8BAA8B,gBAAgB,QAAQ;EACtD,OAAO;CACT;CAEA,kCAAkC;CAClC,+BAA+B,IAAI,6BAA6B;CAChE,6BAA6B,gBAAgB,QAAQ;CACrD,yBAAyB,IAAIC,8BAAAA,oBAAoB;EAC/C,gBAAgB,CAAC,4BAA4B;EAC7C,aAAa,IAAI,uBAAuB;CAC1C,CAAC;CAED,CAAA,GAAA,kBAAA,0BAAA,CAA0B,sBAAsB;CAChD,OAAO;AACT;AAEA,SAAgB,mCAEF;CACZ,OAAO,0BAA0B;AACnC;AAEA,iCAAiC"}
+{"version":3,"file":"instrumentation.cjs","names":["resolveTraceIdSeedForSpan","context","isPresent","traceIdFromSeed","ROOT_CONTEXT","AsyncLocalStorageContextManager","hasLangfuseConfigCredentials","hasLangfuseEnvConfig","hasLangfuseEnvCredentials","createLangfuseSpanProcessor","resolveLangfuseConfigForSpan","createLibreChatTraceAttributes","BasicTracerProvider"],"sources":["../../src/instrumentation.ts"],"sourcesContent":["import { createHash, randomBytes } from 'node:crypto';\nimport { setLangfuseTracerProvider } from '@langfuse/tracing';\nimport { BasicTracerProvider } from '@opentelemetry/sdk-trace-base';\nimport { context, ROOT_CONTEXT, createContextKey } from '@opentelemetry/api';\nimport { AsyncLocalStorageContextManager } from '@opentelemetry/context-async-hooks';\nimport type {\n  IdGenerator,\n  ReadableSpan,\n  Span,\n  SpanProcessor,\n} from '@opentelemetry/sdk-trace-base';\nimport type { LangfuseSpanProcessorParams } from '@langfuse/otel';\nimport type { Context } from '@opentelemetry/api';\nimport type * as t from '@/types';\nimport {\n  createLibreChatTraceAttributes,\n  hasLangfuseConfigCredentials,\n  hasLangfuseEnvCredentials,\n  hasLangfuseEnvConfig,\n} from '@/langfuse';\nimport {\n  resolveLangfuseConfigForSpan,\n  resolveTraceIdSeedForSpan,\n} from '@/langfuseRuntimeScope';\nimport { createLangfuseSpanProcessor } from '@/langfuseToolOutputTracing';\nimport { traceIdFromSeed } from '@/langfuseRuntimeContext';\nimport { isPresent } from '@/utils/misc';\n\n/**\n * Per-run seed for deterministic Langfuse trace ids. When a run opts in\n * (`LangfuseConfig.deterministicTraceId`), it executes its stream inside\n * `runWithTraceIdSeed(runId, ...)` from `./langfuseRuntimeContext`, and the\n * IdGenerator below derives the root trace id from that seed instead of a\n * random one. This lets external systems (e.g. a host app recording user\n * feedback after the fact) attach scores or observations to the trace by\n * regenerating the same id from the run/message id; no trace lookup required.\n * With no active seed it falls back to random ids, so default behavior is\n * unchanged.\n */\nclass SeededTraceIdGenerator implements IdGenerator {\n  generateTraceId(): string {\n    const seed = resolveTraceIdSeedForSpan(context.active());\n    return isPresent(seed)\n      ? traceIdFromSeed(seed)\n      : randomBytes(16).toString('hex');\n  }\n\n  generateSpanId(): string {\n    return randomBytes(8).toString('hex');\n  }\n}\n\nlet langfuseTracerProvider: BasicTracerProvider | undefined;\nlet langfuseRoutingSpanProcessor: RoutingLangfuseSpanProcessor | undefined;\nconst contextManagerProbeKey = createContextKey(\n  'langfuse-context-manager-probe'\n);\n\nfunction hasActiveContextManager(): boolean {\n  return context.with(\n    ROOT_CONTEXT.setValue(contextManagerProbeKey, true),\n    () => context.active().getValue(contextManagerProbeKey) === true\n  );\n}\n\nexport function ensureOpenTelemetryContextManager(): void {\n  if (hasActiveContextManager()) {\n    return;\n  }\n\n  const contextManager = new AsyncLocalStorageContextManager();\n  contextManager.enable();\n  if (!context.setGlobalContextManager(contextManager)) {\n    contextManager.disable();\n  }\n}\n\nfunction resolveLangfuseEnvironment(\n  langfuse?: t.LangfuseConfig\n): string | undefined {\n  const candidates = [\n    langfuse?.environment,\n    process.env.LANGFUSE_TRACING_ENVIRONMENT,\n    process.env.NODE_ENV,\n  ];\n  for (const candidate of candidates) {\n    if (candidate != null && candidate.trim() !== '') {\n      return candidate.trim();\n    }\n  }\n  return undefined;\n}\n\nfunction getLangfuseSpanProcessorParams(\n  langfuse?: t.LangfuseConfig\n): LangfuseSpanProcessorParams | undefined {\n  if (langfuse?.enabled === false) {\n    return undefined;\n  }\n  const environment = resolveLangfuseEnvironment(langfuse);\n  if (hasLangfuseConfigCredentials(langfuse)) {\n    return {\n      publicKey: langfuse.publicKey,\n      secretKey: langfuse.secretKey,\n      ...(isPresent(langfuse.baseUrl) ? { baseUrl: langfuse.baseUrl } : {}),\n      ...(isPresent(environment) ? { environment } : {}),\n    };\n  }\n  if (hasLangfuseEnvConfig()) {\n    const baseUrl =\n      langfuse?.baseUrl ??\n      process.env.LANGFUSE_BASE_URL ??\n      process.env.LANGFUSE_BASEURL;\n    return {\n      publicKey: process.env.LANGFUSE_PUBLIC_KEY as string,\n      secretKey: process.env.LANGFUSE_SECRET_KEY as string,\n      ...(isPresent(baseUrl) ? { baseUrl } : {}),\n      ...(isPresent(environment) ? { environment } : {}),\n    };\n  }\n  if (isPresent(langfuse?.baseUrl) && hasLangfuseEnvCredentials()) {\n    return {\n      publicKey: process.env.LANGFUSE_PUBLIC_KEY as string,\n      secretKey: process.env.LANGFUSE_SECRET_KEY as string,\n      baseUrl: langfuse.baseUrl,\n      ...(isPresent(environment) ? { environment } : {}),\n    };\n  }\n  return undefined;\n}\n\nfunction hashCacheKeyValue(value: string | undefined): string | undefined {\n  return isPresent(value)\n    ? createHash('sha256').update(value, 'utf8').digest('hex')\n    : undefined;\n}\n\nfunction getLangfuseTracerProviderKey(\n  params: LangfuseSpanProcessorParams,\n  langfuse?: t.LangfuseConfig\n): string {\n  return JSON.stringify({\n    publicKey: params.publicKey,\n    secretKeyHash: hashCacheKeyValue(params.secretKey),\n    baseUrl: params.baseUrl,\n    environment: params.environment,\n    toolOutputTracing: langfuse?.toolOutputTracing,\n  });\n}\n\nclass RoutingLangfuseSpanProcessor implements SpanProcessor {\n  // Processors live for the process lifetime. LibreChat tenant Langfuse\n  // destinations are expected to be a bounded admin-managed set, and shutdown\n  // drains every cached processor when the provider is disposed.\n  private readonly processors = new Map<string, SpanProcessor>();\n  private readonly spanProcessors = new WeakMap<object, SpanProcessor>();\n\n  ensureProcessor(langfuse?: t.LangfuseConfig): SpanProcessor | undefined {\n    const params = getLangfuseSpanProcessorParams(langfuse);\n    if (params == null) {\n      return undefined;\n    }\n\n    const processorKey = getLangfuseTracerProviderKey(params, langfuse);\n    const existing = this.processors.get(processorKey);\n    if (existing != null) {\n      return existing;\n    }\n\n    const processor = createLangfuseSpanProcessor(params, langfuse);\n    this.processors.set(processorKey, processor);\n    return processor;\n  }\n\n  onStart(span: Span, parentContext: Context): void {\n    const langfuse = resolveLangfuseConfigForSpan(parentContext);\n    const processor = this.ensureProcessor(langfuse);\n    if (processor == null) {\n      return;\n    }\n\n    const librechatTraceAttributes = createLibreChatTraceAttributes(\n      langfuse?.librechatTraceAttributes ?? {}\n    );\n    if (Object.keys(librechatTraceAttributes).length > 0) {\n      span.setAttributes(librechatTraceAttributes);\n    }\n\n    this.spanProcessors.set(span, processor);\n    processor.onStart(span, parentContext);\n  }\n\n  onEnd(span: ReadableSpan): void {\n    this.spanProcessors.get(span)?.onEnd(span);\n  }\n\n  async forceFlush(): Promise<void> {\n    await Promise.all(\n      Array.from(this.processors.values(), (processor) =>\n        processor.forceFlush()\n      )\n    );\n  }\n\n  async shutdown(): Promise<void> {\n    await Promise.all(\n      Array.from(this.processors.values(), (processor) => processor.shutdown())\n    );\n  }\n}\n\nexport function initializeLangfuseTracing(\n  langfuse?: t.LangfuseConfig\n): BasicTracerProvider | undefined {\n  const params = getLangfuseSpanProcessorParams(langfuse);\n  if (params == null) {\n    return undefined;\n  }\n\n  if (langfuseTracerProvider != null) {\n    langfuseRoutingSpanProcessor?.ensureProcessor(langfuse);\n    return langfuseTracerProvider;\n  }\n\n  ensureOpenTelemetryContextManager();\n  langfuseRoutingSpanProcessor = new RoutingLangfuseSpanProcessor();\n  langfuseRoutingSpanProcessor.ensureProcessor(langfuse);\n  langfuseTracerProvider = new BasicTracerProvider({\n    spanProcessors: [langfuseRoutingSpanProcessor],\n    idGenerator: new SeededTraceIdGenerator(),\n  });\n\n  setLangfuseTracerProvider(langfuseTracerProvider);\n  return langfuseTracerProvider;\n}\n\nexport function initializeLangfuseTracingFromEnv():\n  | BasicTracerProvider\n  | undefined {\n  return initializeLangfuseTracing();\n}\n\ninitializeLangfuseTracingFromEnv();\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAM,yBAAN,MAAoD;CAClD,kBAA0B;EACxB,MAAM,OAAOA,6BAAAA,0BAA0BC,mBAAAA,QAAQ,OAAO,CAAC;EACvD,OAAOC,aAAAA,UAAU,IAAI,IACjBC,+BAAAA,gBAAgB,IAAI,KAAA,GAAA,YAAA,YAAA,CACR,EAAE,CAAC,CAAC,SAAS,KAAK;CACpC;CAEA,iBAAyB;EACvB,QAAA,GAAA,YAAA,YAAA,CAAmB,CAAC,CAAC,CAAC,SAAS,KAAK;CACtC;AACF;AAEA,IAAI;AACJ,IAAI;AACJ,MAAM,0BAAA,GAAA,mBAAA,iBAAA,CACJ,gCACF;AAEA,SAAS,0BAAmC;CAC1C,OAAOF,mBAAAA,QAAQ,KACbG,mBAAAA,aAAa,SAAS,wBAAwB,IAAI,SAC5CH,mBAAAA,QAAQ,OAAO,CAAC,CAAC,SAAS,sBAAsB,MAAM,IAC9D;AACF;AAEA,SAAgB,oCAA0C;CACxD,IAAI,wBAAwB,GAC1B;CAGF,MAAM,iBAAiB,IAAII,mCAAAA,gCAAgC;CAC3D,eAAe,OAAO;CACtB,IAAI,CAACJ,mBAAAA,QAAQ,wBAAwB,cAAc,GACjD,eAAe,QAAQ;AAE3B;AAEA,SAAS,2BACP,UACoB;CACpB,MAAM,aAAa;EACjB,UAAU;EACV,QAAQ,IAAI;EACZ,QAAQ,IAAI;CACd;CACA,KAAK,MAAM,aAAa,YACtB,IAAI,aAAa,QAAQ,UAAU,KAAK,MAAM,IAC5C,OAAO,UAAU,KAAK;AAI5B;AAEA,SAAS,+BACP,UACyC;CACzC,IAAI,UAAU,YAAY,OACxB;CAEF,MAAM,cAAc,2BAA2B,QAAQ;CACvD,IAAIK,iBAAAA,6BAA6B,QAAQ,GACvC,OAAO;EACL,WAAW,SAAS;EACpB,WAAW,SAAS;EACpB,GAAIJ,aAAAA,UAAU,SAAS,OAAO,IAAI,EAAE,SAAS,SAAS,QAAQ,IAAI,CAAC;EACnE,GAAIA,aAAAA,UAAU,WAAW,IAAI,EAAE,YAAY,IAAI,CAAC;CAClD;CAEF,IAAIK,iBAAAA,qBAAqB,GAAG;EAC1B,MAAM,UACJ,UAAU,WACV,QAAQ,IAAI,qBACZ,QAAQ,IAAI;EACd,OAAO;GACL,WAAW,QAAQ,IAAI;GACvB,WAAW,QAAQ,IAAI;GACvB,GAAIL,aAAAA,UAAU,OAAO,IAAI,EAAE,QAAQ,IAAI,CAAC;GACxC,GAAIA,aAAAA,UAAU,WAAW,IAAI,EAAE,YAAY,IAAI,CAAC;EAClD;CACF;CACA,IAAIA,aAAAA,UAAU,UAAU,OAAO,KAAKM,iBAAAA,0BAA0B,GAC5D,OAAO;EACL,WAAW,QAAQ,IAAI;EACvB,WAAW,QAAQ,IAAI;EACvB,SAAS,SAAS;EAClB,GAAIN,aAAAA,UAAU,WAAW,IAAI,EAAE,YAAY,IAAI,CAAC;CAClD;AAGJ;AAEA,SAAS,kBAAkB,OAA+C;CACxE,OAAOA,aAAAA,UAAU,KAAK,KAAA,GAAA,YAAA,WAAA,CACP,QAAQ,CAAC,CAAC,OAAO,OAAO,MAAM,CAAC,CAAC,OAAO,KAAK,IACvD,KAAA;AACN;AAEA,SAAS,6BACP,QACA,UACQ;CACR,OAAO,KAAK,UAAU;EACpB,WAAW,OAAO;EAClB,eAAe,kBAAkB,OAAO,SAAS;EACjD,SAAS,OAAO;EAChB,aAAa,OAAO;EACpB,mBAAmB,UAAU;CAC/B,CAAC;AACH;AAEA,IAAM,+BAAN,MAA4D;CAI1D,6BAA8B,IAAI,IAA2B;CAC7D,iCAAkC,IAAI,QAA+B;CAErE,gBAAgB,UAAwD;EACtE,MAAM,SAAS,+BAA+B,QAAQ;EACtD,IAAI,UAAU,MACZ;EAGF,MAAM,eAAe,6BAA6B,QAAQ,QAAQ;EAClE,MAAM,WAAW,KAAK,WAAW,IAAI,YAAY;EACjD,IAAI,YAAY,MACd,OAAO;EAGT,MAAM,YAAYO,kCAAAA,4BAA4B,QAAQ,QAAQ;EAC9D,KAAK,WAAW,IAAI,cAAc,SAAS;EAC3C,OAAO;CACT;CAEA,QAAQ,MAAY,eAA8B;EAChD,MAAM,WAAWC,6BAAAA,6BAA6B,aAAa;EAC3D,MAAM,YAAY,KAAK,gBAAgB,QAAQ;EAC/C,IAAI,aAAa,MACf;EAGF,MAAM,2BAA2BC,iBAAAA,+BAC/B,UAAU,4BAA4B,CAAC,CACzC;EACA,IAAI,OAAO,KAAK,wBAAwB,CAAC,CAAC,SAAS,GACjD,KAAK,cAAc,wBAAwB;EAG7C,KAAK,eAAe,IAAI,MAAM,SAAS;EACvC,UAAU,QAAQ,MAAM,aAAa;CACvC;CAEA,MAAM,MAA0B;EAC9B,KAAK,eAAe,IAAI,IAAI,CAAC,EAAE,MAAM,IAAI;CAC3C;CAEA,MAAM,aAA4B;EAChC,MAAM,QAAQ,IACZ,MAAM,KAAK,KAAK,WAAW,OAAO,IAAI,cACpC,UAAU,WAAW,CACvB,CACF;CACF;CAEA,MAAM,WAA0B;EAC9B,MAAM,QAAQ,IACZ,MAAM,KAAK,KAAK,WAAW,OAAO,IAAI,cAAc,UAAU,SAAS,CAAC,CAC1E;CACF;AACF;AAEA,SAAgB,0BACd,UACiC;CAEjC,IADe,+BAA+B,QACrC,KAAK,MACZ;CAGF,IAAI,0BAA0B,MAAM;EAClC,8BAA8B,gBAAgB,QAAQ;EACtD,OAAO;CACT;CAEA,kCAAkC;CAClC,+BAA+B,IAAI,6BAA6B;CAChE,6BAA6B,gBAAgB,QAAQ;CACrD,yBAAyB,IAAIC,8BAAAA,oBAAoB;EAC/C,gBAAgB,CAAC,4BAA4B;EAC7C,aAAa,IAAI,uBAAuB;CAC1C,CAAC;CAED,CAAA,GAAA,kBAAA,0BAAA,CAA0B,sBAAsB;CAChD,OAAO;AACT;AAEA,SAAgB,mCAEF;CACZ,OAAO,0BAA0B;AACnC;AAEA,iCAAiC"}

package/dist/cjs/langfuseToolOutputTracing.cjs

@@ -239,8 +239,7 @@ function hasLangfuseConfigKeys(langfuse) {
 function shouldTraceToolNodeForLangfuse({ runLangfuse, agentLangfuse }) {
 	const langfuse = require_langfuseConfig.resolveLangfuseConfig(runLangfuse, agentLangfuse);
 	if (langfuse?.enabled === false) return false;
-	const explicit = langfuse?.toolNodeTracing?.enabled;
-	if (explicit != null) return explicit && (hasLangfuseConfigKeys(langfuse) || hasLangfuseEnvKeys());
+	if (langfuse?.toolNodeTracing?.enabled !== true) return false;
 	return hasLangfuseConfigKeys(langfuse) || hasLangfuseEnvKeys();
 }
 //#endregion

package/dist/cjs/langfuseToolOutputTracing.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"langfuseToolOutputTracing.cjs","names":["normalizeToolName","LangfuseOtelSpanAttributes","LangfuseSpanProcessor","shouldDropLangfuseSpan","resolveToolOutputTracingConfigForSpan","hasToolOutputTracingConfig","resolveToolOutputTracingConfig","resolveLangfuseConfig"],"sources":["../../src/langfuseToolOutputTracing.ts"],"sourcesContent":["import { LangfuseSpanProcessor } from '@langfuse/otel';\nimport { LangfuseOtelSpanAttributes } from '@langfuse/tracing';\nimport type {\n  ReadableSpan,\n  Span,\n  SpanProcessor,\n} from '@opentelemetry/sdk-trace-base';\nimport type { LangfuseSpanProcessorParams } from '@langfuse/otel';\nimport type { Context } from '@opentelemetry/api';\nimport type { ResolvedLangfuseToolOutputTracingConfig } from '@/langfuseRuntimeContext';\nimport type * as t from '@/types';\nimport {\n  LANGFUSE_TOOL_OUTPUT_REDACTION_TEXT,\n  hasToolOutputTracingConfig,\n  normalizeToolName,\n  resolveLangfuseConfig,\n  resolveToolOutputTracingConfig,\n} from '@/langfuseConfig';\nimport {\n  shapeLangfuseSpan,\n  shouldDropLangfuseSpan,\n} from '@/langfuseTraceShaping';\nimport { resolveToolOutputTracingConfigForSpan } from '@/langfuseRuntimeScope';\n\nexport { LANGFUSE_TOOL_OUTPUT_REDACTION_TEXT, resolveLangfuseConfig };\n\nconst LANGGRAPH_TOOL_NODE_PREFIX = 'tools=';\n\nconst CHAT_ROLES = new Set([\n  'assistant',\n  'developer',\n  'human',\n  'system',\n  'user',\n]);\n\ntype SpanWithAttributes = ReadableSpan & {\n  attributes: Record<string, unknown>;\n};\n\ntype RedactionResult = {\n  value: unknown;\n  changed: boolean;\n};\n\ntype RedactionContext = {\n  toolNamesByCallId: Map<string, string>;\n};\n\nconst TOOL_OUTPUT_FIELD_KEYS = ['content', 'artifact'];\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return value != null && typeof value === 'object' && !Array.isArray(value);\n}\n\nfunction isPresent(value: unknown): value is string {\n  return typeof value === 'string' && value.trim() !== '';\n}\n\nfunction shouldApplyToolOutputRedaction(\n  config: ResolvedLangfuseToolOutputTracingConfig\n): boolean {\n  return config.enabled === false || config.redactedToolNames.size > 0;\n}\n\nfunction toolNameMatches(\n  toolName: string | undefined,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): boolean {\n  if (!isPresent(toolName)) {\n    return false;\n  }\n\n  const normalizedToolName = normalizeToolName(toolName);\n  if (config.redactedToolNameMatchMode === 'partial') {\n    for (const redactedToolName of config.redactedToolNames) {\n      if (normalizedToolName.includes(redactedToolName)) {\n        return true;\n      }\n    }\n    return false;\n  }\n\n  return config.redactedToolNames.has(normalizedToolName);\n}\n\nfunction shouldRedactTool(\n  toolName: string | undefined,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): boolean {\n  return config.enabled === false || toolNameMatches(toolName, config);\n}\n\nfunction getStringField(\n  value: Record<string, unknown>,\n  key: string\n): string | undefined {\n  const field = value[key];\n  return typeof field === 'string' ? field : undefined;\n}\n\nfunction getNestedStringField(\n  value: Record<string, unknown>,\n  objectKey: string,\n  fieldKey: string\n): string | undefined {\n  const nested = value[objectKey];\n  if (!isRecord(nested)) {\n    return undefined;\n  }\n  return getStringField(nested, fieldKey);\n}\n\nfunction getSerializedToolCallId(\n  value: Record<string, unknown>\n): string | undefined {\n  return (\n    getStringField(value, 'tool_call_id') ??\n    getNestedStringField(value, 'kwargs', 'tool_call_id') ??\n    getNestedStringField(value, 'additional_kwargs', 'tool_call_id') ??\n    getNestedStringField(value, 'data', 'tool_call_id') ??\n    (typeof value.id === 'string' ? value.id : undefined)\n  );\n}\n\nfunction getSerializedToolName(\n  value: Record<string, unknown>,\n  redactionContext?: RedactionContext\n): string | undefined {\n  const role = getStringField(value, 'role');\n  const explicitName =\n    getStringField(value, 'name') ??\n    getStringField(value, 'tool_name') ??\n    getNestedStringField(value, 'function', 'name') ??\n    getNestedStringField(value, 'kwargs', 'name') ??\n    getNestedStringField(value, 'additional_kwargs', 'name') ??\n    getNestedStringField(value, 'data', 'name') ??\n    (role != null && role.toLowerCase() !== 'tool' ? role : undefined);\n\n  if (explicitName != null) {\n    return explicitName;\n  }\n\n  const toolCallId = getSerializedToolCallId(value);\n  return toolCallId != null\n    ? redactionContext?.toolNamesByCallId.get(toolCallId)\n    : undefined;\n}\n\nfunction hasToolMessageIdentity(value: Record<string, unknown>): boolean {\n  const type = getStringField(value, 'type') ?? getStringField(value, '_type');\n  if (type === 'tool' || type === 'tool_message') {\n    return true;\n  }\n\n  const id = value.id;\n  if (\n    Array.isArray(id) &&\n    id.some((part) => typeof part === 'string' && part.includes('ToolMessage'))\n  ) {\n    return true;\n  }\n\n  if (\n    'tool_call_id' in value ||\n    getNestedStringField(value, 'kwargs', 'tool_call_id') != null ||\n    getNestedStringField(value, 'additional_kwargs', 'tool_call_id') != null\n  ) {\n    return true;\n  }\n\n  const role = getStringField(value, 'role');\n  return (\n    role != null &&\n    !CHAT_ROLES.has(role.toLowerCase()) &&\n    ('content' in value || isRecord(value.kwargs) || isRecord(value.data))\n  );\n}\n\nfunction redactToolContentFields(\n  value: Record<string, unknown>,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): Record<string, unknown> {\n  const next = { ...value };\n\n  for (const outputKey of TOOL_OUTPUT_FIELD_KEYS) {\n    if (outputKey in next) {\n      next[outputKey] = config.redactionText;\n    }\n  }\n\n  for (const nestedKey of ['kwargs', 'data', 'additional_kwargs']) {\n    const nested = next[nestedKey];\n    if (!isRecord(nested)) {\n      continue;\n    }\n    const nextNested = { ...nested };\n    let changed = false;\n    for (const outputKey of TOOL_OUTPUT_FIELD_KEYS) {\n      if (outputKey in nextNested) {\n        nextNested[outputKey] = config.redactionText;\n        changed = true;\n      }\n    }\n    if (changed) {\n      next[nestedKey] = nextNested;\n    }\n  }\n\n  return next;\n}\n\nfunction collectToolCallNames(\n  value: unknown,\n  redactionContext: RedactionContext\n): void {\n  if (Array.isArray(value)) {\n    for (const item of value) {\n      collectToolCallNames(item, redactionContext);\n    }\n    return;\n  }\n\n  if (!isRecord(value)) {\n    return;\n  }\n\n  const toolCallId = getSerializedToolCallId(value);\n  const toolName = getSerializedToolName(value);\n  if (toolCallId != null && toolName != null) {\n    redactionContext.toolNamesByCallId.set(toolCallId, toolName);\n  }\n\n  for (const child of Object.values(value)) {\n    collectToolCallNames(child, redactionContext);\n  }\n}\n\nfunction redactValue(\n  value: unknown,\n  config: ResolvedLangfuseToolOutputTracingConfig,\n  redactionContext: RedactionContext\n): RedactionResult {\n  if (Array.isArray(value)) {\n    let changed = false;\n    const next: unknown[] = [];\n    for (const item of value) {\n      const result = redactValue(item, config, redactionContext);\n      if (result.changed) {\n        changed = true;\n      }\n      next.push(result.value);\n    }\n    return changed ? { value: next, changed } : { value, changed };\n  }\n\n  if (!isRecord(value)) {\n    return { value, changed: false };\n  }\n\n  const toolName = getSerializedToolName(value, redactionContext);\n  if (hasToolMessageIdentity(value) && shouldRedactTool(toolName, config)) {\n    return {\n      value: redactToolContentFields(value, config),\n      changed: true,\n    };\n  }\n\n  let changed = false;\n  const next: Record<string, unknown> = {};\n  for (const [key, child] of Object.entries(value)) {\n    const result = redactValue(child, config, redactionContext);\n    if (result.changed) {\n      changed = true;\n    }\n    next[key] = result.value;\n  }\n\n  return changed ? { value: next, changed } : { value, changed };\n}\n\nfunction redactSerializedValue(\n  value: unknown,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): RedactionResult {\n  const redactionContext: RedactionContext = {\n    toolNamesByCallId: new Map(),\n  };\n  if (typeof value !== 'string') {\n    collectToolCallNames(value, redactionContext);\n    return redactValue(value, config, redactionContext);\n  }\n\n  const trimmed = value.trim();\n  if (!trimmed.startsWith('{') && !trimmed.startsWith('[')) {\n    return { value, changed: false };\n  }\n\n  try {\n    const parsed = JSON.parse(value) as unknown;\n    collectToolCallNames(parsed, redactionContext);\n    const result = redactValue(parsed, config, redactionContext);\n    return result.changed\n      ? { value: JSON.stringify(result.value), changed: true }\n      : { value, changed: false };\n  } catch {\n    return { value, changed: false };\n  }\n}\n\nfunction redactAttribute(\n  attributes: Record<string, unknown>,\n  key: string,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): void {\n  if (!(key in attributes)) {\n    return;\n  }\n\n  const result = redactSerializedValue(attributes[key], config);\n  if (result.changed) {\n    attributes[key] = result.value;\n  }\n}\n\nfunction isToolObservation(attributes: Record<string, unknown>): boolean {\n  const type = attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE];\n  return typeof type === 'string' && type.toLowerCase() === 'tool';\n}\n\nfunction classifyLangGraphToolNodeSpan(\n  attributes: Record<string, unknown>\n): void {\n  const type = attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE];\n  if (typeof type !== 'string' || type.toLowerCase() !== 'span') {\n    return;\n  }\n\n  const langGraphNode =\n    attributes[\n      `${LangfuseOtelSpanAttributes.OBSERVATION_METADATA}.langgraph_node`\n    ];\n  if (\n    typeof langGraphNode === 'string' &&\n    langGraphNode.startsWith(LANGGRAPH_TOOL_NODE_PREFIX)\n  ) {\n    attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE] = 'tool';\n  }\n}\n\nexport function classifyLangfuseToolNodeSpan(span: ReadableSpan): void {\n  classifyLangGraphToolNodeSpan((span as SpanWithAttributes).attributes);\n}\n\nfunction redactToolObservationOutput(\n  span: ReadableSpan,\n  attributes: Record<string, unknown>,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): void {\n  if (\n    !(\n      isToolObservation(attributes) &&\n      shouldRedactTool(span.name, config) &&\n      LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT in attributes\n    )\n  ) {\n    return;\n  }\n\n  attributes[LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT] =\n    config.redactionText;\n}\n\nexport function redactLangfuseSpanToolOutputs(\n  span: ReadableSpan,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): void {\n  const attributes = (span as SpanWithAttributes).attributes;\n  classifyLangfuseToolNodeSpan(span);\n\n  if (!shouldApplyToolOutputRedaction(config)) {\n    return;\n  }\n\n  redactToolObservationOutput(span, attributes, config);\n\n  for (const key of [\n    LangfuseOtelSpanAttributes.OBSERVATION_INPUT,\n    LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT,\n    LangfuseOtelSpanAttributes.TRACE_INPUT,\n    LangfuseOtelSpanAttributes.TRACE_OUTPUT,\n  ]) {\n    redactAttribute(attributes, key, config);\n  }\n}\n\nclass ToolOutputRedactingLangfuseSpanProcessor implements SpanProcessor {\n  private readonly processor: LangfuseSpanProcessor;\n  private readonly fallbackConfig?: ResolvedLangfuseToolOutputTracingConfig;\n  private readonly spanConfigs = new WeakMap<\n    object,\n    ResolvedLangfuseToolOutputTracingConfig\n  >();\n\n  constructor(\n    params?: LangfuseSpanProcessorParams,\n    fallbackConfig?: ResolvedLangfuseToolOutputTracingConfig\n  ) {\n    this.processor = new LangfuseSpanProcessor(params);\n    this.fallbackConfig = fallbackConfig;\n  }\n\n  onStart(span: Span, parentContext: Context): void {\n    if (shouldDropLangfuseSpan(span.name)) {\n      return;\n    }\n    const config =\n      resolveToolOutputTracingConfigForSpan(parentContext) ??\n      this.fallbackConfig;\n    if (config != null) {\n      this.spanConfigs.set(span, config);\n    }\n    this.processor.onStart(span, parentContext);\n  }\n\n  onEnd(span: ReadableSpan): void {\n    if (shouldDropLangfuseSpan(span.name)) {\n      return;\n    }\n    classifyLangfuseToolNodeSpan(span);\n    shapeLangfuseSpan(span);\n    const config = this.spanConfigs.get(span) ?? this.fallbackConfig;\n    if (config != null) {\n      redactLangfuseSpanToolOutputs(span, config);\n    }\n    this.processor.onEnd(span);\n  }\n\n  forceFlush(): Promise<void> {\n    return this.processor.forceFlush();\n  }\n\n  shutdown(): Promise<void> {\n    return this.processor.shutdown();\n  }\n}\n\nexport function createLangfuseSpanProcessor(\n  params?: LangfuseSpanProcessorParams,\n  runLangfuse?: t.LangfuseConfig,\n  agentLangfuse?: t.LangfuseConfig\n): SpanProcessor {\n  const fallbackConfig = hasToolOutputTracingConfig(runLangfuse, agentLangfuse)\n    ? resolveToolOutputTracingConfig(runLangfuse, agentLangfuse)\n    : undefined;\n  return new ToolOutputRedactingLangfuseSpanProcessor(params, fallbackConfig);\n}\n\nfunction hasLangfuseEnvKeys(): boolean {\n  return (\n    isPresent(process.env.LANGFUSE_SECRET_KEY) &&\n    isPresent(process.env.LANGFUSE_PUBLIC_KEY)\n  );\n}\n\nfunction hasLangfuseConfigKeys(langfuse?: t.LangfuseConfig): boolean {\n  if (langfuse == null) {\n    return false;\n  }\n  return isPresent(langfuse.secretKey) && isPresent(langfuse.publicKey);\n}\n\nexport function shouldTraceToolNodeForLangfuse({\n  runLangfuse,\n  agentLangfuse,\n}: {\n  runLangfuse?: t.LangfuseConfig;\n  agentLangfuse?: t.LangfuseConfig;\n}): boolean {\n  const langfuse = resolveLangfuseConfig(runLangfuse, agentLangfuse);\n  if (langfuse?.enabled === false) {\n    return false;\n  }\n\n  const explicit = langfuse?.toolNodeTracing?.enabled;\n  if (explicit != null) {\n    return (\n      explicit && (hasLangfuseConfigKeys(langfuse) || hasLangfuseEnvKeys())\n    );\n  }\n\n  return hasLangfuseConfigKeys(langfuse) || hasLangfuseEnvKeys();\n}\n"],"mappings":";;;;;;AA0BA,MAAM,6BAA6B;AAEnC,MAAM,aAAa,IAAI,IAAI;CACzB;CACA;CACA;CACA;CACA;AACF,CAAC;AAeD,MAAM,yBAAyB,CAAC,WAAW,UAAU;AAErD,SAAS,SAAS,OAAkD;CAClE,OAAO,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAC3E;AAEA,SAAS,UAAU,OAAiC;CAClD,OAAO,OAAO,UAAU,YAAY,MAAM,KAAK,MAAM;AACvD;AAEA,SAAS,+BACP,QACS;CACT,OAAO,OAAO,YAAY,SAAS,OAAO,kBAAkB,OAAO;AACrE;AAEA,SAAS,gBACP,UACA,QACS;CACT,IAAI,CAAC,UAAU,QAAQ,GACrB,OAAO;CAGT,MAAM,qBAAqBA,uBAAAA,kBAAkB,QAAQ;CACrD,IAAI,OAAO,8BAA8B,WAAW;EAClD,KAAK,MAAM,oBAAoB,OAAO,mBACpC,IAAI,mBAAmB,SAAS,gBAAgB,GAC9C,OAAO;EAGX,OAAO;CACT;CAEA,OAAO,OAAO,kBAAkB,IAAI,kBAAkB;AACxD;AAEA,SAAS,iBACP,UACA,QACS;CACT,OAAO,OAAO,YAAY,SAAS,gBAAgB,UAAU,MAAM;AACrE;AAEA,SAAS,eACP,OACA,KACoB;CACpB,MAAM,QAAQ,MAAM;CACpB,OAAO,OAAO,UAAU,WAAW,QAAQ,KAAA;AAC7C;AAEA,SAAS,qBACP,OACA,WACA,UACoB;CACpB,MAAM,SAAS,MAAM;CACrB,IAAI,CAAC,SAAS,MAAM,GAClB;CAEF,OAAO,eAAe,QAAQ,QAAQ;AACxC;AAEA,SAAS,wBACP,OACoB;CACpB,OACE,eAAe,OAAO,cAAc,KACpC,qBAAqB,OAAO,UAAU,cAAc,KACpD,qBAAqB,OAAO,qBAAqB,cAAc,KAC/D,qBAAqB,OAAO,QAAQ,cAAc,MACjD,OAAO,MAAM,OAAO,WAAW,MAAM,KAAK,KAAA;AAE/C;AAEA,SAAS,sBACP,OACA,kBACoB;CACpB,MAAM,OAAO,eAAe,OAAO,MAAM;CACzC,MAAM,eACJ,eAAe,OAAO,MAAM,KAC5B,eAAe,OAAO,WAAW,KACjC,qBAAqB,OAAO,YAAY,MAAM,KAC9C,qBAAqB,OAAO,UAAU,MAAM,KAC5C,qBAAqB,OAAO,qBAAqB,MAAM,KACvD,qBAAqB,OAAO,QAAQ,MAAM,MACzC,QAAQ,QAAQ,KAAK,YAAY,MAAM,SAAS,OAAO,KAAA;CAE1D,IAAI,gBAAgB,MAClB,OAAO;CAGT,MAAM,aAAa,wBAAwB,KAAK;CAChD,OAAO,cAAc,OACjB,kBAAkB,kBAAkB,IAAI,UAAU,IAClD,KAAA;AACN;AAEA,SAAS,uBAAuB,OAAyC;CACvE,MAAM,OAAO,eAAe,OAAO,MAAM,KAAK,eAAe,OAAO,OAAO;CAC3E,IAAI,SAAS,UAAU,SAAS,gBAC9B,OAAO;CAGT,MAAM,KAAK,MAAM;CACjB,IACE,MAAM,QAAQ,EAAE,KAChB,GAAG,MAAM,SAAS,OAAO,SAAS,YAAY,KAAK,SAAS,aAAa,CAAC,GAE1E,OAAO;CAGT,IACE,kBAAkB,SAClB,qBAAqB,OAAO,UAAU,cAAc,KAAK,QACzD,qBAAqB,OAAO,qBAAqB,cAAc,KAAK,MAEpE,OAAO;CAGT,MAAM,OAAO,eAAe,OAAO,MAAM;CACzC,OACE,QAAQ,QACR,CAAC,WAAW,IAAI,KAAK,YAAY,CAAC,MACjC,aAAa,SAAS,SAAS,MAAM,MAAM,KAAK,SAAS,MAAM,IAAI;AAExE;AAEA,SAAS,wBACP,OACA,QACyB;CACzB,MAAM,OAAO,EAAE,GAAG,MAAM;CAExB,KAAK,MAAM,aAAa,wBACtB,IAAI,aAAa,MACf,KAAK,aAAa,OAAO;CAI7B,KAAK,MAAM,aAAa;EAAC;EAAU;EAAQ;CAAmB,GAAG;EAC/D,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,SAAS,MAAM,GAClB;EAEF,MAAM,aAAa,EAAE,GAAG,OAAO;EAC/B,IAAI,UAAU;EACd,KAAK,MAAM,aAAa,wBACtB,IAAI,aAAa,YAAY;GAC3B,WAAW,aAAa,OAAO;GAC/B,UAAU;EACZ;EAEF,IAAI,SACF,KAAK,aAAa;CAEtB;CAEA,OAAO;AACT;AAEA,SAAS,qBACP,OACA,kBACM;CACN,IAAI,MAAM,QAAQ,KAAK,GAAG;EACxB,KAAK,MAAM,QAAQ,OACjB,qBAAqB,MAAM,gBAAgB;EAE7C;CACF;CAEA,IAAI,CAAC,SAAS,KAAK,GACjB;CAGF,MAAM,aAAa,wBAAwB,KAAK;CAChD,MAAM,WAAW,sBAAsB,KAAK;CAC5C,IAAI,cAAc,QAAQ,YAAY,MACpC,iBAAiB,kBAAkB,IAAI,YAAY,QAAQ;CAG7D,KAAK,MAAM,SAAS,OAAO,OAAO,KAAK,GACrC,qBAAqB,OAAO,gBAAgB;AAEhD;AAEA,SAAS,YACP,OACA,QACA,kBACiB;CACjB,IAAI,MAAM,QAAQ,KAAK,GAAG;EACxB,IAAI,UAAU;EACd,MAAM,OAAkB,CAAC;EACzB,KAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,SAAS,YAAY,MAAM,QAAQ,gBAAgB;GACzD,IAAI,OAAO,SACT,UAAU;GAEZ,KAAK,KAAK,OAAO,KAAK;EACxB;EACA,OAAO,UAAU;GAAE,OAAO;GAAM;EAAQ,IAAI;GAAE;GAAO;EAAQ;CAC/D;CAEA,IAAI,CAAC,SAAS,KAAK,GACjB,OAAO;EAAE;EAAO,SAAS;CAAM;CAGjC,MAAM,WAAW,sBAAsB,OAAO,gBAAgB;CAC9D,IAAI,uBAAuB,KAAK,KAAK,iBAAiB,UAAU,MAAM,GACpE,OAAO;EACL,OAAO,wBAAwB,OAAO,MAAM;EAC5C,SAAS;CACX;CAGF,IAAI,UAAU;CACd,MAAM,OAAgC,CAAC;CACvC,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,KAAK,GAAG;EAChD,MAAM,SAAS,YAAY,OAAO,QAAQ,gBAAgB;EAC1D,IAAI,OAAO,SACT,UAAU;EAEZ,KAAK,OAAO,OAAO;CACrB;CAEA,OAAO,UAAU;EAAE,OAAO;EAAM;CAAQ,IAAI;EAAE;EAAO;CAAQ;AAC/D;AAEA,SAAS,sBACP,OACA,QACiB;CACjB,MAAM,mBAAqC,EACzC,mCAAmB,IAAI,IAAI,EAC7B;CACA,IAAI,OAAO,UAAU,UAAU;EAC7B,qBAAqB,OAAO,gBAAgB;EAC5C,OAAO,YAAY,OAAO,QAAQ,gBAAgB;CACpD;CAEA,MAAM,UAAU,MAAM,KAAK;CAC3B,IAAI,CAAC,QAAQ,WAAW,GAAG,KAAK,CAAC,QAAQ,WAAW,GAAG,GACrD,OAAO;EAAE;EAAO,SAAS;CAAM;CAGjC,IAAI;EACF,MAAM,SAAS,KAAK,MAAM,KAAK;EAC/B,qBAAqB,QAAQ,gBAAgB;EAC7C,MAAM,SAAS,YAAY,QAAQ,QAAQ,gBAAgB;EAC3D,OAAO,OAAO,UACV;GAAE,OAAO,KAAK,UAAU,OAAO,KAAK;GAAG,SAAS;EAAK,IACrD;GAAE;GAAO,SAAS;EAAM;CAC9B,QAAQ;EACN,OAAO;GAAE;GAAO,SAAS;EAAM;CACjC;AACF;AAEA,SAAS,gBACP,YACA,KACA,QACM;CACN,IAAI,EAAE,OAAO,aACX;CAGF,MAAM,SAAS,sBAAsB,WAAW,MAAM,MAAM;CAC5D,IAAI,OAAO,SACT,WAAW,OAAO,OAAO;AAE7B;AAEA,SAAS,kBAAkB,YAA8C;CACvE,MAAM,OAAO,WAAWC,kBAAAA,2BAA2B;CACnD,OAAO,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM;AAC5D;AAEA,SAAS,8BACP,YACM;CACN,MAAM,OAAO,WAAWA,kBAAAA,2BAA2B;CACnD,IAAI,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM,QACrD;CAGF,MAAM,gBACJ,WACE,GAAGA,kBAAAA,2BAA2B,qBAAqB;CAEvD,IACE,OAAO,kBAAkB,YACzB,cAAc,WAAW,0BAA0B,GAEnD,WAAWA,kBAAAA,2BAA2B,oBAAoB;AAE9D;AAEA,SAAgB,6BAA6B,MAA0B;CACrE,8BAA+B,KAA4B,UAAU;AACvE;AAEA,SAAS,4BACP,MACA,YACA,QACM;CACN,IACE,EACE,kBAAkB,UAAU,KAC5B,iBAAiB,KAAK,MAAM,MAAM,KAClCA,kBAAAA,2BAA2B,sBAAsB,aAGnD;CAGF,WAAWA,kBAAAA,2BAA2B,sBACpC,OAAO;AACX;AAEA,SAAgB,8BACd,MACA,QACM;CACN,MAAM,aAAc,KAA4B;CAChD,6BAA6B,IAAI;CAEjC,IAAI,CAAC,+BAA+B,MAAM,GACxC;CAGF,4BAA4B,MAAM,YAAY,MAAM;CAEpD,KAAK,MAAM,OAAO;EAChBA,kBAAAA,2BAA2B;EAC3BA,kBAAAA,2BAA2B;EAC3BA,kBAAAA,2BAA2B;EAC3BA,kBAAAA,2BAA2B;CAC7B,GACE,gBAAgB,YAAY,KAAK,MAAM;AAE3C;AAEA,IAAM,2CAAN,MAAwE;CACtE;CACA;CACA,8BAA+B,IAAI,QAGjC;CAEF,YACE,QACA,gBACA;EACA,KAAK,YAAY,IAAIC,eAAAA,sBAAsB,MAAM;EACjD,KAAK,iBAAiB;CACxB;CAEA,QAAQ,MAAY,eAA8B;EAChD,IAAIC,6BAAAA,uBAAuB,KAAK,IAAI,GAClC;EAEF,MAAM,SACJC,6BAAAA,sCAAsC,aAAa,KACnD,KAAK;EACP,IAAI,UAAU,MACZ,KAAK,YAAY,IAAI,MAAM,MAAM;EAEnC,KAAK,UAAU,QAAQ,MAAM,aAAa;CAC5C;CAEA,MAAM,MAA0B;EAC9B,IAAID,6BAAAA,uBAAuB,KAAK,IAAI,GAClC;EAEF,6BAA6B,IAAI;EACjC,6BAAA,kBAAkB,IAAI;EACtB,MAAM,SAAS,KAAK,YAAY,IAAI,IAAI,KAAK,KAAK;EAClD,IAAI,UAAU,MACZ,8BAA8B,MAAM,MAAM;EAE5C,KAAK,UAAU,MAAM,IAAI;CAC3B;CAEA,aAA4B;EAC1B,OAAO,KAAK,UAAU,WAAW;CACnC;CAEA,WAA0B;EACxB,OAAO,KAAK,UAAU,SAAS;CACjC;AACF;AAEA,SAAgB,4BACd,QACA,aACA,eACe;CAIf,OAAO,IAAI,yCAAyC,QAH7BE,uBAAAA,2BAA2B,aAAa,aAAa,IACxEC,uBAAAA,+BAA+B,aAAa,aAAa,IACzD,KAAA,CACsE;AAC5E;AAEA,SAAS,qBAA8B;CACrC,OACE,UAAU,QAAQ,IAAI,mBAAmB,KACzC,UAAU,QAAQ,IAAI,mBAAmB;AAE7C;AAEA,SAAS,sBAAsB,UAAsC;CACnE,IAAI,YAAY,MACd,OAAO;CAET,OAAO,UAAU,SAAS,SAAS,KAAK,UAAU,SAAS,SAAS;AACtE;AAEA,SAAgB,+BAA+B,EAC7C,aACA,iBAIU;CACV,MAAM,WAAWC,uBAAAA,sBAAsB,aAAa,aAAa;CACjE,IAAI,UAAU,YAAY,OACxB,OAAO;CAGT,MAAM,WAAW,UAAU,iBAAiB;CAC5C,IAAI,YAAY,MACd,OACE,aAAa,sBAAsB,QAAQ,KAAK,mBAAmB;CAIvE,OAAO,sBAAsB,QAAQ,KAAK,mBAAmB;AAC/D"}
+{"version":3,"file":"langfuseToolOutputTracing.cjs","names":["normalizeToolName","LangfuseOtelSpanAttributes","LangfuseSpanProcessor","shouldDropLangfuseSpan","resolveToolOutputTracingConfigForSpan","hasToolOutputTracingConfig","resolveToolOutputTracingConfig","resolveLangfuseConfig"],"sources":["../../src/langfuseToolOutputTracing.ts"],"sourcesContent":["import { LangfuseSpanProcessor } from '@langfuse/otel';\nimport { LangfuseOtelSpanAttributes } from '@langfuse/tracing';\nimport type {\n  ReadableSpan,\n  Span,\n  SpanProcessor,\n} from '@opentelemetry/sdk-trace-base';\nimport type { LangfuseSpanProcessorParams } from '@langfuse/otel';\nimport type { Context } from '@opentelemetry/api';\nimport type { ResolvedLangfuseToolOutputTracingConfig } from '@/langfuseRuntimeContext';\nimport type * as t from '@/types';\nimport {\n  LANGFUSE_TOOL_OUTPUT_REDACTION_TEXT,\n  hasToolOutputTracingConfig,\n  normalizeToolName,\n  resolveLangfuseConfig,\n  resolveToolOutputTracingConfig,\n} from '@/langfuseConfig';\nimport {\n  shapeLangfuseSpan,\n  shouldDropLangfuseSpan,\n} from '@/langfuseTraceShaping';\nimport { resolveToolOutputTracingConfigForSpan } from '@/langfuseRuntimeScope';\n\nexport { LANGFUSE_TOOL_OUTPUT_REDACTION_TEXT, resolveLangfuseConfig };\n\nconst LANGGRAPH_TOOL_NODE_PREFIX = 'tools=';\n\nconst CHAT_ROLES = new Set([\n  'assistant',\n  'developer',\n  'human',\n  'system',\n  'user',\n]);\n\ntype SpanWithAttributes = ReadableSpan & {\n  attributes: Record<string, unknown>;\n};\n\ntype RedactionResult = {\n  value: unknown;\n  changed: boolean;\n};\n\ntype RedactionContext = {\n  toolNamesByCallId: Map<string, string>;\n};\n\nconst TOOL_OUTPUT_FIELD_KEYS = ['content', 'artifact'];\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return value != null && typeof value === 'object' && !Array.isArray(value);\n}\n\nfunction isPresent(value: unknown): value is string {\n  return typeof value === 'string' && value.trim() !== '';\n}\n\nfunction shouldApplyToolOutputRedaction(\n  config: ResolvedLangfuseToolOutputTracingConfig\n): boolean {\n  return config.enabled === false || config.redactedToolNames.size > 0;\n}\n\nfunction toolNameMatches(\n  toolName: string | undefined,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): boolean {\n  if (!isPresent(toolName)) {\n    return false;\n  }\n\n  const normalizedToolName = normalizeToolName(toolName);\n  if (config.redactedToolNameMatchMode === 'partial') {\n    for (const redactedToolName of config.redactedToolNames) {\n      if (normalizedToolName.includes(redactedToolName)) {\n        return true;\n      }\n    }\n    return false;\n  }\n\n  return config.redactedToolNames.has(normalizedToolName);\n}\n\nfunction shouldRedactTool(\n  toolName: string | undefined,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): boolean {\n  return config.enabled === false || toolNameMatches(toolName, config);\n}\n\nfunction getStringField(\n  value: Record<string, unknown>,\n  key: string\n): string | undefined {\n  const field = value[key];\n  return typeof field === 'string' ? field : undefined;\n}\n\nfunction getNestedStringField(\n  value: Record<string, unknown>,\n  objectKey: string,\n  fieldKey: string\n): string | undefined {\n  const nested = value[objectKey];\n  if (!isRecord(nested)) {\n    return undefined;\n  }\n  return getStringField(nested, fieldKey);\n}\n\nfunction getSerializedToolCallId(\n  value: Record<string, unknown>\n): string | undefined {\n  return (\n    getStringField(value, 'tool_call_id') ??\n    getNestedStringField(value, 'kwargs', 'tool_call_id') ??\n    getNestedStringField(value, 'additional_kwargs', 'tool_call_id') ??\n    getNestedStringField(value, 'data', 'tool_call_id') ??\n    (typeof value.id === 'string' ? value.id : undefined)\n  );\n}\n\nfunction getSerializedToolName(\n  value: Record<string, unknown>,\n  redactionContext?: RedactionContext\n): string | undefined {\n  const role = getStringField(value, 'role');\n  const explicitName =\n    getStringField(value, 'name') ??\n    getStringField(value, 'tool_name') ??\n    getNestedStringField(value, 'function', 'name') ??\n    getNestedStringField(value, 'kwargs', 'name') ??\n    getNestedStringField(value, 'additional_kwargs', 'name') ??\n    getNestedStringField(value, 'data', 'name') ??\n    (role != null && role.toLowerCase() !== 'tool' ? role : undefined);\n\n  if (explicitName != null) {\n    return explicitName;\n  }\n\n  const toolCallId = getSerializedToolCallId(value);\n  return toolCallId != null\n    ? redactionContext?.toolNamesByCallId.get(toolCallId)\n    : undefined;\n}\n\nfunction hasToolMessageIdentity(value: Record<string, unknown>): boolean {\n  const type = getStringField(value, 'type') ?? getStringField(value, '_type');\n  if (type === 'tool' || type === 'tool_message') {\n    return true;\n  }\n\n  const id = value.id;\n  if (\n    Array.isArray(id) &&\n    id.some((part) => typeof part === 'string' && part.includes('ToolMessage'))\n  ) {\n    return true;\n  }\n\n  if (\n    'tool_call_id' in value ||\n    getNestedStringField(value, 'kwargs', 'tool_call_id') != null ||\n    getNestedStringField(value, 'additional_kwargs', 'tool_call_id') != null\n  ) {\n    return true;\n  }\n\n  const role = getStringField(value, 'role');\n  return (\n    role != null &&\n    !CHAT_ROLES.has(role.toLowerCase()) &&\n    ('content' in value || isRecord(value.kwargs) || isRecord(value.data))\n  );\n}\n\nfunction redactToolContentFields(\n  value: Record<string, unknown>,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): Record<string, unknown> {\n  const next = { ...value };\n\n  for (const outputKey of TOOL_OUTPUT_FIELD_KEYS) {\n    if (outputKey in next) {\n      next[outputKey] = config.redactionText;\n    }\n  }\n\n  for (const nestedKey of ['kwargs', 'data', 'additional_kwargs']) {\n    const nested = next[nestedKey];\n    if (!isRecord(nested)) {\n      continue;\n    }\n    const nextNested = { ...nested };\n    let changed = false;\n    for (const outputKey of TOOL_OUTPUT_FIELD_KEYS) {\n      if (outputKey in nextNested) {\n        nextNested[outputKey] = config.redactionText;\n        changed = true;\n      }\n    }\n    if (changed) {\n      next[nestedKey] = nextNested;\n    }\n  }\n\n  return next;\n}\n\nfunction collectToolCallNames(\n  value: unknown,\n  redactionContext: RedactionContext\n): void {\n  if (Array.isArray(value)) {\n    for (const item of value) {\n      collectToolCallNames(item, redactionContext);\n    }\n    return;\n  }\n\n  if (!isRecord(value)) {\n    return;\n  }\n\n  const toolCallId = getSerializedToolCallId(value);\n  const toolName = getSerializedToolName(value);\n  if (toolCallId != null && toolName != null) {\n    redactionContext.toolNamesByCallId.set(toolCallId, toolName);\n  }\n\n  for (const child of Object.values(value)) {\n    collectToolCallNames(child, redactionContext);\n  }\n}\n\nfunction redactValue(\n  value: unknown,\n  config: ResolvedLangfuseToolOutputTracingConfig,\n  redactionContext: RedactionContext\n): RedactionResult {\n  if (Array.isArray(value)) {\n    let changed = false;\n    const next: unknown[] = [];\n    for (const item of value) {\n      const result = redactValue(item, config, redactionContext);\n      if (result.changed) {\n        changed = true;\n      }\n      next.push(result.value);\n    }\n    return changed ? { value: next, changed } : { value, changed };\n  }\n\n  if (!isRecord(value)) {\n    return { value, changed: false };\n  }\n\n  const toolName = getSerializedToolName(value, redactionContext);\n  if (hasToolMessageIdentity(value) && shouldRedactTool(toolName, config)) {\n    return {\n      value: redactToolContentFields(value, config),\n      changed: true,\n    };\n  }\n\n  let changed = false;\n  const next: Record<string, unknown> = {};\n  for (const [key, child] of Object.entries(value)) {\n    const result = redactValue(child, config, redactionContext);\n    if (result.changed) {\n      changed = true;\n    }\n    next[key] = result.value;\n  }\n\n  return changed ? { value: next, changed } : { value, changed };\n}\n\nfunction redactSerializedValue(\n  value: unknown,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): RedactionResult {\n  const redactionContext: RedactionContext = {\n    toolNamesByCallId: new Map(),\n  };\n  if (typeof value !== 'string') {\n    collectToolCallNames(value, redactionContext);\n    return redactValue(value, config, redactionContext);\n  }\n\n  const trimmed = value.trim();\n  if (!trimmed.startsWith('{') && !trimmed.startsWith('[')) {\n    return { value, changed: false };\n  }\n\n  try {\n    const parsed = JSON.parse(value) as unknown;\n    collectToolCallNames(parsed, redactionContext);\n    const result = redactValue(parsed, config, redactionContext);\n    return result.changed\n      ? { value: JSON.stringify(result.value), changed: true }\n      : { value, changed: false };\n  } catch {\n    return { value, changed: false };\n  }\n}\n\nfunction redactAttribute(\n  attributes: Record<string, unknown>,\n  key: string,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): void {\n  if (!(key in attributes)) {\n    return;\n  }\n\n  const result = redactSerializedValue(attributes[key], config);\n  if (result.changed) {\n    attributes[key] = result.value;\n  }\n}\n\nfunction isToolObservation(attributes: Record<string, unknown>): boolean {\n  const type = attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE];\n  return typeof type === 'string' && type.toLowerCase() === 'tool';\n}\n\nfunction classifyLangGraphToolNodeSpan(\n  attributes: Record<string, unknown>\n): void {\n  const type = attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE];\n  if (typeof type !== 'string' || type.toLowerCase() !== 'span') {\n    return;\n  }\n\n  const langGraphNode =\n    attributes[\n      `${LangfuseOtelSpanAttributes.OBSERVATION_METADATA}.langgraph_node`\n    ];\n  if (\n    typeof langGraphNode === 'string' &&\n    langGraphNode.startsWith(LANGGRAPH_TOOL_NODE_PREFIX)\n  ) {\n    attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE] = 'tool';\n  }\n}\n\nexport function classifyLangfuseToolNodeSpan(span: ReadableSpan): void {\n  classifyLangGraphToolNodeSpan((span as SpanWithAttributes).attributes);\n}\n\nfunction redactToolObservationOutput(\n  span: ReadableSpan,\n  attributes: Record<string, unknown>,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): void {\n  if (\n    !(\n      isToolObservation(attributes) &&\n      shouldRedactTool(span.name, config) &&\n      LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT in attributes\n    )\n  ) {\n    return;\n  }\n\n  attributes[LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT] =\n    config.redactionText;\n}\n\nexport function redactLangfuseSpanToolOutputs(\n  span: ReadableSpan,\n  config: ResolvedLangfuseToolOutputTracingConfig\n): void {\n  const attributes = (span as SpanWithAttributes).attributes;\n  classifyLangfuseToolNodeSpan(span);\n\n  if (!shouldApplyToolOutputRedaction(config)) {\n    return;\n  }\n\n  redactToolObservationOutput(span, attributes, config);\n\n  for (const key of [\n    LangfuseOtelSpanAttributes.OBSERVATION_INPUT,\n    LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT,\n    LangfuseOtelSpanAttributes.TRACE_INPUT,\n    LangfuseOtelSpanAttributes.TRACE_OUTPUT,\n  ]) {\n    redactAttribute(attributes, key, config);\n  }\n}\n\nclass ToolOutputRedactingLangfuseSpanProcessor implements SpanProcessor {\n  private readonly processor: LangfuseSpanProcessor;\n  private readonly fallbackConfig?: ResolvedLangfuseToolOutputTracingConfig;\n  private readonly spanConfigs = new WeakMap<\n    object,\n    ResolvedLangfuseToolOutputTracingConfig\n  >();\n\n  constructor(\n    params?: LangfuseSpanProcessorParams,\n    fallbackConfig?: ResolvedLangfuseToolOutputTracingConfig\n  ) {\n    this.processor = new LangfuseSpanProcessor(params);\n    this.fallbackConfig = fallbackConfig;\n  }\n\n  onStart(span: Span, parentContext: Context): void {\n    if (shouldDropLangfuseSpan(span.name)) {\n      return;\n    }\n    const config =\n      resolveToolOutputTracingConfigForSpan(parentContext) ??\n      this.fallbackConfig;\n    if (config != null) {\n      this.spanConfigs.set(span, config);\n    }\n    this.processor.onStart(span, parentContext);\n  }\n\n  onEnd(span: ReadableSpan): void {\n    if (shouldDropLangfuseSpan(span.name)) {\n      return;\n    }\n    classifyLangfuseToolNodeSpan(span);\n    shapeLangfuseSpan(span);\n    const config = this.spanConfigs.get(span) ?? this.fallbackConfig;\n    if (config != null) {\n      redactLangfuseSpanToolOutputs(span, config);\n    }\n    this.processor.onEnd(span);\n  }\n\n  forceFlush(): Promise<void> {\n    return this.processor.forceFlush();\n  }\n\n  shutdown(): Promise<void> {\n    return this.processor.shutdown();\n  }\n}\n\nexport function createLangfuseSpanProcessor(\n  params?: LangfuseSpanProcessorParams,\n  runLangfuse?: t.LangfuseConfig,\n  agentLangfuse?: t.LangfuseConfig\n): SpanProcessor {\n  const fallbackConfig = hasToolOutputTracingConfig(runLangfuse, agentLangfuse)\n    ? resolveToolOutputTracingConfig(runLangfuse, agentLangfuse)\n    : undefined;\n  return new ToolOutputRedactingLangfuseSpanProcessor(params, fallbackConfig);\n}\n\nfunction hasLangfuseEnvKeys(): boolean {\n  return (\n    isPresent(process.env.LANGFUSE_SECRET_KEY) &&\n    isPresent(process.env.LANGFUSE_PUBLIC_KEY)\n  );\n}\n\nfunction hasLangfuseConfigKeys(langfuse?: t.LangfuseConfig): boolean {\n  if (langfuse == null) {\n    return false;\n  }\n  return isPresent(langfuse.secretKey) && isPresent(langfuse.publicKey);\n}\n\nexport function shouldTraceToolNodeForLangfuse({\n  runLangfuse,\n  agentLangfuse,\n}: {\n  runLangfuse?: t.LangfuseConfig;\n  agentLangfuse?: t.LangfuseConfig;\n}): boolean {\n  const langfuse = resolveLangfuseConfig(runLangfuse, agentLangfuse);\n  if (langfuse?.enabled === false) {\n    return false;\n  }\n\n  const explicit = langfuse?.toolNodeTracing?.enabled;\n  if (explicit !== true) {\n    return false;\n  }\n\n  return hasLangfuseConfigKeys(langfuse) || hasLangfuseEnvKeys();\n}\n"],"mappings":";;;;;;AA0BA,MAAM,6BAA6B;AAEnC,MAAM,aAAa,IAAI,IAAI;CACzB;CACA;CACA;CACA;CACA;AACF,CAAC;AAeD,MAAM,yBAAyB,CAAC,WAAW,UAAU;AAErD,SAAS,SAAS,OAAkD;CAClE,OAAO,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAC3E;AAEA,SAAS,UAAU,OAAiC;CAClD,OAAO,OAAO,UAAU,YAAY,MAAM,KAAK,MAAM;AACvD;AAEA,SAAS,+BACP,QACS;CACT,OAAO,OAAO,YAAY,SAAS,OAAO,kBAAkB,OAAO;AACrE;AAEA,SAAS,gBACP,UACA,QACS;CACT,IAAI,CAAC,UAAU,QAAQ,GACrB,OAAO;CAGT,MAAM,qBAAqBA,uBAAAA,kBAAkB,QAAQ;CACrD,IAAI,OAAO,8BAA8B,WAAW;EAClD,KAAK,MAAM,oBAAoB,OAAO,mBACpC,IAAI,mBAAmB,SAAS,gBAAgB,GAC9C,OAAO;EAGX,OAAO;CACT;CAEA,OAAO,OAAO,kBAAkB,IAAI,kBAAkB;AACxD;AAEA,SAAS,iBACP,UACA,QACS;CACT,OAAO,OAAO,YAAY,SAAS,gBAAgB,UAAU,MAAM;AACrE;AAEA,SAAS,eACP,OACA,KACoB;CACpB,MAAM,QAAQ,MAAM;CACpB,OAAO,OAAO,UAAU,WAAW,QAAQ,KAAA;AAC7C;AAEA,SAAS,qBACP,OACA,WACA,UACoB;CACpB,MAAM,SAAS,MAAM;CACrB,IAAI,CAAC,SAAS,MAAM,GAClB;CAEF,OAAO,eAAe,QAAQ,QAAQ;AACxC;AAEA,SAAS,wBACP,OACoB;CACpB,OACE,eAAe,OAAO,cAAc,KACpC,qBAAqB,OAAO,UAAU,cAAc,KACpD,qBAAqB,OAAO,qBAAqB,cAAc,KAC/D,qBAAqB,OAAO,QAAQ,cAAc,MACjD,OAAO,MAAM,OAAO,WAAW,MAAM,KAAK,KAAA;AAE/C;AAEA,SAAS,sBACP,OACA,kBACoB;CACpB,MAAM,OAAO,eAAe,OAAO,MAAM;CACzC,MAAM,eACJ,eAAe,OAAO,MAAM,KAC5B,eAAe,OAAO,WAAW,KACjC,qBAAqB,OAAO,YAAY,MAAM,KAC9C,qBAAqB,OAAO,UAAU,MAAM,KAC5C,qBAAqB,OAAO,qBAAqB,MAAM,KACvD,qBAAqB,OAAO,QAAQ,MAAM,MACzC,QAAQ,QAAQ,KAAK,YAAY,MAAM,SAAS,OAAO,KAAA;CAE1D,IAAI,gBAAgB,MAClB,OAAO;CAGT,MAAM,aAAa,wBAAwB,KAAK;CAChD,OAAO,cAAc,OACjB,kBAAkB,kBAAkB,IAAI,UAAU,IAClD,KAAA;AACN;AAEA,SAAS,uBAAuB,OAAyC;CACvE,MAAM,OAAO,eAAe,OAAO,MAAM,KAAK,eAAe,OAAO,OAAO;CAC3E,IAAI,SAAS,UAAU,SAAS,gBAC9B,OAAO;CAGT,MAAM,KAAK,MAAM;CACjB,IACE,MAAM,QAAQ,EAAE,KAChB,GAAG,MAAM,SAAS,OAAO,SAAS,YAAY,KAAK,SAAS,aAAa,CAAC,GAE1E,OAAO;CAGT,IACE,kBAAkB,SAClB,qBAAqB,OAAO,UAAU,cAAc,KAAK,QACzD,qBAAqB,OAAO,qBAAqB,cAAc,KAAK,MAEpE,OAAO;CAGT,MAAM,OAAO,eAAe,OAAO,MAAM;CACzC,OACE,QAAQ,QACR,CAAC,WAAW,IAAI,KAAK,YAAY,CAAC,MACjC,aAAa,SAAS,SAAS,MAAM,MAAM,KAAK,SAAS,MAAM,IAAI;AAExE;AAEA,SAAS,wBACP,OACA,QACyB;CACzB,MAAM,OAAO,EAAE,GAAG,MAAM;CAExB,KAAK,MAAM,aAAa,wBACtB,IAAI,aAAa,MACf,KAAK,aAAa,OAAO;CAI7B,KAAK,MAAM,aAAa;EAAC;EAAU;EAAQ;CAAmB,GAAG;EAC/D,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,SAAS,MAAM,GAClB;EAEF,MAAM,aAAa,EAAE,GAAG,OAAO;EAC/B,IAAI,UAAU;EACd,KAAK,MAAM,aAAa,wBACtB,IAAI,aAAa,YAAY;GAC3B,WAAW,aAAa,OAAO;GAC/B,UAAU;EACZ;EAEF,IAAI,SACF,KAAK,aAAa;CAEtB;CAEA,OAAO;AACT;AAEA,SAAS,qBACP,OACA,kBACM;CACN,IAAI,MAAM,QAAQ,KAAK,GAAG;EACxB,KAAK,MAAM,QAAQ,OACjB,qBAAqB,MAAM,gBAAgB;EAE7C;CACF;CAEA,IAAI,CAAC,SAAS,KAAK,GACjB;CAGF,MAAM,aAAa,wBAAwB,KAAK;CAChD,MAAM,WAAW,sBAAsB,KAAK;CAC5C,IAAI,cAAc,QAAQ,YAAY,MACpC,iBAAiB,kBAAkB,IAAI,YAAY,QAAQ;CAG7D,KAAK,MAAM,SAAS,OAAO,OAAO,KAAK,GACrC,qBAAqB,OAAO,gBAAgB;AAEhD;AAEA,SAAS,YACP,OACA,QACA,kBACiB;CACjB,IAAI,MAAM,QAAQ,KAAK,GAAG;EACxB,IAAI,UAAU;EACd,MAAM,OAAkB,CAAC;EACzB,KAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,SAAS,YAAY,MAAM,QAAQ,gBAAgB;GACzD,IAAI,OAAO,SACT,UAAU;GAEZ,KAAK,KAAK,OAAO,KAAK;EACxB;EACA,OAAO,UAAU;GAAE,OAAO;GAAM;EAAQ,IAAI;GAAE;GAAO;EAAQ;CAC/D;CAEA,IAAI,CAAC,SAAS,KAAK,GACjB,OAAO;EAAE;EAAO,SAAS;CAAM;CAGjC,MAAM,WAAW,sBAAsB,OAAO,gBAAgB;CAC9D,IAAI,uBAAuB,KAAK,KAAK,iBAAiB,UAAU,MAAM,GACpE,OAAO;EACL,OAAO,wBAAwB,OAAO,MAAM;EAC5C,SAAS;CACX;CAGF,IAAI,UAAU;CACd,MAAM,OAAgC,CAAC;CACvC,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,KAAK,GAAG;EAChD,MAAM,SAAS,YAAY,OAAO,QAAQ,gBAAgB;EAC1D,IAAI,OAAO,SACT,UAAU;EAEZ,KAAK,OAAO,OAAO;CACrB;CAEA,OAAO,UAAU;EAAE,OAAO;EAAM;CAAQ,IAAI;EAAE;EAAO;CAAQ;AAC/D;AAEA,SAAS,sBACP,OACA,QACiB;CACjB,MAAM,mBAAqC,EACzC,mCAAmB,IAAI,IAAI,EAC7B;CACA,IAAI,OAAO,UAAU,UAAU;EAC7B,qBAAqB,OAAO,gBAAgB;EAC5C,OAAO,YAAY,OAAO,QAAQ,gBAAgB;CACpD;CAEA,MAAM,UAAU,MAAM,KAAK;CAC3B,IAAI,CAAC,QAAQ,WAAW,GAAG,KAAK,CAAC,QAAQ,WAAW,GAAG,GACrD,OAAO;EAAE;EAAO,SAAS;CAAM;CAGjC,IAAI;EACF,MAAM,SAAS,KAAK,MAAM,KAAK;EAC/B,qBAAqB,QAAQ,gBAAgB;EAC7C,MAAM,SAAS,YAAY,QAAQ,QAAQ,gBAAgB;EAC3D,OAAO,OAAO,UACV;GAAE,OAAO,KAAK,UAAU,OAAO,KAAK;GAAG,SAAS;EAAK,IACrD;GAAE;GAAO,SAAS;EAAM;CAC9B,QAAQ;EACN,OAAO;GAAE;GAAO,SAAS;EAAM;CACjC;AACF;AAEA,SAAS,gBACP,YACA,KACA,QACM;CACN,IAAI,EAAE,OAAO,aACX;CAGF,MAAM,SAAS,sBAAsB,WAAW,MAAM,MAAM;CAC5D,IAAI,OAAO,SACT,WAAW,OAAO,OAAO;AAE7B;AAEA,SAAS,kBAAkB,YAA8C;CACvE,MAAM,OAAO,WAAWC,kBAAAA,2BAA2B;CACnD,OAAO,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM;AAC5D;AAEA,SAAS,8BACP,YACM;CACN,MAAM,OAAO,WAAWA,kBAAAA,2BAA2B;CACnD,IAAI,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM,QACrD;CAGF,MAAM,gBACJ,WACE,GAAGA,kBAAAA,2BAA2B,qBAAqB;CAEvD,IACE,OAAO,kBAAkB,YACzB,cAAc,WAAW,0BAA0B,GAEnD,WAAWA,kBAAAA,2BAA2B,oBAAoB;AAE9D;AAEA,SAAgB,6BAA6B,MAA0B;CACrE,8BAA+B,KAA4B,UAAU;AACvE;AAEA,SAAS,4BACP,MACA,YACA,QACM;CACN,IACE,EACE,kBAAkB,UAAU,KAC5B,iBAAiB,KAAK,MAAM,MAAM,KAClCA,kBAAAA,2BAA2B,sBAAsB,aAGnD;CAGF,WAAWA,kBAAAA,2BAA2B,sBACpC,OAAO;AACX;AAEA,SAAgB,8BACd,MACA,QACM;CACN,MAAM,aAAc,KAA4B;CAChD,6BAA6B,IAAI;CAEjC,IAAI,CAAC,+BAA+B,MAAM,GACxC;CAGF,4BAA4B,MAAM,YAAY,MAAM;CAEpD,KAAK,MAAM,OAAO;EAChBA,kBAAAA,2BAA2B;EAC3BA,kBAAAA,2BAA2B;EAC3BA,kBAAAA,2BAA2B;EAC3BA,kBAAAA,2BAA2B;CAC7B,GACE,gBAAgB,YAAY,KAAK,MAAM;AAE3C;AAEA,IAAM,2CAAN,MAAwE;CACtE;CACA;CACA,8BAA+B,IAAI,QAGjC;CAEF,YACE,QACA,gBACA;EACA,KAAK,YAAY,IAAIC,eAAAA,sBAAsB,MAAM;EACjD,KAAK,iBAAiB;CACxB;CAEA,QAAQ,MAAY,eAA8B;EAChD,IAAIC,6BAAAA,uBAAuB,KAAK,IAAI,GAClC;EAEF,MAAM,SACJC,6BAAAA,sCAAsC,aAAa,KACnD,KAAK;EACP,IAAI,UAAU,MACZ,KAAK,YAAY,IAAI,MAAM,MAAM;EAEnC,KAAK,UAAU,QAAQ,MAAM,aAAa;CAC5C;CAEA,MAAM,MAA0B;EAC9B,IAAID,6BAAAA,uBAAuB,KAAK,IAAI,GAClC;EAEF,6BAA6B,IAAI;EACjC,6BAAA,kBAAkB,IAAI;EACtB,MAAM,SAAS,KAAK,YAAY,IAAI,IAAI,KAAK,KAAK;EAClD,IAAI,UAAU,MACZ,8BAA8B,MAAM,MAAM;EAE5C,KAAK,UAAU,MAAM,IAAI;CAC3B;CAEA,aAA4B;EAC1B,OAAO,KAAK,UAAU,WAAW;CACnC;CAEA,WAA0B;EACxB,OAAO,KAAK,UAAU,SAAS;CACjC;AACF;AAEA,SAAgB,4BACd,QACA,aACA,eACe;CAIf,OAAO,IAAI,yCAAyC,QAH7BE,uBAAAA,2BAA2B,aAAa,aAAa,IACxEC,uBAAAA,+BAA+B,aAAa,aAAa,IACzD,KAAA,CACsE;AAC5E;AAEA,SAAS,qBAA8B;CACrC,OACE,UAAU,QAAQ,IAAI,mBAAmB,KACzC,UAAU,QAAQ,IAAI,mBAAmB;AAE7C;AAEA,SAAS,sBAAsB,UAAsC;CACnE,IAAI,YAAY,MACd,OAAO;CAET,OAAO,UAAU,SAAS,SAAS,KAAK,UAAU,SAAS,SAAS;AACtE;AAEA,SAAgB,+BAA+B,EAC7C,aACA,iBAIU;CACV,MAAM,WAAWC,uBAAAA,sBAAsB,aAAa,aAAa;CACjE,IAAI,UAAU,YAAY,OACxB,OAAO;CAIT,IADiB,UAAU,iBAAiB,YAC3B,MACf,OAAO;CAGT,OAAO,sBAAsB,QAAQ,KAAK,mBAAmB;AAC/D"}

package/dist/cjs/langfuseTraceShaping.cjs

@@ -4,8 +4,13 @@ const LANGGRAPH_START_NODE = "__start__";
 const ANONYMOUS_LAMBDA_NAME = "RunnableLambda";
 const LANGGRAPH_AGENT_NODE_PREFIX = "agent=";
 const LANGGRAPH_TOOL_NODE_PREFIX = "tools=";
-const TOOL_BATCH_RUN_NAME = "tool_batch";
 const AGENT_NODE_SPAN_NAME = "agent";
+const TOOL_DISPATCH_SPAN_NAME = "tool-dispatch";
+const GENERATION_SPAN_NAME = "llm";
+const ROOT_OBSERVATION_TYPE = "agent";
+const CHAIN_OBSERVATION_TYPE = "chain";
+const AGENT_TRACE_TAG = "agent";
+const TITLE_TRACE_TAG = "title";
 function isRecord(value) {
 	return value != null && typeof value === "object" && !Array.isArray(value);
 }
@@ -113,57 +118,79 @@ function isRootSpan(span) {
 * LangGraph plumbing observations that add noise without information:
 * the duplicated `__start__` channel-seed nodes and anonymous
 * `RunnableLambda` pass-throughs (Langfuse team feedback items 4 & 5).
+* Internal ToolNode batch spans are disabled at their source so traced child
+* tools retain an exported parent. Explicitly traced ToolNodes are preserved.
 */
 function shouldDropLangfuseSpan(spanName) {
 	return spanName === LANGGRAPH_START_NODE || spanName === ANONYMOUS_LAMBDA_NAME;
 }
 function shapeToolNodeSpan(span) {
 	const inputKey = _langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_INPUT;
+	span.name = TOOL_DISPATCH_SPAN_NAME;
+	span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_TYPE] = CHAIN_OBSERVATION_TYPE;
 	const calls = findPendingToolCalls(parseAttributeValue(span.attributes[inputKey]));
 	if (calls.length === 0) return;
-	span.name = [...new Set(calls.map((call) => call.name))].join(", ");
 	span.attributes[inputKey] = JSON.stringify(calls.map(({ name, args }) => ({
 		name,
 		args
 	})));
 }
+function shapeAgentNodeSpan(span) {
+	span.name = AGENT_NODE_SPAN_NAME;
+	span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_TYPE] = ROOT_OBSERVATION_TYPE;
+}
 function shapeRootSpan(span) {
 	const inputKey = _langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_INPUT;
 	const outputKey = _langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT;
 	const question = findLastMessageText(parseAttributeValue(span.attributes[inputKey]), "user");
 	const answer = findLastMessageText(parseAttributeValue(span.attributes[outputKey]), "assistant");
-	if (question != null) {
-		span.attributes[inputKey] = question;
-		span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.TRACE_INPUT] = question;
-	}
-	if (answer != null) {
-		span.attributes[outputKey] = answer;
-		span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.TRACE_OUTPUT] = answer;
+	if (question != null) span.attributes[inputKey] = question;
+	if (answer != null) span.attributes[outputKey] = answer;
+	const traceInput = question ?? span.attributes[inputKey];
+	const traceOutput = answer ?? span.attributes[outputKey];
+	if (traceInput != null) span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.TRACE_INPUT] = traceInput;
+	if (traceOutput != null) span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.TRACE_OUTPUT] = traceOutput;
+}
+function isGenerationSpan(span) {
+	const type = span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_TYPE];
+	return typeof type === "string" && type.toLowerCase() === "generation";
+}
+function hasTraceTag(span, expectedTag) {
+	const tags = parseAttributeValue(span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.TRACE_TAGS]);
+	return Array.isArray(tags) && tags.some((tag) => typeof tag === "string" && tag === expectedTag);
+}
+function shapeRootObservationType(span) {
+	if (isGenerationSpan(span)) return;
+	if (hasTraceTag(span, AGENT_TRACE_TAG)) {
+		span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_TYPE] = ROOT_OBSERVATION_TYPE;
+		return;
 	}
+	if (hasTraceTag(span, TITLE_TRACE_TAG)) span.attributes[_langfuse_tracing.LangfuseOtelSpanAttributes.OBSERVATION_TYPE] = CHAIN_OBSERVATION_TYPE;
 }
 /**
 * Reshapes spans per Langfuse-team feedback before export:
 * - `agent=<id>` / `tools=<id>` node names carry the ephemeral agent id
 *   (`provider__model`) — strip it so switching models doesn't break
 *   name-based logic (item 1).
-* - Tool node spans are renamed to the actual tool name(s) and their
-*   input scoped to the pending tool-call args instead of the whole
-*   chat history (items 3 & 4).
-* - The root span (and trace) input/output become the user question and
-*   assistant response so the session view reads as a conversation
-*   (item 2).
+* - LLM generation spans keep the provider client class name (`ChatOpenAI`,
+*   `AzureChatOpenAI`, …); rename them to a provider-agnostic `llm` so the
+*   name reflects the operation, not the model (the model stays on the
+*   generation's model attribute).
+* - Agent nodes become `agent` observations, while tool-dispatch nodes become
+*   stable `chain` observations whose input is scoped to the pending calls.
+*   Individual child calls remain `tool` observations (items 3 & 4).
+* - Agent trace roots become `agent` observations and title trace roots become
+*   `chain` observations. Root and trace input/output are reduced to the user
+*   question and assistant response when chat messages are available (item 2).
 */
 function shapeLangfuseSpan(span) {
 	const mutable = span;
-	if (mutable.name.startsWith(LANGGRAPH_AGENT_NODE_PREFIX)) {
-		mutable.name = AGENT_NODE_SPAN_NAME;
-		return;
-	}
-	if (mutable.name.startsWith(LANGGRAPH_TOOL_NODE_PREFIX) || mutable.name === TOOL_BATCH_RUN_NAME) {
-		shapeToolNodeSpan(mutable);
-		return;
-	}
-	if (isRootSpan(span)) shapeRootSpan(mutable);
+	if (mutable.name.startsWith(LANGGRAPH_AGENT_NODE_PREFIX)) shapeAgentNodeSpan(mutable);
+	else if (mutable.name.startsWith(LANGGRAPH_TOOL_NODE_PREFIX)) shapeToolNodeSpan(mutable);
+	else if (isGenerationSpan(mutable)) mutable.name = GENERATION_SPAN_NAME;
+	if (!isRootSpan(span)) return;
+	shapeRootObservationType(mutable);
+	shapeRootSpan(mutable);
 }
 //#endregion
 exports.shapeLangfuseSpan = shapeLangfuseSpan;

package/dist/cjs/langfuseTraceShaping.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"langfuseTraceShaping.cjs","names":["LangfuseOtelSpanAttributes"],"sources":["../../src/langfuseTraceShaping.ts"],"sourcesContent":["import { LangfuseOtelSpanAttributes } from '@langfuse/tracing';\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\n\nconst LANGGRAPH_START_NODE = '__start__';\nconst ANONYMOUS_LAMBDA_NAME = 'RunnableLambda';\nconst LANGGRAPH_AGENT_NODE_PREFIX = 'agent=';\nconst LANGGRAPH_TOOL_NODE_PREFIX = 'tools=';\nconst TOOL_BATCH_RUN_NAME = 'tool_batch';\nconst AGENT_NODE_SPAN_NAME = 'agent';\n\ntype MutableSpan = ReadableSpan & {\n  name: string;\n  attributes: Record<string, unknown>;\n};\n\ntype SerializedToolCall = {\n  name: string;\n  args: unknown;\n  id?: string;\n};\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return value != null && typeof value === 'object' && !Array.isArray(value);\n}\n\nfunction parseAttributeValue(value: unknown): unknown {\n  if (typeof value !== 'string') {\n    return value;\n  }\n  const trimmed = value.trim();\n  if (!trimmed.startsWith('{') && !trimmed.startsWith('[')) {\n    return value;\n  }\n  try {\n    return JSON.parse(value) as unknown;\n  } catch {\n    return value;\n  }\n}\n\nfunction getMessageArray(\n  value: unknown\n): Record<string, unknown>[] | undefined {\n  if (Array.isArray(value)) {\n    const records = value.filter(isRecord);\n    return records.length > 0 ? records : undefined;\n  }\n  if (!isRecord(value)) {\n    return undefined;\n  }\n  return (\n    getMessageArray(value.messages) ??\n    getMessageArray(value.input) ??\n    getMessageArray(value.output)\n  );\n}\n\nfunction getMessageRole(message: Record<string, unknown>): string | undefined {\n  const id = message.id;\n  if (Array.isArray(id)) {\n    const className = id[id.length - 1];\n    if (typeof className === 'string') {\n      if (className.includes('Human')) {\n        return 'user';\n      }\n      if (className.includes('AI')) {\n        return 'assistant';\n      }\n      if (className.includes('System')) {\n        return 'system';\n      }\n      if (className.includes('Tool')) {\n        return 'tool';\n      }\n    }\n  }\n  const rawRole =\n    message.type ?? message._type ?? message.role ?? message.sender;\n  if (typeof rawRole !== 'string') {\n    const kwargs = message.kwargs;\n    return isRecord(kwargs) ? getMessageRole(kwargs) : undefined;\n  }\n  const normalized = rawRole.toLowerCase();\n  if (normalized === 'human') {\n    return 'user';\n  }\n  if (normalized === 'ai') {\n    return 'assistant';\n  }\n  return normalized;\n}\n\nfunction getMessageText(message: Record<string, unknown>): string | undefined {\n  const content =\n    message.content ??\n    (isRecord(message.kwargs) ? message.kwargs.content : undefined) ??\n    (isRecord(message.data) ? message.data.content : undefined);\n  if (typeof content === 'string') {\n    return content;\n  }\n  if (!Array.isArray(content)) {\n    return undefined;\n  }\n  const text = content\n    .filter(isRecord)\n    .map((part) => (typeof part.text === 'string' ? part.text : ''))\n    .join('');\n  return text === '' ? undefined : text;\n}\n\nfunction findLastMessageText(value: unknown, role: string): string | undefined {\n  const messages = getMessageArray(value);\n  if (messages == null) {\n    return undefined;\n  }\n  for (let i = messages.length - 1; i >= 0; i--) {\n    if (getMessageRole(messages[i]) !== role) {\n      continue;\n    }\n    const text = getMessageText(messages[i]);\n    if (text != null && text.trim() !== '') {\n      return text;\n    }\n  }\n  return undefined;\n}\n\nfunction normalizeToolCall(value: unknown): SerializedToolCall | undefined {\n  if (!isRecord(value)) {\n    return undefined;\n  }\n  const fn = value.function;\n  if (isRecord(fn) && typeof fn.name === 'string') {\n    return {\n      name: fn.name,\n      args: parseAttributeValue(fn.arguments),\n      ...(typeof value.id === 'string' ? { id: value.id } : {}),\n    };\n  }\n  if (typeof value.name !== 'string') {\n    return undefined;\n  }\n  return {\n    name: value.name,\n    args: value.args ?? {},\n    ...(typeof value.id === 'string' ? { id: value.id } : {}),\n  };\n}\n\nfunction getMessageToolCalls(\n  message: Record<string, unknown>\n): SerializedToolCall[] {\n  const rawCalls =\n    message.tool_calls ??\n    (isRecord(message.kwargs) ? message.kwargs.tool_calls : undefined) ??\n    (isRecord(message.additional_kwargs)\n      ? message.additional_kwargs.tool_calls\n      : undefined) ??\n    (isRecord(message.data) ? message.data.tool_calls : undefined);\n  if (!Array.isArray(rawCalls)) {\n    return [];\n  }\n  const calls: SerializedToolCall[] = [];\n  for (const rawCall of rawCalls) {\n    const call = normalizeToolCall(rawCall);\n    if (call != null) {\n      calls.push(call);\n    }\n  }\n  return calls;\n}\n\n/** Latest assistant turn's tool calls — the calls this tool node is executing. */\nfunction findPendingToolCalls(value: unknown): SerializedToolCall[] {\n  const messages = getMessageArray(value);\n  if (messages == null) {\n    return [];\n  }\n  for (let i = messages.length - 1; i >= 0; i--) {\n    if (getMessageRole(messages[i]) !== 'assistant') {\n      continue;\n    }\n    const calls = getMessageToolCalls(messages[i]);\n    if (calls.length > 0) {\n      return calls;\n    }\n  }\n  return [];\n}\n\nfunction getRootSpanParentId(span: ReadableSpan): string | undefined {\n  const legacyParent = (span as { parentSpanId?: string }).parentSpanId;\n  if (typeof legacyParent === 'string' && legacyParent !== '') {\n    return legacyParent;\n  }\n  const parentContext = (span as { parentSpanContext?: { spanId?: string } })\n    .parentSpanContext;\n  const spanId = parentContext?.spanId;\n  return typeof spanId === 'string' && spanId !== '' ? spanId : undefined;\n}\n\nfunction isRootSpan(span: ReadableSpan): boolean {\n  return getRootSpanParentId(span) == null;\n}\n\n/**\n * LangGraph plumbing observations that add noise without information:\n * the duplicated `__start__` channel-seed nodes and anonymous\n * `RunnableLambda` pass-throughs (Langfuse team feedback items 4 & 5).\n */\nexport function shouldDropLangfuseSpan(spanName: string): boolean {\n  return (\n    spanName === LANGGRAPH_START_NODE || spanName === ANONYMOUS_LAMBDA_NAME\n  );\n}\n\nfunction shapeToolNodeSpan(span: MutableSpan): void {\n  const inputKey = LangfuseOtelSpanAttributes.OBSERVATION_INPUT;\n  const calls = findPendingToolCalls(\n    parseAttributeValue(span.attributes[inputKey])\n  );\n  if (calls.length === 0) {\n    return;\n  }\n  span.name = [...new Set(calls.map((call) => call.name))].join(', ');\n  span.attributes[inputKey] = JSON.stringify(\n    calls.map(({ name, args }) => ({ name, args }))\n  );\n}\n\nfunction shapeRootSpan(span: MutableSpan): void {\n  const inputKey = LangfuseOtelSpanAttributes.OBSERVATION_INPUT;\n  const outputKey = LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT;\n  const question = findLastMessageText(\n    parseAttributeValue(span.attributes[inputKey]),\n    'user'\n  );\n  const answer = findLastMessageText(\n    parseAttributeValue(span.attributes[outputKey]),\n    'assistant'\n  );\n  if (question != null) {\n    span.attributes[inputKey] = question;\n    span.attributes[LangfuseOtelSpanAttributes.TRACE_INPUT] = question;\n  }\n  if (answer != null) {\n    span.attributes[outputKey] = answer;\n    span.attributes[LangfuseOtelSpanAttributes.TRACE_OUTPUT] = answer;\n  }\n}\n\n/**\n * Reshapes spans per Langfuse-team feedback before export:\n * - `agent=<id>` / `tools=<id>` node names carry the ephemeral agent id\n *   (`provider__model`) — strip it so switching models doesn't break\n *   name-based logic (item 1).\n * - Tool node spans are renamed to the actual tool name(s) and their\n *   input scoped to the pending tool-call args instead of the whole\n *   chat history (items 3 & 4).\n * - The root span (and trace) input/output become the user question and\n *   assistant response so the session view reads as a conversation\n *   (item 2).\n */\nexport function shapeLangfuseSpan(span: ReadableSpan): void {\n  const mutable = span as MutableSpan;\n  if (mutable.name.startsWith(LANGGRAPH_AGENT_NODE_PREFIX)) {\n    mutable.name = AGENT_NODE_SPAN_NAME;\n    return;\n  }\n  if (\n    mutable.name.startsWith(LANGGRAPH_TOOL_NODE_PREFIX) ||\n    mutable.name === TOOL_BATCH_RUN_NAME\n  ) {\n    shapeToolNodeSpan(mutable);\n    return;\n  }\n  if (isRootSpan(span)) {\n    shapeRootSpan(mutable);\n  }\n}\n"],"mappings":";;AAGA,MAAM,uBAAuB;AAC7B,MAAM,wBAAwB;AAC9B,MAAM,8BAA8B;AACpC,MAAM,6BAA6B;AACnC,MAAM,sBAAsB;AAC5B,MAAM,uBAAuB;AAa7B,SAAS,SAAS,OAAkD;CAClE,OAAO,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAC3E;AAEA,SAAS,oBAAoB,OAAyB;CACpD,IAAI,OAAO,UAAU,UACnB,OAAO;CAET,MAAM,UAAU,MAAM,KAAK;CAC3B,IAAI,CAAC,QAAQ,WAAW,GAAG,KAAK,CAAC,QAAQ,WAAW,GAAG,GACrD,OAAO;CAET,IAAI;EACF,OAAO,KAAK,MAAM,KAAK;CACzB,QAAQ;EACN,OAAO;CACT;AACF;AAEA,SAAS,gBACP,OACuC;CACvC,IAAI,MAAM,QAAQ,KAAK,GAAG;EACxB,MAAM,UAAU,MAAM,OAAO,QAAQ;EACrC,OAAO,QAAQ,SAAS,IAAI,UAAU,KAAA;CACxC;CACA,IAAI,CAAC,SAAS,KAAK,GACjB;CAEF,OACE,gBAAgB,MAAM,QAAQ,KAC9B,gBAAgB,MAAM,KAAK,KAC3B,gBAAgB,MAAM,MAAM;AAEhC;AAEA,SAAS,eAAe,SAAsD;CAC5E,MAAM,KAAK,QAAQ;CACnB,IAAI,MAAM,QAAQ,EAAE,GAAG;EACrB,MAAM,YAAY,GAAG,GAAG,SAAS;EACjC,IAAI,OAAO,cAAc,UAAU;GACjC,IAAI,UAAU,SAAS,OAAO,GAC5B,OAAO;GAET,IAAI,UAAU,SAAS,IAAI,GACzB,OAAO;GAET,IAAI,UAAU,SAAS,QAAQ,GAC7B,OAAO;GAET,IAAI,UAAU,SAAS,MAAM,GAC3B,OAAO;EAEX;CACF;CACA,MAAM,UACJ,QAAQ,QAAQ,QAAQ,SAAS,QAAQ,QAAQ,QAAQ;CAC3D,IAAI,OAAO,YAAY,UAAU;EAC/B,MAAM,SAAS,QAAQ;EACvB,OAAO,SAAS,MAAM,IAAI,eAAe,MAAM,IAAI,KAAA;CACrD;CACA,MAAM,aAAa,QAAQ,YAAY;CACvC,IAAI,eAAe,SACjB,OAAO;CAET,IAAI,eAAe,MACjB,OAAO;CAET,OAAO;AACT;AAEA,SAAS,eAAe,SAAsD;CAC5E,MAAM,UACJ,QAAQ,YACP,SAAS,QAAQ,MAAM,IAAI,QAAQ,OAAO,UAAU,KAAA,OACpD,SAAS,QAAQ,IAAI,IAAI,QAAQ,KAAK,UAAU,KAAA;CACnD,IAAI,OAAO,YAAY,UACrB,OAAO;CAET,IAAI,CAAC,MAAM,QAAQ,OAAO,GACxB;CAEF,MAAM,OAAO,QACV,OAAO,QAAQ,CAAC,CAChB,KAAK,SAAU,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO,EAAG,CAAC,CAC/D,KAAK,EAAE;CACV,OAAO,SAAS,KAAK,KAAA,IAAY;AACnC;AAEA,SAAS,oBAAoB,OAAgB,MAAkC;CAC7E,MAAM,WAAW,gBAAgB,KAAK;CACtC,IAAI,YAAY,MACd;CAEF,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;EAC7C,IAAI,eAAe,SAAS,EAAE,MAAM,MAClC;EAEF,MAAM,OAAO,eAAe,SAAS,EAAE;EACvC,IAAI,QAAQ,QAAQ,KAAK,KAAK,MAAM,IAClC,OAAO;CAEX;AAEF;AAEA,SAAS,kBAAkB,OAAgD;CACzE,IAAI,CAAC,SAAS,KAAK,GACjB;CAEF,MAAM,KAAK,MAAM;CACjB,IAAI,SAAS,EAAE,KAAK,OAAO,GAAG,SAAS,UACrC,OAAO;EACL,MAAM,GAAG;EACT,MAAM,oBAAoB,GAAG,SAAS;EACtC,GAAI,OAAO,MAAM,OAAO,WAAW,EAAE,IAAI,MAAM,GAAG,IAAI,CAAC;CACzD;CAEF,IAAI,OAAO,MAAM,SAAS,UACxB;CAEF,OAAO;EACL,MAAM,MAAM;EACZ,MAAM,MAAM,QAAQ,CAAC;EACrB,GAAI,OAAO,MAAM,OAAO,WAAW,EAAE,IAAI,MAAM,GAAG,IAAI,CAAC;CACzD;AACF;AAEA,SAAS,oBACP,SACsB;CACtB,MAAM,WACJ,QAAQ,eACP,SAAS,QAAQ,MAAM,IAAI,QAAQ,OAAO,aAAa,KAAA,OACvD,SAAS,QAAQ,iBAAiB,IAC/B,QAAQ,kBAAkB,aAC1B,KAAA,OACH,SAAS,QAAQ,IAAI,IAAI,QAAQ,KAAK,aAAa,KAAA;CACtD,IAAI,CAAC,MAAM,QAAQ,QAAQ,GACzB,OAAO,CAAC;CAEV,MAAM,QAA8B,CAAC;CACrC,KAAK,MAAM,WAAW,UAAU;EAC9B,MAAM,OAAO,kBAAkB,OAAO;EACtC,IAAI,QAAQ,MACV,MAAM,KAAK,IAAI;CAEnB;CACA,OAAO;AACT;;AAGA,SAAS,qBAAqB,OAAsC;CAClE,MAAM,WAAW,gBAAgB,KAAK;CACtC,IAAI,YAAY,MACd,OAAO,CAAC;CAEV,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;EAC7C,IAAI,eAAe,SAAS,EAAE,MAAM,aAClC;EAEF,MAAM,QAAQ,oBAAoB,SAAS,EAAE;EAC7C,IAAI,MAAM,SAAS,GACjB,OAAO;CAEX;CACA,OAAO,CAAC;AACV;AAEA,SAAS,oBAAoB,MAAwC;CACnE,MAAM,eAAgB,KAAmC;CACzD,IAAI,OAAO,iBAAiB,YAAY,iBAAiB,IACvD,OAAO;CAIT,MAAM,SAFiB,KACpB,mBAC2B;CAC9B,OAAO,OAAO,WAAW,YAAY,WAAW,KAAK,SAAS,KAAA;AAChE;AAEA,SAAS,WAAW,MAA6B;CAC/C,OAAO,oBAAoB,IAAI,KAAK;AACtC;;;;;;AAOA,SAAgB,uBAAuB,UAA2B;CAChE,OACE,aAAa,wBAAwB,aAAa;AAEtD;AAEA,SAAS,kBAAkB,MAAyB;CAClD,MAAM,WAAWA,kBAAAA,2BAA2B;CAC5C,MAAM,QAAQ,qBACZ,oBAAoB,KAAK,WAAW,SAAS,CAC/C;CACA,IAAI,MAAM,WAAW,GACnB;CAEF,KAAK,OAAO,CAAC,GAAG,IAAI,IAAI,MAAM,KAAK,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI;CAClE,KAAK,WAAW,YAAY,KAAK,UAC/B,MAAM,KAAK,EAAE,MAAM,YAAY;EAAE;EAAM;CAAK,EAAE,CAChD;AACF;AAEA,SAAS,cAAc,MAAyB;CAC9C,MAAM,WAAWA,kBAAAA,2BAA2B;CAC5C,MAAM,YAAYA,kBAAAA,2BAA2B;CAC7C,MAAM,WAAW,oBACf,oBAAoB,KAAK,WAAW,SAAS,GAC7C,MACF;CACA,MAAM,SAAS,oBACb,oBAAoB,KAAK,WAAW,UAAU,GAC9C,WACF;CACA,IAAI,YAAY,MAAM;EACpB,KAAK,WAAW,YAAY;EAC5B,KAAK,WAAWA,kBAAAA,2BAA2B,eAAe;CAC5D;CACA,IAAI,UAAU,MAAM;EAClB,KAAK,WAAW,aAAa;EAC7B,KAAK,WAAWA,kBAAAA,2BAA2B,gBAAgB;CAC7D;AACF;;;;;;;;;;;;;AAcA,SAAgB,kBAAkB,MAA0B;CAC1D,MAAM,UAAU;CAChB,IAAI,QAAQ,KAAK,WAAW,2BAA2B,GAAG;EACxD,QAAQ,OAAO;EACf;CACF;CACA,IACE,QAAQ,KAAK,WAAW,0BAA0B,KAClD,QAAQ,SAAS,qBACjB;EACA,kBAAkB,OAAO;EACzB;CACF;CACA,IAAI,WAAW,IAAI,GACjB,cAAc,OAAO;AAEzB"}
+{"version":3,"file":"langfuseTraceShaping.cjs","names":["LangfuseOtelSpanAttributes"],"sources":["../../src/langfuseTraceShaping.ts"],"sourcesContent":["import { LangfuseOtelSpanAttributes } from '@langfuse/tracing';\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\n\nconst LANGGRAPH_START_NODE = '__start__';\nconst ANONYMOUS_LAMBDA_NAME = 'RunnableLambda';\nconst LANGGRAPH_AGENT_NODE_PREFIX = 'agent=';\nconst LANGGRAPH_TOOL_NODE_PREFIX = 'tools=';\nconst AGENT_NODE_SPAN_NAME = 'agent';\nconst TOOL_DISPATCH_SPAN_NAME = 'tool-dispatch';\nconst GENERATION_SPAN_NAME = 'llm';\nconst ROOT_OBSERVATION_TYPE = 'agent';\nconst CHAIN_OBSERVATION_TYPE = 'chain';\nconst AGENT_TRACE_TAG = 'agent';\nconst TITLE_TRACE_TAG = 'title';\n\ntype MutableSpan = ReadableSpan & {\n  name: string;\n  attributes: Record<string, unknown>;\n};\n\ntype SerializedToolCall = {\n  name: string;\n  args: unknown;\n  id?: string;\n};\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return value != null && typeof value === 'object' && !Array.isArray(value);\n}\n\nfunction parseAttributeValue(value: unknown): unknown {\n  if (typeof value !== 'string') {\n    return value;\n  }\n  const trimmed = value.trim();\n  if (!trimmed.startsWith('{') && !trimmed.startsWith('[')) {\n    return value;\n  }\n  try {\n    return JSON.parse(value) as unknown;\n  } catch {\n    return value;\n  }\n}\n\nfunction getMessageArray(\n  value: unknown\n): Record<string, unknown>[] | undefined {\n  if (Array.isArray(value)) {\n    const records = value.filter(isRecord);\n    return records.length > 0 ? records : undefined;\n  }\n  if (!isRecord(value)) {\n    return undefined;\n  }\n  return (\n    getMessageArray(value.messages) ??\n    getMessageArray(value.input) ??\n    getMessageArray(value.output)\n  );\n}\n\nfunction getMessageRole(message: Record<string, unknown>): string | undefined {\n  const id = message.id;\n  if (Array.isArray(id)) {\n    const className = id[id.length - 1];\n    if (typeof className === 'string') {\n      if (className.includes('Human')) {\n        return 'user';\n      }\n      if (className.includes('AI')) {\n        return 'assistant';\n      }\n      if (className.includes('System')) {\n        return 'system';\n      }\n      if (className.includes('Tool')) {\n        return 'tool';\n      }\n    }\n  }\n  const rawRole =\n    message.type ?? message._type ?? message.role ?? message.sender;\n  if (typeof rawRole !== 'string') {\n    const kwargs = message.kwargs;\n    return isRecord(kwargs) ? getMessageRole(kwargs) : undefined;\n  }\n  const normalized = rawRole.toLowerCase();\n  if (normalized === 'human') {\n    return 'user';\n  }\n  if (normalized === 'ai') {\n    return 'assistant';\n  }\n  return normalized;\n}\n\nfunction getMessageText(message: Record<string, unknown>): string | undefined {\n  const content =\n    message.content ??\n    (isRecord(message.kwargs) ? message.kwargs.content : undefined) ??\n    (isRecord(message.data) ? message.data.content : undefined);\n  if (typeof content === 'string') {\n    return content;\n  }\n  if (!Array.isArray(content)) {\n    return undefined;\n  }\n  const text = content\n    .filter(isRecord)\n    .map((part) => (typeof part.text === 'string' ? part.text : ''))\n    .join('');\n  return text === '' ? undefined : text;\n}\n\nfunction findLastMessageText(value: unknown, role: string): string | undefined {\n  const messages = getMessageArray(value);\n  if (messages == null) {\n    return undefined;\n  }\n  for (let i = messages.length - 1; i >= 0; i--) {\n    if (getMessageRole(messages[i]) !== role) {\n      continue;\n    }\n    const text = getMessageText(messages[i]);\n    if (text != null && text.trim() !== '') {\n      return text;\n    }\n  }\n  return undefined;\n}\n\nfunction normalizeToolCall(value: unknown): SerializedToolCall | undefined {\n  if (!isRecord(value)) {\n    return undefined;\n  }\n  const fn = value.function;\n  if (isRecord(fn) && typeof fn.name === 'string') {\n    return {\n      name: fn.name,\n      args: parseAttributeValue(fn.arguments),\n      ...(typeof value.id === 'string' ? { id: value.id } : {}),\n    };\n  }\n  if (typeof value.name !== 'string') {\n    return undefined;\n  }\n  return {\n    name: value.name,\n    args: value.args ?? {},\n    ...(typeof value.id === 'string' ? { id: value.id } : {}),\n  };\n}\n\nfunction getMessageToolCalls(\n  message: Record<string, unknown>\n): SerializedToolCall[] {\n  const rawCalls =\n    message.tool_calls ??\n    (isRecord(message.kwargs) ? message.kwargs.tool_calls : undefined) ??\n    (isRecord(message.additional_kwargs)\n      ? message.additional_kwargs.tool_calls\n      : undefined) ??\n    (isRecord(message.data) ? message.data.tool_calls : undefined);\n  if (!Array.isArray(rawCalls)) {\n    return [];\n  }\n  const calls: SerializedToolCall[] = [];\n  for (const rawCall of rawCalls) {\n    const call = normalizeToolCall(rawCall);\n    if (call != null) {\n      calls.push(call);\n    }\n  }\n  return calls;\n}\n\n/** Latest assistant turn's tool calls — the calls this tool node is executing. */\nfunction findPendingToolCalls(value: unknown): SerializedToolCall[] {\n  const messages = getMessageArray(value);\n  if (messages == null) {\n    return [];\n  }\n  for (let i = messages.length - 1; i >= 0; i--) {\n    if (getMessageRole(messages[i]) !== 'assistant') {\n      continue;\n    }\n    const calls = getMessageToolCalls(messages[i]);\n    if (calls.length > 0) {\n      return calls;\n    }\n  }\n  return [];\n}\n\nfunction getRootSpanParentId(span: ReadableSpan): string | undefined {\n  const legacyParent = (span as { parentSpanId?: string }).parentSpanId;\n  if (typeof legacyParent === 'string' && legacyParent !== '') {\n    return legacyParent;\n  }\n  const parentContext = (span as { parentSpanContext?: { spanId?: string } })\n    .parentSpanContext;\n  const spanId = parentContext?.spanId;\n  return typeof spanId === 'string' && spanId !== '' ? spanId : undefined;\n}\n\nfunction isRootSpan(span: ReadableSpan): boolean {\n  return getRootSpanParentId(span) == null;\n}\n\n/**\n * LangGraph plumbing observations that add noise without information:\n * the duplicated `__start__` channel-seed nodes and anonymous\n * `RunnableLambda` pass-throughs (Langfuse team feedback items 4 & 5).\n * Internal ToolNode batch spans are disabled at their source so traced child\n * tools retain an exported parent. Explicitly traced ToolNodes are preserved.\n */\nexport function shouldDropLangfuseSpan(spanName: string): boolean {\n  return (\n    spanName === LANGGRAPH_START_NODE || spanName === ANONYMOUS_LAMBDA_NAME\n  );\n}\n\nfunction shapeToolNodeSpan(span: MutableSpan): void {\n  const inputKey = LangfuseOtelSpanAttributes.OBSERVATION_INPUT;\n  span.name = TOOL_DISPATCH_SPAN_NAME;\n  span.attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE] =\n    CHAIN_OBSERVATION_TYPE;\n  const calls = findPendingToolCalls(\n    parseAttributeValue(span.attributes[inputKey])\n  );\n  if (calls.length === 0) {\n    return;\n  }\n  span.attributes[inputKey] = JSON.stringify(\n    calls.map(({ name, args }) => ({ name, args }))\n  );\n}\n\nfunction shapeAgentNodeSpan(span: MutableSpan): void {\n  span.name = AGENT_NODE_SPAN_NAME;\n  span.attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE] =\n    ROOT_OBSERVATION_TYPE;\n}\n\nfunction shapeRootSpan(span: MutableSpan): void {\n  const inputKey = LangfuseOtelSpanAttributes.OBSERVATION_INPUT;\n  const outputKey = LangfuseOtelSpanAttributes.OBSERVATION_OUTPUT;\n  const question = findLastMessageText(\n    parseAttributeValue(span.attributes[inputKey]),\n    'user'\n  );\n  const answer = findLastMessageText(\n    parseAttributeValue(span.attributes[outputKey]),\n    'assistant'\n  );\n  if (question != null) {\n    span.attributes[inputKey] = question;\n  }\n  if (answer != null) {\n    span.attributes[outputKey] = answer;\n  }\n  const traceInput = question ?? span.attributes[inputKey];\n  const traceOutput = answer ?? span.attributes[outputKey];\n  if (traceInput != null) {\n    span.attributes[LangfuseOtelSpanAttributes.TRACE_INPUT] = traceInput;\n  }\n  if (traceOutput != null) {\n    span.attributes[LangfuseOtelSpanAttributes.TRACE_OUTPUT] = traceOutput;\n  }\n}\n\nfunction isGenerationSpan(span: MutableSpan): boolean {\n  const type = span.attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE];\n  return typeof type === 'string' && type.toLowerCase() === 'generation';\n}\n\nfunction hasTraceTag(span: MutableSpan, expectedTag: string): boolean {\n  const tags = parseAttributeValue(\n    span.attributes[LangfuseOtelSpanAttributes.TRACE_TAGS]\n  );\n  return (\n    Array.isArray(tags) &&\n    tags.some((tag) => typeof tag === 'string' && tag === expectedTag)\n  );\n}\n\nfunction shapeRootObservationType(span: MutableSpan): void {\n  if (isGenerationSpan(span)) {\n    return;\n  }\n  if (hasTraceTag(span, AGENT_TRACE_TAG)) {\n    span.attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE] =\n      ROOT_OBSERVATION_TYPE;\n    return;\n  }\n  if (hasTraceTag(span, TITLE_TRACE_TAG)) {\n    span.attributes[LangfuseOtelSpanAttributes.OBSERVATION_TYPE] =\n      CHAIN_OBSERVATION_TYPE;\n  }\n}\n\n/**\n * Reshapes spans per Langfuse-team feedback before export:\n * - `agent=<id>` / `tools=<id>` node names carry the ephemeral agent id\n *   (`provider__model`) — strip it so switching models doesn't break\n *   name-based logic (item 1).\n * - LLM generation spans keep the provider client class name (`ChatOpenAI`,\n *   `AzureChatOpenAI`, …); rename them to a provider-agnostic `llm` so the\n *   name reflects the operation, not the model (the model stays on the\n *   generation's model attribute).\n * - Agent nodes become `agent` observations, while tool-dispatch nodes become\n *   stable `chain` observations whose input is scoped to the pending calls.\n *   Individual child calls remain `tool` observations (items 3 & 4).\n * - Agent trace roots become `agent` observations and title trace roots become\n *   `chain` observations. Root and trace input/output are reduced to the user\n *   question and assistant response when chat messages are available (item 2).\n */\nexport function shapeLangfuseSpan(span: ReadableSpan): void {\n  const mutable = span as MutableSpan;\n  if (mutable.name.startsWith(LANGGRAPH_AGENT_NODE_PREFIX)) {\n    shapeAgentNodeSpan(mutable);\n  } else if (mutable.name.startsWith(LANGGRAPH_TOOL_NODE_PREFIX)) {\n    shapeToolNodeSpan(mutable);\n  } else if (isGenerationSpan(mutable)) {\n    mutable.name = GENERATION_SPAN_NAME;\n  }\n  if (!isRootSpan(span)) {\n    return;\n  }\n  shapeRootObservationType(mutable);\n  shapeRootSpan(mutable);\n}\n"],"mappings":";;AAGA,MAAM,uBAAuB;AAC7B,MAAM,wBAAwB;AAC9B,MAAM,8BAA8B;AACpC,MAAM,6BAA6B;AACnC,MAAM,uBAAuB;AAC7B,MAAM,0BAA0B;AAChC,MAAM,uBAAuB;AAC7B,MAAM,wBAAwB;AAC9B,MAAM,yBAAyB;AAC/B,MAAM,kBAAkB;AACxB,MAAM,kBAAkB;AAaxB,SAAS,SAAS,OAAkD;CAClE,OAAO,SAAS,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAC3E;AAEA,SAAS,oBAAoB,OAAyB;CACpD,IAAI,OAAO,UAAU,UACnB,OAAO;CAET,MAAM,UAAU,MAAM,KAAK;CAC3B,IAAI,CAAC,QAAQ,WAAW,GAAG,KAAK,CAAC,QAAQ,WAAW,GAAG,GACrD,OAAO;CAET,IAAI;EACF,OAAO,KAAK,MAAM,KAAK;CACzB,QAAQ;EACN,OAAO;CACT;AACF;AAEA,SAAS,gBACP,OACuC;CACvC,IAAI,MAAM,QAAQ,KAAK,GAAG;EACxB,MAAM,UAAU,MAAM,OAAO,QAAQ;EACrC,OAAO,QAAQ,SAAS,IAAI,UAAU,KAAA;CACxC;CACA,IAAI,CAAC,SAAS,KAAK,GACjB;CAEF,OACE,gBAAgB,MAAM,QAAQ,KAC9B,gBAAgB,MAAM,KAAK,KAC3B,gBAAgB,MAAM,MAAM;AAEhC;AAEA,SAAS,eAAe,SAAsD;CAC5E,MAAM,KAAK,QAAQ;CACnB,IAAI,MAAM,QAAQ,EAAE,GAAG;EACrB,MAAM,YAAY,GAAG,GAAG,SAAS;EACjC,IAAI,OAAO,cAAc,UAAU;GACjC,IAAI,UAAU,SAAS,OAAO,GAC5B,OAAO;GAET,IAAI,UAAU,SAAS,IAAI,GACzB,OAAO;GAET,IAAI,UAAU,SAAS,QAAQ,GAC7B,OAAO;GAET,IAAI,UAAU,SAAS,MAAM,GAC3B,OAAO;EAEX;CACF;CACA,MAAM,UACJ,QAAQ,QAAQ,QAAQ,SAAS,QAAQ,QAAQ,QAAQ;CAC3D,IAAI,OAAO,YAAY,UAAU;EAC/B,MAAM,SAAS,QAAQ;EACvB,OAAO,SAAS,MAAM,IAAI,eAAe,MAAM,IAAI,KAAA;CACrD;CACA,MAAM,aAAa,QAAQ,YAAY;CACvC,IAAI,eAAe,SACjB,OAAO;CAET,IAAI,eAAe,MACjB,OAAO;CAET,OAAO;AACT;AAEA,SAAS,eAAe,SAAsD;CAC5E,MAAM,UACJ,QAAQ,YACP,SAAS,QAAQ,MAAM,IAAI,QAAQ,OAAO,UAAU,KAAA,OACpD,SAAS,QAAQ,IAAI,IAAI,QAAQ,KAAK,UAAU,KAAA;CACnD,IAAI,OAAO,YAAY,UACrB,OAAO;CAET,IAAI,CAAC,MAAM,QAAQ,OAAO,GACxB;CAEF,MAAM,OAAO,QACV,OAAO,QAAQ,CAAC,CAChB,KAAK,SAAU,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO,EAAG,CAAC,CAC/D,KAAK,EAAE;CACV,OAAO,SAAS,KAAK,KAAA,IAAY;AACnC;AAEA,SAAS,oBAAoB,OAAgB,MAAkC;CAC7E,MAAM,WAAW,gBAAgB,KAAK;CACtC,IAAI,YAAY,MACd;CAEF,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;EAC7C,IAAI,eAAe,SAAS,EAAE,MAAM,MAClC;EAEF,MAAM,OAAO,eAAe,SAAS,EAAE;EACvC,IAAI,QAAQ,QAAQ,KAAK,KAAK,MAAM,IAClC,OAAO;CAEX;AAEF;AAEA,SAAS,kBAAkB,OAAgD;CACzE,IAAI,CAAC,SAAS,KAAK,GACjB;CAEF,MAAM,KAAK,MAAM;CACjB,IAAI,SAAS,EAAE,KAAK,OAAO,GAAG,SAAS,UACrC,OAAO;EACL,MAAM,GAAG;EACT,MAAM,oBAAoB,GAAG,SAAS;EACtC,GAAI,OAAO,MAAM,OAAO,WAAW,EAAE,IAAI,MAAM,GAAG,IAAI,CAAC;CACzD;CAEF,IAAI,OAAO,MAAM,SAAS,UACxB;CAEF,OAAO;EACL,MAAM,MAAM;EACZ,MAAM,MAAM,QAAQ,CAAC;EACrB,GAAI,OAAO,MAAM,OAAO,WAAW,EAAE,IAAI,MAAM,GAAG,IAAI,CAAC;CACzD;AACF;AAEA,SAAS,oBACP,SACsB;CACtB,MAAM,WACJ,QAAQ,eACP,SAAS,QAAQ,MAAM,IAAI,QAAQ,OAAO,aAAa,KAAA,OACvD,SAAS,QAAQ,iBAAiB,IAC/B,QAAQ,kBAAkB,aAC1B,KAAA,OACH,SAAS,QAAQ,IAAI,IAAI,QAAQ,KAAK,aAAa,KAAA;CACtD,IAAI,CAAC,MAAM,QAAQ,QAAQ,GACzB,OAAO,CAAC;CAEV,MAAM,QAA8B,CAAC;CACrC,KAAK,MAAM,WAAW,UAAU;EAC9B,MAAM,OAAO,kBAAkB,OAAO;EACtC,IAAI,QAAQ,MACV,MAAM,KAAK,IAAI;CAEnB;CACA,OAAO;AACT;;AAGA,SAAS,qBAAqB,OAAsC;CAClE,MAAM,WAAW,gBAAgB,KAAK;CACtC,IAAI,YAAY,MACd,OAAO,CAAC;CAEV,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;EAC7C,IAAI,eAAe,SAAS,EAAE,MAAM,aAClC;EAEF,MAAM,QAAQ,oBAAoB,SAAS,EAAE;EAC7C,IAAI,MAAM,SAAS,GACjB,OAAO;CAEX;CACA,OAAO,CAAC;AACV;AAEA,SAAS,oBAAoB,MAAwC;CACnE,MAAM,eAAgB,KAAmC;CACzD,IAAI,OAAO,iBAAiB,YAAY,iBAAiB,IACvD,OAAO;CAIT,MAAM,SAFiB,KACpB,mBAC2B;CAC9B,OAAO,OAAO,WAAW,YAAY,WAAW,KAAK,SAAS,KAAA;AAChE;AAEA,SAAS,WAAW,MAA6B;CAC/C,OAAO,oBAAoB,IAAI,KAAK;AACtC;;;;;;;;AASA,SAAgB,uBAAuB,UAA2B;CAChE,OACE,aAAa,wBAAwB,aAAa;AAEtD;AAEA,SAAS,kBAAkB,MAAyB;CAClD,MAAM,WAAWA,kBAAAA,2BAA2B;CAC5C,KAAK,OAAO;CACZ,KAAK,WAAWA,kBAAAA,2BAA2B,oBACzC;CACF,MAAM,QAAQ,qBACZ,oBAAoB,KAAK,WAAW,SAAS,CAC/C;CACA,IAAI,MAAM,WAAW,GACnB;CAEF,KAAK,WAAW,YAAY,KAAK,UAC/B,MAAM,KAAK,EAAE,MAAM,YAAY;EAAE;EAAM;CAAK,EAAE,CAChD;AACF;AAEA,SAAS,mBAAmB,MAAyB;CACnD,KAAK,OAAO;CACZ,KAAK,WAAWA,kBAAAA,2BAA2B,oBACzC;AACJ;AAEA,SAAS,cAAc,MAAyB;CAC9C,MAAM,WAAWA,kBAAAA,2BAA2B;CAC5C,MAAM,YAAYA,kBAAAA,2BAA2B;CAC7C,MAAM,WAAW,oBACf,oBAAoB,KAAK,WAAW,SAAS,GAC7C,MACF;CACA,MAAM,SAAS,oBACb,oBAAoB,KAAK,WAAW,UAAU,GAC9C,WACF;CACA,IAAI,YAAY,MACd,KAAK,WAAW,YAAY;CAE9B,IAAI,UAAU,MACZ,KAAK,WAAW,aAAa;CAE/B,MAAM,aAAa,YAAY,KAAK,WAAW;CAC/C,MAAM,cAAc,UAAU,KAAK,WAAW;CAC9C,IAAI,cAAc,MAChB,KAAK,WAAWA,kBAAAA,2BAA2B,eAAe;CAE5D,IAAI,eAAe,MACjB,KAAK,WAAWA,kBAAAA,2BAA2B,gBAAgB;AAE/D;AAEA,SAAS,iBAAiB,MAA4B;CACpD,MAAM,OAAO,KAAK,WAAWA,kBAAAA,2BAA2B;CACxD,OAAO,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM;AAC5D;AAEA,SAAS,YAAY,MAAmB,aAA8B;CACpE,MAAM,OAAO,oBACX,KAAK,WAAWA,kBAAAA,2BAA2B,WAC7C;CACA,OACE,MAAM,QAAQ,IAAI,KAClB,KAAK,MAAM,QAAQ,OAAO,QAAQ,YAAY,QAAQ,WAAW;AAErE;AAEA,SAAS,yBAAyB,MAAyB;CACzD,IAAI,iBAAiB,IAAI,GACvB;CAEF,IAAI,YAAY,MAAM,eAAe,GAAG;EACtC,KAAK,WAAWA,kBAAAA,2BAA2B,oBACzC;EACF;CACF;CACA,IAAI,YAAY,MAAM,eAAe,GACnC,KAAK,WAAWA,kBAAAA,2BAA2B,oBACzC;AAEN;;;;;;;;;;;;;;;;;AAkBA,SAAgB,kBAAkB,MAA0B;CAC1D,MAAM,UAAU;CAChB,IAAI,QAAQ,KAAK,WAAW,2BAA2B,GACrD,mBAAmB,OAAO;MACrB,IAAI,QAAQ,KAAK,WAAW,0BAA0B,GAC3D,kBAAkB,OAAO;MACpB,IAAI,iBAAiB,OAAO,GACjC,QAAQ,OAAO;CAEjB,IAAI,CAAC,WAAW,IAAI,GAClB;CAEF,yBAAyB,OAAO;CAChC,cAAc,OAAO;AACvB"}

package/dist/cjs/llm/anthropic/index.cjs

@@ -1,5 +1,6 @@
 const require_message_inputs = require("./utils/message_inputs.cjs");
 const require_message_outputs = require("./utils/message_outputs.cjs");
+const require_stream_events = require("./utils/stream_events.cjs");
 const require_tools = require("./utils/tools.cjs");
 let _langchain_core_messages = require("@langchain/core/messages");
 let _langchain_core_outputs = require("@langchain/core/outputs");
@@ -140,6 +141,15 @@ async function waitForStreamDelay(delay, signal) {
 function isSignalAborted(signal) {
 	return signal?.aborted === true;
 }
+async function* abortableAnthropicStream(source, signal) {
+	for await (const data of source) {
+		if (isSignalAborted(signal)) {
+			source.controller?.abort();
+			return;
+		}
+		yield data;
+	}
+}
 function extractToken(chunk) {
 	if (typeof chunk.content === "string") return [chunk.content, "string"];
 	else if (Array.isArray(chunk.content) && chunk.content.length >= 1 && "input" in chunk.content[0]) return typeof chunk.content[0].input === "string" ? [chunk.content[0].input, "input"] : [JSON.stringify(chunk.content[0].input), "input"];
@@ -161,20 +171,47 @@ function cloneChunk(text, tokenType, chunk) {
 	else if (typeof content.type === "string" && content.type.startsWith("thinking")) return new _langchain_core_messages.AIMessageChunk(Object.assign({}, chunk, { content: [Object.assign({}, content, { thinking: text })] }));
 	return chunk;
 }
-function withIncrementalMessageDeltaUsage(chunk, previousOutputTokens) {
+function resolveCumulativeUsageValue(current, previous) {
+	if (current == null) return {
+		cumulative: previous,
+		increment: 0
+	};
+	const cumulative = Math.max(previous, current);
+	return {
+		cumulative,
+		increment: cumulative - previous
+	};
+}
+function withIncrementalMessageDeltaUsage(chunk, previousUsage, cumulativeUsage) {
 	const usage = chunk.usage_metadata;
 	if (usage == null) return {
 		chunk,
-		outputTokens: previousOutputTokens
+		usage: previousUsage
+	};
+	const inputTokens = resolveCumulativeUsageValue(cumulativeUsage.input_tokens, previousUsage.inputTokens);
+	const cacheCreationInputTokens = resolveCumulativeUsageValue(cumulativeUsage.cache_creation_input_tokens, previousUsage.cacheCreationInputTokens);
+	const cacheReadInputTokens = resolveCumulativeUsageValue(cumulativeUsage.cache_read_input_tokens, previousUsage.cacheReadInputTokens);
+	const outputTokens = resolveCumulativeUsageValue(cumulativeUsage.output_tokens, previousUsage.outputTokens);
+	const incrementalInputTokens = inputTokens.increment + cacheCreationInputTokens.increment + cacheReadInputTokens.increment;
+	const hasCacheUsage = cumulativeUsage.cache_creation_input_tokens != null || cumulativeUsage.cache_read_input_tokens != null;
+	const inputTokenDetails = {
+		...cumulativeUsage.cache_creation_input_tokens != null && { cache_creation: cacheCreationInputTokens.increment },
+		...cumulativeUsage.cache_read_input_tokens != null && { cache_read: cacheReadInputTokens.increment }
 	};
-	const outputTokens = Math.max(0, usage.output_tokens - previousOutputTokens);
 	return {
 		chunk: new _langchain_core_messages.AIMessageChunk(Object.assign({}, chunk, { usage_metadata: {
 			...usage,
-			output_tokens: outputTokens,
-			total_tokens: usage.input_tokens + outputTokens
+			input_tokens: incrementalInputTokens,
+			output_tokens: outputTokens.increment,
+			total_tokens: incrementalInputTokens + outputTokens.increment,
+			input_token_details: hasCacheUsage ? inputTokenDetails : void 0
 		} })),
-		outputTokens: usage.output_tokens
+		usage: {
+			inputTokens: inputTokens.cumulative,
+			cacheCreationInputTokens: cacheCreationInputTokens.cumulative,
+			cacheReadInputTokens: cacheReadInputTokens.cumulative,
+			outputTokens: outputTokens.cumulative
+		}
 	};
 }
 var CustomAnthropic = class extends _langchain_anthropic.ChatAnthropicMessages {
@@ -281,6 +318,21 @@ var CustomAnthropic = class extends _langchain_anthropic.ChatAnthropicMessages {
 	async completionWithRetry(request, options) {
 		return super.completionWithRetry(require_message_inputs.stripUnsupportedAssistantPrefill(request), options);
 	}
+	async *_streamChatModelEvents(messages, options, _runManager) {
+		const params = this.invocationParams(options);
+		const formattedMessages = require_message_inputs._convertMessagesToAnthropicPayload(messages);
+		const payload = require_message_inputs.stripUnsupportedAssistantPrefill({
+			...params,
+			...formattedMessages,
+			stream: true
+		});
+		const stream = await this.createStreamWithRetry(payload, {
+			headers: options.headers,
+			signal: options.signal
+		});
+		const shouldStreamUsage = options.streamUsage ?? this.streamUsage;
+		yield* require_stream_events.convertAnthropicStream(abortableAnthropicStream(stream, options.signal), { streamUsage: shouldStreamUsage });
+	}
 	async *_streamResponseChunks(messages, options, runManager) {
 		this.resetTokenEvents();
 		const params = this.invocationParams(options);
@@ -296,7 +348,12 @@ var CustomAnthropic = class extends _langchain_anthropic.ChatAnthropicMessages {
 			signal: options.signal
 		});
 		const shouldStreamUsage = options.streamUsage ?? this.streamUsage;
-		let messageDeltaOutputTokens = 0;
+		let streamUsage = {
+			inputTokens: 0,
+			cacheCreationInputTokens: 0,
+			cacheReadInputTokens: 0,
+			outputTokens: 0
+		};
 		const queuedChunks = [];
 		const producerState = { done: false };
 		let queuedChunkIndex = 0;
@@ -407,10 +464,17 @@ var CustomAnthropic = class extends _langchain_anthropic.ChatAnthropicMessages {
 					});
 					if (!result) continue;
 					let { chunk } = result;
+					if (data.type === "message_start") streamUsage = {
+						...streamUsage,
+						inputTokens: data.message.usage.input_tokens,
+						outputTokens: data.message.usage.output_tokens,
+						cacheCreationInputTokens: data.message.usage.cache_creation_input_tokens ?? 0,
+						cacheReadInputTokens: data.message.usage.cache_read_input_tokens ?? 0
+					};
 					if (data.type === "message_delta") {
-						const incremental = withIncrementalMessageDeltaUsage(chunk, messageDeltaOutputTokens);
+						const incremental = withIncrementalMessageDeltaUsage(chunk, streamUsage, data.usage);
 						chunk = incremental.chunk;
-						messageDeltaOutputTokens = incremental.outputTokens;
+						streamUsage = incremental.usage;
 					}
 					const [token = "", tokenType] = extractToken(chunk);
 					if (!tokenType || tokenType === "input" || token === "" && (chunk.usage_metadata != null || chunk.id != null)) {