fix internet search icons / text + assistants tab (#1862)

web/src/app/assistants/SidebarWrapper.tsx

@@ -143,6 +143,7 @@ export default function SidebarWrapper<T extends object>({
           <div className="mt-4 mx-auto">{content(contentProps)}</div>
         </div>
       </div>
+      <FixedLogo />
     </div>
   );
 }

web/src/app/assistants/mine/page.tsx

@@ -67,9 +67,6 @@ export default async function GalleryPage({
           user={user}
           assistants={assistants}
         />
-
-        {/* Temporary - fixed logo */}
-        <FixedLogo />
       </ChatProvider>
     </>
   );

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

@@ -57,6 +57,8 @@ import {
 import { ValidSources } from "@/lib/types";
 import { Tooltip } from "@/components/tooltip/Tooltip";
 import { useMouseTracking } from "./hooks";
+import { InternetSearchIcon } from "@/components/InternetSearchIcon";
+import { getTitleFromDocument } from "@/lib/sources";
 
 const TOOLS_WITH_CUSTOM_HANDLING = [
   SEARCH_TOOL_NAME,
@@ -443,17 +445,17 @@ export const AIMessage = ({
                                   >
                                     <Citation link={doc.link} index={ind + 1} />
                                     <p className="shrink truncate ellipsis break-all ">
-                                      {
+                                      {getTitleFromDocument(doc)}
-                                        doc.document_id.split("/")[
-                                          doc.document_id.split("/").length - 1
-                                        ]
-                                      }
                                     </p>
                                     <div className="ml-auto flex-none">
-                                      <SourceIcon
+                                      {doc.is_internet ? (
-                                        sourceType={doc.source_type}
+                                        <InternetSearchIcon url={doc.link} />
-                                        iconSize={18}
+                                      ) : (
-                                      />
+                                        <SourceIcon
+                                          sourceType={doc.source_type}
+                                          iconSize={18}
+                                        />
+                                      )}
                                     </div>
                                   </a>
                                   <div className="flex overscroll-x-scroll mt-.5">

web/src/lib/sources.ts

@@ -38,7 +38,11 @@ import {
   ColorSlackIcon,
 } from "@/components/icons/icons";
 import { ValidSources } from "./types";
-import { SourceCategory, SourceMetadata } from "./search/interfaces";
+import {
+  DanswerDocument,
+  SourceCategory,
+  SourceMetadata,
+} from "./search/interfaces";
 import { Persona } from "@/app/admin/assistants/interfaces";
 import { FaAccessibleIcon, FaSlack } from "react-icons/fa";
 
@@ -290,3 +294,16 @@ export function getSourcesForPersona(persona: Persona): ValidSources[] {
   });
   return personaSources;
 }
+
+function stripTrailingSlash(str: string) {
+  if (str.substr(-1) === "/") {
+    return str.substr(0, str.length - 1);
+  }
+  return str;
+}
+
+export const getTitleFromDocument = (document: DanswerDocument) => {
+  return stripTrailingSlash(document.document_id).split("/")[
+    stripTrailingSlash(document.document_id).split("/").length - 1
+  ];
+};