diff --git a/apps/desktop/src/components/EnhancedChatMessageV2.tsx b/apps/desktop/src/components/EnhancedChatMessageV2.tsx
index 7319b8e..8aa855c 100644
--- a/apps/desktop/src/components/EnhancedChatMessageV2.tsx
+++ b/apps/desktop/src/components/EnhancedChatMessageV2.tsx
@@ -117,7 +117,7 @@ export const EnhancedChatMessageV2: React.FC<EnhancedChatMessageV2Props> = ({
       </div>
 
       {/* 消息内容 */}
-      <div className={`flex-1 max-w-3xl ${isUser ? 'text-right' : 'text-left'}`}>
+      <div className={`flex-1 max-w-full ${isUser ? 'text-right' : 'text-left'}`}>
         {/* 消息气泡 */}
         <div className={`
           inline-block p-4 rounded-2xl chat-message-bubble
@@ -203,7 +203,7 @@ export const EnhancedChatMessageV2: React.FC<EnhancedChatMessageV2Props> = ({
             </div>
 
             {expandedSources && (
-              <div className="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 gap-4">
+              <div className="grid grid-cols-1 sm:grid-cols-2 md:grid-cols-3 lg:grid-cols-4 xl:grid-cols-5 2xl:grid-cols-6 gap-4">
                 {sources.map((source, index) => (
                   <ImageCard
                     key={index}
diff --git a/apps/desktop/src/components/EnhancedMarkdownRenderer.tsx b/apps/desktop/src/components/EnhancedMarkdownRenderer.tsx
index d77da3b..c36e9b8 100644
--- a/apps/desktop/src/components/EnhancedMarkdownRenderer.tsx
+++ b/apps/desktop/src/components/EnhancedMarkdownRenderer.tsx
@@ -456,7 +456,7 @@ export const EnhancedMarkdownRenderer: React.FC<EnhancedMarkdownRendererProps> =
 
           <div className="bg-gradient-to-br from-gray-50 to-gray-100 rounded-xl p-6 shadow-inner">
             {expandedGrounding.groundingInfo?.sources && expandedGrounding.groundingInfo.sources.length > 0 ? (
-              <div className="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 gap-6">
+              <div className="grid grid-cols-1 sm:grid-cols-2 md:grid-cols-3 lg:grid-cols-4 xl:grid-cols-5 2xl:grid-cols-6 gap-6">
                 {expandedGrounding.groundingInfo.sources.map((source: any, index: number) => (
                   <div key={index} className="relative transform hover:scale-105 transition-transform duration-200">
                     <ImageCard