hash user id

packages/compass-generative-ai/src/atlas-ai-service.spec.ts

@@ -1058,11 +1058,12 @@ describe('AtlasAiService', function () {
 
             const requestBody = JSON.parse(args[1].body as string);
             expect(requestBody.model).to.equal('mongodb-chat-latest');
-            expect(requestBody.metadata).to.deep.equal({
-              userId: '1234',
+            const { userId, ...restOfMetadata } = requestBody.metadata;
+            expect(restOfMetadata).to.deep.equal({
               store: 'true',
               sensitiveStorage: 'sensitive',
             });
+            expect(userId).to.be.a('string').that.is.not.empty;
             expect(requestBody.instructions).to.be.a('string');
             expect(requestBody.input).to.be.an('array');
 

packages/compass-generative-ai/src/atlas-ai-service.ts

@@ -260,12 +260,35 @@ export type MockDataSchemaResponse = z.infer<
   typeof MockDataSchemaResponseShape
 >;
 
-function getActiveUserId(preferences: PreferencesAccess): string {
+async function getHashedActiveUserId(
+  preferences: PreferencesAccess,
+  logger: Logger
+): Promise<string> {
   const { currentUserId, telemetryAnonymousId, telemetryAtlasUserId } =
     preferences.getPreferences();
-  return (
-    currentUserId ?? telemetryAnonymousId ?? telemetryAtlasUserId ?? 'unknown'
-  );
+  const userId = currentUserId ?? telemetryAnonymousId ?? telemetryAtlasUserId;
+  if (!userId) {
+    return 'unknown';
+  }
+  try {
+    const data = new TextEncoder().encode(userId);
+    const hashBuffer = await crypto.subtle.digest('SHA-256', data);
+    const hashArray = Array.from(new Uint8Array(hashBuffer));
+    const hashHex = hashArray
+      .map((b) => b.toString(16).padStart(2, '0'))
+      .join('');
+    return hashHex;
+  } catch (e) {
+    logger.log.warn(
+      logger.mongoLogId(1_001_000_385),
+      'AtlasAiService',
+      'Failed to hash user id for AI request',
+      {
+        error: (e as Error).message,
+      }
+    );
+    return 'unknown';
+  }
 }
 
 /**
@@ -462,7 +485,7 @@ export class AtlasAiService {
     if (this.preferences.getPreferences().enableChatbotEndpointForGenAI) {
       const message = buildAggregateQueryPrompt({
         ...input,
-        userId: getActiveUserId(this.preferences),
+        userId: await getHashedActiveUserId(this.preferences, this.logger),
       });
       return this.generateQueryUsingChatbot(
         message,
@@ -487,7 +510,7 @@ export class AtlasAiService {
     if (this.preferences.getPreferences().enableChatbotEndpointForGenAI) {
       const message = buildFindQueryPrompt({
         ...input,
-        userId: getActiveUserId(this.preferences),
+        userId: await getHashedActiveUserId(this.preferences, this.logger),
       });
       return this.generateQueryUsingChatbot(message, validateAIQueryResponse, {
         signal: input.signal,