dd-trace/packages/dd-trace/src/llmobs/plugins/langchain/handlers/chain.js

Datadog APM tracing client for JavaScript

'use strict'

const LangChainLLMObsHandler = require('.')
const { spanHasError } = require('../../../util')

class LangChainLLMObsChainHandler extends LangChainLLMObsHandler {
  setMetaTags ({ span, inputs, results }) {
    let input, output
    if (inputs) {
      input = this.formatIO(inputs)
    }

    if (!results || spanHasError(span)) {
      output = ''
    } else {
      output = this.formatIO(results)
    }

    // chain spans will always be workflows
    this._tagger.tagTextIO(span, input, output)
  }
}

module.exports = LangChainLLMObsChainHandler