hotfix to stop the loader from always spinning on code blocks

Author: sestinj

gui/src/components/StyledMarkdownPreview/StepContainerPreToolbar/index.tsx

@@ -52,7 +52,7 @@ export interface StepContainerPreToolbarProps {
   codeBlockContent: string;
   language: string | null;
   relativeFilepath?: string;
-  isGeneratingCodeBlock: boolean;
+  isFinalCodeblock: boolean;
   codeBlockIndex: number; // To track which codeblock we are applying
   codeBlockStreamId: string;
   range?: string;
@@ -65,7 +65,7 @@ export function StepContainerPreToolbar({
   codeBlockContent,
   language,
   relativeFilepath,
-  isGeneratingCodeBlock,
+  isFinalCodeblock,
   codeBlockIndex,
   codeBlockStreamId,
   range,
@@ -115,6 +115,9 @@ export function StepContainerPreToolbar({
   const hasFileExtension =
     relativeFilepath && /\.[0-9a-z]+$/i.test(relativeFilepath);
 
+  const isStreaming = useAppSelector((store) => store.session.isStreaming);
+  const isGeneratingCodeBlock = isFinalCodeblock && isStreaming;
+
   // If we are creating a file, we already render that in the button
   // so we don't want to dispaly it twice here
   const displayFilepath = relativeFilepath ?? appliedFileUri;

gui/src/components/StyledMarkdownPreview/index.tsx

@@ -310,6 +310,9 @@ const StyledMarkdownPreview = memo(function StyledMarkdownPreview(
 
           const language = getLanguageFromClassName(className);
 
+          const isFinalCodeblock =
+            preChildProps["data-islastcodeblock"] && isLastItemRef.current;
+
           const isGeneratingCodeBlock =
             preChildProps["data-islastcodeblock"] &&
             isLastItemRef.current &&
@@ -331,7 +334,7 @@ const StyledMarkdownPreview = memo(function StyledMarkdownPreview(
               codeBlockIndex={codeBlockIndex}
               language={language}
               relativeFilepath={relativeFilePath}
-              isGeneratingCodeBlock={isGeneratingCodeBlock}
+              isFinalCodeblock={isFinalCodeblock}
               range={range}
               codeBlockStreamId={
                 codeblockState.current[codeBlockIndex].streamId
@@ -381,7 +384,7 @@ const StyledMarkdownPreview = memo(function StyledMarkdownPreview(
   const codeWrapState = uiConfig?.codeWrap ? "pre-wrap" : "pre";
   return (
     <StyledMarkdown
-      contentEditable='false'
+      contentEditable="false"
       fontSize={getFontSize()}
       whiteSpace={codeWrapState}
       bgColor={props.useParentBackgroundColor ? "" : vscBackground}