@@ -20,7 +20,7 @@ import { useElapsedTime } from './hooks/use-elapsed-time'
2020import { useInputHistory } from './hooks/use-input-history'
2121import { useKeyboardHandlers } from './hooks/use-keyboard-handlers'
2222import { useMessageQueue } from './hooks/use-message-queue'
23- import { useMessageRenderer } from './hooks/use- message-renderer'
23+ import { MessageRenderer } from './components/ message-renderer'
2424import { useChatScrollbox } from './hooks/use-scroll-management'
2525import { useSendMessage } from './hooks/use-send-message'
2626import { useSuggestionEngine } from './hooks/use-suggestion-engine'
@@ -761,22 +761,7 @@ export const Chat = ({
761761 topLevelMessages . length - virtualTopLevelMessages . length ,
762762 )
763763
764- const messageItems = useMessageRenderer ( {
765- messages,
766- messageTree,
767- topLevelMessages : virtualTopLevelMessages ,
768- availableWidth : separatorWidth ,
769- theme,
770- markdownPalette,
771- collapsedAgents,
772- streamingAgents,
773- isWaitingForResponse,
774- timer : mainAgentTimer ,
775- setCollapsedAgents,
776- setFocusedAgentId,
777- userOpenedAgents,
778- setUserOpenedAgents,
779- } )
764+
780765
781766 const virtualizationNotice =
782767 shouldVirtualize && hiddenTopLevelCount > 0 ? (
@@ -864,7 +849,22 @@ export const Chat = ({
864849 >
865850 { headerContent }
866851 { virtualizationNotice }
867- { messageItems }
852+ < MessageRenderer
853+ messages = { messages }
854+ messageTree = { messageTree }
855+ topLevelMessages = { virtualTopLevelMessages }
856+ availableWidth = { separatorWidth }
857+ theme = { theme }
858+ markdownPalette = { markdownPalette }
859+ collapsedAgents = { collapsedAgents }
860+ streamingAgents = { streamingAgents }
861+ isWaitingForResponse = { isWaitingForResponse }
862+ timer = { mainAgentTimer }
863+ setCollapsedAgents = { setCollapsedAgents }
864+ setFocusedAgentId = { setFocusedAgentId }
865+ userOpenedAgents = { userOpenedAgents }
866+ setUserOpenedAgents = { setUserOpenedAgents }
867+ />
868868 </ scrollbox >
869869 </ box >
870870
0 commit comments