diff --git a/apps/desktop/layer/renderer/src/modules/ai-chat/components/message/AIChatMessage.tsx b/apps/desktop/layer/renderer/src/modules/ai-chat/components/message/AIChatMessage.tsx
index 139301a06..52d53d41d 100644
--- a/apps/desktop/layer/renderer/src/modules/ai-chat/components/message/AIChatMessage.tsx
+++ b/apps/desktop/layer/renderer/src/modules/ai-chat/components/message/AIChatMessage.tsx
@@ -105,9 +105,9 @@ export const AIChatMessage: React.FC<AIChatMessageProps> = React.memo(
 
     return (
       <div onContextMenu={stopPropagation} className="group flex justify-start">
-        <div className="relative flex max-w-full flex-col gap-2 text-text">
+        <div className="relative flex w-full max-w-full flex-col gap-2 text-text">
           {/* Normal message display */}
-          <div className="text-text">
+          <div className="w-full text-text">
             <div className="flex cursor-text select-text flex-col gap-2 text-sm">
               <AIMessageParts message={message} isLastMessage={isLastMessage} />
             </div>