From 94788cda53fa52783fdf92b17550bd91e7e743f9 Mon Sep 17 00:00:00 2001
From: pablonyx <pablo@danswer.ai>
Date: Fri, 7 Feb 2025 18:07:47 -0800
Subject: [PATCH] Update display (#3934)

* update display

* quick nit
---
 backend/onyx/db/models.py                      | 2 +-
 web/src/app/chat/ChatPage.tsx                  | 2 +-
 web/src/app/chat/message/Messages.tsx          | 6 +++++-
 web/src/app/chat/message/SourcesDisplay.tsx    | 1 -
 web/src/components/chat/sources/SourceCard.tsx | 4 ++--
 5 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/backend/onyx/db/models.py b/backend/onyx/db/models.py
index 3256fbebe..4931b37de 100644
--- a/backend/onyx/db/models.py
+++ b/backend/onyx/db/models.py
@@ -152,7 +152,7 @@ class User(SQLAlchemyBaseUserTableUUID, Base):
     # if not specified, all assistants are shown
     temperature_override_enabled: Mapped[bool] = mapped_column(Boolean, default=False)
     auto_scroll: Mapped[bool] = mapped_column(Boolean, default=True)
-    shortcut_enabled: Mapped[bool] = mapped_column(Boolean, default=True)
+    shortcut_enabled: Mapped[bool] = mapped_column(Boolean, default=False)
     chosen_assistants: Mapped[list[int] | None] = mapped_column(
         postgresql.JSONB(), nullable=True, default=None
     )
diff --git a/web/src/app/chat/ChatPage.tsx b/web/src/app/chat/ChatPage.tsx
index adb591185..0915f22ec 100644
--- a/web/src/app/chat/ChatPage.tsx
+++ b/web/src/app/chat/ChatPage.tsx
@@ -1612,7 +1612,7 @@ export function ChatPage({
                 second_level_message: second_level_answer,
                 type: error ? "error" : "assistant",
                 retrievalType,
-                query: finalMessage?.rephrased_query || query,
+                query: finalMessage?.rephrased_query,
                 documents: documents,
                 citations: finalMessage?.citations || {},
                 files: finalMessage?.files || aiMessageImages || [],
diff --git a/web/src/app/chat/message/Messages.tsx b/web/src/app/chat/message/Messages.tsx
index 3f4cf20fc..242f89039 100644
--- a/web/src/app/chat/message/Messages.tsx
+++ b/web/src/app/chat/message/Messages.tsx
@@ -483,7 +483,11 @@ export const AIMessage = ({
                       )}
 
                     {docs && docs.length > 0 && (
-                      <div className="mobile:hidden mt-2 -mx-8 w-full mb-4 flex relative">
+                      <div
+                        className={`mobile:hidden ${
+                          query && "mt-2"
+                        }  -mx-8 w-full mb-4 flex relative`}
+                      >
                         <div className="w-full">
                           <div className="px-8 flex gap-x-2">
                             {!settings?.isMobile &&
diff --git a/web/src/app/chat/message/SourcesDisplay.tsx b/web/src/app/chat/message/SourcesDisplay.tsx
index d3c2b1220..bd2111473 100644
--- a/web/src/app/chat/message/SourcesDisplay.tsx
+++ b/web/src/app/chat/message/SourcesDisplay.tsx
@@ -102,7 +102,6 @@ export const SourcesDisplay: React.FC<SourcesDisplayProps> = ({
             />
           </div>
         ))}
-
         {hasMoreDocuments && (
           <SeeMoreBlock
             fullWidth
diff --git a/web/src/components/chat/sources/SourceCard.tsx b/web/src/components/chat/sources/SourceCard.tsx
index 1494578cf..13be7651c 100644
--- a/web/src/components/chat/sources/SourceCard.tsx
+++ b/web/src/components/chat/sources/SourceCard.tsx
@@ -148,7 +148,7 @@ export function SeeMoreBlock({
   toggled,
   fullWidth = false,
 }: SeeMoreBlockProps) {
-  const iconsToRender = getUniqueIcons(docs);
+  const iconsToRender = docs.length > 2 ? getUniqueIcons(docs) : [];
 
   return (
     <button
@@ -157,7 +157,7 @@ export function SeeMoreBlock({
         h-[80px] p-3 border border-[1.5px] border-[#D9D1c0] bg-[#f1eee8] text-left hover:bg-[#ebe7de] cursor-pointer rounded-lg flex flex-col justify-between overflow-hidden`}
     >
       <div className="flex items-center gap-1">
-        {iconsToRender.map((icon, index) => icon)}
+        {docs.length > 2 && iconsToRender.map((icon, index) => icon)}
       </div>
       <div className="text-text-darker text-xs font-semibold">
         {toggled ? "Hide Results" : "Show All"}