diff --git a/docs/components/floating-ai-search.tsx b/docs/components/floating-ai-search.tsx
index a59320da..0db2b575 100644
--- a/docs/components/floating-ai-search.tsx
+++ b/docs/components/floating-ai-search.tsx
@@ -319,6 +319,18 @@ function Message({
 		}
 	}
 
+	// Fix incomplete code blocks for better rendering during streaming
+	const codeBlockCount = (markdown.match(/```/g) || []).length;
+	if (codeBlockCount % 2 !== 0) {
+		// Odd number of ``` means there's an unclosed code block
+		markdown += "\n```";
+	}
+
+	// Ensure proper spacing around code blocks
+	markdown = markdown
+		.replace(/```(\w+)?\n/g, "\n```$1\n")
+		.replace(/\n```\n/g, "\n```\n\n");
+
 	return (
 		<div {...props}>
 			<p