diff --git a/src/renderer/src/pages/home/Messages/Message.tsx b/src/renderer/src/pages/home/Messages/Message.tsx
index 29df6acfbc..baa41390e0 100644
--- a/src/renderer/src/pages/home/Messages/Message.tsx
+++ b/src/renderer/src/pages/home/Messages/Message.tsx
@@ -174,7 +174,8 @@ const MessageItem: FC<Props> = ({
             display: 'flex',
             flexDirection: 'column',
             alignSelf: isAssistantMessage ? 'flex-start' : 'flex-end',
-            flex: isBubbleStyle ? undefined : 1
+            flex: 1,
+            maxWidth: '100%'
           }}>
           <MessageHeader
             message={message}
diff --git a/src/renderer/src/pages/home/Messages/MessageGroup.tsx b/src/renderer/src/pages/home/Messages/MessageGroup.tsx
index 0c04122f99..efc9cd8f5d 100644
--- a/src/renderer/src/pages/home/Messages/MessageGroup.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageGroup.tsx
@@ -180,7 +180,8 @@ const MessageGroup = ({ messages, topic, registerMessageElement }: Props) => {
           $isGrouped={isGrouped}
           key={message.id}
           className={classNames({
-            'group-message-wrapper': message.role === 'assistant' && isHorizontal && isGrouped,
+            // 加个卡片布局
+            'group-message-wrapper': message.role === 'assistant' && (isHorizontal || isGrid) && isGrouped,
             [multiModelMessageStyle]: isGrouped,
             selected: message.id === selectedMessageId
           })}>