fix(copilot): show custom block names in read tool rows

Author: j15z

apps/sim/lib/copilot/tools/client/read-block.test.ts

@@ -5,9 +5,18 @@ import { describe, expect, it, vi } from 'vitest'
 import { getReadTargetBlock } from '@/lib/copilot/tools/client/read-block'
 
 const gmailBlock = { type: 'gmail_v2', name: 'Gmail', icon: () => null }
+const customBlock = {
+  type: 'custom_block_invoice_parser',
+  name: 'Invoice Parser',
+  icon: () => null,
+}
 
 vi.mock('@/blocks/registry', () => ({
-  getBlock: vi.fn((type: string) => (type === 'gmail_v2' ? gmailBlock : undefined)),
+  getBlock: vi.fn((type: string) => {
+    if (type === 'gmail_v2') return gmailBlock
+    if (type === 'custom_block_invoice_parser') return customBlock
+    return undefined
+  }),
   getLatestBlock: vi.fn((baseType: string) => (baseType === 'gmail' ? gmailBlock : undefined)),
 }))
 
@@ -21,6 +30,12 @@ describe('getReadTargetBlock', () => {
     expect(getReadTargetBlock('components/integrations/gmail')?.name).toBe('Gmail')
   })
 
+  it('resolves an organization custom-block read to its block', () => {
+    expect(
+      getReadTargetBlock('organization/custom-blocks/custom_block_invoice_parser.json')?.name
+    ).toBe('Invoice Parser')
+  })
+
   it('returns undefined for unknown blocks and non-component paths', () => {
     expect(getReadTargetBlock('components/blocks/unknown_block.json')).toBeUndefined()
     expect(getReadTargetBlock('workflows/My Workflow/meta.json')).toBeUndefined()

apps/sim/lib/copilot/tools/client/read-block.ts

@@ -2,16 +2,20 @@ import { getBlock, getLatestBlock } from '@/blocks/registry'
 import type { BlockConfig } from '@/blocks/types'
 
 /**
- * Resolves the block a copilot `read` call targets when the path is a
- * component schema — `components/blocks/{type}.json` or
- * `components/integrations/{service}/{operation}.json` — so tool rows can show
- * the block's display name and brand icon instead of the raw type id
- * (e.g. "Gmail" instead of `gmail_v2`). Returns undefined for every other
- * path, leaving the generic read-target labeling untouched.
+ * Resolves the block a copilot `read` call targets when the path references a
+ * component schema or organization custom block, so tool rows can show the
+ * block's display name and brand icon instead of its raw type id. Returns
+ * undefined for every other path, leaving generic read-target labeling
+ * untouched.
  */
 export function getReadTargetBlock(path: string | undefined): BlockConfig | undefined {
   if (!path) return undefined
   const segments = path.trim().split('/').filter(Boolean)
+
+  if (segments[0] === 'organization' && segments[1] === 'custom-blocks' && segments.length === 3) {
+    return getBlock(segments[2].replace(/\.json$/, ''))
+  }
+
   if (segments[0] !== 'components' || segments.length < 3) return undefined
   if (segments[1] === 'blocks' && segments.length === 3) {
     return getBlock(segments[2].replace(/\.json$/, ''))

apps/sim/lib/copilot/tools/client/store-utils.test.ts

@@ -8,9 +8,18 @@ import { resolveToolDisplay } from './store-utils'
 import { ClientToolCallState } from './tool-call-state'
 
 const gmailBlock = { type: 'gmail_v2', name: 'Gmail', icon: () => null }
+const customBlock = {
+  type: 'custom_block_invoice_parser',
+  name: 'Invoice Parser',
+  icon: () => null,
+}
 
 vi.mock('@/blocks/registry', () => ({
-  getBlock: vi.fn((type: string) => (type === 'gmail_v2' ? gmailBlock : undefined)),
+  getBlock: vi.fn((type: string) => {
+    if (type === 'gmail_v2') return gmailBlock
+    if (type === 'custom_block_invoice_parser') return customBlock
+    return undefined
+  }),
   getLatestBlock: vi.fn((baseType: string) => (baseType === 'gmail' ? gmailBlock : undefined)),
 }))
 
@@ -180,7 +189,7 @@ describe('resolveToolDisplay', () => {
     ).toBe('Read style details for deck.pptx')
   })
 
-  it('shows the block display name for block and integration schema reads', () => {
+  it('shows the block display name for block, integration, and custom-block reads', () => {
     expect(
       resolveToolDisplay(ReadTool.id, ClientToolCallState.success, {
         path: 'components/blocks/gmail_v2.json',
@@ -198,6 +207,12 @@ describe('resolveToolDisplay', () => {
         path: 'components/blocks/unknown_block.json',
       })?.text
     ).toBe('Read Unknown block')
+
+    expect(
+      resolveToolDisplay(ReadTool.id, ClientToolCallState.success, {
+        path: 'organization/custom-blocks/custom_block_invoice_parser.json',
+      })?.text
+    ).toBe('Read Invoice Parser')
   })
 
   it('humanizes internal VFS resource identifiers', () => {