dd-trace/packages/datadog-plugin-langchain/src/handlers/language_models.js

Datadog APM tracing client for JavaScript

'use strict'

const { getSegment } = require('../../../dd-trace/src/util')
const LangChainHandler = require('./default')

class LangChainLanguageModelHandler extends LangChainHandler {
  extractProvider (instance) {
    return typeof instance._llmType === 'function' && getSegment(instance._llmType(), '-', 0)
  }

  extractModel (instance) {
    for (const attr of ['model', 'modelName', 'modelId', 'modelKey', 'repoId']) {
      const modelName = instance[attr]
      if (modelName) return modelName
    }
  }
}

module.exports = LangChainLanguageModelHandler