feat(posthog): correlate task events with Go logs via request_id

Auto-injects server request_id into all PostHog server events from
AsyncLocalStorage context. Adds task_request_started client event fired
when SSE traceparent response header arrives, carrying the trace ID used
by Go for log correlation. task_generation_aborted now includes the
in-flight request_id when available.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/home/home.tsx

@@ -143,12 +143,21 @@ export function Home({ chatId }: HomeProps = {}) {
     editQueuedMessage,
     previewSession,
     genericResourceData,
+    getCurrentRequestId,
   } = useChat(
     workspaceId,
     chatId,
     getMothershipUseChatOptions({
       onResourceEvent: handleResourceEvent,
       initialActiveResourceId: initialResourceId,
+      onRequestStarted: ({ requestId, userMessageId }) => {
+        captureEvent(posthogRef.current, 'task_request_started', {
+          workspace_id: workspaceId,
+          view: 'mothership',
+          request_id: requestId,
+          user_message_id: userMessageId,
+        })
+      },
     })
   )
 
@@ -198,6 +207,7 @@ export function Home({ chatId }: HomeProps = {}) {
     captureEvent(posthogRef.current, 'task_generation_aborted', {
       workspace_id: workspaceId,
       view: 'mothership',
+      request_id: getCurrentRequestId(),
     })
     void stopGeneration().catch(() => {})
   }

apps/sim/app/workspace/[workspaceId]/home/hooks/use-chat.ts

@@ -167,6 +167,7 @@ export interface UseChatReturn {
   editQueuedMessage: (id: string) => QueuedMessage | undefined
   previewSession: FilePreviewSession | null
   genericResourceData: GenericResourceData | null
+  getCurrentRequestId: () => string | undefined
 }
 
 const DEPLOY_TOOL_NAMES: Set<string> = new Set([
@@ -1278,6 +1279,13 @@ export interface UseChatOptions {
   onTitleUpdate?: () => void
   onStreamEnd?: (chatId: string, messages: ChatMessage[]) => void
   initialActiveResourceId?: string | null
+  /**
+   * Fired once per chat send as soon as the server's `traceparent`
+   * response header arrives (i.e. before any stream content). Used by
+   * callers to emit a follow-up PostHog event carrying the request ID
+   * for correlation with Go-side logs.
+   */
+  onRequestStarted?: (info: { requestId: string; userMessageId: string }) => void
 }
 
 interface ActiveStreamRecovery {
@@ -1293,7 +1301,10 @@ interface StopGenerationOptions {
 }
 
 export function getMothershipUseChatOptions(
-  options: Pick<UseChatOptions, 'onResourceEvent' | 'onStreamEnd' | 'initialActiveResourceId'> = {}
+  options: Pick<
+    UseChatOptions,
+    'onResourceEvent' | 'onStreamEnd' | 'initialActiveResourceId' | 'onRequestStarted'
+  > = {}
 ): UseChatOptions {
   return {
     apiPath: MOTHERSHIP_CHAT_API_PATH,
@@ -1305,7 +1316,7 @@ export function getMothershipUseChatOptions(
 export function getWorkflowCopilotUseChatOptions(
   options: Pick<
     UseChatOptions,
-    'workflowId' | 'onToolResult' | 'onTitleUpdate' | 'onStreamEnd'
+    'workflowId' | 'onToolResult' | 'onTitleUpdate' | 'onStreamEnd' | 'onRequestStarted'
   > = {}
 ): UseChatOptions {
   return {
@@ -1351,6 +1362,10 @@ export function useChat(
   onTitleUpdateRef.current = options?.onTitleUpdate
   const onStreamEndRef = useRef(options?.onStreamEnd)
   onStreamEndRef.current = options?.onStreamEnd
+  const onRequestStartedRef = useRef(options?.onRequestStarted)
+  onRequestStartedRef.current = options?.onRequestStarted
+
+  const getCurrentRequestId = useCallback(() => streamRequestIdRef.current, [])
 
   const clearQueueDispatchState = useCallback(() => {
     queueDispatchEpochRef.current++
@@ -4209,6 +4224,15 @@ export function useChat(
         if (traceparent) {
           streamTraceparentRef.current = traceparent
           setCurrentChatTraceparent(traceparent)
+          const parts = traceparent.split('-')
+          const traceId = parts.length === 4 ? parts[1] : ''
+          if (/^[0-9a-f]{32}$/.test(traceId) && traceId !== '0'.repeat(32)) {
+            try {
+              onRequestStartedRef.current?.({ requestId: traceId, userMessageId })
+            } catch (callbackError) {
+              logger.warn('onRequestStarted callback threw', { error: callbackError })
+            }
+          }
         }
 
         if (!response.ok) {
@@ -5103,5 +5127,6 @@ export function useChat(
     editQueuedMessage,
     previewSession,
     genericResourceData,
+    getCurrentRequestId,
   }
 }

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/components/panel/panel.tsx

@@ -346,13 +346,22 @@ export const Panel = memo(function Panel({ workspaceId: propWorkspaceId }: Panel
     removeFromQueue: copilotRemoveFromQueue,
     sendNow: copilotSendNow,
     editQueuedMessage: copilotEditQueuedMessage,
+    getCurrentRequestId: getCopilotCurrentRequestId,
   } = useChat(
     workspaceId,
     copilotChatId,
     getWorkflowCopilotUseChatOptions({
       workflowId: activeWorkflowId || undefined,
       onTitleUpdate: loadCopilotChats,
       onToolResult: handleCopilotToolResult,
+      onRequestStarted: ({ requestId, userMessageId }) => {
+        captureEvent(posthogRef.current, 'task_request_started', {
+          workspace_id: workspaceId,
+          view: 'copilot',
+          request_id: requestId,
+          user_message_id: userMessageId,
+        })
+      },
     })
   )
 
@@ -413,9 +422,10 @@ export const Panel = memo(function Panel({ workspaceId: propWorkspaceId }: Panel
     captureEvent(posthogRef.current, 'task_generation_aborted', {
       workspace_id: workspaceId,
       view: 'copilot',
+      request_id: getCopilotCurrentRequestId(),
     })
     copilotStopGeneration()
-  }, [copilotStopGeneration, workspaceId])
+  }, [copilotStopGeneration, getCopilotCurrentRequestId, workspaceId])
 
   const handleCopilotSubmit = useCallback(
     (text: string, fileAttachments?: FileAttachmentForApi[], contexts?: ChatContext[]) => {

apps/sim/lib/posthog/events.ts

@@ -418,6 +418,7 @@ export interface PostHogEventMap {
   task_generation_aborted: {
     workspace_id: string
     view: 'mothership' | 'copilot'
+    request_id?: string
   }
 
   task_message_sent: {
@@ -427,6 +428,20 @@ export interface PostHogEventMap {
     is_new_task: boolean
   }
 
+  /**
+   * Fired when the server has assigned a request ID for a chat send (i.e.
+   * the SSE response headers have arrived with a `traceparent`). Pairs
+   * one-to-one with `task_message_sent` (or with a queued/replay dispatch)
+   * via timestamp proximity, and carries the `request_id` used by Go for
+   * log correlation.
+   */
+  task_request_started: {
+    workspace_id: string
+    view: 'mothership' | 'copilot'
+    request_id: string
+    user_message_id: string
+  }
+
   tour_started: {
     tour_type: 'nav' | 'workflow'
   }

apps/sim/lib/posthog/server.ts

@@ -1,4 +1,4 @@
-import { createLogger } from '@sim/logger'
+import { createLogger, getRequestContext } from '@sim/logger'
 import type { PostHog } from 'posthog-node'
 import type { PostHogEventMap, PostHogEventName } from '@/lib/posthog/events'
 
@@ -71,10 +71,12 @@ export function captureServerEvent<E extends PostHogEventName>(
     const client = getClient()
     if (!client) return
 
+    const requestId = getRequestContext()?.requestId
     client.capture({
       distinctId,
       event,
       properties: {
+        ...(requestId ? { request_id: requestId } : {}),
         ...properties,
         ...(options?.groups ? { $groups: options.groups } : {}),
         ...(options?.set ? { $set: options.set } : {}),