Add the Databricks Opentelemetry

Author: poad

agents/agent-ai-sdk/package.json

@@ -19,22 +19,22 @@
     "@types/aws-lambda": "^8.10.161",
     "@types/node": "^24.12.2",
     "@vitest/eslint-plugin": "^1.6.16",
-    "aws-cdk": "^2.1118.2",
+    "aws-cdk": "^2.1118.4",
     "esbuild": "^0.27.7",
-    "eslint": "^10.2.0",
+    "eslint": "^10.2.1",
     "eslint-import-resolver-typescript": "^4.4.4",
     "eslint-plugin-awscdk": "^4.3.2",
     "eslint-plugin-import-x": "^4.16.2",
     "eslint-plugin-promise": "^7.2.1",
     "jiti": "^2.6.1",
     "tsx": "^4.21.0",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.58.2",
-    "vite": "^8.0.8",
-    "vitest": "^4.1.4"
+    "typescript-eslint": "^8.59.0",
+    "vite": "^8.0.9",
+    "vitest": "^4.1.5"
   },
   "dependencies": {
-    "@ai-sdk/amazon-bedrock": "^4.0.95",
+    "@ai-sdk/amazon-bedrock": "^4.0.96",
     "@ai-sdk/azure": "^3.0.54",
     "@llm-ts-example/common-backend-core": "workspace:*",
     "@llm-ts-example/common-core": "workspace:*",

agents/agent-claude-agent-sdk/package.json

@@ -21,19 +21,19 @@
     "@vitest/eslint-plugin": "^1.6.16",
     "dotenv": "^17.4.2",
     "esbuild": "^0.27.7",
-    "eslint": "^10.2.0",
+    "eslint": "^10.2.1",
     "eslint-import-resolver-typescript": "^4.4.4",
     "eslint-plugin-import-x": "^4.16.2",
     "eslint-plugin-promise": "^7.2.1",
     "jiti": "^2.6.1",
     "tsx": "^4.21.0",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.58.2",
-    "vite": "^8.0.8",
-    "vitest": "^4.1.4"
+    "typescript-eslint": "^8.59.0",
+    "vite": "^8.0.9",
+    "vitest": "^4.1.5"
   },
   "dependencies": {
-    "@anthropic-ai/claude-agent-sdk": "^0.2.112",
+    "@anthropic-ai/claude-agent-sdk": "^0.2.117",
     "source-map-support": "^0.5.21",
     "uuid": "^14.0.0",
     "zod": "^4.3.6"

agents/agent-deepagents/package.json

@@ -19,10 +19,10 @@
     "@opentelemetry/exporter-trace-otlp-http": "^0.214.0",
     "@opentelemetry/exporter-trace-otlp-proto": "^0.214.0",
     "@opentelemetry/instrumentation": "^0.214.0",
-    "@opentelemetry/resources": "^2.6.1",
+    "@opentelemetry/resources": "^2.7.0",
     "@opentelemetry/sdk-node": "^0.214.0",
-    "@opentelemetry/sdk-trace-base": "^2.6.1",
-    "@opentelemetry/sdk-trace-node": "^2.6.1",
+    "@opentelemetry/sdk-trace-base": "^2.7.0",
+    "@opentelemetry/sdk-trace-node": "^2.7.0",
     "@opentelemetry/semantic-conventions": "^1.40.0",
     "@traceloop/node-server-sdk": "^0.25.0",
     "deepagents": "^1.9.0",
@@ -34,15 +34,15 @@
     "@eslint/js": "^10.0.1",
     "@stylistic/eslint-plugin": "^5.10.0",
     "@types/node": "^24.12.2",
-    "eslint": "^10.2.0",
+    "eslint": "^10.2.1",
     "eslint-import-resolver-typescript": "^4.4.4",
     "eslint-plugin-import-x": "^4.16.2",
     "eslint-plugin-promise": "^7.2.1",
     "langchain": "^1.3.3",
     "tsx": "^4.21.0",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.58.2",
-    "vite": "^8.0.8",
-    "vitest": "^4.1.4"
+    "typescript-eslint": "^8.59.0",
+    "vite": "^8.0.9",
+    "vitest": "^4.1.5"
   }
 }

agents/agent-langchain-nextjs/package.json

@@ -12,8 +12,8 @@
   "dependencies": {
     "@ai-sdk/langchain": "^2.0.174",
     "@ai-sdk/react": "^3.0.170",
-    "@arizeai/openinference-instrumentation-bedrock": "^0.4.10",
-    "@arizeai/openinference-instrumentation-langchain": "^4.0.8",
+    "@arizeai/openinference-instrumentation-bedrock": "^0.4.11",
+    "@arizeai/openinference-instrumentation-langchain": "^4.0.9",
     "@aws-lambda-powertools/logger": "^2.33.0",
     "@aws-sdk/credential-provider-node": "^3.972.36",
     "@langchain/aws": "^1.3.5",
@@ -28,12 +28,12 @@
     "@opentelemetry/exporter-trace-otlp-http": "^0.214.0",
     "@opentelemetry/exporter-trace-otlp-proto": "^0.214.0",
     "@opentelemetry/instrumentation": "^0.214.0",
-    "@opentelemetry/resources": "^2.6.1",
+    "@opentelemetry/resources": "^2.7.0",
     "@opentelemetry/sdk-logs": "^0.214.0",
-    "@opentelemetry/sdk-metrics": "^2.6.1",
+    "@opentelemetry/sdk-metrics": "^2.7.0",
     "@opentelemetry/sdk-node": "^0.214.0",
-    "@opentelemetry/sdk-trace-base": "^2.6.1",
-    "@opentelemetry/sdk-trace-node": "^2.6.1",
+    "@opentelemetry/sdk-trace-base": "^2.7.0",
+    "@opentelemetry/sdk-trace-node": "^2.7.0",
     "@opentelemetry/semantic-conventions": "^1.40.0",
     "@radix-ui/react-accordion": "^1.2.12",
     "@radix-ui/react-avatar": "^1.1.11",
@@ -51,7 +51,7 @@
     "@radix-ui/react-tabs": "^1.1.13",
     "@radix-ui/react-tooltip": "^1.2.8",
     "@radix-ui/react-use-controllable-state": "^1.2.2",
-    "@rive-app/react-webgl2": "^4.28.0",
+    "@rive-app/react-webgl2": "^4.28.1",
     "@smithy/eventstream-codec": "^4.2.14",
     "@streamdown/cjk": "^1.0.3",
     "@streamdown/code": "^1.1.1",
@@ -83,20 +83,20 @@
     "@eslint/compat": "^2.0.5",
     "@eslint/js": "^10.0.1",
     "@stylistic/eslint-plugin": "^5.10.0",
-    "@tailwindcss/postcss": "^4.2.2",
+    "@tailwindcss/postcss": "^4.2.4",
     "@types/node": "^24.12.2",
     "@types/react": "^19.2.14",
     "@types/react-dom": "^19.2.3",
-    "eslint": "^10.2.0",
+    "eslint": "^10.2.1",
     "eslint-config-next": "^16.2.4",
     "eslint-import-resolver-typescript": "^4.4.4",
     "eslint-plugin-import-x": "^4.16.2",
     "eslint-plugin-promise": "^7.2.1",
-    "tailwindcss": "^4.2.2",
+    "tailwindcss": "^4.2.4",
     "tw-animate-css": "^1.4.0",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.58.2",
-    "vite": "^8.0.8",
-    "vitest": "^4.1.4"
+    "typescript-eslint": "^8.59.0",
+    "vite": "^8.0.9",
+    "vitest": "^4.1.5"
   }
 }

agents/agent-langchain/package.json

@@ -20,42 +20,52 @@
     "@types/aws-lambda": "^8.10.161",
     "@types/node": "^24.12.2",
     "@vitest/eslint-plugin": "^1.6.16",
-    "dotenv": "^17.4.2",
-    "eslint": "^10.2.0",
+    "eslint": "^10.2.1",
     "eslint-import-resolver-typescript": "^4.4.4",
     "eslint-plugin-import-x": "^4.16.2",
     "eslint-plugin-promise": "^7.2.1",
     "jiti": "^2.6.1",
     "tsx": "^4.21.0",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.58.2",
-    "vite": "^8.0.8",
-    "vitest": "^4.1.4"
+    "typescript-eslint": "^8.59.0",
+    "vite": "^8.0.9",
+    "vitest": "^4.1.5"
   },
   "dependencies": {
-    "@arizeai/openinference-instrumentation-bedrock": "^0.4.10",
-    "@arizeai/openinference-instrumentation-langchain": "^4.0.8",
+    "@arizeai/openinference-instrumentation-bedrock": "^0.4.11",
+    "@arizeai/openinference-instrumentation-langchain": "^4.0.9",
     "@aws-lambda-powertools/logger": "^2.33.0",
     "@aws-sdk/credential-provider-node": "^3.972.36",
+    "@dotenvx/dotenvx": "^1.61.4",
+    "@grpc/grpc-js": "^1.14.3",
     "@langchain/aws": "^1.3.5",
     "@langchain/classic": "^1.0.31",
     "@langchain/core": "^1.1.40",
     "@langchain/langgraph": "^1.2.9",
     "@langchain/mcp-adapters": "^1.1.3",
     "@langchain/openai": "^1.4.4",
     "@opentelemetry/api": "^1.9.1",
-    "@opentelemetry/exporter-trace-otlp-http": "^0.214.0",
-    "@opentelemetry/exporter-trace-otlp-proto": "^0.214.0",
-    "@opentelemetry/instrumentation": "^0.214.0",
-    "@opentelemetry/resources": "^2.6.1",
-    "@opentelemetry/sdk-node": "^0.214.0",
-    "@opentelemetry/sdk-trace-base": "^2.6.1",
-    "@opentelemetry/sdk-trace-node": "^2.6.1",
+    "@opentelemetry/api-logs": "^0.215.0",
+    "@opentelemetry/auto-instrumentations-node": "^0.69.0",
+    "@opentelemetry/exporter-logs-otlp-grpc": "^0.215.0",
+    "@opentelemetry/exporter-logs-otlp-proto": "^0.215.0",
+    "@opentelemetry/exporter-metrics-otlp-grpc": "^0.215.0",
+    "@opentelemetry/exporter-metrics-otlp-http": "^0.215.0",
+    "@opentelemetry/exporter-metrics-otlp-proto": "^0.215.0",
+    "@opentelemetry/exporter-trace-otlp-grpc": "^0.215.0",
+    "@opentelemetry/exporter-trace-otlp-proto": "^0.215.0",
+    "@opentelemetry/instrumentation": "^0.215.0",
+    "@opentelemetry/otlp-exporter-base": "^0.215.0",
+    "@opentelemetry/resources": "^2.7.0",
+    "@opentelemetry/sdk-logs": "^0.215.0",
+    "@opentelemetry/sdk-metrics": "^2.7.0",
+    "@opentelemetry/sdk-node": "^0.215.0",
+    "@opentelemetry/sdk-trace-base": "^2.7.0",
+    "@opentelemetry/sdk-trace-node": "^2.7.0",
     "@opentelemetry/semantic-conventions": "^1.40.0",
     "@smithy/eventstream-codec": "^4.2.14",
     "constructs": "^10.6.0",
     "langchain": "^1.3.3",
-    "source-map-support": "^0.5.21",
     "uuid": "^14.0.0",
     "zod": "^4.3.6"
   }

agents/agent-langchain/src/index.ts

@@ -1,5 +1,14 @@
 import { MultiServerMCPClient } from "@langchain/mcp-adapters";
 import { createAgent } from "langchain";
+import dotenv from '@dotenvx/dotenvx';
+
+dotenv.config({ path: ['.env', '.env.test'], override: true });
+
+import { initialize } from './instrumentation.js';
+
+const exporters = process.env.ENABLED_OPENINFERENCE_TELEMETRY === 'true' ? await initialize() : { flush: async () => {
+  // pass
+ } };
 
 const client = new MultiServerMCPClient({
   'aws-knowledge-mcp-server': {
@@ -16,7 +25,7 @@ const client = new MultiServerMCPClient({
 const tools = await client.getTools();
 
 const agent = createAgent({
-  model: "bedrock:global.amazon.nova-2-lite-v1:0",
+  model: "bedrock:us.amazon.nova-lite-v1:0",
   systemPrompt: `
       You are an assistant that helps architects design systems using Amazon Web Services (AWS). Your primary function is to answer user questions based on AWS knowledge and propose system architectures. When responding, follow these guidelines:
 
@@ -37,6 +46,7 @@ const stream = await agent.stream({
 
 }, { streamMode: "values" });
 for await (const chunk of stream) {
-  console.log('Agent', chunk.messages);
+  console.info('Agent', { messages: chunk.messages });
 }
+await exporters.flush()
 process.exit(0);

agents/agent-langchain/src/instrumentation.ts

@@ -0,0 +1,125 @@
+import { logger } from './logging.js';
+import * as opentelemetry from '@opentelemetry/api';
+import { registerInstrumentations } from '@opentelemetry/instrumentation';
+import {
+  NodeTracerProvider,
+  SimpleSpanProcessor,
+} from '@opentelemetry/sdk-trace-node';
+import {
+  ATTR_SERVICE_NAME,
+  ATTR_SERVICE_VERSION,
+} from '@opentelemetry/semantic-conventions';
+import { diag, DiagConsoleLogger, DiagLogLevel } from '@opentelemetry/api';
+import { LangChainInstrumentation } from '@arizeai/openinference-instrumentation-langchain';
+import * as lcCallbackManager from '@langchain/core/callbacks/manager';
+import {
+  defaultResource,
+  resourceFromAttributes,
+} from '@opentelemetry/resources';
+import { getNodeAutoInstrumentations } from '@opentelemetry/auto-instrumentations-node';
+import {
+  MeterProvider,
+  PeriodicExportingMetricReader,
+} from '@opentelemetry/sdk-metrics';
+
+import { logs } from '@opentelemetry/api-logs';
+import {
+  LoggerProvider,
+  SimpleLogRecordProcessor,
+} from '@opentelemetry/sdk-logs';
+
+import type { Exporters } from './otel/otel-exporters.js';
+import { Provider, createExporters } from './otel/otel-exporters.js';
+import dotenv from '@dotenvx/dotenvx';
+
+dotenv.config({ path: ['.env', '.env.test'], override: true });
+
+const initialize = async (): Promise<Exporters & { flush?: () => Promise<void> }> => {
+
+  diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.WARN);
+
+  logger.info('begin initialize opentelemetry exporters');
+
+  const exporters = await createExporters(
+    process.env.OTEL_PROVIDER ?? Provider.PROVIDER_DATABRICS,
+    { useGrpc: process.env.ENABLED_OTEL_EXPORTER_OTLP_GRPC === 'true' });
+
+  const resource = defaultResource().merge(
+    resourceFromAttributes({
+      [ATTR_SERVICE_NAME]: 'agent-langchain',
+      [ATTR_SERVICE_VERSION]: '0.1.0',
+      ['openinference.project.name']: 'default',
+    }),
+  );
+
+  if (exporters.trace) {
+    logger.info('Enable tracing');
+
+    const provider = new NodeTracerProvider({
+      spanProcessors: [new SimpleSpanProcessor(
+        exporters.trace,
+      ),
+      ],
+      resource,
+    });
+
+    registerInstrumentations({
+      instrumentations: [],
+    });
+
+    const lcInstrumentation = new LangChainInstrumentation();
+    lcInstrumentation.manuallyInstrument(lcCallbackManager);
+
+
+    registerInstrumentations({
+      instrumentations: [
+        getNodeAutoInstrumentations({
+        }),
+      ],
+    });
+    provider.register();
+  }
+
+  if (exporters.metric) {
+    logger.info('Enable metrics');
+
+    const metricReader = new PeriodicExportingMetricReader({
+      exporter: exporters.metric,
+      exportIntervalMillis: 10000,
+    });
+
+    const myServiceMeterProvider = new MeterProvider({
+      resource: resource,
+      readers: [metricReader],
+    });
+
+    opentelemetry.metrics.setGlobalMeterProvider(myServiceMeterProvider);
+  }
+
+  if (exporters.logs) {
+    logger.info('Enable logs');
+
+    const loggerProvider = new LoggerProvider({
+      processors: [
+        new SimpleLogRecordProcessor(
+          exporters.logs,
+        ),
+      ],
+    });
+
+    logs.setGlobalLoggerProvider(loggerProvider);
+  }
+
+  logger.info('👀 OpenInference initialized');
+
+  return {
+    ...exporters,
+    flush: async () => {
+      await exporters.trace?.forceFlush?.();
+      await exporters.logs?.forceFlush?.();
+      await exporters.metric?.forceFlush?.();
+    }
+  };
+}
+
+export { initialize };

agents/agent-langchain/src/logging.ts

@@ -0,0 +1,4 @@
+import * as powertools from '@aws-lambda-powertools/logger';
+
+const logger = new powertools.Logger();
+export { logger };