diff --git a/src/components/chat/ChatHeader.tsx b/src/components/chat/ChatHeader.tsx
index 7a75080..2c8d806 100644
--- a/src/components/chat/ChatHeader.tsx
+++ b/src/components/chat/ChatHeader.tsx
@@ -26,6 +26,8 @@ import { useStreamChat } from "@/hooks/useStreamChat";
import { useCurrentBranch } from "@/hooks/useCurrentBranch";
import { useCheckoutVersion } from "@/hooks/useCheckoutVersion";
import { useRenameBranch } from "@/hooks/useRenameBranch";
+import { isAnyCheckoutVersionInProgressAtom } from "@/store/appAtoms";
+import { LoadingBar } from "../ui/LoadingBar";
interface ChatHeaderProps {
isVersionPaneOpen: boolean;
@@ -46,6 +48,9 @@ export function ChatHeader({
const [selectedChatId, setSelectedChatId] = useAtom(selectedChatIdAtom);
const { refreshChats } = useChats(appId);
const { isStreaming } = useStreamChat();
+ const isAnyCheckoutVersionInProgress = useAtomValue(
+ isAnyCheckoutVersionInProgressAtom,
+ );
const {
branchInfo,
@@ -102,6 +107,7 @@ export function ChatHeader({
return (
+
{/* If the version pane is open, it's expected to not always be on the main branch. */}
{isNotMainBranch && !isVersionPaneOpen && (
@@ -161,7 +167,8 @@ export function ChatHeader({
)}
-
+ {/* Why is this pt-0.5? Because the loading bar is h-1 (it always takes space) and we want the vertical spacing to be consistent.*/}
+