feat: WIP add opentelemetry configuration (#58)

Azure-Samples · Sep 29, 2023 · 154e262 · 154e262
1 parent a21d368
commit 154e262
Show file tree

Hide file tree

Showing 6 changed files with 219 additions and 45 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/indexer/package.json b/packages/indexer/package.json
@@ -24,7 +24,7 @@
   },
   "dependencies": {
     "@azure/identity": "^3.3.0",
-    "@azure/monitor-opentelemetry": "^1.0.0-beta.2",
+    "@azure/monitor-opentelemetry": "^1.0.0",
     "@azure/search-documents": "^12.0.0-beta.3",
     "@azure/storage-blob": "^12.15.0",
     "@dqbd/tiktoken": "^1.0.7",
@@ -33,6 +33,8 @@
     "@fastify/multipart": "^7.7.3",
     "@fastify/sensible": "^5.0.0",
     "@fastify/type-provider-json-schema-to-ts": "^2.2.2",
+    "@opentelemetry/instrumentation": "^0.43.0",
+    "@opentelemetry/instrumentation-fastify": "^0.32.2",
     "commander": "^11.0.0",
     "dotenv": "^16.3.1",
     "fastify": "^4.22.2",

diff --git a/packages/indexer/src/plugins/opentelemetry.ts b/packages/indexer/src/plugins/opentelemetry.ts
@@ -0,0 +1,32 @@
+import fp from 'fastify-plugin';
+import { useAzureMonitor, type AzureMonitorOpenTelemetryOptions } from '@azure/monitor-opentelemetry';
+import { registerInstrumentations } from '@opentelemetry/instrumentation';
+import { FastifyInstrumentation } from '@opentelemetry/instrumentation-fastify';
+
+export default fp(
+  async (fastify) => {
+    const { applicationInsightsConnectionString } = fastify.config;
+    if (applicationInsightsConnectionString) {
+      const options: AzureMonitorOpenTelemetryOptions = {
+        azureMonitorExporterOptions: {
+          connectionString: applicationInsightsConnectionString,
+        },
+      };
+      useAzureMonitor(options);
+
+      registerInstrumentations({
+        instrumentations: [
+          // Azure monitor already instruments HTTP layer by default,
+          // so we don't need to add instrumentation for it
+          new FastifyInstrumentation(),
+        ],
+      });
+    } else {
+      fastify.log.info('Application Insights configuration not found, OpenTelemetry disabled');
+    }
+  },
+  {
+    name: 'opentelemetry',
+    dependencies: ['config'],
+  },
+);
diff --git a/packages/search/package.json b/packages/search/package.json
@@ -21,7 +21,7 @@
   },
   "dependencies": {
     "@azure/identity": "^3.3.0",
-    "@azure/monitor-opentelemetry": "^1.0.0-beta.2",
+    "@azure/monitor-opentelemetry": "^1.0.0",
     "@azure/search-documents": "^12.0.0-beta.3",
     "@azure/storage-blob": "^12.15.0",
     "@dqbd/tiktoken": "^1.0.7",
@@ -31,6 +31,8 @@
     "@fastify/swagger": "^8.10.0",
     "@fastify/swagger-ui": "^1.9.3",
     "@fastify/type-provider-json-schema-to-ts": "^2.2.2",
+    "@opentelemetry/instrumentation": "^0.43.0",
+    "@opentelemetry/instrumentation-fastify": "^0.32.2",
     "dotenv": "^16.3.1",
     "fastify": "^4.0.0",
     "fastify-cli": "^5.7.0",

diff --git a/packages/search/src/plugins/opentelemetry.ts b/packages/search/src/plugins/opentelemetry.ts
@@ -0,0 +1,33 @@
+import fp from 'fastify-plugin';
+import { useAzureMonitor, type AzureMonitorOpenTelemetryOptions } from '@azure/monitor-opentelemetry';
+import { registerInstrumentations } from '@opentelemetry/instrumentation';
+import { FastifyInstrumentation } from '@opentelemetry/instrumentation-fastify';
+
+export default fp(
+  async (fastify) => {
+    const { applicationInsightsConnectionString } = fastify.config;
+    if (applicationInsightsConnectionString) {
+      fastify.log.info('Application Insights configuration found, OpenTelemetry enabled');
+      const options: AzureMonitorOpenTelemetryOptions = {
+        azureMonitorExporterOptions: {
+          connectionString: applicationInsightsConnectionString,
+        },
+      };
+      useAzureMonitor(options);
+
+      registerInstrumentations({
+        instrumentations: [
+          // Azure monitor already instruments HTTP layer by default,
+          // so we don't need to add instrumentation for it
+          new FastifyInstrumentation(),
+        ],
+      });
+    } else {
+      fastify.log.info('Application Insights configuration not found, OpenTelemetry disabled');
+    }
+  },
+  {
+    name: 'opentelemetry',
+    dependencies: ['config'],
+  },
+);
diff --git a/packages/search/test.http b/packages/search/test.http
@@ -15,7 +15,7 @@ POST {{api_host}}/chat
 Content-Type: application/json
 
 {
-  "history": [{ "user": "What happens is a rental doesn't fit the description?" }],
+  "history": [{ "user": "What happens if a rental doesn't fit the description?" }],
   "approach":"rrr",
   "overrides": {
     "retrieval_mode": "hybrid",