diff --git a/components/AIAssist.tsx b/components/AIAssist.tsx
index 28b4780..18bc235 100644
--- a/components/AIAssist.tsx
+++ b/components/AIAssist.tsx
@@ -973,9 +973,10 @@ export default function AIAssist() {
                                             remarkPlugins={[remarkGfm]}
                                             rehypePlugins={[rehypeHighlight]}
                                             components={{
-                                                code({ node, inline, className, children, ...props }) {
+                                                code({ node, className, children, ...props }: any) {
                                                     const match = /language-(\w+)/.exec(className || "");
-                                                    return !inline ? (
+                                                    const isInline = !match && !String(children).includes('\n');
+                                                    return !isInline ? (
                                                         <div className="relative group/code w-full overflow-hidden">
                                                             <pre className={cn(
                                                                 "p-4 rounded-xl bg-slate-950/50 border border-white/5 overflow-x-auto scrollbar-thin scrollbar-thumb-white/10",
@@ -993,6 +994,7 @@ export default function AIAssist() {
                                                     );
                                                 },
                                             }}
+
                                         >
                                             {parseStreamingContent(msg.content, msg.agent || "general").chatDisplay || (msg.role === "assistant" ? "..." : "")}
                                         </ReactMarkdown>