Runnables subscriber

Author: amychisholm03

lib/instrumentation/langchain/callback-manager.js

@@ -376,10 +376,11 @@ class Subscriber {
   subscribe() {
     if (!this.events || !Array.isArray(this.events)) {
       this.logger.debug('Subscriber does not have valid events to subscribe to.')
+      return
     }
     this.subscriptions = this.events.reduce((events, curr) => {
       try { events[curr] = this[curr].bind(this) } catch {
-        this.logger.debug(`Failed to bind subscriber event '${curr}'.`)
+        this.logger.debug(`Failed to bind subscriber event '${curr}'. Is there a handler for this event?`)
       }
       return events
     }, {})
@@ -388,7 +389,7 @@ class Subscriber {
   }
 
   /**
-   * Unsubscribes from the events defined in the `events` array..
+   * Unsubscribes from the events defined in the `events` array.
    */
   unsubscribe() {
     this.channel.unsubscribe(this.subscriptions)

lib/instrumentation/langchain/nr-hooks.js

@@ -17,7 +17,7 @@ class LangchainSubscriber extends Subscriber {
 
   get enabled() {
     if (!this.agent?.config?.ai_monitoring?.enabled) {
-      this.logger.debug('langchain instrumentation is disabled.  To enable set `config.ai_monitoring.enabled` to true')
+      this.logger.debug('Langchain instrumentation is disabled. To enable, set `config.ai_monitoring.enabled` to true.')
     }
     return super.enabled && this.agent?.config?.ai_monitoring?.enabled
   }

lib/instrumentation/langchain/tools.js

@@ -83,6 +83,7 @@ const chainCallbackManager = {
       channelName: 'nr_handleChainStart',
       module: { name: '@langchain/core', versionRange: '>=1.0.0', filePath: 'dist/callbacks/manager.cjs' },
       functionQuery: {
+        className: 'CallbackManager',
         methodName: 'handleChainStart',
         kind: 'Async'
       }
@@ -92,13 +93,46 @@ const chainCallbackManager = {
       channelName: 'nr_handleChainStart',
       module: { name: '@langchain/core', versionRange: '>=1.0.0', filePath: 'dist/callbacks/manager.js' },
       functionQuery: {
+        className: 'CallbackManager',
         methodName: 'handleChainStart',
         kind: 'Async'
       }
     },
   ]
 }
 
+const runnableInvoke = {
+  path: './langchain/runnable.js',
+  instrumentations: [
+    // CommonJs
+    {
+      channelName: 'nr_invoke',
+      module: { name: '@langchain/core', versionRange: '>=1.0.0', filePath: 'dist/runnables/base.cjs' },
+      functionQuery: {
+        className: 'RunnableSequence',
+        methodName: 'invoke',
+        kind: 'Async'
+      }
+    },
+    // ESM
+    {
+      channelName: 'nr_invoke',
+      module: { name: '@langchain/core', versionRange: '>=1.0.0', filePath: 'dist/runnables/base.js' },
+      functionQuery: {
+        className: 'RunnableSequence',
+        methodName: 'invoke',
+        kind: 'Async'
+      }
+    },
+  ]
+}
+
 module.exports = {
-  '@langchain/core': [similaritySearch, toolCall, toolCallbackManager, chainCallbackManager]
+  '@langchain/core': [
+    similaritySearch,
+    toolCall,
+    toolCallbackManager,
+    chainCallbackManager,
+    runnableInvoke
+  ]
 }