k

web/src/app/chat/message/AgenticMessage.tsx

@@ -48,7 +48,6 @@ import rehypeKatex from "rehype-katex";
 import "katex/dist/katex.min.css";
 import SubQuestionsDisplay from "./SubQuestionsDisplay";
 import { StatusRefinement } from "../Refinement";
-import SubQuestionProgress from "./SubQuestionProgress";
 
 export const AgenticMessage = ({
   docSidebarToggled,
@@ -445,6 +444,7 @@ export const AgenticMessage = ({
                         <div className="flex items-center gap-x-2 px-4">
                           <div className="text-black text-lg font-medium">
                             Answer
+                            {isViewingInitialAnswer ? "Original" : "Refined"}
                           </div>
 
                           <StatusRefinement

web/src/app/chat/message/SubQuestionsDisplay.tsx

@@ -525,9 +525,12 @@ const SubQuestionsDisplay: React.FC<SubQuestionsDisplayProps> = ({
 
   useEffect(() => {
     if (documents && documents.length > 0) {
-      setTimeout(() => {
-        setShownDocuments(documents);
-      }, 800);
+      setTimeout(
+        () => {
+          setShownDocuments(documents);
+        },
+        finishedGenerating ? 0 : 800
+      );
     }
   }, [documents]);
 

web/src/app/globals.css

@@ -656,6 +656,11 @@ ul > li > p {
   color: white;
 }
 
-.dark li {
+.dark li,
+.dark h1,
+.dark h2,
+.dark h3,
+.dark h4,
+.dark h5 {
   color: #e5e5e5;
 }