From 181acde135e026e174f08cd51344a88cbb204c94 Mon Sep 17 00:00:00 2001
From: Bereket Engida <Bekacru@gmail.com>
Date: Fri, 10 Oct 2025 19:10:53 -0700
Subject: [PATCH] docs: fix code formatting

---
 docs/components/floating-ai-search.tsx | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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