langwatch 0.3.2 → 0.4.0

package/dist/observability-sdk/instrumentation/langchain/index.d.ts

@@ -0,0 +1,69 @@
+import * as _opentelemetry_api from '@opentelemetry/api';
+import { Attributes } from '@opentelemetry/api';
+import { L as LangWatchSpan, a as LangWatchTracer, C as ChatMessage } from '../../../types-C9k5gGhU.js';
+import { AgentAction, AgentFinish } from '@langchain/core/agents';
+import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
+import { DocumentInterface } from '@langchain/core/documents';
+import { Serialized } from '@langchain/core/load/serializable';
+import { BaseMessage } from '@langchain/core/messages';
+import { LLMResult } from '@langchain/core/outputs';
+import { ChainValues } from '@langchain/core/utils/types';
+import '../../../types-DRiQaKFG.js';
+import '@opentelemetry/api-logs';
+import '@opentelemetry/semantic-conventions';
+import '@opentelemetry/semantic-conventions/incubating';
+
+type RunKind = "llm" | "chat" | "chain" | "tool" | "retriever";
+declare class LangWatchCallbackHandler extends BaseCallbackHandler {
+    name: string;
+    tracer: LangWatchTracer;
+    private spans;
+    private parentOf;
+    private skipped;
+    private seenStarts;
+    private startRunSpan;
+    private finishRun;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, name?: string): Promise<void>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, name?: string): Promise<void>;
+    handleLLMEnd(response: LLMResult, runId: string, parentRunId?: string): Promise<void>;
+    handleLLMError(err: Error, runId: string, parentRunId?: string): Promise<void>;
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, _runType?: string, name?: string): Promise<void>;
+    handleChainEnd(output: ChainValues, runId: string, parentRunId?: string): Promise<void>;
+    handleChainError(err: Error, runId: string, parentRunId?: string, tags?: string[], kwargs?: {
+        inputs?: Record<string, unknown> | undefined;
+    }): Promise<void>;
+    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, name?: string): Promise<void>;
+    handleToolEnd(output: string, runId: string, parentRunId?: string): Promise<void>;
+    handleToolError(err: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, name?: string): Promise<void>;
+    handleRetrieverEnd(documents: DocumentInterface<Record<string, any>>[], runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleRetrieverError(err: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleAgentAction(_action: AgentAction, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleAgentEnd(action: AgentFinish, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+}
+declare function convertFromLangChainMessages(messages: BaseMessage[]): ChatMessage[];
+declare function className(serialized?: Serialized): string;
+declare function shorten(str: string, max?: number): string;
+declare function previewInput(v: unknown): string | undefined;
+declare function ctxSkip(serialized?: Serialized, tags?: string[]): boolean;
+declare function wrapNonScalarValues(value: unknown): string | number | boolean | undefined;
+declare function addLangChainEvent(span: LangWatchSpan, eventName: string, runId: string, parentRunId: string | undefined, tags?: string[], metadata?: Record<string, unknown>, attributes?: Attributes): void;
+declare function setLangGraphAttributes(span: LangWatchSpan, metadata?: Record<string, unknown>): void;
+declare function buildLangChainMetadataAttributes(metadata: Record<string, unknown>): {
+    [k: string]: string | number | boolean | undefined;
+};
+declare function applyGenAIAttrs(span: LangWatchSpan, metadata?: Record<string, unknown>, extraParams?: Record<string, unknown>): void;
+declare function getResolvedParentContext(runId: string | undefined, spans: Record<string, LangWatchSpan | undefined>, parentOf: Record<string, string | undefined>): _opentelemetry_api.Context;
+declare function deriveNameAndType(opts: {
+    runType: RunKind;
+    name?: string;
+    serialized?: Serialized;
+    metadata?: Record<string, unknown>;
+    tags?: string[];
+    inputs?: unknown;
+}): {
+    name: string;
+    type: "llm" | "chain" | "tool" | "rag" | "component";
+};
+
+export { LangWatchCallbackHandler, addLangChainEvent, applyGenAIAttrs, buildLangChainMetadataAttributes, className, convertFromLangChainMessages, ctxSkip, deriveNameAndType, getResolvedParentContext, previewInput, setLangGraphAttributes, shorten, wrapNonScalarValues };

package/dist/observability-sdk/instrumentation/langchain/index.js

@@ -0,0 +1,518 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+var _chunkM4VUHTT2js = require('../../../chunk-M4VUHTT2.js');
+
+
+
+var _chunkDISMHYXCjs = require('../../../chunk-DISMHYXC.js');
+
+
+var _chunkOXBO24RBjs = require('../../../chunk-OXBO24RB.js');
+
+// src/observability-sdk/instrumentation/langchain/index.ts
+var _base = require('@langchain/core/callbacks/base');
+
+
+
+
+var _api = require('@opentelemetry/api');
+var LANGGRAPH_METADATA_KEYS = /* @__PURE__ */ new Set([
+  "thread_id",
+  "langgraph_step",
+  "langgraph_node",
+  "langgraph_triggers",
+  "langgraph_path",
+  "langgraph_checkpoint_ns",
+  "__pregel_task_id",
+  "checkpoint_ns"
+]);
+var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
+  constructor() {
+    super(...arguments);
+    this.name = "LangWatchCallbackHandler";
+    this.tracer = _chunkM4VUHTT2js.getLangWatchTracer.call(void 0, "langwatch.instrumentation.langchain");
+    this.spans = {};
+    this.parentOf = {};
+    this.skipped = {};
+    this.seenStarts = /* @__PURE__ */ new Set();
+  }
+  startRunSpan(args) {
+    var _a;
+    const { runId, parentRunId, serialized, tags } = args;
+    this.parentOf[runId] = parentRunId;
+    if (ctxSkip(serialized, tags)) {
+      this.skipped[runId] = true;
+      return;
+    }
+    if (this.seenStarts.has(runId)) return;
+    this.seenStarts.add(runId);
+    const parentCtx = getResolvedParentContext(
+      parentRunId,
+      this.spans,
+      this.parentOf
+    );
+    const parentSpan = parentRunId ? this.spans[parentRunId] : void 0;
+    const links = parentSpan ? [{ context: parentSpan.spanContext() }] : void 0;
+    const { name, type } = deriveNameAndType({
+      runType: args.kind,
+      name: args.name,
+      serialized: args.serialized,
+      metadata: args.metadata,
+      tags: args.tags,
+      inputs: args.input
+    });
+    const span = this.tracer.startSpan(name, { links }, parentCtx);
+    span.setType(type);
+    if ((_a = args.tags) == null ? void 0 : _a.length)
+      span.setAttribute("langwatch.langchain.run.tags", args.tags.slice(0, 50));
+    if (_chunkDISMHYXCjs.shouldCaptureInput.call(void 0, ) && args.input !== void 0) {
+      const i = args.input;
+      if (i && typeof i === "object" && "type" in i && "value" in i) {
+        span.setInput(i.type, i.value);
+      } else {
+        span.setInput(i);
+      }
+    }
+    if (args.extraParams) {
+      span.setAttributes(
+        Object.fromEntries(
+          Object.entries(args.extraParams).map(([k, v]) => [
+            `langwatch.langchain.run.extra_params.${k}`,
+            wrapNonScalarValues(v)
+          ])
+        )
+      );
+    }
+    if (args.metadata) {
+      applyGenAIAttrs(span, args.metadata, args.extraParams);
+      setLangGraphAttributes(span, args.metadata);
+      span.setAttributes(buildLangChainMetadataAttributes(args.metadata));
+    }
+    this.spans[runId] = span;
+  }
+  finishRun(runId, end) {
+    const span = this.spans[runId];
+    if (!span) return;
+    addLangChainEvent(
+      span,
+      end.event,
+      runId,
+      end.parentRunId,
+      end.tags,
+      end.md,
+      end.extra
+    );
+    if (end.err) {
+      span.recordException(end.err);
+      span.setStatus({ code: _api.SpanStatusCode.ERROR, message: end.err.message });
+    } else if (_chunkDISMHYXCjs.shouldCaptureOutput.call(void 0, ) && end.output !== void 0) {
+      span.setOutput(end.output);
+    }
+    span.end();
+    delete this.spans[runId];
+    delete this.parentOf[runId];
+    delete this.skipped[runId];
+    this.seenStarts.delete(runId);
+  }
+  async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
+    const input = _chunkDISMHYXCjs.shouldCaptureInput.call(void 0, ) && prompts ? {
+      type: "list",
+      value: prompts.map((p) => ({ type: "text", value: p }))
+    } : void 0;
+    this.startRunSpan({
+      kind: "llm",
+      runId,
+      parentRunId,
+      serialized: llm,
+      metadata,
+      tags,
+      name,
+      extraParams,
+      input
+    });
+  }
+  async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
+    const input = _chunkDISMHYXCjs.shouldCaptureInput.call(void 0, ) ? {
+      type: "chat_messages",
+      value: messages.flatMap(convertFromLangChainMessages)
+    } : void 0;
+    this.startRunSpan({
+      kind: "chat",
+      runId,
+      parentRunId,
+      serialized: llm,
+      metadata,
+      tags,
+      name,
+      extraParams,
+      input
+    });
+  }
+  async handleLLMEnd(response, runId, parentRunId) {
+    var _a, _b, _c, _d;
+    const span = this.spans[runId];
+    const tu = (_a = response.llmOutput) == null ? void 0 : _a.tokenUsage;
+    if (span && tu) {
+      span.setAttributes({
+        "gen_ai.usage.prompt_tokens": (_b = tu.promptTokens) != null ? _b : 0,
+        "gen_ai.usage.completion_tokens": (_c = tu.completionTokens) != null ? _c : 0,
+        "gen_ai.usage.total_tokens": (_d = tu.totalTokens) != null ? _d : 0
+      });
+    }
+    const outputs = _chunkDISMHYXCjs.shouldCaptureOutput.call(void 0, ) ? response.generations.flat().map((g) => {
+      if ("message" in g && g.message) {
+        return convertFromLangChainMessages([
+          g.message
+        ]);
+      } else if ("text" in g && g.text) {
+        return g.text;
+      }
+      return g;
+    }) : void 0;
+    this.finishRun(runId, {
+      output: outputs,
+      event: "handleLLMEnd",
+      parentRunId
+    });
+  }
+  async handleLLMError(err, runId, parentRunId) {
+    this.finishRun(runId, { err, event: "handleLLMError", parentRunId });
+  }
+  async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, _runType, name) {
+    this.startRunSpan({
+      kind: "chain",
+      runId,
+      parentRunId,
+      serialized: chain,
+      metadata,
+      tags,
+      name,
+      input: _chunkDISMHYXCjs.shouldCaptureInput.call(void 0, ) ? inputs : void 0
+    });
+    if (_runType) {
+      const span = this.spans[runId];
+      if (span) span.setAttribute("langwatch.langchain.run.type", _runType);
+    }
+  }
+  async handleChainEnd(output, runId, parentRunId) {
+    this.finishRun(runId, { output, event: "handleChainEnd", parentRunId });
+  }
+  async handleChainError(err, runId, parentRunId, tags, kwargs) {
+    this.finishRun(runId, {
+      err,
+      event: "handleChainError",
+      parentRunId,
+      tags,
+      md: kwargs
+    });
+  }
+  async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
+    this.startRunSpan({
+      kind: "tool",
+      runId,
+      parentRunId,
+      serialized: tool,
+      metadata,
+      tags,
+      name,
+      input: _chunkDISMHYXCjs.shouldCaptureInput.call(void 0, ) ? { type: "text", value: input } : void 0
+    });
+    const span = this.spans[runId];
+    if (span) {
+      span.setAttributes({
+        "langwatch.langchain.run.id": runId,
+        "langwatch.langchain.run.parent_id": parentRunId
+      });
+    }
+  }
+  async handleToolEnd(output, runId, parentRunId) {
+    this.finishRun(runId, {
+      output: _chunkDISMHYXCjs.shouldCaptureOutput.call(void 0, ) ? { type: "text", value: output } : void 0,
+      event: "handleToolEnd",
+      parentRunId
+    });
+  }
+  async handleToolError(err, runId, parentRunId, tags) {
+    this.finishRun(runId, { err, event: "handleToolError", parentRunId, tags });
+  }
+  async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
+    this.startRunSpan({
+      kind: "retriever",
+      runId,
+      parentRunId,
+      serialized: retriever,
+      metadata,
+      tags,
+      name,
+      input: _chunkDISMHYXCjs.shouldCaptureInput.call(void 0, ) ? { type: "text", value: query } : void 0
+    });
+    const span = this.spans[runId];
+    if (span) {
+      span.setAttributes({
+        "langwatch.langchain.run.id": runId,
+        "langwatch.langchain.run.parent_id": parentRunId
+      });
+    }
+  }
+  async handleRetrieverEnd(documents, runId, parentRunId, tags) {
+    const span = this.spans[runId];
+    if (span && _chunkDISMHYXCjs.shouldCaptureOutput.call(void 0, )) {
+      span.setOutput(documents);
+    }
+    if (span && _chunkDISMHYXCjs.shouldCaptureInput.call(void 0, )) {
+      span.setRAGContexts(
+        documents.map((document) => ({
+          document_id: document.metadata.id,
+          chunk_id: document.metadata.chunk_id,
+          content: document.pageContent
+        }))
+      );
+    }
+    this.finishRun(runId, { event: "handleRetrieverEnd", parentRunId, tags });
+  }
+  async handleRetrieverError(err, runId, parentRunId, tags) {
+    this.finishRun(runId, {
+      err,
+      event: "handleRetrieverError",
+      parentRunId,
+      tags
+    });
+  }
+  async handleAgentAction(_action, runId, parentRunId, tags) {
+    const span = this.spans[runId];
+    if (span) {
+      addLangChainEvent(span, "handleAgentAction", runId, parentRunId, tags);
+      span.setType("agent");
+    }
+  }
+  async handleAgentEnd(action, runId, parentRunId, tags) {
+    this.finishRun(runId, {
+      output: _chunkDISMHYXCjs.shouldCaptureOutput.call(void 0, ) ? { type: "json", value: action.returnValues } : void 0,
+      event: "handleAgentEnd",
+      parentRunId,
+      tags
+    });
+  }
+};
+function convertFromLangChainMessages(messages) {
+  const out = [];
+  for (const message of messages) {
+    out.push(
+      convertFromLangChainMessage(message)
+    );
+  }
+  return out;
+}
+function convertFromLangChainMessage(message) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+  let role = "user";
+  const msgType = message.type;
+  if (msgType === "human") role = "user";
+  else if (msgType === "ai") role = "assistant";
+  else if (msgType === "system") role = "system";
+  else if (msgType === "function") role = "function";
+  else if (msgType === "tool") role = "tool";
+  else {
+    if (((_a = message == null ? void 0 : message._getType) == null ? void 0 : _a.call(message)) === "human" || ((_b = message.id) == null ? void 0 : _b[message.id.length - 1]) === "HumanMessage") {
+      role = "user";
+    } else if (((_c = message == null ? void 0 : message._getType) == null ? void 0 : _c.call(message)) === "ai" || ((_d = message.id) == null ? void 0 : _d[message.id.length - 1]) === "AIMessage") {
+      role = "assistant";
+    } else if (((_e = message == null ? void 0 : message._getType) == null ? void 0 : _e.call(message)) === "system" || ((_f = message.id) == null ? void 0 : _f[message.id.length - 1]) === "SystemMessage") {
+      role = "system";
+    } else if (((_g = message == null ? void 0 : message._getType) == null ? void 0 : _g.call(message)) === "function" || ((_h = message.id) == null ? void 0 : _h[message.id.length - 1]) === "FunctionMessage") {
+      role = "function";
+    } else if (((_i = message == null ? void 0 : message._getType) == null ? void 0 : _i.call(message)) === "tool" || ((_j = message.id) == null ? void 0 : _j[message.id.length - 1]) === "ToolMessage") {
+      role = "tool";
+    }
+  }
+  const content = typeof message.content === "string" ? message.content : message.content == null ? null : Array.isArray(message.content) ? message.content.map(
+    (c) => (c == null ? void 0 : c.type) === "text" ? { type: "text", text: c.text } : (c == null ? void 0 : c.type) === "image_url" ? { type: "image_url", image_url: c.image_url } : { type: "text", text: JSON.stringify(c) }
+  ) : JSON.stringify(message.content);
+  const functionCall = message.additional_kwargs;
+  return _chunkOXBO24RBjs.__spreadValues.call(void 0, {
+    role,
+    content
+  }, functionCall && typeof functionCall === "object" && Object.keys(functionCall).length > 0 ? { function_call: functionCall } : {});
+}
+function className(serialized) {
+  const id = serialized == null ? void 0 : serialized.id;
+  if (Array.isArray(id) && id.length) return String(id[id.length - 1]);
+  const ns = serialized == null ? void 0 : serialized.lc_namespace;
+  if (Array.isArray(ns) && ns.length) return String(ns[ns.length - 1]);
+  return "";
+}
+function shorten(str, max = 120) {
+  return typeof str === "string" && str.length > max ? str.slice(0, max - 1) + "\u2026" : str;
+}
+function previewInput(v) {
+  if (typeof v === "string") {
+    const s = v.trim();
+    return s ? shorten(s, 120) : void 0;
+  }
+  return void 0;
+}
+function ctxSkip(serialized, tags) {
+  var _a;
+  const cls = className(serialized);
+  return cls.startsWith("ChannelWrite") || ((_a = tags == null ? void 0 : tags.includes("langsmith:hidden")) != null ? _a : false);
+}
+function wrapNonScalarValues(value) {
+  if (value === void 0) return void 0;
+  if (value === null) return JSON.stringify(null);
+  if (typeof value === "string" || typeof value === "number" || typeof value === "boolean")
+    return value;
+  const chatMessages = _chunkM4VUHTT2js.chatMessageSchema.array().safeParse(value);
+  if (Array.isArray(value) && chatMessages.success) {
+    return JSON.stringify({ type: "chat_messages", value: chatMessages.data });
+  }
+  try {
+    const seen = /* @__PURE__ */ new WeakSet();
+    const json = JSON.stringify(value, (k, val) => {
+      if (typeof val === "object" && val !== null) {
+        if (seen.has(val)) return "[Circular]";
+        seen.add(val);
+      }
+      return val;
+    });
+    return json;
+  } catch (e) {
+    return JSON.stringify({ type: "raw", value: "[Non-Serializable]" });
+  }
+}
+function addLangChainEvent(span, eventName, runId, parentRunId, tags, metadata, attributes) {
+  const attrs = _chunkOXBO24RBjs.__spreadValues.call(void 0, {
+    "langwatch.langchain.run.id": runId,
+    "langwatch.langchain.run.parent_id": parentRunId,
+    "langwatch.langchain.event.name": eventName
+  }, attributes);
+  if (tags == null ? void 0 : tags.length) attrs["langwatch.langchain.run.tags"] = tags.slice(0, 50);
+  if (metadata) {
+    Object.entries(metadata).forEach(([key, value]) => {
+      attrs[key] = wrapNonScalarValues(value);
+    });
+  }
+  span.addEvent("langwatch.langchain.callback", attrs);
+}
+function setLangGraphAttributes(span, metadata) {
+  if (!metadata) return;
+  const keys = Object.keys(metadata);
+  for (const key of keys) {
+    const value = metadata[key];
+    if (value !== void 0) {
+      const wrapped = wrapNonScalarValues(value);
+      if (wrapped !== void 0) {
+        span.setAttribute(
+          `langwatch.langgraph.${key}`,
+          wrapped
+        );
+      }
+    }
+  }
+}
+function buildLangChainMetadataAttributes(metadata) {
+  if (!metadata) return {};
+  return Object.fromEntries(
+    Object.entries(metadata).filter(([key]) => !LANGGRAPH_METADATA_KEYS.has(key)).map(([key, value]) => [
+      `langwatch.langchain.run.metadata.${key}`,
+      wrapNonScalarValues(value)
+    ])
+  );
+}
+function applyGenAIAttrs(span, metadata, extraParams) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+  const md = metadata != null ? metadata : {};
+  const ex = extraParams != null ? extraParams : {};
+  const provider = md.ls_provider;
+  const requestModel = (_d = (_b = md.ls_model_name) != null ? _b : (_a = md.kwargs) == null ? void 0 : _a.model) != null ? _d : (_c = ex.kwargs) == null ? void 0 : _c.model;
+  const temperature = (_h = (_f = md.ls_temperature) != null ? _f : (_e = md.kwargs) == null ? void 0 : _e.temperature) != null ? _h : (_g = ex.kwargs) == null ? void 0 : _g.temperature;
+  const responseModel = (_i = md.response_metadata) == null ? void 0 : _i.model_name;
+  if (provider) span.setAttribute("gen_ai.system", provider);
+  if (requestModel) span.setAttribute("gen_ai.request.model", requestModel);
+  if (typeof temperature === "number")
+    span.setAttribute("gen_ai.request.temperature", temperature);
+  if (responseModel) span.setAttribute("gen_ai.response.model", responseModel);
+}
+function getResolvedParentContext(runId, spans, parentOf) {
+  let cur = runId;
+  while (cur) {
+    const s = spans[cur];
+    if (s) return _api.trace.setSpan(_api.context.active(), s);
+    cur = parentOf[cur];
+  }
+  return _api.context.active();
+}
+function deriveNameAndType(opts) {
+  var _a, _b, _c, _d, _e, _f;
+  const { runType, name, serialized, metadata, inputs } = opts;
+  const hardName = (_a = name == null ? void 0 : name.trim()) != null ? _a : metadata == null ? void 0 : metadata.operation_name;
+  if (hardName) {
+    return {
+      name: hardName,
+      type: runType === "tool" ? "tool" : runType === "retriever" ? "rag" : runType === "llm" || runType === "chat" ? "llm" : "chain"
+    };
+  }
+  const cls = className(serialized);
+  const md = metadata != null ? metadata : {};
+  const hasNode = (md == null ? void 0 : md.langgraph_node) != null;
+  const hasTriggers = Array.isArray(md == null ? void 0 : md.langgraph_triggers) && md.langgraph_triggers.length > 0;
+  const isRouter = cls.startsWith("Branch<") || hasTriggers;
+  const isGraphRunner = (md == null ? void 0 : md.langgraph_path) && !(md == null ? void 0 : md.langgraph_node);
+  if (runType === "llm" || runType === "chat") {
+    const prov = (_b = md == null ? void 0 : md.ls_provider) != null ? _b : "LLM";
+    const model = (_c = md == null ? void 0 : md.ls_model_name) != null ? _c : cls || "call";
+    const temp = md == null ? void 0 : md.ls_temperature;
+    const tempStr = temp != null ? typeof temp === "number" ? temp.toString() : JSON.stringify(temp) : null;
+    const nm = tempStr != null ? `${prov} ${model} (temp ${tempStr})` : `${prov} ${model}`;
+    return { name: nm, type: "llm" };
+  }
+  if (isRouter) {
+    const pathArr = md == null ? void 0 : md.langgraph_path;
+    const fromNode = Array.isArray(pathArr) && pathArr.length ? pathArr[pathArr.length - 1] : void 0;
+    const decision = Array.isArray(md == null ? void 0 : md.langgraph_triggers) ? String(
+      (_d = md.langgraph_triggers.find(
+        (t) => String(t).startsWith("branch:")
+      )) != null ? _d : ""
+    ).replace(/^branch:(to:)?/, "") : void 0;
+    const nm = `Route: ${fromNode != null ? fromNode : "unknown"} \u2192 ${decision != null ? decision : "unknown"}`;
+    return { name: nm, type: "component" };
+  }
+  if (hasNode) {
+    const step = md == null ? void 0 : md.langgraph_step;
+    const nm = `Node: ${md.langgraph_node}${step != null ? ` (step ${String(step)})` : ""}`;
+    return { name: nm, type: "component" };
+  }
+  if (isGraphRunner && runType === "chain") {
+    return { name: "Graph: LangGraph", type: "chain" };
+  }
+  if (runType === "tool") {
+    const tool = (_e = metadata == null ? void 0 : metadata.name) != null ? _e : cls || "tool";
+    const prev = (_f = previewInput(inputs)) != null ? _f : previewInput(serialized == null ? void 0 : serialized.input);
+    return {
+      name: prev ? `Tool: ${tool} \u2014 ${prev}` : `Tool: ${tool}`,
+      type: "tool"
+    };
+  }
+  if (runType === "retriever") return { name: "Retriever", type: "rag" };
+  if (cls.includes("Agent"))
+    return { name: `Agent: ${cls}`, type: "component" };
+  if (cls.startsWith("Runnable"))
+    return { name: `Runnable: ${cls.replace(/^Runnable/, "")}`, type: "chain" };
+  return { name: cls || "LangChain operation", type: "chain" };
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+exports.LangWatchCallbackHandler = LangWatchCallbackHandler; exports.addLangChainEvent = addLangChainEvent; exports.applyGenAIAttrs = applyGenAIAttrs; exports.buildLangChainMetadataAttributes = buildLangChainMetadataAttributes; exports.className = className; exports.convertFromLangChainMessages = convertFromLangChainMessages; exports.ctxSkip = ctxSkip; exports.deriveNameAndType = deriveNameAndType; exports.getResolvedParentContext = getResolvedParentContext; exports.previewInput = previewInput; exports.setLangGraphAttributes = setLangGraphAttributes; exports.shorten = shorten; exports.wrapNonScalarValues = wrapNonScalarValues;
+//# sourceMappingURL=index.js.map

package/dist/observability-sdk/instrumentation/langchain/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/home/runner/work/langwatch/langwatch/typescript-sdk/dist/observability-sdk/instrumentation/langchain/index.js","../../../../src/observability-sdk/instrumentation/langchain/index.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,6DAAmC;AACnC;AACE;AACA;AACF,6DAAmC;AACnC;AACE;AACF,6DAAmC;AACnC;AACA;ACXA,sDAAoC;AAYpC;AACE;AACA;AACA;AAAA,yCAEK;AAkBP,IAAM,wBAAA,kBAA0B,IAAI,GAAA,CAAY;AAAA,EAC9C,WAAA;AAAA,EACA,gBAAA;AAAA,EACA,gBAAA;AAAA,EACA,oBAAA;AAAA,EACA,gBAAA;AAAA,EACA,yBAAA;AAAA,EACA,kBAAA;AAAA,EACA;AACF,CAAC,CAAA;AAEM,IAAM,yBAAA,EAAN,MAAA,QAAuC,0BAAoB;AAAA,EAA3D,WAAA,CAAA,EAAA;AAAA,IAAA,KAAA,CAAA,GAAA,SAAA,CAAA;AACL,IAAA,IAAA,CAAA,KAAA,EAAO,0BAAA;AACP,IAAA,IAAA,CAAA,OAAA,EAAS,iDAAA,qCAAwD,CAAA;AAEjE,IAAA,IAAA,CAAQ,MAAA,EAAmD,CAAC,CAAA;AAC5D,IAAA,IAAA,CAAQ,SAAA,EAA+C,CAAC,CAAA;AACxD,IAAA,IAAA,CAAQ,QAAA,EAA4C,CAAC,CAAA;AACrD,IAAA,IAAA,CAAQ,WAAA,kBAAa,IAAI,GAAA,CAAY,CAAA;AAAA,EAAA;AAAA,EAE7B,YAAA,CAAa,IAAA,EAAiB;AAxDxC,IAAA,IAAA,EAAA;AAyDI,IAAA,MAAM,EAAE,KAAA,EAAO,WAAA,EAAa,UAAA,EAAY,KAAK,EAAA,EAAI,IAAA;AACjD,IAAA,IAAA,CAAK,QAAA,CAAS,KAAK,EAAA,EAAI,WAAA;AAGvB,IAAA,GAAA,CAAI,OAAA,CAAQ,UAAA,EAAY,IAAI,CAAA,EAAG;AAC7B,MAAA,IAAA,CAAK,OAAA,CAAQ,KAAK,EAAA,EAAI,IAAA;AACtB,MAAA,MAAA;AAAA,IACF;AACA,IAAA,GAAA,CAAI,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA,EAAG,MAAA;AAChC,IAAA,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAEzB,IAAA,MAAM,UAAA,EAAY,wBAAA;AAAA,MAChB,WAAA;AAAA,MACA,IAAA,CAAK,KAAA;AAAA,MACL,IAAA,CAAK;AAAA,IACP,CAAA;AACA,IAAA,MAAM,WAAA,EAAa,YAAA,EAAc,IAAA,CAAK,KAAA,CAAM,WAAW,EAAA,EAAI,KAAA,CAAA;AAC3D,IAAA,MAAM,MAAA,EAAQ,WAAA,EACV,CAAC,EAAE,OAAA,EAAS,UAAA,CAAW,WAAA,CAAY,EAAE,CAAC,EAAA,EACtC,KAAA,CAAA;AAEJ,IAAA,MAAM,EAAE,IAAA,EAAM,KAAK,EAAA,EAAI,iBAAA,CAAkB;AAAA,MACvC,OAAA,EAAS,IAAA,CAAK,IAAA;AAAA,MACd,IAAA,EAAM,IAAA,CAAK,IAAA;AAAA,MACX,UAAA,EAAY,IAAA,CAAK,UAAA;AAAA,MACjB,QAAA,EAAU,IAAA,CAAK,QAAA;AAAA,MACf,IAAA,EAAM,IAAA,CAAK,IAAA;AAAA,MACX,MAAA,EAAQ,IAAA,CAAK;AAAA,IACf,CAAC,CAAA;AAED,IAAA,MAAM,KAAA,EAAO,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,IAAA,EAAM,EAAE,MAAM,CAAA,EAAG,SAAS,CAAA;AAC7D,IAAA,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AAEjB,IAAA,GAAA,CAAA,CAAI,GAAA,EAAA,IAAA,CAAK,IAAA,EAAA,GAAL,KAAA,EAAA,KAAA,EAAA,EAAA,EAAA,CAAW,MAAA;AACb,MAAA,IAAA,CAAK,YAAA,CAAa,8BAAA,EAAgC,IAAA,CAAK,IAAA,CAAK,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA;AAE1E,IAAA,GAAA,CAAI,iDAAA,EAAmB,GAAK,IAAA,CAAK,MAAA,IAAU,KAAA,CAAA,EAAQ;AACjD,MAAA,MAAM,EAAA,EAAS,IAAA,CAAK,KAAA;AACpB,MAAA,GAAA,CAAI,EAAA,GAAK,OAAO,EAAA,IAAM,SAAA,GAAY,OAAA,GAAU,EAAA,GAAK,QAAA,GAAW,CAAA,EAAG;AAC7D,QAAA,IAAA,CAAK,QAAA,CAAS,CAAA,CAAE,IAAA,EAAM,CAAA,CAAE,KAAK,CAAA;AAAA,MAC/B,EAAA,KAAO;AACL,QAAA,IAAA,CAAK,QAAA,CAAS,CAAC,CAAA;AAAA,MACjB;AAAA,IACF;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK,WAAA,EAAa;AACpB,MAAA,IAAA,CAAK,aAAA;AAAA,QACH,MAAA,CAAO,WAAA;AAAA,UACL,MAAA,CAAO,OAAA,CAAQ,IAAA,CAAK,WAAW,CAAA,CAAE,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,CAAA,EAAA,GAAM;AAAA,YAC/C,CAAA,qCAAA,EAAwC,CAAC,CAAA,CAAA;AACpB,YAAA;AACtB,UAAA;AACH,QAAA;AACF,MAAA;AACF,IAAA;AAEmB,IAAA;AACyB,MAAA;AACA,MAAA;AACvB,MAAA;AACrB,IAAA;AAEoB,IAAA;AACtB,EAAA;AAaE,EAAA;AAC6B,IAAA;AAClB,IAAA;AAEX,IAAA;AACE,MAAA;AACI,MAAA;AACJ,MAAA;AACI,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACN,IAAA;AAEa,IAAA;AACiB,MAAA;AACiB,MAAA;AACP,IAAA;AACN,MAAA;AAClC,IAAA;AAES,IAAA;AAEc,IAAA;AACG,IAAA;AACD,IAAA;AACG,IAAA;AAC9B,EAAA;AAME,EAAA;AAOI,IAAA;AACQ,MAAA;AACqC,MAAA;AAE7C,IAAA;AAEc,IAAA;AACV,MAAA;AACN,MAAA;AACA,MAAA;AACY,MAAA;AACZ,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACH,EAAA;AAME,EAAA;AAOI,IAAA;AACQ,MAAA;AACkB,MAAA;AAE1B,IAAA;AAEc,IAAA;AACV,MAAA;AACN,MAAA;AACA,MAAA;AACY,MAAA;AACZ,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACH,EAAA;AAMiB,EAAA;AAlOnB,IAAA;AAmOiC,IAAA;AACjB,IAAA;AAOI,IAAA;AACK,MAAA;AACgB,QAAA;AACI,QAAA;AACL,QAAA;AACjC,MAAA;AACH,IAAA;AAGa,IAAA;AAC0B,MAAA;AACK,QAAA;AACZ,UAAA;AACvB,QAAA;AAC+B,MAAA;AACvB,QAAA;AACX,MAAA;AACO,MAAA;AAET,IAAA;AAEkB,IAAA;AACZ,MAAA;AACD,MAAA;AACP,MAAA;AACD,IAAA;AACH,EAAA;AAMiB,EAAA;AACqB,IAAA;AACtC,EAAA;AAME,EAAA;AAMkB,IAAA;AACV,MAAA;AACN,MAAA;AACA,MAAA;AACY,MAAA;AACZ,MAAA;AACA,MAAA;AACA,MAAA;AACuC,MAAA;AACxC,IAAA;AAEa,IAAA;AACiB,MAAA;AACD,MAAA;AAC9B,IAAA;AACF,EAAA;AAMiB,EAAA;AACwB,IAAA;AACzC,EAAA;AAME,EAAA;AAGsB,IAAA;AACpB,MAAA;AACO,MAAA;AACP,MAAA;AACA,MAAA;AACI,MAAA;AACL,IAAA;AACH,EAAA;AAME,EAAA;AAKkB,IAAA;AACV,MAAA;AACN,MAAA;AACA,MAAA;AACY,MAAA;AACZ,MAAA;AACA,MAAA;AACA,MAAA;AAC8C,MAAA;AAC/C,IAAA;AAE4B,IAAA;AACnB,IAAA;AACW,MAAA;AACa,QAAA;AACO,QAAA;AACtC,MAAA;AACH,IAAA;AACF,EAAA;AAMiB,EAAA;AACO,IAAA;AAEA,MAAA;AAEb,MAAA;AACP,MAAA;AACD,IAAA;AACH,EAAA;AAME,EAAA;AAEoC,IAAA;AACtC,EAAA;AAME,EAAA;AAKkB,IAAA;AACV,MAAA;AACN,MAAA;AACA,MAAA;AACY,MAAA;AACZ,MAAA;AACA,MAAA;AACA,MAAA;AAC8C,MAAA;AAC/C,IAAA;AAE4B,IAAA;AACnB,IAAA;AACW,MAAA;AACa,QAAA;AACO,QAAA;AACtC,MAAA;AACH,IAAA;AACF,EAAA;AAKE,EAAA;AAG6B,IAAA;AAEM,IAAA;AACT,MAAA;AAC1B,IAAA;AAEkC,IAAA;AAC3B,MAAA;AAC0B,QAAA;AACI,UAAA;AACH,UAAA;AACV,UAAA;AAClB,QAAA;AACJ,MAAA;AACF,IAAA;AAE+B,IAAA;AACjC,EAAA;AAME,EAAA;AAEsB,IAAA;AACpB,MAAA;AACO,MAAA;AACP,MAAA;AACA,MAAA;AACD,IAAA;AACH,EAAA;AAKE,EAAA;AAG6B,IAAA;AACnB,IAAA;AACqC,MAAA;AACzB,MAAA;AACtB,IAAA;AACF,EAAA;AAME,EAAA;AAEsB,IAAA;AAEA,MAAA;AAEb,MAAA;AACP,MAAA;AACA,MAAA;AACD,IAAA;AACH,EAAA;AACF;AAIiB;AACa,EAAA;AACI,EAAA;AAC1B,IAAA;AACoE,MAAA;AACxE,IAAA;AACF,EAAA;AACO,EAAA;AACT;AAIe;AAtef,EAAA;AAuekC,EAAA;AAEC,EAAA;AACD,EAAA;AACE,EAAA;AACI,EAAA;AACE,EAAA;AACJ,EAAA;AAC/B,EAAA;AAEiB,IAAA;AAGX,MAAA;AAEN,IAAA;AAGM,MAAA;AAEN,IAAA;AAGM,MAAA;AAEN,IAAA;AAGM,MAAA;AAEN,IAAA;AAGM,MAAA;AACT,IAAA;AACF,EAAA;AAGsC,EAAA;AAQxB,IAAA;AAK+B,EAAA;AAEP,EAAA;AAE/B,EAAA;AACL,IAAA;AACA,IAAA;AAEwB,EAAA;AAK5B;AAEoD;AAClB,EAAA;AACkB,EAAA;AAClB,EAAA;AACkB,EAAA;AAE3C,EAAA;AACT;AAEiD;AACA,EAAA;AAGjD;AAEsD;AACzB,EAAA;AACR,IAAA;AACY,IAAA;AAC/B,EAAA;AACO,EAAA;AACT;AAE2D;AAhkB3D,EAAA;AAikBkC,EAAA;AAG7B,EAAA;AAEL;AAIyC;AACV,EAAA;AACiB,EAAA;AAG3B,EAAA;AAGV,IAAA;AAGsC,EAAA;AACG,EAAA;AACD,IAAA;AACjD,EAAA;AAEI,EAAA;AACuB,IAAA;AAC6B,IAAA;AACP,MAAA;AACjB,QAAA;AACd,QAAA;AACd,MAAA;AACO,MAAA;AACR,IAAA;AACM,IAAA;AACD,EAAA;AACsC,IAAA;AAC9C,EAAA;AACF;AAME;AAK0B,EAAA;AACM,IAAA;AACO,IAAA;AACH,IAAA;AAC/B,EAAA;AAGmB,EAAA;AACV,EAAA;AACiC,IAAA;AACL,MAAA;AACvC,IAAA;AACH,EAAA;AAEmD,EAAA;AACrD;AAKE;AACe,EAAA;AACkB,EAAA;AACT,EAAA;AACa,IAAA;AACV,IAAA;AACkB,MAAA;AACd,MAAA;AACpB,QAAA;AACuB,UAAA;AAC1B,UAAA;AACF,QAAA;AACF,MAAA;AACF,IAAA;AACF,EAAA;AACF;AAE6E;AACpD,EAAA;AACT,EAAA;AAEU,IAAA;AAEqB,MAAA;AACd,MAAA;AAC1B,IAAA;AACL,EAAA;AACF;AAME;AAvqBF,EAAA;AAwqB2B,EAAA;AACG,EAAA;AAER,EAAA;AACI,EAAA;AAEnB,EAAA;AACoB,EAAA;AAEwB,EAAA;AACb,EAAA;AACT,EAAA;AACuB,IAAA;AACb,EAAA;AACvC;AAKE;AAEU,EAAA;AACE,EAAA;AACS,IAAA;AAC4B,IAAA;AAC7B,IAAA;AACpB,EAAA;AACsB,EAAA;AACxB;AAS2E;AA7sB3E,EAAA;AA8sBsD,EAAA;AAG5B,EAAA;AAGV,EAAA;AACL,IAAA;AACC,MAAA;AAIY,MAAA;AAKpB,IAAA;AACF,EAAA;AAEgC,EAAA;AACP,EAAA;AAGL,EAAA;AACc,EAAA;AAEH,EAAA;AACL,EAAA;AAGmB,EAAA;AACzB,IAAA;AACC,IAAA;AACF,IAAA;AAGK,IAAA;AAMA,IAAA;AAES,IAAA;AACjC,EAAA;AAGc,EAAA;AACQ,IAAA;AAEQ,IAAA;AAGG,IAAA;AAEH,MAAA;AACU,QAAA;AAC3B,MAAA;AAEP,IAAA;AACiB,IAAA;AACgB,IAAA;AACvC,EAAA;AAEa,EAAA;AACM,IAAA;AAEP,IAAA;AAE2B,IAAA;AACvC,EAAA;AAC0C,EAAA;AACS,IAAA;AACnD,EAAA;AAGwB,EAAA;AACR,IAAA;AAEZ,IAAA;AACK,IAAA;AACoC,MAAA;AACnC,MAAA;AACR,IAAA;AACF,EAAA;AAG4C,EAAA;AAGpB,EAAA;AAC4B,IAAA;AACvB,EAAA;AACa,IAAA;AACS,EAAA;AACrD;ADzTsD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/langwatch/langwatch/typescript-sdk/dist/observability-sdk/instrumentation/langchain/index.js","sourcesContent":[null,"import type { AgentAction, AgentFinish } from \"@langchain/core/agents\";\nimport { BaseCallbackHandler } from \"@langchain/core/callbacks/base\";\nimport { type DocumentInterface } from \"@langchain/core/documents\";\nimport type { Serialized } from \"@langchain/core/load/serializable\";\nimport { type BaseMessage } from \"@langchain/core/messages\";\nimport type { ChatGeneration, LLMResult } from \"@langchain/core/outputs\";\nimport type {\n  ChatMessage,\n  ChatRichContent,\n} from \"../../../internal/generated/types/tracer\";\nimport type { ChainValues } from \"@langchain/core/utils/types\";\nimport { getLangWatchTracer } from \"../../tracer\";\nimport type { LangWatchSpan } from \"../../span\";\nimport {\n  context,\n  trace,\n  SpanStatusCode,\n  type Attributes,\n} from \"@opentelemetry/api\";\nimport { chatMessageSchema } from \"../../../internal/generated/types/tracer.generated\";\nimport { shouldCaptureInput, shouldCaptureOutput } from \"../../config\";\n\ntype RunKind = \"llm\" | \"chat\" | \"chain\" | \"tool\" | \"retriever\";\n\ntype StartArgs = {\n  kind: RunKind;\n  runId: string;\n  parentRunId?: string;\n  serialized?: Serialized;\n  metadata?: Record<string, unknown>;\n  tags?: string[];\n  name?: string;\n  extraParams?: Record<string, unknown>;\n  input?: unknown; // already prepared for setInput()\n};\n\nconst LANGGRAPH_METADATA_KEYS = new Set<string>([\n  \"thread_id\",\n  \"langgraph_step\",\n  \"langgraph_node\",\n  \"langgraph_triggers\",\n  \"langgraph_path\",\n  \"langgraph_checkpoint_ns\",\n  \"__pregel_task_id\",\n  \"checkpoint_ns\",\n]);\n\nexport class LangWatchCallbackHandler extends BaseCallbackHandler {\n  name = \"LangWatchCallbackHandler\";\n  tracer = getLangWatchTracer(\"langwatch.instrumentation.langchain\");\n\n  private spans: Record<string, LangWatchSpan | undefined> = {};\n  private parentOf: Record<string, string | undefined> = {};\n  private skipped: Record<string, true | undefined> = {};\n  private seenStarts = new Set<string>();\n\n  private startRunSpan(args: StartArgs) {\n    const { runId, parentRunId, serialized, tags } = args;\n    this.parentOf[runId] = parentRunId;\n\n    // if we want to skip, we record as such as we cn do context matching latet\n    if (ctxSkip(serialized, tags)) {\n      this.skipped[runId] = true;\n      return;\n    }\n    if (this.seenStarts.has(runId)) return;\n    this.seenStarts.add(runId);\n\n    const parentCtx = getResolvedParentContext(\n      parentRunId,\n      this.spans,\n      this.parentOf\n    );\n    const parentSpan = parentRunId ? this.spans[parentRunId] : void 0;\n    const links = parentSpan\n      ? [{ context: parentSpan.spanContext() }]\n      : void 0;\n\n    const { name, type } = deriveNameAndType({\n      runType: args.kind,\n      name: args.name,\n      serialized: args.serialized,\n      metadata: args.metadata,\n      tags: args.tags,\n      inputs: args.input,\n    });\n\n    const span = this.tracer.startSpan(name, { links }, parentCtx);\n    span.setType(type);\n\n    if (args.tags?.length)\n      span.setAttribute(\"langwatch.langchain.run.tags\", args.tags.slice(0, 50));\n\n    if (shouldCaptureInput() && args.input !== void 0) {\n      const i: any = args.input as any;\n      if (i && typeof i === \"object\" && \"type\" in i && \"value\" in i) {\n        span.setInput(i.type, i.value);\n      } else {\n        span.setInput(i);\n      }\n    }\n\n    if (args.extraParams) {\n      span.setAttributes(\n        Object.fromEntries(\n          Object.entries(args.extraParams).map(([k, v]) => [\n            `langwatch.langchain.run.extra_params.${k}`,\n            wrapNonScalarValues(v),\n          ])\n        )\n      );\n    }\n\n    if (args.metadata) {\n      applyGenAIAttrs(span, args.metadata, args.extraParams);\n      setLangGraphAttributes(span, args.metadata);\n      span.setAttributes(buildLangChainMetadataAttributes(args.metadata));\n    }\n\n    this.spans[runId] = span;\n  }\n\n  private finishRun(\n    runId: string,\n    end: {\n      output?: unknown;\n      err?: Error;\n      event: string;\n      parentRunId?: string;\n      extra?: Attributes;\n      tags?: string[];\n      md?: Record<string, unknown>;\n    }\n  ) {\n    const span = this.spans[runId];\n    if (!span) return;\n\n    addLangChainEvent(\n      span,\n      end.event,\n      runId,\n      end.parentRunId,\n      end.tags,\n      end.md,\n      end.extra\n    );\n\n    if (end.err) {\n      span.recordException(end.err);\n      span.setStatus({ code: SpanStatusCode.ERROR, message: end.err.message });\n    } else if (shouldCaptureOutput() && end.output !== undefined) {\n      span.setOutput(end.output as any);\n    }\n\n    span.end();\n\n    delete this.spans[runId];\n    delete this.parentOf[runId];\n    delete this.skipped[runId];\n    this.seenStarts.delete(runId);\n  }\n\n  async handleLLMStart(\n    llm: Serialized,\n    prompts: string[],\n    runId: string,\n    parentRunId?: string,\n    extraParams?: Record<string, unknown>,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    name?: string\n  ): Promise<void> {\n    const input = shouldCaptureInput() && prompts\n      ? {\n          type: \"list\",\n          value: prompts.map((p) => ({ type: \"text\", value: p })),\n        }\n      : void 0;\n\n    this.startRunSpan({\n      kind: \"llm\",\n      runId,\n      parentRunId,\n      serialized: llm,\n      metadata,\n      tags,\n      name,\n      extraParams,\n      input,\n    });\n  }\n\n  async handleChatModelStart(\n    llm: Serialized,\n    messages: BaseMessage[][],\n    runId: string,\n    parentRunId?: string,\n    extraParams?: Record<string, unknown>,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    name?: string\n  ): Promise<void> {\n    const input = shouldCaptureInput()\n      ? {\n          type: \"chat_messages\",\n          value: messages.flatMap(convertFromLangChainMessages),\n        }\n      : void 0;\n\n    this.startRunSpan({\n      kind: \"chat\",\n      runId,\n      parentRunId,\n      serialized: llm,\n      metadata,\n      tags,\n      name,\n      extraParams,\n      input,\n    });\n  }\n\n  async handleLLMEnd(\n    response: LLMResult,\n    runId: string,\n    parentRunId?: string\n  ): Promise<void> {\n    const span = this.spans[runId];\n    const tu = (response.llmOutput as any)?.tokenUsage as\n      | {\n          promptTokens?: number;\n          completionTokens?: number;\n          totalTokens?: number;\n        }\n      | undefined;\n    if (span && tu) {\n      span.setAttributes({\n        \"gen_ai.usage.prompt_tokens\": tu.promptTokens ?? 0,\n        \"gen_ai.usage.completion_tokens\": tu.completionTokens ?? 0,\n        \"gen_ai.usage.total_tokens\": tu.totalTokens ?? 0,\n      });\n    }\n\n    const outputs = shouldCaptureOutput()\n      ? response.generations.flat().map((g) => {\n          if (\"message\" in g && g.message) {\n            return convertFromLangChainMessages([\n              (g as ChatGeneration).message,\n            ]);\n          } else if (\"text\" in g && g.text) {\n            return g.text;\n          }\n          return g;\n        })\n      : undefined;\n\n    this.finishRun(runId, {\n      output: outputs,\n      event: \"handleLLMEnd\",\n      parentRunId,\n    });\n  }\n\n  async handleLLMError(\n    err: Error,\n    runId: string,\n    parentRunId?: string\n  ): Promise<void> {\n    this.finishRun(runId, { err, event: \"handleLLMError\", parentRunId });\n  }\n\n  async handleChainStart(\n    chain: Serialized,\n    inputs: ChainValues,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    _runType?: string,\n    name?: string\n  ): Promise<void> {\n    this.startRunSpan({\n      kind: \"chain\",\n      runId,\n      parentRunId,\n      serialized: chain,\n      metadata,\n      tags,\n      name,\n      input: shouldCaptureInput() ? inputs : void 0,\n    });\n\n    if (_runType) {\n      const span = this.spans[runId];\n      if (span) span.setAttribute(\"langwatch.langchain.run.type\", _runType);\n    }\n  }\n\n  async handleChainEnd(\n    output: ChainValues,\n    runId: string,\n    parentRunId?: string\n  ): Promise<void> {\n    this.finishRun(runId, { output, event: \"handleChainEnd\", parentRunId });\n  }\n\n  async handleChainError(\n    err: Error,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[],\n    kwargs?: { inputs?: Record<string, unknown> | undefined }\n  ): Promise<void> {\n    this.finishRun(runId, {\n      err,\n      event: \"handleChainError\",\n      parentRunId,\n      tags,\n      md: kwargs,\n    });\n  }\n\n  async handleToolStart(\n    tool: Serialized,\n    input: string,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    name?: string\n  ): Promise<void> {\n    this.startRunSpan({\n      kind: \"tool\",\n      runId,\n      parentRunId,\n      serialized: tool,\n      metadata,\n      tags,\n      name,\n      input: shouldCaptureInput() ? { type: \"text\", value: input } : void 0,\n    });\n\n    const span = this.spans[runId];\n    if (span) {\n      span.setAttributes({\n        \"langwatch.langchain.run.id\": runId,\n        \"langwatch.langchain.run.parent_id\": parentRunId,\n      });\n    }\n  }\n\n  async handleToolEnd(\n    output: string,\n    runId: string,\n    parentRunId?: string\n  ): Promise<void> {\n    this.finishRun(runId, {\n      output: shouldCaptureOutput()\n        ? { type: \"text\", value: output }\n        : void 0,\n      event: \"handleToolEnd\",\n      parentRunId,\n    });\n  }\n\n  async handleToolError(\n    err: Error,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[]\n  ): Promise<void> {\n    this.finishRun(runId, { err, event: \"handleToolError\", parentRunId, tags });\n  }\n\n  async handleRetrieverStart(\n    retriever: Serialized,\n    query: string,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    name?: string\n  ) {\n    this.startRunSpan({\n      kind: \"retriever\",\n      runId,\n      parentRunId,\n      serialized: retriever,\n      metadata,\n      tags,\n      name,\n      input: shouldCaptureInput() ? { type: \"text\", value: query } : void 0,\n    });\n\n    const span = this.spans[runId];\n    if (span) {\n      span.setAttributes({\n        \"langwatch.langchain.run.id\": runId,\n        \"langwatch.langchain.run.parent_id\": parentRunId,\n      });\n    }\n  }\n\n  async handleRetrieverEnd(\n    documents: DocumentInterface<Record<string, any>>[],\n    runId: string,\n    parentRunId?: string,\n    tags?: string[]\n  ) {\n    const span = this.spans[runId];\n\n    if (span && shouldCaptureOutput()) {\n      span.setOutput(documents);\n    }\n\n    if (span && shouldCaptureInput()) {\n      span.setRAGContexts(\n        documents.map((document) => ({\n          document_id: document.metadata.id,\n          chunk_id: document.metadata.chunk_id,\n          content: document.pageContent,\n        }))\n      );\n    }\n\n    this.finishRun(runId, { event: \"handleRetrieverEnd\", parentRunId, tags });\n  }\n\n  async handleRetrieverError(\n    err: Error,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[]\n  ) {\n    this.finishRun(runId, {\n      err,\n      event: \"handleRetrieverError\",\n      parentRunId,\n      tags,\n    });\n  }\n\n  async handleAgentAction(\n    _action: AgentAction,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[]\n  ): Promise<void> {\n    const span = this.spans[runId];\n    if (span) {\n      addLangChainEvent(span, \"handleAgentAction\", runId, parentRunId, tags);\n      span.setType(\"agent\");\n    }\n  }\n\n  async handleAgentEnd(\n    action: AgentFinish,\n    runId: string,\n    parentRunId?: string,\n    tags?: string[]\n  ): Promise<void> {\n    this.finishRun(runId, {\n      output: shouldCaptureOutput()\n        ? { type: \"json\", value: action.returnValues }\n        : void 0,\n      event: \"handleAgentEnd\",\n      parentRunId,\n      tags,\n    });\n  }\n}\n\nexport function convertFromLangChainMessages(\n  messages: BaseMessage[]\n): ChatMessage[] {\n  const out: ChatMessage[] = [];\n  for (const message of messages) {\n    out.push(\n      convertFromLangChainMessage(message as BaseMessage & { id?: string[] })\n    );\n  }\n  return out;\n}\n\nfunction convertFromLangChainMessage(\n  message: BaseMessage & { id?: string[] }\n): ChatMessage {\n  let role: ChatMessage[\"role\"] = \"user\";\n\n  const msgType = (message as any).type as string | undefined;\n  if (msgType === \"human\") role = \"user\";\n  else if (msgType === \"ai\") role = \"assistant\";\n  else if (msgType === \"system\") role = \"system\";\n  else if (msgType === \"function\") role = \"function\";\n  else if (msgType === \"tool\") role = \"tool\";\n  else {\n    if (\n      (message as any)?._getType?.() === \"human\" ||\n      message.id?.[message.id.length - 1] === \"HumanMessage\"\n    ) {\n      role = \"user\";\n    } else if (\n      (message as any)?._getType?.() === \"ai\" ||\n      message.id?.[message.id.length - 1] === \"AIMessage\"\n    ) {\n      role = \"assistant\";\n    } else if (\n      (message as any)?._getType?.() === \"system\" ||\n      message.id?.[message.id.length - 1] === \"SystemMessage\"\n    ) {\n      role = \"system\";\n    } else if (\n      (message as any)?._getType?.() === \"function\" ||\n      message.id?.[message.id.length - 1] === \"FunctionMessage\"\n    ) {\n      role = \"function\";\n    } else if (\n      (message as any)?._getType?.() === \"tool\" ||\n      message.id?.[message.id.length - 1] === \"ToolMessage\"\n    ) {\n      role = \"tool\";\n    }\n  }\n\n  const content: ChatMessage[\"content\"] =\n    typeof (message as any).content === \"string\"\n      ? ((message as any).content as string)\n      : (message as any).content == null\n      ? null\n      : Array.isArray((message as any).content)\n      ? (message as any).content.map(\n          (c: any): ChatRichContent =>\n            c?.type === \"text\"\n              ? { type: \"text\", text: c.text }\n              : c?.type === \"image_url\"\n              ? { type: \"image_url\", image_url: c.image_url }\n              : { type: \"text\", text: JSON.stringify(c) }\n        )\n      : JSON.stringify((message as any).content);\n\n  const functionCall = (message as any).additional_kwargs;\n\n  return {\n    role,\n    content,\n    ...(functionCall &&\n    typeof functionCall === \"object\" &&\n    Object.keys(functionCall).length > 0\n      ? { function_call: functionCall }\n      : {}),\n  };\n}\n\nfunction className(serialized?: Serialized): string {\n  const id = (serialized as any)?.id;\n  if (Array.isArray(id) && id.length) return String(id[id.length - 1]);\n  const ns = (serialized as any)?.lc_namespace;\n  if (Array.isArray(ns) && ns.length) return String(ns[ns.length - 1]);\n\n  return \"\";\n}\n\nfunction shorten(str: string, max = 120): string {\n  return typeof str === \"string\" && str.length > max\n    ? str.slice(0, max - 1) + \"…\"\n    : str;\n}\n\nfunction previewInput(v: unknown): string | undefined {\n  if (typeof v === \"string\") {\n    const s = v.trim();\n    return s ? shorten(s, 120) : void 0;\n  }\n  return void 0;\n}\n\nfunction ctxSkip(serialized?: Serialized, tags?: string[]) {\n  const cls = className(serialized);\n  return (\n    cls.startsWith(\"ChannelWrite\") ||\n    (tags?.includes(\"langsmith:hidden\") ?? false)\n  );\n}\n\nfunction wrapNonScalarValues(\n  value: unknown\n): string | number | boolean | undefined {\n  if (value === void 0) return void 0;\n  if (value === null) return JSON.stringify(null);\n  if (\n    typeof value === \"string\" ||\n    typeof value === \"number\" ||\n    typeof value === \"boolean\"\n  )\n    return value;\n\n  // Special-case: ChatMessage[] detection via zod schema the project already has\n  const chatMessages = chatMessageSchema.array().safeParse(value as any);\n  if (Array.isArray(value) && chatMessages.success) {\n    return JSON.stringify({ type: \"chat_messages\", value: chatMessages.data });\n  }\n\n  try {\n    const seen = new WeakSet();\n    const json = JSON.stringify(value as any, (k, val) => {\n      if (typeof val === \"object\" && val !== null) {\n        if (seen.has(val)) return \"[Circular]\";\n        seen.add(val);\n      }\n      return val;\n    });\n    return json;\n  } catch {\n    return JSON.stringify({ type: \"raw\", value: \"[Non-Serializable]\" });\n  }\n}\n\nfunction addLangChainEvent(\n  span: LangWatchSpan,\n  eventName: string,\n  runId: string,\n  parentRunId: string | undefined,\n  tags?: string[],\n  metadata?: Record<string, unknown>,\n  attributes?: Attributes\n) {\n  const attrs: Attributes = {\n    \"langwatch.langchain.run.id\": runId,\n    \"langwatch.langchain.run.parent_id\": parentRunId,\n    \"langwatch.langchain.event.name\": eventName,\n    ...attributes,\n  };\n\n  if (tags?.length) attrs[\"langwatch.langchain.run.tags\"] = tags.slice(0, 50);\n  if (metadata) {\n    Object.entries(metadata).forEach(([key, value]) => {\n      attrs[key] = wrapNonScalarValues(value);\n    });\n  }\n\n  span.addEvent(\"langwatch.langchain.callback\", attrs);\n}\n\nfunction setLangGraphAttributes(\n  span: LangWatchSpan,\n  metadata?: Record<string, unknown>\n) {\n  if (!metadata) return;\n  const keys = Object.keys(metadata);\n  for (const key of keys) {\n    const value = (metadata as any)[key];\n    if (value !== undefined) {\n      const wrapped = wrapNonScalarValues(value);\n      if (wrapped !== undefined) {\n        span.setAttribute(\n          `langwatch.langgraph.${key}` as const,\n          wrapped as any\n        );\n      }\n    }\n  }\n}\n\nfunction buildLangChainMetadataAttributes(metadata: Record<string, unknown>) {\n  if (!metadata) return {};\n  return Object.fromEntries(\n    Object.entries(metadata)\n      .filter(([key]) => !LANGGRAPH_METADATA_KEYS.has(key))\n      .map(([key, value]) => [\n        `langwatch.langchain.run.metadata.${key}`,\n        wrapNonScalarValues(value),\n      ])\n  );\n}\n\nfunction applyGenAIAttrs(\n  span: LangWatchSpan,\n  metadata?: Record<string, unknown>,\n  extraParams?: Record<string, unknown>\n) {\n  const md = (metadata ?? {}) as any;\n  const ex = (extraParams ?? {}) as any;\n\n  const provider = md.ls_provider as string | undefined;\n  const requestModel = md.ls_model_name ?? md.kwargs?.model ?? ex.kwargs?.model;\n  const temperature =\n    md.ls_temperature ?? md.kwargs?.temperature ?? ex.kwargs?.temperature;\n  const responseModel = md.response_metadata?.model_name as string | undefined;\n\n  if (provider) span.setAttribute(\"gen_ai.system\", provider);\n  if (requestModel) span.setAttribute(\"gen_ai.request.model\", requestModel);\n  if (typeof temperature === \"number\")\n    span.setAttribute(\"gen_ai.request.temperature\", temperature);\n  if (responseModel) span.setAttribute(\"gen_ai.response.model\", responseModel);\n}\n\nfunction getResolvedParentContext(\n  runId: string | undefined,\n  spans: Record<string, LangWatchSpan | undefined>,\n  parentOf: Record<string, string | undefined>\n) {\n  let cur = runId;\n  while (cur) {\n    const s = spans[cur];\n    if (s) return trace.setSpan(context.active(), s);\n    cur = parentOf[cur];\n  }\n  return context.active();\n}\n\nfunction deriveNameAndType(opts: {\n  runType: RunKind;\n  name?: string;\n  serialized?: Serialized;\n  metadata?: Record<string, unknown>;\n  tags?: string[];\n  inputs?: unknown;\n}): { name: string; type: \"llm\" | \"chain\" | \"tool\" | \"rag\" | \"component\" } {\n  const { runType, name, serialized, metadata, inputs } = opts;\n\n  // user-specified name / metadata override\n  const hardName = (name?.trim() ?? (metadata as any)?.operation_name) as\n    | string\n    | undefined;\n  if (hardName) {\n    return {\n      name: hardName,\n      type:\n        runType === \"tool\"\n          ? \"tool\"\n          : runType === \"retriever\"\n          ? \"rag\"\n          : runType === \"llm\" || runType === \"chat\"\n          ? \"llm\"\n          : \"chain\",\n    };\n  }\n\n  const cls = className(serialized);\n  const md = (metadata ?? {}) as any;\n\n  // LangGraph node / router - prioritize routers over nodes\n  const hasNode = md?.langgraph_node != null;\n  const hasTriggers = Array.isArray(md?.langgraph_triggers) && md.langgraph_triggers.length > 0;\n  const isRouter =\n    cls.startsWith(\"Branch<\") || hasTriggers;\n  const isGraphRunner = md?.langgraph_path && !md?.langgraph_node;\n\n  // LLM / Chat - always prioritize runType over metadata\n  if (runType === \"llm\" || runType === \"chat\") {\n    const prov = (md?.ls_provider as string) ?? \"LLM\";\n    const model = (md?.ls_model_name as string) ?? (cls || \"call\");\n    const temp = md?.ls_temperature;\n    const tempStr =\n      temp != null\n        ? typeof temp === \"number\"\n          ? temp.toString()\n          : JSON.stringify(temp)\n        : null;\n    const nm =\n      tempStr != null\n        ? `${prov} ${model} (temp ${tempStr})`\n        : `${prov} ${model}`;\n    return { name: nm, type: \"llm\" };\n  }\n\n  // Prioritize LangGraph routers over nodes (but after LLM/Chat)\n  if (isRouter) {\n    const pathArr = md?.langgraph_path as string[] | undefined;\n    const fromNode =\n      Array.isArray(pathArr) && pathArr.length\n        ? pathArr[pathArr.length - 1]\n        : undefined;\n    const decision = Array.isArray(md?.langgraph_triggers)\n      ? String(\n          md.langgraph_triggers.find((t: any) =>\n            String(t).startsWith(\"branch:\")\n          ) ?? \"\"\n        ).replace(/^branch:(to:)?/, \"\")\n      : undefined;\n    const nm = `Route: ${fromNode ?? \"unknown\"} → ${decision ?? \"unknown\"}`;\n    return { name: nm, type: \"component\" };\n  }\n\n  if (hasNode) {\n    const step = md?.langgraph_step;\n    const nm = `Node: ${md.langgraph_node}${\n      step != null ? ` (step ${String(step)})` : \"\"\n    }`;\n    return { name: nm, type: \"component\" };\n  }\n  if (isGraphRunner && runType === \"chain\") {\n    return { name: \"Graph: LangGraph\", type: \"chain\" };\n  }\n\n  // Tool\n  if (runType === \"tool\") {\n    const tool = (metadata as any)?.name ?? (cls || \"tool\");\n    const prev =\n      previewInput(inputs) ?? previewInput((serialized as any)?.input);\n    return {\n      name: prev ? `Tool: ${tool} — ${prev}` : `Tool: ${tool}`,\n      type: \"tool\",\n    };\n  }\n\n  // Retriever\n  if (runType === \"retriever\") return { name: \"Retriever\", type: \"rag\" };\n\n  // Fallbacks\n  if (cls.includes(\"Agent\"))\n    return { name: `Agent: ${cls}`, type: \"component\" };\n  if (cls.startsWith(\"Runnable\"))\n    return { name: `Runnable: ${cls.replace(/^Runnable/, \"\")}`, type: \"chain\" };\n  return { name: cls || \"LangChain operation\", type: \"chain\" };\n}\n\n// Export helper functions for testing\nexport {\n  className,\n  shorten,\n  previewInput,\n  ctxSkip,\n  wrapNonScalarValues,\n  addLangChainEvent,\n  setLangGraphAttributes,\n  buildLangChainMetadataAttributes,\n  applyGenAIAttrs,\n  getResolvedParentContext,\n  deriveNameAndType,\n};\n"]}