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
    if (inputs) {
      input = this.formatIO(inputs)
    }

    const output = !results || spanHasError(span) ? '' : this.formatIO(results)

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

module.exports = LangChainLLMObsChainHandler