From a05203644096e836aca712cd30123cd7e8d5d97e Mon Sep 17 00:00:00 2001
From: Nate Kelley <nate@buster.so>
Date: Sat, 19 Apr 2025 23:22:24 -0600
Subject: [PATCH] prefetch chat home

---
 .../ChatResponseReasoning.tsx                     |  2 +-
 .../FileContainerHeader/FileContainerHeader.tsx   | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/web/src/layouts/ChatLayout/ChatContainer/ChatContent/ChatResponseMessages/ChatResponseReasoning.tsx b/web/src/layouts/ChatLayout/ChatContainer/ChatContent/ChatResponseMessages/ChatResponseReasoning.tsx
index 614dbcd1e..f4ac9897b 100644
--- a/web/src/layouts/ChatLayout/ChatContainer/ChatContent/ChatResponseMessages/ChatResponseReasoning.tsx
+++ b/web/src/layouts/ChatLayout/ChatContainer/ChatContent/ChatResponseMessages/ChatResponseReasoning.tsx
@@ -65,7 +65,7 @@ export const ChatResponseReasoning: React.FC<{
   }, [isReasonginFileSelected, messageId, chatId]);
 
   return (
-    <Link href={href}>
+    <Link href={href} prefetch>
       <AnimatePresence initial={!isCompletedStream} mode="wait">
         <motion.div
           {...animations}
diff --git a/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeader.tsx b/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeader.tsx
index 717a18a42..57243a47d 100644
--- a/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeader.tsx
+++ b/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeader.tsx
@@ -6,6 +6,9 @@ import { useChatLayoutContextSelector } from '../../ChatLayoutContext';
 import { FileContainerHeaderVersionHistory } from './FileContainerHeaderVersionHistory';
 import { SelectedFileButtonsRecord, SelectedFileSegmentRecord } from './config';
 import { FileType } from '@/api/asset_interfaces/chat';
+import { useRouter } from 'next/router';
+import { useMount } from '@/hooks';
+import { BusterRoutes, createBusterRoute } from '@/routes';
 
 export const FileContainerHeader: React.FC = React.memo(() => {
   const selectedFileType = useChatLayoutContextSelector((x) => x.selectedFile?.type);
@@ -28,6 +31,7 @@ const FileContainerHeaderStandard: React.FC<{
   selectedFileId: string | undefined;
   selectedFileType: FileType | undefined;
 }> = ({ selectedFileId, selectedFileType }) => {
+  const router = useRouter();
   const chatId = useChatLayoutContextSelector((x) => x.chatId);
   const selectedFileView = useChatLayoutContextSelector((x) => x.selectedFileView);
   const onCollapseFileClick = useChatLayoutContextSelector((state) => state.onCollapseFileClick);
@@ -50,6 +54,17 @@ const FileContainerHeaderStandard: React.FC<{
     [selectedFileType]
   );
 
+  useMount(() => {
+    if (chatId) {
+      router.prefetch(
+        createBusterRoute({
+          route: BusterRoutes.APP_CHAT_ID,
+          chatId
+        })
+      );
+    }
+  });
+
   return (
     <>
       <div className="flex min-w-0 shrink items-center gap-1.5 overflow-hidden">