feat(@langchain/langgraph): initial integration support

.github/workflows/apm-integrations.yml

@@ -671,6 +671,17 @@ jobs:
         with:
           dd_api_key: ${{ secrets.DD_API_KEY }}
 
+  langgraph:
+    runs-on: ubuntu-latest
+    env:
+      PLUGINS: langgraph
+    steps:
+      - uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
+      - uses: ./.github/actions/plugins/test
+        with:
+          dd_api_key: ${{ secrets.DD_API_KEY }}
+
+
   ldapjs:
     runs-on: 'ubuntu-latest'
     env:

docs/API.md

@@ -67,6 +67,7 @@ tracer.use('pg', {
 <h5 id="knex"></h5>
 <h5 id="koa"></h5>
 <h5 id="langchain"></h5>
+<h5 id="langgraph"></h5>
 <h5 id="mariadb"></h5>
 <h5 id="memcached"></h5>
 <h5 id="microgateway-core"></h5>
@@ -145,6 +146,7 @@ tracer.use('pg', {
 * [knex](./interfaces/export_.plugins.knex.html)
 * [koa](./interfaces/export_.plugins.koa.html)
 * [langchain](./interfaces/export_.plugins.langchain.html)
+* [langgraph](./interfaces/export_.plugins.langgraph.html)
 * [mariadb](./interfaces/export_.plugins.mariadb.html)
 * [memcached](./interfaces/export_.plugins.memcached.html)
 * [microgateway-core](./interfaces/export_.plugins.microgateway_core.html)

docs/add-redirects.sh

@@ -53,6 +53,7 @@ declare -a plugins=(
   "knex"
   "koa"
   "langchain"
+  "langgraph"
   "ldapjs"
   "mariadb"
   "memcached"

docs/test.ts

@@ -377,6 +377,7 @@ tracer.use('koa');
 tracer.use('koa', httpServerOptions);
 tracer.use('langchain');
 tracer.use('mariadb', { service: () => `my-custom-mariadb` })
+tracer.use('langgraph');
 tracer.use('memcached');
 tracer.use('microgateway-core');
 tracer.use('microgateway-core', httpServerOptions);

index.d.ts

@@ -260,6 +260,7 @@ interface Plugins {
   "knex": tracer.plugins.knex;
   "koa": tracer.plugins.koa;
   "langchain": tracer.plugins.langchain;
+  "langgraph": tracer.plugins.langgraph;
   "mariadb": tracer.plugins.mariadb;
   "memcached": tracer.plugins.memcached;
   "microgateway-core": tracer.plugins.microgateway_core;
@@ -2268,6 +2269,12 @@ declare namespace tracer {
      * This plugin automatically instruments the
      * [ldapjs](https://github.com/ldapjs/node-ldapjs/) module.
      */
+    /**
+     * This plugin automatically instruments the
+     * [langgraph](https://github.com/npmjs/package/langgraph) library.
+     */
+    interface langgraph extends Instrumentation {}
+
     interface ldapjs extends Instrumentation {}
 
     /**

packages/datadog-instrumentations/src/helpers/hooks.js

@@ -4,6 +4,7 @@ module.exports = {
   '@anthropic-ai/sdk': { esmFirst: true, fn: () => require('../anthropic') },
   '@apollo/server': () => require('../apollo-server'),
   '@apollo/gateway': () => require('../apollo'),
+  '@langchain/langgraph': { esmFirst: true, fn: () => require('../langgraph') },
   'apollo-server-core': () => require('../apollo-server-core'),
   '@aws-sdk/smithy-client': () => require('../aws-sdk'),
   '@azure/event-hubs': () => require('../azure-event-hubs'),

packages/datadog-instrumentations/src/helpers/rewriter/instrumentations/index.js

@@ -3,4 +3,5 @@
 module.exports = [
   ...require('./langchain'),
   ...require('./bullmq'),
+  ...require('./langgraph'),
 ]

packages/datadog-instrumentations/src/helpers/rewriter/instrumentations/langgraph.js

@@ -0,0 +1,56 @@
+'use strict'
+
+module.exports = [
+  {
+    module: {
+      name: '@langchain/langgraph',
+      versionRange: '>=1.1.2',
+      filePath: 'dist/pregel/index.js',
+    },
+    functionQuery: {
+      methodName: 'invoke',
+      className: 'Pregel',
+      kind: 'Async',
+    },
+    channelName: 'Pregel_invoke',
+  },
+  {
+    module: {
+      name: '@langchain/langgraph',
+      versionRange: '>=1.1.2',
+      filePath: 'dist/pregel/index.cjs',
+    },
+    functionQuery: {
+      methodName: 'invoke',
+      className: 'Pregel',
+      kind: 'Async',
+    },
+    channelName: 'Pregel_invoke',
+  },
+  {
+    module: {
+      name: '@langchain/langgraph',
+      versionRange: '>=1.1.2',
+      filePath: 'dist/pregel/index.js',
+    },
+    functionQuery: {
+      methodName: 'stream',
+      className: 'Pregel',
+      kind: 'Async',
+    },
+    channelName: 'Pregel_stream',
+  },
+  {
+    module: {
+      name: '@langchain/langgraph',
+      versionRange: '>=1.1.2',
+      filePath: 'dist/pregel/index.cjs',
+    },
+    functionQuery: {
+      methodName: 'stream',
+      className: 'Pregel',
+      kind: 'Async',
+    },
+    channelName: 'Pregel_stream',
+  },
+]

packages/datadog-instrumentations/src/langgraph.js

@@ -0,0 +1,7 @@
+'use strict'
+
+const { addHook, getHooks } = require('./helpers/instrument')
+
+for (const hook of getHooks('@langchain/langgraph')) {
+  addHook(hook, exports => exports)
+}

packages/datadog-plugin-langgraph/src/index.js

@@ -0,0 +1,24 @@
+'use strict'
+
+const CompositePlugin = require('../../dd-trace/src/plugins/composite')
+const langgraphLLMObsPlugins = require('../../dd-trace/src/llmobs/plugins/langgraph')
+const internalPlugin = require('./internal')
+
+const plugins = {}
+
+// CRITICAL: LLMObs plugins MUST come first
+for (const Plugin of langgraphLLMObsPlugins) {
+  plugins[Plugin.id] = Plugin
+}
+
+// Tracing plugins second
+for (const [id, Plugin] of Object.entries(internalPlugin)) {
+  plugins[id] = Plugin
+}
+
+class LanggraphPlugin extends CompositePlugin {
+  static id = 'langgraph'
+  static plugins = plugins
+}
+
+module.exports = LanggraphPlugin

packages/datadog-plugin-langgraph/src/internal.js

@@ -0,0 +1,99 @@
+'use strict'
+
+const TracingPlugin = require('../../dd-trace/src/plugins/tracing')
+
+class BaseLanggraphInternalPlugin extends TracingPlugin {
+  static id = 'langgraph'
+  static prefix = 'tracing:orchestrion:@langchain/langgraph:Pregel_invoke'
+
+  bindStart (ctx) {
+    const meta = this.getTags(ctx)
+
+    this.startSpan('langgraph.invoke', {
+      service: this.config.service,
+      meta,
+    }, ctx)
+
+    return ctx.currentStore
+  }
+
+  getTags (ctx) {
+    return {
+      component: 'langgraph',
+      'span.kind': 'internal',
+    }
+  }
+
+  asyncEnd (ctx) {
+    this.finish(ctx)
+  }
+}
+
+class PregelStreamPlugin extends TracingPlugin {
+  static prefix = 'tracing:orchestrion:@langchain/langgraph:Pregel_stream'
+
+  bindStart (ctx) {
+    this.startSpan('langgraph.stream', {
+      service: this.config.service,
+      component: 'langgraph',
+      'span.kind': 'internal',
+    }, ctx)
+
+    return ctx.currentStore
+  }
+
+  asyncStart (ctx) {
+    const span = ctx.currentStore?.span
+    if (!span) {
+      return
+    }
+
+    const asyncIterable = ctx.result
+
+    const originalAsyncIterator = asyncIterable[Symbol.asyncIterator].bind(asyncIterable)
+
+    asyncIterable[Symbol.asyncIterator] = function () {
+      const originalIterator = originalAsyncIterator()
+
+      return {
+        async next (...args) {
+          try {
+            const result = await originalIterator.next(...args)
+
+            if (result.done) span.finish()
+
+            return result
+          } catch (error) {
+            span.setTag('error', error)
+            span.finish()
+            throw error
+          }
+        },
+
+        async return (...args) {
+          span.finish()
+
+          if (originalIterator.return) {
+            return await originalIterator.return(...args)
+          }
+          return { done: true, value: undefined }
+        },
+
+        async throw (error) {
+          span.setTag('error', error)
+          span.finish()
+
+          if (originalIterator.throw) {
+            return await originalIterator.throw(error)
+          }
+          throw error
+        },
+      }
+    }
+  }
+}
+
+module.exports = {
+  BaseLanggraphInternalPlugin,
+  PregelStreamPlugin,
+}