feat(OTel): Push Amplitude IDs to Baggage (#7238)

Author: khvn26

frontend/common/service.ts

@@ -1,3 +1,4 @@
+import * as amplitude from '@amplitude/analytics-browser'
 import { createApi, fetchBaseQuery } from '@reduxjs/toolkit/dist/query/react'
 
 import { FetchBaseQueryArgs } from '@reduxjs/toolkit/dist/query/fetchBaseQuery'
@@ -36,6 +37,17 @@ const baseApiOptions = (queryArgs?: Partial<FetchBaseQueryArgs>) => {
           } catch (e) {}
         }
 
+        const deviceId = amplitude.getDeviceId()
+        const sessionId = amplitude.getSessionId()
+        const userId = amplitude.getUserId()
+        if (deviceId || sessionId || userId) {
+          const entries: string[] = []
+          if (deviceId) entries.push(`amplitude.device_id=${deviceId}`)
+          if (sessionId) entries.push(`amplitude.session_id=${sessionId}`)
+          if (userId) entries.push(`amplitude.user_id=${userId}`)
+          headers.set('baggage', entries.join(','))
+        }
+
         return headers
       },
       ...queryArgs,

frontend/common/types/responses.ts

@@ -694,6 +694,7 @@ export type Account = {
   pylon_email_signature: string
   sign_up_type: SignupType
   id: number
+  uuid: string
   email: string
   auth_type: AuthType
   is_superuser: boolean

frontend/web/project/api.ts

@@ -112,7 +112,8 @@ const API = {
     }
 
     if (Project.amplitude) {
-      amplitude.setUserId(id)
+      const amplitudeUserId = user.uuid || id
+      amplitude.setUserId(amplitudeUserId)
       API.trackTraits({ email: id })
       if (window.engagement) {
         window.engagement.boot({
@@ -122,7 +123,7 @@ const API = {
                 amplitude.track(event.event_type, event.event_properties),
             },
           ],
-          user: { user_id: id, user_properties: {} },
+          user: { user_id: amplitudeUserId, user_properties: {} },
         })
       }
     }