langwatch 1.19.0 → 1.20.0

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

@@ -1,20 +1,21 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
-var _chunkO4J3ZPZYjs = require('../../../chunk-O4J3ZPZY.js');
 
+var _chunk5ZNEXSABjs = require('../../../chunk-5ZNEXSAB.js');
 
 
 
-var _chunkLCG5Y2EGjs = require('../../../chunk-LCG5Y2EG.js');
-
-// src/observability-sdk/instrumentation/langchain/index.ts
-var _base = require('@langchain/core/callbacks/base');
+var _chunk5VYEG3RSjs = require('../../../chunk-5VYEG3RS.js');
 
 
 
+var _chunkN76BG37Wjs = require('../../../chunk-N76BG37W.js');
 
+// src/observability-sdk/instrumentation/langchain/index.ts
+var _base = require('@langchain/core/callbacks/base');
 var _api = require('@opentelemetry/api');
+var _core = require('@opentelemetry/core');
 var LANGGRAPH_METADATA_KEYS = /* @__PURE__ */ new Set([
   "thread_id",
   "langgraph_step",
@@ -29,7 +30,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
   constructor() {
     super(...arguments);
     this.name = "LangWatchCallbackHandler";
-    this.tracer = _chunkO4J3ZPZYjs.getLangWatchTracer.call(void 0, "langwatch.instrumentation.langchain");
+    this.tracer = _chunk5ZNEXSABjs.getLangWatchTracer.call(void 0, "langwatch.instrumentation.langchain");
     this.spans = {};
     this.parentOf = {};
     this.skipped = {};
@@ -45,11 +46,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
     }
     if (this.seenStarts.has(runId)) return;
     this.seenStarts.add(runId);
-    const parentCtx = getResolvedParentContext(
-      parentRunId,
-      this.spans,
-      this.parentOf
-    );
+    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({
@@ -64,13 +61,8 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
     span.setType(type);
     if ((_a = args.tags) == null ? void 0 : _a.length)
       span.setAttribute("langwatch.langchain.run.tags", args.tags.slice(0, 50));
-    if (_chunkLCG5Y2EGjs.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 (_chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, ) && args.input !== void 0) {
+      setRunInput(span, args.input);
     }
     if (args.extraParams) {
       span.setAttributes(
@@ -92,19 +84,11 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
   finishRun(runId, end) {
     const span = this.spans[runId];
     if (!span) return;
-    addLangChainEvent(
-      span,
-      end.event,
-      runId,
-      end.parentRunId,
-      end.tags,
-      end.md,
-      end.extra
-    );
+    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 (_chunkLCG5Y2EGjs.shouldCaptureOutput.call(void 0, ) && end.output !== void 0) {
+    } else if (_chunk5VYEG3RSjs.shouldCaptureOutput.call(void 0, ) && end.output !== void 0) {
       span.setOutput(end.output);
     }
     span.end();
@@ -114,7 +98,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
     this.seenStarts.delete(runId);
   }
   async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
-    const input = _chunkLCG5Y2EGjs.shouldCaptureInput.call(void 0, ) && prompts ? {
+    const input = _chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, ) && prompts ? {
       type: "list",
       value: prompts.map((p) => ({ type: "text", value: p }))
     } : void 0;
@@ -131,7 +115,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
     });
   }
   async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
-    const input = _chunkLCG5Y2EGjs.shouldCaptureInput.call(void 0, ) ? {
+    const input = _chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, ) ? {
       type: "chat_messages",
       value: messages.flatMap(convertFromLangChainMessages)
     } : void 0;
@@ -158,11 +142,9 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
         "gen_ai.usage.total_tokens": (_d = tu.totalTokens) != null ? _d : 0
       });
     }
-    const outputs = _chunkLCG5Y2EGjs.shouldCaptureOutput.call(void 0, ) ? response.generations.flat().map((g) => {
+    const outputs = _chunk5VYEG3RSjs.shouldCaptureOutput.call(void 0, ) ? response.generations.flat().map((g) => {
       if ("message" in g && g.message) {
-        return convertFromLangChainMessages([
-          g.message
-        ]);
+        return convertFromLangChainMessages([g.message]);
       } else if ("text" in g && g.text) {
         return g.text;
       }
@@ -186,7 +168,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
       metadata,
       tags,
       name,
-      input: _chunkLCG5Y2EGjs.shouldCaptureInput.call(void 0, ) ? inputs : void 0
+      input: _chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, ) ? inputs : void 0
     });
     if (_runType) {
       const span = this.spans[runId];
@@ -214,7 +196,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
       metadata,
       tags,
       name,
-      input: _chunkLCG5Y2EGjs.shouldCaptureInput.call(void 0, ) ? { type: "text", value: input } : void 0
+      input: _chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, ) ? { type: "text", value: input } : void 0
     });
     const span = this.spans[runId];
     if (span) {
@@ -226,7 +208,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
   }
   async handleToolEnd(output, runId, parentRunId) {
     this.finishRun(runId, {
-      output: _chunkLCG5Y2EGjs.shouldCaptureOutput.call(void 0, ) ? { type: "text", value: output } : void 0,
+      output: _chunk5VYEG3RSjs.shouldCaptureOutput.call(void 0, ) ? { type: "text", value: output } : void 0,
       event: "handleToolEnd",
       parentRunId
     });
@@ -243,7 +225,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
       metadata,
       tags,
       name,
-      input: _chunkLCG5Y2EGjs.shouldCaptureInput.call(void 0, ) ? { type: "text", value: query } : void 0
+      input: _chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, ) ? { type: "text", value: query } : void 0
     });
     const span = this.spans[runId];
     if (span) {
@@ -255,10 +237,10 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
   }
   async handleRetrieverEnd(documents, runId, parentRunId, tags) {
     const span = this.spans[runId];
-    if (span && _chunkLCG5Y2EGjs.shouldCaptureOutput.call(void 0, )) {
+    if (span && _chunk5VYEG3RSjs.shouldCaptureOutput.call(void 0, )) {
       span.setOutput(documents);
     }
-    if (span && _chunkLCG5Y2EGjs.shouldCaptureInput.call(void 0, )) {
+    if (span && _chunk5VYEG3RSjs.shouldCaptureInput.call(void 0, )) {
       span.setRAGContexts(
         documents.map((document) => ({
           document_id: document.metadata.id,
@@ -286,7 +268,7 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
   }
   async handleAgentEnd(action, runId, parentRunId, tags) {
     this.finishRun(runId, {
-      output: _chunkLCG5Y2EGjs.shouldCaptureOutput.call(void 0, ) ? { type: "json", value: action.returnValues } : void 0,
+      output: _chunk5VYEG3RSjs.shouldCaptureOutput.call(void 0, ) ? { type: "json", value: action.returnValues } : void 0,
       event: "handleAgentEnd",
       parentRunId,
       tags
@@ -296,50 +278,84 @@ var LangWatchCallbackHandler = class extends _base.BaseCallbackHandler {
 function convertFromLangChainMessages(messages) {
   const out = [];
   for (const message of messages) {
-    out.push(
-      convertFromLangChainMessage(message)
-    );
+    out.push(convertFromLangChainMessage(message));
   }
   return out;
 }
+function messageRoleOf(message) {
+  const msgType = "type" in message ? message.type : void 0;
+  if (msgType === "human") return "user";
+  if (msgType === "ai") return "assistant";
+  if (msgType === "system") return "system";
+  if (msgType === "function") return "function";
+  if (msgType === "tool") return "tool";
+  return legacyMessageRoleOf(message);
+}
+function legacyMessageRoleOf(message) {
+  var _a, _b;
+  const legacyType = (_a = message._getType) == null ? void 0 : _a.call(message);
+  const lastId = (_b = message.id) == null ? void 0 : _b[message.id.length - 1];
+  if (legacyType === "human" || lastId === "HumanMessage") return "user";
+  if (legacyType === "ai" || lastId === "AIMessage") return "assistant";
+  if (legacyType === "system" || lastId === "SystemMessage") return "system";
+  if (legacyType === "function" || lastId === "FunctionMessage") return "function";
+  if (legacyType === "tool" || lastId === "ToolMessage") return "tool";
+  return "user";
+}
 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 role = messageRoleOf(message);
+  let content;
+  if (typeof message.content === "string") {
+    content = message.content;
+  } else if (message.content == null) {
+    content = null;
+  } else if (Array.isArray(message.content)) {
+    content = message.content.map((c) => {
+      if ((c == null ? void 0 : c.type) === "text") {
+        return { type: "text", text: c.text };
+      }
+      if ((c == null ? void 0 : c.type) === "image_url") {
+        return { type: "image_url", image_url: c.image_url };
+      }
+      return { type: "text", text: JSON.stringify(c) };
+    });
+  } else {
+    content = JSON.stringify(message.content);
   }
-  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 _chunkLCG5Y2EGjs.__spreadValues.call(void 0, {
+  return _chunkN76BG37Wjs.__spreadValues.call(void 0, {
     role,
     content
-  }, functionCall && typeof functionCall === "object" && Object.keys(functionCall).length > 0 ? { function_call: functionCall } : {});
+  }, functionCall && typeof functionCall === "object" && Object.keys(functionCall).length > 0 && isFunctionCallShaped(functionCall) ? { function_call: functionCall } : {});
+}
+function isFunctionCallShaped(kwargs) {
+  const isOptionalString = (value) => value === void 0 || typeof value === "string";
+  return isOptionalString(Reflect.get(kwargs, "name")) && isOptionalString(Reflect.get(kwargs, "arguments"));
+}
+function setRunInput(span, input) {
+  if (input && typeof input === "object" && "type" in input && "value" in input) {
+    span.setAttribute(
+      _chunkN76BG37Wjs.ATTR_LANGWATCH_INPUT,
+      JSON.stringify(_chunk5ZNEXSABjs.processSpanInputOutput.call(void 0, input.type, input.value))
+    );
+    return;
+  }
+  span.setInput(input);
 }
 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;
+  const ns = serialized && "lc_namespace" in serialized ? serialized.lc_namespace : void 0;
   if (Array.isArray(ns) && ns.length) return String(ns[ns.length - 1]);
   return "";
 }
+function displayOf(value) {
+  var _a;
+  if (typeof value === "string") return value;
+  if (typeof value === "number" || typeof value === "boolean" || typeof value === "bigint")
+    return String(value);
+  return (_a = JSON.stringify(value)) != null ? _a : String(value);
+}
 function shorten(str, max = 120) {
   return typeof str === "string" && str.length > max ? str.slice(0, max - 1) + "\u2026" : str;
 }
@@ -360,7 +376,7 @@ function wrapNonScalarValues(value) {
   if (value === null) return JSON.stringify(null);
   if (typeof value === "string" || typeof value === "number" || typeof value === "boolean")
     return value;
-  const chatMessages = _chunkO4J3ZPZYjs.chatMessageSchema.array().safeParse(value);
+  const chatMessages = _chunk5ZNEXSABjs.chatMessageSchema.array().safeParse(value);
   if (Array.isArray(value) && chatMessages.success) {
     return JSON.stringify({ type: "chat_messages", value: chatMessages.data });
   }
@@ -379,7 +395,7 @@ function wrapNonScalarValues(value) {
   }
 }
 function addLangChainEvent(span, eventName, runId, parentRunId, tags, metadata, attributes) {
-  const attrs = _chunkLCG5Y2EGjs.__spreadValues.call(void 0, {
+  const attrs = _chunkN76BG37Wjs.__spreadValues.call(void 0, {
     "langwatch.langchain.run.id": runId,
     "langwatch.langchain.run.parent_id": parentRunId,
     "langwatch.langchain.event.name": eventName
@@ -400,10 +416,7 @@ function setLangGraphAttributes(span, metadata) {
     if (value !== void 0) {
       const wrapped = wrapNonScalarValues(value);
       if (wrapped !== void 0) {
-        span.setAttribute(
-          `langwatch.langgraph.${key}`,
-          wrapped
-        );
+        span.setAttribute(`langwatch.langgraph.${key}`, wrapped);
       }
     }
   }
@@ -418,18 +431,22 @@ function buildLangChainMetadataAttributes(metadata) {
   );
 }
 function applyGenAIAttrs(span, metadata, extraParams) {
-  var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+  var _a, _b, _c, _d;
   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);
+  const requestModel = (_b = (_a = md.ls_model_name) != null ? _a : propertyOf(md.kwargs, "model")) != null ? _b : propertyOf(extraParams == null ? void 0 : extraParams.kwargs, "model");
+  const temperature = (_d = (_c = md.ls_temperature) != null ? _c : propertyOf(md.kwargs, "temperature")) != null ? _d : propertyOf(extraParams == null ? void 0 : extraParams.kwargs, "temperature");
+  const responseModel = propertyOf(md.response_metadata, "model_name");
+  if (provider && _core.isAttributeValue.call(void 0, provider)) span.setAttribute("gen_ai.system", provider);
+  if (requestModel && _core.isAttributeValue.call(void 0, requestModel))
+    span.setAttribute("gen_ai.request.model", requestModel);
+  if (typeof temperature === "number") span.setAttribute("gen_ai.request.temperature", temperature);
+  if (responseModel && _core.isAttributeValue.call(void 0, responseModel))
+    span.setAttribute("gen_ai.response.model", responseModel);
+}
+function propertyOf(value, key) {
+  if (value === null || value === void 0) return void 0;
+  return Reflect.get(Object(value), key);
 }
 function getResolvedParentContext(runId, spans, parentOf) {
   let cur = runId;
@@ -440,14 +457,61 @@ function getResolvedParentContext(runId, spans, parentOf) {
   }
   return _api.context.active();
 }
+function typeFromRunKind(runType) {
+  if (runType === "tool") {
+    return "tool";
+  }
+  if (runType === "retriever") {
+    return "rag";
+  }
+  if (runType === "llm" || runType === "chat") {
+    return "llm";
+  }
+  return "chain";
+}
+function llmRunName({ md, cls }) {
+  var _a, _b;
+  const prov = (_a = md == null ? void 0 : md.ls_provider) != null ? _a : "LLM";
+  const model = (_b = md == null ? void 0 : md.ls_model_name) != null ? _b : cls || "call";
+  const temp = md == null ? void 0 : md.ls_temperature;
+  let tempStr = null;
+  if (typeof temp === "number") {
+    tempStr = temp.toString();
+  } else if (temp != null) {
+    tempStr = JSON.stringify(temp);
+  }
+  return tempStr != null ? `${prov} ${model} (temp ${tempStr})` : `${prov} ${model}`;
+}
+function routerRunName(md) {
+  var _a;
+  const pathArr = md == null ? void 0 : md.langgraph_path;
+  const fromNode = Array.isArray(pathArr) && pathArr.length ? pathArr[pathArr.length - 1] : void 0;
+  const triggers = md == null ? void 0 : md.langgraph_triggers;
+  const decision = Array.isArray(triggers) ? String((_a = triggers.find((t) => String(t).startsWith("branch:"))) != null ? _a : "").replace(
+    /^branch:(to:)?/,
+    ""
+  ) : void 0;
+  return `Route: ${fromNode != null ? fromNode : "unknown"} \u2192 ${decision != null ? decision : "unknown"}`;
+}
+function toolRunName({
+  metadata,
+  cls,
+  inputs,
+  serialized
+}) {
+  var _a, _b;
+  const tool = displayOf((_a = metadata == null ? void 0 : metadata.name) != null ? _a : cls || "tool");
+  const prev = (_b = previewInput(inputs)) != null ? _b : previewInput(serialized && "input" in serialized ? serialized.input : void 0);
+  return prev ? `Tool: ${tool} \u2014 ${prev}` : `Tool: ${tool}`;
+}
 function deriveNameAndType(opts) {
-  var _a, _b, _c, _d, _e, _f;
+  var _a;
   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"
+      type: typeFromRunKind(runType)
     };
   }
   const cls = className(serialized);
@@ -457,43 +521,27 @@ function deriveNameAndType(opts) {
   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" };
+    return { name: llmRunName({ md, cls }), 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" };
+    return { name: routerRunName(md), type: "component" };
   }
   if (hasNode) {
     const step = md == null ? void 0 : md.langgraph_step;
-    const nm = `Node: ${md.langgraph_node}${step != null ? ` (step ${String(step)})` : ""}`;
+    const nm = `Node: ${displayOf(md.langgraph_node)}${step != null ? ` (step ${displayOf(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"
-    };
+    return { name: toolRunName({ metadata, cls, inputs, serialized }), type: "tool" };
   }
+  return fallbackNameAndType({ runType, cls });
+}
+function fallbackNameAndType({ runType, cls }) {
   if (runType === "retriever") return { name: "Retriever", type: "rag" };
-  if (cls.includes("Agent"))
-    return { name: `Agent: ${cls}`, type: "component" };
+  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" };

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

@@ -1 +1 @@
-{"version":3,"sources":["/home/runner/work/langwatch/langwatch/sdks/typescript/dist/observability-sdk/instrumentation/langchain/index.js","../../../../src/observability-sdk/instrumentation/langchain/index.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,6DAAmC;AACnC;AACE;AACA;AACA;AACF,6DAAmC;AACnC;AACA;ACTA,sDAAoC;AAapC;AACE;AACA;AACA;AAAA,yCAEK;AAiBP,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;AD3TsD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/langwatch/langwatch/sdks/typescript/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 {\n  chatMessageSchema,\n  type ChatMessage,\n  type 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 { 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"]}
+{"version":3,"sources":["/home/runner/work/langwatch/langwatch/sdks/typescript/dist/observability-sdk/instrumentation/langchain/index.js","../../../../src/observability-sdk/instrumentation/langchain/index.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACF,6DAAmC;AACnC;AACE;AACA;AACF,6DAAmC;AACnC;AACE;AACA;AACF,6DAAmC;AACnC;AACA;ACbA,sDAAoC;AAMpC,yCAAgE;AAChE,2CAAiC;AA0BjC,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;AAtDxC,IAAA,IAAA,EAAA;AAuDI,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,CAAyB,WAAA,EAAa,IAAA,CAAK,KAAA,EAAO,IAAA,CAAK,QAAQ,CAAA;AACjF,IAAA,MAAM,WAAA,EAAa,YAAA,EAAc,IAAA,CAAK,KAAA,CAAM,WAAW,EAAA,EAAI,KAAA,CAAA;AAC3D,IAAA,MAAM,MAAA,EAAQ,WAAA,EAAa,CAAC,EAAE,OAAA,EAAS,UAAA,CAAW,WAAA,CAAY,EAAE,CAAC,EAAA,EAAI,KAAA,CAAA;AAErE,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,WAAA,CAAY,IAAA,EAAM,IAAA,CAAK,KAAK,CAAA;AAAA,IAC9B;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;AAEmC,IAAA;AAEjC,IAAA;AACiB,MAAA;AACiB,MAAA;AACP,IAAA;AACb,MAAA;AAC3B,IAAA;AAES,IAAA;AAEc,IAAA;AACG,IAAA;AACD,IAAA;AACG,IAAA;AAC9B,EAAA;AAME,EAAA;AAQM,IAAA;AACQ,MAAA;AACqC,MAAA;AAE7C,IAAA;AAEY,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;AAE4F,EAAA;AA1M9F,IAAA;AA2MiC,IAAA;AAClB,IAAA;AAOK,IAAA;AACK,MAAA;AACgB,QAAA;AACI,QAAA;AACL,QAAA;AACjC,MAAA;AACH,IAAA;AAGa,IAAA;AAC0B,MAAA;AACmC,QAAA;AAClC,MAAA;AACvB,QAAA;AACX,MAAA;AACO,MAAA;AAET,IAAA;AAEkB,IAAA;AACZ,MAAA;AACD,MAAA;AACP,MAAA;AACD,IAAA;AACH,EAAA;AAEqF,EAAA;AAC/C,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;AAE8F,EAAA;AACrD,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;AAEwF,EAAA;AAChE,IAAA;AAC4B,MAAA;AACzC,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;AAE4E,EAAA;AACpD,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;AAC4B,MAAA;AACzC,MAAA;AACP,MAAA;AACA,MAAA;AACD,IAAA;AACH,EAAA;AACF;AAEqF;AACvD,EAAA;AACI,EAAA;AACkD,IAAA;AAClF,EAAA;AACO,EAAA;AACT;AAEsF;AACjC,EAAA;AACnB,EAAA;AACH,EAAA;AACI,EAAA;AACE,EAAA;AACJ,EAAA;AACG,EAAA;AACpC;AAE4F;AA3b5F,EAAA;AA4bqB,EAAA;AACJ,EAAA;AAC0B,EAAA;AACU,EAAA;AACT,EAAA;AACE,EAAA;AACJ,EAAA;AACjC,EAAA;AACT;AAE4F;AACxD,EAAA;AAE9B,EAAA;AACqC,EAAA;AACrB,IAAA;AACgB,EAAA;AACxB,IAAA;AAC+B,EAAA;AACa,IAAA;AAC5B,MAAA;AACc,QAAA;AACtC,MAAA;AACgB,MAAA;AAC2B,QAAA;AAC3C,MAAA;AAC+C,MAAA;AAChD,IAAA;AACI,EAAA;AACmC,IAAA;AAC1C,EAAA;AAE6B,EAAA;AAEtB,EAAA;AACL,IAAA;AACA,IAAA;AAEwB,EAAA;AAM5B;AAKuD;AACE,EAAA;AAET,EAAA;AAGhD;AAG0D;AACJ,EAAA;AAC7C,IAAA;AACH,MAAA;AAC4C,MAAA;AAC9C,IAAA;AACA,IAAA;AACF,EAAA;AACmB,EAAA;AACrB;AAEoD;AAC3B,EAAA;AAC2B,EAAA;AACP,EAAA;AACO,EAAA;AAE3C,EAAA;AACT;AAE2C;AA1gB3C,EAAA;AA2gBwC,EAAA;AACY,EAAA;AAC7B,IAAA;AACW,EAAA;AAClC;AAEiD;AACA,EAAA;AACjD;AAEsD;AACzB,EAAA;AACR,IAAA;AACY,IAAA;AAC/B,EAAA;AACO,EAAA;AACT;AAE2D;AA7hB3D,EAAA;AA8hBkC,EAAA;AACU,EAAA;AAC5C;AAEoF;AACrD,EAAA;AACiB,EAAA;AACI,EAAA;AACzC,IAAA;AAGsC,EAAA;AACG,EAAA;AACD,IAAA;AACjD,EAAA;AAEI,EAAA;AACuB,IAAA;AACsB,IAAA;AACA,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;AAEyF;AACxE,EAAA;AACkB,EAAA;AACT,EAAA;AACI,IAAA;AACD,IAAA;AACkB,MAAA;AACd,MAAA;AACmB,QAAA;AAC9C,MAAA;AACF,IAAA;AACF,EAAA;AACF;AAE6E;AACpD,EAAA;AACT,EAAA;AAEU,IAAA;AAEqB,MAAA;AACd,MAAA;AAC1B,IAAA;AACL,EAAA;AACF;AAME;AArnBF,EAAA;AAsnB0B,EAAA;AAEJ,EAAA;AAEf,EAAA;AAEA,EAAA;AAG+B,EAAA;AAEa,EAAA;AACA,EAAA;AACL,IAAA;AACF,EAAA;AACS,EAAA;AACN,IAAA;AAC/C;AAE0D;AACN,EAAA;AACb,EAAA;AACvC;AAKE;AAEU,EAAA;AACE,EAAA;AACS,IAAA;AAC4B,IAAA;AAC7B,IAAA;AACpB,EAAA;AACsB,EAAA;AACxB;AAE2F;AACjE,EAAA;AACf,IAAA;AACT,EAAA;AAC6B,EAAA;AACpB,IAAA;AACT,EAAA;AAC6C,EAAA;AACpC,IAAA;AACT,EAAA;AACO,EAAA;AACT;AAEuF;AAzqBvF,EAAA;AA0qBoB,EAAA;AACC,EAAA;AACF,EAAA;AACY,EAAA;AACC,EAAA;AACJ,IAAA;AACD,EAAA;AACM,IAAA;AAC/B,EAAA;AACmD,EAAA;AACrD;AAE4D;AAtrB5D,EAAA;AAurBsB,EAAA;AAEgB,EAAA;AACf,EAAA;AAEV,EAAA;AACL,IAAA;AACA,IAAA;AAEF,EAAA;AACyB,EAAA;AAC/B;AAEqB;AACnB,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AAMS;AA9sBX,EAAA;AA+sByB,EAAA;AAErB,EAAA;AAEwC,EAAA;AAC5C;AAS2E;AA7tB3E,EAAA;AA8tBsD,EAAA;AAG5B,EAAA;AACV,EAAA;AACL,IAAA;AACC,MAAA;AACuB,MAAA;AAC/B,IAAA;AACF,EAAA;AAEgC,EAAA;AACR,EAAA;AAGJ,EAAA;AACc,EAAA;AACY,EAAA;AACpB,EAAA;AAGmB,EAAA;AACG,IAAA;AAChD,EAAA;AAGc,EAAA;AAC4B,IAAA;AAC1C,EAAA;AAEa,EAAA;AACM,IAAA;AAC+B,IAAA;AACX,IAAA;AACvC,EAAA;AAC0C,EAAA;AACS,IAAA;AACnD,EAAA;AAGwB,EAAA;AACsB,IAAA;AAC9C,EAAA;AAE2C,EAAA;AAC7C;AAKE;AAC4C,EAAA;AAEQ,EAAA;AACvB,EAAA;AACa,IAAA;AACS,EAAA;AACrD;ADnPsD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/langwatch/langwatch/sdks/typescript/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 { ChainValues } from \"@langchain/core/utils/types\";\nimport { context, trace, SpanStatusCode, type Attributes } from \"@opentelemetry/api\";\nimport { isAttributeValue } from \"@opentelemetry/core\";\n\nimport {\n  chatMessageSchema,\n  type ChatMessage,\n  type ChatRichContent,\n} from \"../../../internal/generated/types/tracer\";\nimport { shouldCaptureInput, shouldCaptureOutput } from \"../../config\";\nimport { ATTR_LANGWATCH_INPUT } from \"../../semconv/attributes\";\nimport { processSpanInputOutput, type LangWatchSpan } from \"../../span\";\nimport { getLangWatchTracer } from \"../../tracer\";\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(parentRunId, this.spans, this.parentOf);\n    const parentSpan = parentRunId ? this.spans[parentRunId] : void 0;\n    const links = parentSpan ? [{ context: parentSpan.spanContext() }] : 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      setRunInput(span, args.input);\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(span, end.event, runId, end.parentRunId, end.tags, end.md, end.extra);\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);\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 =\n      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(response: LLMResult, runId: string, parentRunId?: string): Promise<void> {\n    const span = this.spans[runId];\n    const tu = response.llmOutput?.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([(g as ChatGeneration).message]);\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(err: Error, runId: string, parentRunId?: string): 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(output: ChainValues, runId: string, parentRunId?: string): 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(output: string, runId: string, parentRunId?: string): Promise<void> {\n    this.finishRun(runId, {\n      output: shouldCaptureOutput() ? { type: \"text\", value: output } : 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[],\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(err: Error, runId: string, parentRunId?: string, tags?: string[]) {\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() ? { type: \"json\", value: action.returnValues } : void 0,\n      event: \"handleAgentEnd\",\n      parentRunId,\n      tags,\n    });\n  }\n}\n\nexport function convertFromLangChainMessages(messages: BaseMessage[]): ChatMessage[] {\n  const out: ChatMessage[] = [];\n  for (const message of messages) {\n    out.push(convertFromLangChainMessage(message as BaseMessage & { id?: string[] }));\n  }\n  return out;\n}\n\nfunction messageRoleOf(message: BaseMessage & { id?: string[] }): ChatMessage[\"role\"] {\n  const msgType = \"type\" in message ? message.type : undefined;\n  if (msgType === \"human\") return \"user\";\n  if (msgType === \"ai\") return \"assistant\";\n  if (msgType === \"system\") return \"system\";\n  if (msgType === \"function\") return \"function\";\n  if (msgType === \"tool\") return \"tool\";\n  return legacyMessageRoleOf(message);\n}\n\nfunction legacyMessageRoleOf(message: BaseMessage & { id?: string[] }): ChatMessage[\"role\"] {\n  const legacyType = message._getType?.();\n  const lastId = message.id?.[message.id.length - 1];\n  if (legacyType === \"human\" || lastId === \"HumanMessage\") return \"user\";\n  if (legacyType === \"ai\" || lastId === \"AIMessage\") return \"assistant\";\n  if (legacyType === \"system\" || lastId === \"SystemMessage\") return \"system\";\n  if (legacyType === \"function\" || lastId === \"FunctionMessage\") return \"function\";\n  if (legacyType === \"tool\" || lastId === \"ToolMessage\") return \"tool\";\n  return \"user\";\n}\n\nfunction convertFromLangChainMessage(message: BaseMessage & { id?: string[] }): ChatMessage {\n  const role = messageRoleOf(message);\n\n  let content: ChatMessage[\"content\"];\n  if (typeof message.content === \"string\") {\n    content = message.content;\n  } else if (message.content == null) {\n    content = null;\n  } else if (Array.isArray(message.content)) {\n    content = message.content.map((c): ChatRichContent => {\n      if (c?.type === \"text\") {\n        return { type: \"text\", text: c.text };\n      }\n      if (c?.type === \"image_url\") {\n        return { type: \"image_url\", image_url: c.image_url };\n      }\n      return { type: \"text\", text: JSON.stringify(c) };\n    });\n  } else {\n    content = JSON.stringify(message.content);\n  }\n\n  const functionCall = message.additional_kwargs;\n\n  return {\n    role,\n    content,\n    ...(functionCall &&\n    typeof functionCall === \"object\" &&\n    Object.keys(functionCall).length > 0 &&\n    isFunctionCallShaped(functionCall)\n      ? { function_call: functionCall }\n      : {}),\n  };\n}\n\n/** All of `additional_kwargs` travels as `function_call`; its name/arguments must be strings. */\nfunction isFunctionCallShaped<T extends object>(\n  kwargs: T,\n): kwargs is T & { name?: string; arguments?: string } {\n  const isOptionalString = (value: unknown) => value === undefined || typeof value === \"string\";\n  return (\n    isOptionalString(Reflect.get(kwargs, \"name\")) &&\n    isOptionalString(Reflect.get(kwargs, \"arguments\"))\n  );\n}\n\n/** A `{ type, value }` input keeps its declared type, as `span.setInput(type, value)` would. */\nfunction setRunInput(span: LangWatchSpan, input: unknown) {\n  if (input && typeof input === \"object\" && \"type\" in input && \"value\" in input) {\n    span.setAttribute(\n      ATTR_LANGWATCH_INPUT,\n      JSON.stringify(processSpanInputOutput(input.type, input.value)),\n    );\n    return;\n  }\n  span.setInput(input);\n}\n\nfunction className(serialized?: Serialized): string {\n  const id = serialized?.id;\n  if (Array.isArray(id) && id.length) return String(id[id.length - 1]);\n  const ns = serialized && \"lc_namespace\" in serialized ? serialized.lc_namespace : undefined;\n  if (Array.isArray(ns) && ns.length) return String(ns[ns.length - 1]);\n\n  return \"\";\n}\n\nfunction displayOf(value: unknown): string {\n  if (typeof value === \"string\") return value;\n  if (typeof value === \"number\" || typeof value === \"boolean\" || typeof value === \"bigint\")\n    return String(value);\n  return JSON.stringify(value) ?? String(value);\n}\n\nfunction shorten(str: string, max = 120): string {\n  return typeof str === \"string\" && str.length > max ? str.slice(0, max - 1) + \"…\" : 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 cls.startsWith(\"ChannelWrite\") || (tags?.includes(\"langsmith:hidden\") ?? false);\n}\n\nfunction wrapNonScalarValues(value: unknown): string | number | boolean | undefined {\n  if (value === void 0) return void 0;\n  if (value === null) return JSON.stringify(null);\n  if (typeof value === \"string\" || typeof value === \"number\" || typeof value === \"boolean\")\n    return value;\n\n  // Special-case: ChatMessage[] detection via zod schema the project already has\n  const chatMessages = chatMessageSchema.array().safeParse(value);\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, (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(span: LangWatchSpan, metadata?: Record<string, unknown>) {\n  if (!metadata) return;\n  const keys = Object.keys(metadata);\n  for (const key of keys) {\n    const value = metadata[key];\n    if (value !== undefined) {\n      const wrapped = wrapNonScalarValues(value);\n      if (wrapped !== undefined) {\n        span.setAttribute(`langwatch.langgraph.${key}` as const, wrapped);\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 ?? {};\n\n  const provider = md.ls_provider;\n  const requestModel =\n    md.ls_model_name ?? propertyOf(md.kwargs, \"model\") ?? propertyOf(extraParams?.kwargs, \"model\");\n  const temperature =\n    md.ls_temperature ??\n    propertyOf(md.kwargs, \"temperature\") ??\n    propertyOf(extraParams?.kwargs, \"temperature\");\n  const responseModel = propertyOf(md.response_metadata, \"model_name\");\n\n  if (provider && isAttributeValue(provider)) span.setAttribute(\"gen_ai.system\", provider);\n  if (requestModel && isAttributeValue(requestModel))\n    span.setAttribute(\"gen_ai.request.model\", requestModel);\n  if (typeof temperature === \"number\") span.setAttribute(\"gen_ai.request.temperature\", temperature);\n  if (responseModel && isAttributeValue(responseModel))\n    span.setAttribute(\"gen_ai.response.model\", responseModel);\n}\n\nfunction propertyOf(value: unknown, key: string): unknown {\n  if (value === null || value === undefined) return undefined;\n  return Reflect.get(Object(value), key);\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 typeFromRunKind(runType: RunKind): \"llm\" | \"chain\" | \"tool\" | \"rag\" | \"component\" {\n  if (runType === \"tool\") {\n    return \"tool\";\n  }\n  if (runType === \"retriever\") {\n    return \"rag\";\n  }\n  if (runType === \"llm\" || runType === \"chat\") {\n    return \"llm\";\n  }\n  return \"chain\";\n}\n\nfunction llmRunName({ md, cls }: { md: Record<string, unknown>; cls: string }): string {\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  let tempStr: string | null = null;\n  if (typeof temp === \"number\") {\n    tempStr = temp.toString();\n  } else if (temp != null) {\n    tempStr = JSON.stringify(temp);\n  }\n  return tempStr != null ? `${prov} ${model} (temp ${tempStr})` : `${prov} ${model}`;\n}\n\nfunction routerRunName(md: Record<string, unknown>): string {\n  const pathArr = md?.langgraph_path as string[] | undefined;\n  const fromNode =\n    Array.isArray(pathArr) && pathArr.length ? pathArr[pathArr.length - 1] : undefined;\n  const triggers = md?.langgraph_triggers;\n  const decision = Array.isArray(triggers)\n    ? String(triggers.find((t) => String(t).startsWith(\"branch:\")) ?? \"\").replace(\n        /^branch:(to:)?/,\n        \"\",\n      )\n    : undefined;\n  return `Route: ${fromNode ?? \"unknown\"} → ${decision ?? \"unknown\"}`;\n}\n\nfunction toolRunName({\n  metadata,\n  cls,\n  inputs,\n  serialized,\n}: {\n  metadata?: Record<string, unknown>;\n  cls: string;\n  inputs?: unknown;\n  serialized?: Serialized;\n}): string {\n  const tool = displayOf(metadata?.name ?? (cls || \"tool\"));\n  const prev =\n    previewInput(inputs) ??\n    previewInput(serialized && \"input\" in serialized ? serialized.input : undefined);\n  return prev ? `Tool: ${tool} — ${prev}` : `Tool: ${tool}`;\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?.operation_name) as string | undefined;\n  if (hardName) {\n    return {\n      name: hardName,\n      type: typeFromRunKind(runType),\n    };\n  }\n\n  const cls = className(serialized);\n  const md = metadata ?? {};\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 = 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    return { name: llmRunName({ md, cls }), type: \"llm\" };\n  }\n\n  // Prioritize LangGraph routers over nodes (but after LLM/Chat)\n  if (isRouter) {\n    return { name: routerRunName(md), type: \"component\" };\n  }\n\n  if (hasNode) {\n    const step = md?.langgraph_step;\n    const nm = `Node: ${displayOf(md.langgraph_node)}${step != null ? ` (step ${displayOf(step)})` : \"\"}`;\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    return { name: toolRunName({ metadata, cls, inputs, serialized }), type: \"tool\" };\n  }\n\n  return fallbackNameAndType({ runType, cls });\n}\n\nfunction fallbackNameAndType({ runType, cls }: { runType: RunKind; cls: string }): {\n  name: string;\n  type: \"llm\" | \"chain\" | \"tool\" | \"rag\" | \"component\";\n} {\n  if (runType === \"retriever\") return { name: \"Retriever\", type: \"rag\" };\n\n  if (cls.includes(\"Agent\")) 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"]}