fix aiwrite rerender

Author: tcm390

packages/client/src/components/chat.tsx

@@ -39,6 +39,7 @@ import { Tabs, TabsContent, TabsList, TabsTrigger } from './ui/tabs';
 import { Tooltip, TooltipContent, TooltipTrigger } from './ui/tooltip';
 import { AgentMemoryViewer } from './memory-viewer';
 import { Collapsible, CollapsibleContent, CollapsibleTrigger } from '@radix-ui/react-collapsible';
+import React from 'react';
 
 const SOURCE_NAME = 'client_chat';
 
@@ -50,6 +51,8 @@ type ExtraContentFields = {
 
 type ContentWithUser = Content & ExtraContentFields;
 
+const MemoizedMessageContent = React.memo(MessageContent);
+
 function MessageContent({
   message,
   agentId,
@@ -475,7 +478,7 @@ export default function Page({ agentId }: { agentId: UUID }) {
                       </Avatar>
                     )}
 
-                    <MessageContent
+                    <MemoizedMessageContent
                       message={message}
                       agentId={agentId}
                       isLastMessage={index === messages.length - 1}