dd-trace/packages/datadog-instrumentations/src/helpers/rewriter/instrumentations/langchain.js

Datadog APM tracing client for JavaScript

'use strict'

module.exports = [
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/runnables/base.cjs',
    },
    functionQuery: {
      methodName: 'invoke',
      kind: 'Async',
      className: 'RunnableSequence',
    },
    channelName: 'RunnableSequence_invoke',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/runnables/base.cjs',
    },
    functionQuery: {
      methodName: 'batch',
      kind: 'Async',
      className: 'RunnableSequence',
    },
    channelName: 'RunnableSequence_batch',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/language_models/chat_models.cjs',
    },
    functionQuery: {
      methodName: 'generate',
      kind: 'Async',
      className: 'BaseChatModel',
    },
    channelName: 'BaseChatModel_generate',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/language_models/llms.cjs',
    },
    functionQuery: {
      methodName: 'generate',
      kind: 'Async',
    },
    channelName: 'BaseLLM_generate',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/embeddings.cjs',
    },
    functionQuery: {
      methodName: 'embedQuery',
      kind: 'Async',
      className: 'Embeddings',
    },
    channelName: 'Embeddings_embedQuery',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/embeddings.cjs',
    },
    functionQuery: {
      methodName: 'embedDocuments',
      kind: 'Async',
      className: 'Embeddings',
    },
    channelName: 'Embeddings_embedDocuments',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/tools/index.cjs',
    },
    functionQuery: {
      methodName: 'invoke',
      kind: 'Async',
      className: 'StructuredTool',
    },
    channelName: 'Tool_invoke',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/vectorstores.cjs',
    },
    functionQuery: {
      methodName: 'similaritySearch',
      kind: 'Async',
      className: 'VectorStore',
    },
    channelName: 'VectorStore_similaritySearch',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/vectorstores.cjs',
    },
    functionQuery: {
      methodName: 'similaritySearchWithScore',
      kind: 'Async',
      className: 'VectorStore',
    },
    channelName: 'VectorStore_similaritySearchWithScore',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/runnables/base.js',
    },
    functionQuery: {
      methodName: 'invoke',
      kind: 'Async',
      className: 'RunnableSequence',
    },
    channelName: 'RunnableSequence_invoke',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/runnables/base.js',
    },
    functionQuery: {
      methodName: 'batch',
      kind: 'Async',
      className: 'RunnableSequence',
    },
    channelName: 'RunnableSequence_batch',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/language_models/chat_models.js',
    },
    functionQuery: {
      methodName: 'generate',
      kind: 'Async',
      className: 'BaseChatModel',
    },
    channelName: 'BaseChatModel_generate',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/language_models/llms.js',
    },
    functionQuery: {
      methodName: 'generate',
      kind: 'Async',
    },
    channelName: 'BaseLLM_generate',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/embeddings.js',
    },
    functionQuery: {
      methodName: 'embedQuery',
      kind: 'Async',
      className: 'Embeddings',
    },
    channelName: 'Embeddings_embedQuery',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/embeddings.js',
    },
    functionQuery: {
      methodName: 'embedDocuments',
      kind: 'Async',
      className: 'Embeddings',
    },
    channelName: 'Embeddings_embedDocuments',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/tools/index.js',
    },
    functionQuery: {
      methodName: 'invoke',
      kind: 'Async',
      className: 'StructuredTool',
    },
    channelName: 'Tool_invoke',
  },
  {
    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/vectorstores.js',
    },
    functionQuery: {
      methodName: 'similaritySearch',
      kind: 'Async',
      className: 'VectorStore',
    },
    channelName: 'VectorStore_similaritySearch',
  },
  {

    module: {
      name: '@langchain/core',
      versionRange: '>=0.1',
      filePath: 'dist/vectorstores.js',
    },
    functionQuery: {
      methodName: 'similaritySearchWithScore',
      kind: 'Async',
      className: 'VectorStore',
    },
    channelName: 'VectorStore_similaritySearchWithScore',
  },
]