@librechat/agents 3.2.56 → 3.2.58

package/dist/esm/langfuseToolOutputTracing.mjs

@@ -1,5 +1,6 @@
 import { hasToolOutputTracingConfig, normalizeToolName, resolveLangfuseConfig, resolveToolOutputTracingConfig } from "./langfuseConfig.mjs";
 import { resolveToolOutputTracingConfigForSpan } from "./langfuseRuntimeScope.mjs";
+import { shapeLangfuseSpan, shouldDropLangfuseSpan } from "./langfuseTraceShaping.mjs";
 import { LangfuseOtelSpanAttributes } from "@langfuse/tracing";
 import { LangfuseSpanProcessor } from "@langfuse/otel";
 //#region src/langfuseToolOutputTracing.ts
@@ -205,12 +206,15 @@ var ToolOutputRedactingLangfuseSpanProcessor = class {
 		this.fallbackConfig = fallbackConfig;
 	}
 	onStart(span, parentContext) {
+		if (shouldDropLangfuseSpan(span.name)) return;
 		const config = resolveToolOutputTracingConfigForSpan(parentContext) ?? this.fallbackConfig;
 		if (config != null) this.spanConfigs.set(span, config);
 		this.processor.onStart(span, parentContext);
 	}
 	onEnd(span) {
+		if (shouldDropLangfuseSpan(span.name)) return;
 		classifyLangfuseToolNodeSpan(span);
+		shapeLangfuseSpan(span);
 		const config = this.spanConfigs.get(span) ?? this.fallbackConfig;
 		if (config != null) redactLangfuseSpanToolOutputs(span, config);
 		this.processor.onEnd(span);

package/dist/esm/langfuseToolOutputTracing.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"langfuseToolOutputTracing.mjs","names":[],"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 { 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    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    classifyLangfuseToolNodeSpan(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":";;;;;AAsBA,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,qBAAqB,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,WAAW,2BAA2B;CACnD,OAAO,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM;AAC5D;AAEA,SAAS,8BACP,YACM;CACN,MAAM,OAAO,WAAW,2BAA2B;CACnD,IAAI,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM,QACrD;CAGF,MAAM,gBACJ,WACE,GAAG,2BAA2B,qBAAqB;CAEvD,IACE,OAAO,kBAAkB,YACzB,cAAc,WAAW,0BAA0B,GAEnD,WAAW,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,KAClC,2BAA2B,sBAAsB,aAGnD;CAGF,WAAW,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;EAChB,2BAA2B;EAC3B,2BAA2B;EAC3B,2BAA2B;EAC3B,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,IAAI,sBAAsB,MAAM;EACjD,KAAK,iBAAiB;CACxB;CAEA,QAAQ,MAAY,eAA8B;EAChD,MAAM,SACJ,sCAAsC,aAAa,KACnD,KAAK;EACP,IAAI,UAAU,MACZ,KAAK,YAAY,IAAI,MAAM,MAAM;EAEnC,KAAK,UAAU,QAAQ,MAAM,aAAa;CAC5C;CAEA,MAAM,MAA0B;EAC9B,6BAA6B,IAAI;EACjC,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,QAH7B,2BAA2B,aAAa,aAAa,IACxE,+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,WAAW,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.mjs","names":[],"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,qBAAqB,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,WAAW,2BAA2B;CACnD,OAAO,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM;AAC5D;AAEA,SAAS,8BACP,YACM;CACN,MAAM,OAAO,WAAW,2BAA2B;CACnD,IAAI,OAAO,SAAS,YAAY,KAAK,YAAY,MAAM,QACrD;CAGF,MAAM,gBACJ,WACE,GAAG,2BAA2B,qBAAqB;CAEvD,IACE,OAAO,kBAAkB,YACzB,cAAc,WAAW,0BAA0B,GAEnD,WAAW,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,KAClC,2BAA2B,sBAAsB,aAGnD;CAGF,WAAW,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;EAChB,2BAA2B;EAC3B,2BAA2B;EAC3B,2BAA2B;EAC3B,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,IAAI,sBAAsB,MAAM;EACjD,KAAK,iBAAiB;CACxB;CAEA,QAAQ,MAAY,eAA8B;EAChD,IAAI,uBAAuB,KAAK,IAAI,GAClC;EAEF,MAAM,SACJ,sCAAsC,aAAa,KACnD,KAAK;EACP,IAAI,UAAU,MACZ,KAAK,YAAY,IAAI,MAAM,MAAM;EAEnC,KAAK,UAAU,QAAQ,MAAM,aAAa;CAC5C;CAEA,MAAM,MAA0B;EAC9B,IAAI,uBAAuB,KAAK,IAAI,GAClC;EAEF,6BAA6B,IAAI;EACjC,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,QAH7B,2BAA2B,aAAa,aAAa,IACxE,+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,WAAW,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"}

package/dist/esm/langfuseTraceShaping.mjs

@@ -0,0 +1,171 @@
+import { LangfuseOtelSpanAttributes } from "@langfuse/tracing";
+//#region src/langfuseTraceShaping.ts
+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";
+function isRecord(value) {
+	return value != null && typeof value === "object" && !Array.isArray(value);
+}
+function parseAttributeValue(value) {
+	if (typeof value !== "string") return value;
+	const trimmed = value.trim();
+	if (!trimmed.startsWith("{") && !trimmed.startsWith("[")) return value;
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
+}
+function getMessageArray(value) {
+	if (Array.isArray(value)) {
+		const records = value.filter(isRecord);
+		return records.length > 0 ? records : void 0;
+	}
+	if (!isRecord(value)) return;
+	return getMessageArray(value.messages) ?? getMessageArray(value.input) ?? getMessageArray(value.output);
+}
+function getMessageRole(message) {
+	const id = message.id;
+	if (Array.isArray(id)) {
+		const className = id[id.length - 1];
+		if (typeof className === "string") {
+			if (className.includes("Human")) return "user";
+			if (className.includes("AI")) return "assistant";
+			if (className.includes("System")) return "system";
+			if (className.includes("Tool")) return "tool";
+		}
+	}
+	const rawRole = message.type ?? message._type ?? message.role ?? message.sender;
+	if (typeof rawRole !== "string") {
+		const kwargs = message.kwargs;
+		return isRecord(kwargs) ? getMessageRole(kwargs) : void 0;
+	}
+	const normalized = rawRole.toLowerCase();
+	if (normalized === "human") return "user";
+	if (normalized === "ai") return "assistant";
+	return normalized;
+}
+function getMessageText(message) {
+	const content = message.content ?? (isRecord(message.kwargs) ? message.kwargs.content : void 0) ?? (isRecord(message.data) ? message.data.content : void 0);
+	if (typeof content === "string") return content;
+	if (!Array.isArray(content)) return;
+	const text = content.filter(isRecord).map((part) => typeof part.text === "string" ? part.text : "").join("");
+	return text === "" ? void 0 : text;
+}
+function findLastMessageText(value, role) {
+	const messages = getMessageArray(value);
+	if (messages == null) return;
+	for (let i = messages.length - 1; i >= 0; i--) {
+		if (getMessageRole(messages[i]) !== role) continue;
+		const text = getMessageText(messages[i]);
+		if (text != null && text.trim() !== "") return text;
+	}
+}
+function normalizeToolCall(value) {
+	if (!isRecord(value)) return;
+	const fn = value.function;
+	if (isRecord(fn) && typeof fn.name === "string") return {
+		name: fn.name,
+		args: parseAttributeValue(fn.arguments),
+		...typeof value.id === "string" ? { id: value.id } : {}
+	};
+	if (typeof value.name !== "string") return;
+	return {
+		name: value.name,
+		args: value.args ?? {},
+		...typeof value.id === "string" ? { id: value.id } : {}
+	};
+}
+function getMessageToolCalls(message) {
+	const rawCalls = message.tool_calls ?? (isRecord(message.kwargs) ? message.kwargs.tool_calls : void 0) ?? (isRecord(message.additional_kwargs) ? message.additional_kwargs.tool_calls : void 0) ?? (isRecord(message.data) ? message.data.tool_calls : void 0);
+	if (!Array.isArray(rawCalls)) return [];
+	const calls = [];
+	for (const rawCall of rawCalls) {
+		const call = normalizeToolCall(rawCall);
+		if (call != null) calls.push(call);
+	}
+	return calls;
+}
+/** Latest assistant turn's tool calls — the calls this tool node is executing. */
+function findPendingToolCalls(value) {
+	const messages = getMessageArray(value);
+	if (messages == null) return [];
+	for (let i = messages.length - 1; i >= 0; i--) {
+		if (getMessageRole(messages[i]) !== "assistant") continue;
+		const calls = getMessageToolCalls(messages[i]);
+		if (calls.length > 0) return calls;
+	}
+	return [];
+}
+function getRootSpanParentId(span) {
+	const legacyParent = span.parentSpanId;
+	if (typeof legacyParent === "string" && legacyParent !== "") return legacyParent;
+	const spanId = span.parentSpanContext?.spanId;
+	return typeof spanId === "string" && spanId !== "" ? spanId : void 0;
+}
+function isRootSpan(span) {
+	return getRootSpanParentId(span) == null;
+}
+/**
+* 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).
+*/
+function shouldDropLangfuseSpan(spanName) {
+	return spanName === LANGGRAPH_START_NODE || spanName === ANONYMOUS_LAMBDA_NAME;
+}
+function shapeToolNodeSpan(span) {
+	const inputKey = LangfuseOtelSpanAttributes.OBSERVATION_INPUT;
+	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 shapeRootSpan(span) {
+	const inputKey = LangfuseOtelSpanAttributes.OBSERVATION_INPUT;
+	const outputKey = 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[LangfuseOtelSpanAttributes.TRACE_INPUT] = question;
+	}
+	if (answer != null) {
+		span.attributes[outputKey] = answer;
+		span.attributes[LangfuseOtelSpanAttributes.TRACE_OUTPUT] = answer;
+	}
+}
+/**
+* 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).
+*/
+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);
+}
+//#endregion
+export { shapeLangfuseSpan, shouldDropLangfuseSpan };
+
+//# sourceMappingURL=langfuseTraceShaping.mjs.map

package/dist/esm/langfuseTraceShaping.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"langfuseTraceShaping.mjs","names":[],"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,WAAW,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,WAAW,2BAA2B;CAC5C,MAAM,YAAY,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,WAAW,2BAA2B,eAAe;CAC5D;CACA,IAAI,UAAU,MAAM;EAClB,KAAK,WAAW,aAAa;EAC7B,KAAK,WAAW,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"}

package/dist/esm/run.mjs

@@ -73,6 +73,7 @@ var Run = class Run {
 	langfuse;
 	toolOutputReferences;
 	eagerEventToolExecution;
+	codeSessionToolNames;
 	toolExecution;
 	subagentUsageSink;
 	indexTokenCountMap;
@@ -113,6 +114,7 @@ var Run = class Run {
 		this.langfuse = config.langfuse;
 		this.toolOutputReferences = config.toolOutputReferences;
 		this.eagerEventToolExecution = config.eagerEventToolExecution;
+		this.codeSessionToolNames = config.codeSessionToolNames;
 		this.toolExecution = config.toolExecution;
 		this.subagentUsageSink = config.subagentUsageSink;
 		if (!config.graphConfig) throw new Error("Graph config not provided");
@@ -170,6 +172,7 @@ var Run = class Run {
 		standardGraph.humanInTheLoop = this.humanInTheLoop;
 		standardGraph.toolOutputReferences = this.toolOutputReferences;
 		standardGraph.eagerEventToolExecution = this.eagerEventToolExecution;
+		standardGraph.codeSessionToolNames = this.codeSessionToolNames;
 		standardGraph.toolExecution = this.toolExecution;
 		this.Graph = standardGraph;
 		return standardGraph.createWorkflow();
@@ -192,6 +195,7 @@ var Run = class Run {
 		multiAgentGraph.humanInTheLoop = this.humanInTheLoop;
 		multiAgentGraph.toolOutputReferences = this.toolOutputReferences;
 		multiAgentGraph.eagerEventToolExecution = this.eagerEventToolExecution;
+		multiAgentGraph.codeSessionToolNames = this.codeSessionToolNames;
 		multiAgentGraph.toolExecution = this.toolExecution;
 		this.Graph = multiAgentGraph;
 		return multiAgentGraph.createWorkflow();
@@ -913,10 +917,10 @@ var Run = class Run {
 			convo: promptValue.value,
 			inputText,
 			skipLanguage
-		}) }).withConfig({ runName: "ConvoTransform" });
+		}) }).withConfig({ runName: "PrepareTitleInput" });
 		const titleChain = titleMethod === "completion" ? await createCompletionTitleRunnable(model, titlePrompt) : await createTitleRunnable(model, titlePrompt);
 		/** Pipes `convoTemplate` -> `transformer` -> `titleChain` */
-		const fullChain = convoTemplate.withConfig({ runName: "ConvoTemplate" }).pipe(convoToTitleInput).pipe(titleChain).withConfig({ runName: "TitleChain" });
+		const fullChain = convoTemplate.withConfig({ runName: "FormatConversation" }).pipe(convoToTitleInput).pipe(titleChain).withConfig({ runName: "GenerateConversationTitle" });
 		const invokeConfig = Object.assign({}, chainOptions, {
 			run_id: this.id,
 			runId: this.id,