dd-trace/packages/datadog-plugin-langchain/src/index.js

Datadog APM tracing client for JavaScript

'use strict'

const langChainTracingPlugins = require('./tracing')
const langChainLLMObsPlugins = require('../../dd-trace/src/llmobs/plugins/langchain')
const CompositePlugin = require('../../dd-trace/src/plugins/composite')

const plugins = {}

// ordering here is important - the llm observability plugin must come first
// so that we can add annotations associated with the span before it finishes.
// however, because the tracing plugin uses `bindStart` vs the llmobs' `start`,
// the span is guaranteed to be created in the tracing plugin before the llmobs one is called
for (const Plugin of langChainLLMObsPlugins) {
  plugins[Plugin.id] = Plugin
}

for (const Plugin of langChainTracingPlugins) {
  plugins[Plugin.id] = Plugin
}

class LangChainPlugin extends CompositePlugin {
  static id = 'langchain'
  static plugins = plugins
}

module.exports = LangChainPlugin