fix: re-enable otel tracing for agents

Author: sinedied

package-lock.json

@@ -16,6 +16,7 @@
   "author": "Microsoft",
   "license": "MIT",
   "dependencies": {
+    "@arizeai/openinference-instrumentation-langchain": "^4.0.0",
     "@azure/cosmos": "^4.6.0",
     "@azure/functions": "^4.0.0",
     "@azure/identity": "^4.2.0",

packages/agent-api/package.json

@@ -5,7 +5,7 @@ import { createAgent, AIMessage, HumanMessage } from 'langchain';
 import { ChatOpenAI } from '@langchain/openai';
 import { AzureCosmsosDBNoSQLChatMessageHistory } from '@langchain/azure-cosmosdb';
 import { loadMcpTools } from '@langchain/mcp-adapters';
-import { StreamEvent } from '@langchain/core/dist/tracers/log_stream.js';
+import { StreamEvent } from '@langchain/core/tracers/log_stream';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { getAzureOpenAiTokenProvider, getCredentials, getInternalUserId } from '../auth.js';

packages/agent-api/src/functions/chats-post.ts

@@ -6,8 +6,8 @@ import { useAzureMonitor } from '@azure/monitor-opentelemetry';
 import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
 import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
 import { SimpleSpanProcessor } from '@opentelemetry/sdk-trace-base';
-// import { LangChainInstrumentation } from '@arizeai/openinference-instrumentation-langchain';
-// import * as CallbackManagerModule from 'langchain';
+import { LangChainInstrumentation } from '@arizeai/openinference-instrumentation-langchain';
+import * as CallbackManagerModule from "@langchain/core/callbacks/manager";
 
 let isTracingInitialized = false;
 if (!isTracingInitialized) {
@@ -35,9 +35,8 @@ if (!isTracingInitialized) {
   }
 
   // Manually instrument LangChain's CallbackManager to capture traces
-  // TODO: temporarily disabled due to compatibility issues with LangChain v1 (PR in progress)
-  // const langchainInstrumentation = new LangChainInstrumentation();
-  // langchainInstrumentation.manuallyInstrument(CallbackManagerModule);
+  const langchainInstrumentation = new LangChainInstrumentation();
+  langchainInstrumentation.manuallyInstrument(CallbackManagerModule);
 
   isTracingInitialized = true;
 }

packages/agent-api/src/tracing.ts

@@ -1,6 +1,6 @@
 {
   "compilerOptions": {
-    "module": "CommonJS",
+    "module": "nodenext",
     "target": "ESNext",
     "incremental": true,
     "composite": true,
@@ -14,7 +14,7 @@
     "noUnusedParameters": true,
     "noImplicitReturns": true,
     "noImplicitOverride": true,
-    "moduleResolution": "node",
+    "moduleResolution": "nodenext",
     "esModuleInterop": true,
     "lib": ["ESNext"]
   },