diff --git a/src/uncloseai-modal-builtin.css b/src/uncloseai-modal-builtin.css
index f7d27a9..4a786df 100644
--- a/src/uncloseai-modal-builtin.css
+++ b/src/uncloseai-modal-builtin.css
@@ -1122,21 +1122,13 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .uncloseai-chat-box {
 /* Code blocks within AI messages - allow horizontal scroll, preserve all whitespace */
 .ai-message pre {
   overflow-x: auto;
-  max-width: 340px;
+  max-width: 100%;
+  box-sizing: border-box;
 }
 
 .ai-message code {
-  max-width: 340px;
-}
-
-@media (min-width: 768px) {
-  .ai-message pre {
-    max-width: 720px;
-  }
-
-  .ai-message code {
-    max-width: 720px;
-  }
+  max-width: 100%;
+  box-sizing: border-box;
 }
 
 .message-text {
diff --git a/src/uncloseai-modal-pico.css b/src/uncloseai-modal-pico.css
index 80d7ab0..b511d37 100644
--- a/src/uncloseai-modal-pico.css
+++ b/src/uncloseai-modal-pico.css
@@ -1195,21 +1195,13 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .uncloseai-chat-box {
 /* Code blocks within AI messages - allow horizontal scroll, preserve all whitespace */
 .ai-message pre {
   overflow-x: auto;
-  max-width: 340px;
+  max-width: 100%;
+  box-sizing: border-box;
 }
 
 .ai-message code {
-  max-width: 340px;
-}
-
-@media (min-width: 768px) {
-  .ai-message pre {
-    max-width: 720px;
-  }
-
-  .ai-message code {
-    max-width: 720px;
-  }
+  max-width: 100%;
+  box-sizing: border-box;
 }
 
 .message-text {