Remove useless invoke subscriber

Author: amychisholm03

lib/subscribers/langgraph/config.js

@@ -6,32 +6,6 @@
 // `Pregel` is the base class of the `CompiledStateGraph` (and others),
 // which is the class encapsulating a LangGraph agent.
 
-const graphInvoke = {
-  path: './langgraph/graph-invoke.js',
-  instrumentations: [
-    // CommonJs
-    {
-      channelName: 'nr_invoke',
-      module: { name: '@langchain/langgraph', versionRange: '>=1.0.0', filePath: 'dist/pregel/index.cjs' },
-      functionQuery: {
-        index: 1,
-        methodName: 'invoke',
-        kind: 'Async',
-      }
-    },
-    // ESM
-    {
-      channelName: 'nr_invoke',
-      module: { name: '@langchain/langgraph', versionRange: '>=1.0.0', filePath: 'dist/pregel/index.js' },
-      functionQuery: {
-        index: 1,
-        methodName: 'invoke',
-        kind: 'Async',
-      }
-    },
-  ]
-}
-
 const graphStream = {
   path: './langgraph/graph-stream.js',
   instrumentations: [
@@ -60,7 +34,6 @@ const graphStream = {
 
 module.exports = {
   '@langchain/langgraph': [
-    graphInvoke,
     graphStream
   ]
 }

lib/subscribers/langgraph/graph-invoke.js

@@ -9,13 +9,16 @@ const test = require('node:test')
 const assert = require('node:assert')
 
 const { removeModules } = require('../../lib/cache-buster')
-const { assertSegments } = require('../../lib/custom-assertions')
+const { assertSegments, match } = require('../../lib/custom-assertions')
 const createOpenAIMockServer = require('../openai/mock-server')
 const helper = require('../../lib/agent_helper')
 
 const config = {
   ai_monitoring: {
-    enabled: true
+    enabled: true,
+    streaming: {
+      enabled: true
+    }
   }
 }
 
@@ -80,24 +83,50 @@ test('should create span on successful CompiledStateGraph.invoke', async (t) =>
   const { agent, langgraphAgent } = t.nr
 
   await helper.runInTransaction(agent, async (tx) => {
-    let content
-    try {
-      const result = await langgraphAgent.invoke(
-        { messages: [{ role: 'user', content: 'You are a scientist.' }] }
-      )
-      content = result?.messages?.[1]?.content ?? ''
-    } catch (err) {
-      assert.fail(err)
-    }
+    const result = await langgraphAgent.invoke(
+      { messages: [{ role: 'user', content: 'You are a scientist.' }] }
+    )
+    const content = result?.messages?.[1]?.content
     assert.equal(content, '212 degrees Fahrenheit is equal to 100 degrees Celsius.', 'should output correct content')
-    assertSegments(tx.trace, tx.trace.root, ['Llm/agent/LangGraph/invoke/LangGraphReactAgent'], {
+    assertSegments(tx.trace, tx.trace.root, ['Llm/agent/LangGraph/stream/LangGraphReactAgent'], {
       exact: false
     })
 
     tx.end()
   })
 })
 
+test('should create LlmAgent event for CompiledStateGraph.invoke', async (t) => {
+  const { agent, langgraphAgent } = t.nr
+
+  await helper.runInTransaction(agent, async (tx) => {
+    const result = await langgraphAgent.invoke(
+      { messages: [{ role: 'user', content: 'You are a scientist.' }] }
+    )
+    const content = result?.messages?.[1]?.content
+    assert.ok(content)
+    // Check for LlmAgent event
+    const events = agent.customEventAggregator.events.toArray()
+    const agentEvents = events.filter((e) => e[0].type === 'LlmAgent')
+    assert.ok(agentEvents.length > 0)
+
+    const [[{ type }, agentEvent]] = agentEvents
+    assert.equal(type, 'LlmAgent')
+    const [segment] = tx.trace.getChildren(tx.trace.root.id)
+
+    match(agentEvent, {
+      id: /[a-f0-9]{36}/,
+      name: 'LangGraphReactAgent',
+      span_id: segment.id,
+      trace_id: tx.traceId,
+      ingest_source: 'Node',
+      vendor: 'langgraph'
+    })
+
+    tx.end()
+  })
+})
+
 test('should not create segment or events when ai_monitoring.enabled is false', async (t) => {
   const { agent, langgraphAgent } = t.nr
 

test/versioned/langgraph/graph-invoke.test.js

@@ -300,7 +300,7 @@ test('should add subcomponent attribute to span', async (t) => {
   })
 })
 
-test('should create LlmError event when stream fails', async (t) => {
+test('should create LlmError event when given bad input', async (t) => {
   const { agent, langgraphAgent } = t.nr
 
   await helper.runInTransaction(agent, async (tx) => {

test/versioned/langgraph/graph-stream.test.js

@@ -4,7 +4,7 @@
     {
       "name": "@langchain/langgraph",
       "minSupported": "1.0.0",
-      "minAgentVersion": "13.11.0"
+      "minAgentVersion": "13.12.0"
     }
   ],
   "version": "0.0.0",