fix typecheck

Author: charleslien

cli/src/components/__tests__/message-block.completion.test.tsx

@@ -1,11 +1,11 @@
-import React from 'react'
-
 import { describe, test, expect } from 'bun:test'
+import React from 'react'
 import { renderToStaticMarkup } from 'react-dom/server'
 
-import { MessageBlock } from '../message-block'
 import { initializeThemeStore } from '../../hooks/use-theme'
 import { chatThemes, createMarkdownPalette } from '../../utils/theme-system'
+import { MessageBlock } from '../message-block'
+
 import type { MarkdownPalette } from '../../utils/markdown-renderer'
 
 initializeThemeStore()
@@ -41,10 +41,13 @@ const baseProps = {
   availableWidth: 80,
   markdownPalette: basePalette,
   collapsedAgents: new Set<string>(),
+  autoCollapsedAgents: new Set<string>(),
   streamingAgents: new Set<string>(),
   onToggleCollapsed: () => {},
   onBuildFast: () => {},
   onBuildMax: () => {},
+  setCollapsedAgents: () => {},
+  addAutoCollapsedAgent: () => {},
 }
 
 describe('MessageBlock completion time', () => {

cli/src/components/__tests__/message-block.streaming.test.tsx

@@ -1,11 +1,11 @@
-import React from 'react'
-
 import { describe, test, expect } from 'bun:test'
+import React from 'react'
 import { renderToStaticMarkup } from 'react-dom/server'
 
-import { MessageBlock } from '../message-block'
 import { initializeThemeStore } from '../../hooks/use-theme'
 import { chatThemes, createMarkdownPalette } from '../../utils/theme-system'
+import { MessageBlock } from '../message-block'
+
 import type { MarkdownPalette } from '../../utils/markdown-renderer'
 
 const theme = chatThemes.dark
@@ -37,10 +37,13 @@ const baseProps = {
   availableWidth: 80,
   markdownPalette: basePalette,
   collapsedAgents: new Set<string>(),
+  autoCollapsedAgents: new Set<string>(),
   streamingAgents: new Set<string>(),
   onToggleCollapsed: () => {},
   onBuildFast: () => {},
   onBuildMax: () => {},
+  setCollapsedAgents: () => {},
+  addAutoCollapsedAgent: () => {},
 }
 
 const createTimerStartTime = (elapsedSeconds: number): number | null =>