chore(chat): enforce activity icon catalog coverage

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/home/components/message-content/utils.ts

@@ -34,7 +34,7 @@ import type { ToolCallStatus } from '@/app/workspace/[workspaceId]/home/types'
 
 export type IconComponent = ComponentType<SVGProps<SVGSVGElement>>
 
-const TOOL_ICONS: Record<string, IconComponent> = {
+export const TOOL_ICONS: Readonly<Record<string, IconComponent>> = {
   agent: Brain,
   apply_file_edit: File,
   auth: Integration,
@@ -87,6 +87,7 @@ const TOOL_ICONS: Record<string, IconComponent> = {
   diff_workflows: Layout,
   download_file: File,
   edit_workflow: Pencil,
+  extensions: Brain,
   extract_doc_assets: File,
   ffmpeg: Wrench,
   file: File,

apps/sim/lib/copilot/tools/tool-activity.test.ts

@@ -6,14 +6,26 @@ import { describe, expect, it } from 'vitest'
 import { TOOL_CATALOG } from '@/lib/copilot/generated/tool-catalog-v1'
 import { isToolHiddenInUi } from '@/lib/copilot/tools/client/hidden-tools'
 import { getToolActivityLabel, TOOL_ACTIVITIES } from '@/lib/copilot/tools/tool-activity'
+import { TOOL_ICONS } from '@/app/workspace/[workspaceId]/home/components/message-content/utils'
 
 const visibleTools = Object.values(TOOL_CATALOG).filter(
   (tool) => tool.route !== 'subagent' && !tool.hidden && !isToolHiddenInUi(tool.id)
 )
 
 describe('tool activity catalog coverage', () => {
+  it('covers subagent icons under their dispatch and stream names', () => {
+    for (const tool of Object.values(TOOL_CATALOG)) {
+      if (tool.route !== 'subagent') continue
+      expect(Object.hasOwn(TOOL_ICONS, tool.id), tool.id).toBe(true)
+      if (tool.subagentId) {
+        expect(Object.hasOwn(TOOL_ICONS, tool.subagentId), tool.subagentId).toBe(true)
+      }
+    }
+  })
+
   it.each(visibleTools)('explicitly describes $id and every declared operation', (tool) => {
-    expect(Object.hasOwn(TOOL_ACTIVITIES, tool.id), tool.id).toBe(true)
+    expect(Object.hasOwn(TOOL_ACTIVITIES, tool.id), `${tool.id} summary`).toBe(true)
+    expect(Object.hasOwn(TOOL_ICONS, tool.id), `${tool.id} icon`).toBe(true)
     const activity = TOOL_ACTIVITIES[tool.id]
     const properties = isRecordLike(tool.parameters) ? tool.parameters.properties : undefined
     if (!isRecordLike(properties)) return