feat(copilot): attribute tool spans to agents

Author: j15z

apps/sim/lib/copilot/request/otel.ts

@@ -284,6 +284,7 @@ export async function withCopilotToolSpan<T>(
   input: {
     toolName: string
     toolCallId: string
+    agentName: string
     runId?: string
     chatId?: string
     argsBytes?: number
@@ -299,6 +300,7 @@ export async function withCopilotToolSpan<T>(
         [TraceAttr.ToolName]: input.toolName,
         [TraceAttr.ToolCallId]: input.toolCallId,
         [TraceAttr.ToolExecutor]: 'sim',
+        [TraceAttr.GenAiAgentName]: input.agentName,
         ...(input.runId ? { [TraceAttr.RunId]: input.runId } : {}),
         ...(input.chatId ? { [TraceAttr.ChatId]: input.chatId } : {}),
         ...(typeof input.argsBytes === 'number'

apps/sim/lib/copilot/request/tools/executor.test.ts

@@ -2,20 +2,24 @@ import '@sim/testing/mocks/executor'
 
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
-const { recordSimToolMetric, setAttribute } = vi.hoisted(() => ({
-  recordSimToolMetric: vi.fn(),
-  setAttribute: vi.fn(),
-}))
+const { recordSimToolMetric, setAttribute, withCopilotToolSpan } = vi.hoisted(() => {
+  const setAttribute = vi.fn()
+  return {
+    recordSimToolMetric: vi.fn(),
+    setAttribute,
+    withCopilotToolSpan: vi.fn(
+      (_input: unknown, fn: (span: { setAttribute: typeof setAttribute }) => Promise<unknown>) =>
+        fn({ setAttribute })
+    ),
+  }
+})
 
 vi.mock('@/lib/copilot/request/metrics', () => ({
   recordSimToolMetric,
 }))
 
 vi.mock('@/lib/copilot/request/otel', () => ({
-  withCopilotToolSpan: (
-    _input: unknown,
-    fn: (span: { setAttribute: typeof setAttribute }) => Promise<unknown>
-  ) => fn({ setAttribute }),
+  withCopilotToolSpan,
 }))
 
 import { TOOL_WATCHDOG_DEFAULT_MS, TOOL_WATCHDOG_LONG_RUNNING_MS } from '@/lib/copilot/constants'
@@ -110,6 +114,10 @@ describe('executeToolAndReport metrics', () => {
       MothershipStreamV1ToolOutcome.success,
       expect.any(Number)
     )
+    expect(withCopilotToolSpan).toHaveBeenCalledWith(
+      expect.objectContaining({ agentName: 'workflow' }),
+      expect.any(Function)
+    )
   })
 
   it.each([
@@ -138,6 +146,10 @@ describe('executeToolAndReport metrics', () => {
         MothershipStreamV1ToolOutcome.error,
         expect.any(Number)
       )
+      expect(withCopilotToolSpan).toHaveBeenCalledWith(
+        expect.objectContaining({ agentName: expectedAgentId }),
+        expect.any(Function)
+      )
     }
   )
 })

apps/sim/lib/copilot/request/tools/executor.ts

@@ -408,6 +408,7 @@ export async function executeToolAndReport(
     {
       toolName: toolCall.name,
       toolCallId: toolCall.id,
+      agentName: toolCall.agentId ?? 'main',
       runId: context.runId,
       chatId: execContext.chatId,
       argsBytes: argsPayload?.length,