quick nits

Author: pablonyx

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

@@ -3,38 +3,15 @@ import { LoadedOnyxDocument, OnyxDocument } from "@/lib/search/interfaces";
 import React, { memo } from "react";
 import isEqual from "lodash/isEqual";
 import { SourceIcon } from "@/components/SourceIcon";
-import { SearchResultIcon } from "@/components/SearchResultIcon";
 import { WebResultIcon } from "@/components/WebResultIcon";
 
-const FALLBACK_ICON = "web.svg";
-
-// Updated error handler
-const handleError = (
-  e: React.SyntheticEvent<HTMLImageElement>,
-  associatedDoc: OnyxDocument
-) => {
-  // Prevent future error events
-  e.currentTarget.onerror = null;
-
-  // Repoint image source to fallback
-  e.currentTarget.src = FALLBACK_ICON;
-
-  // If you prefer, log fewer times or log once.
-  // For demonstration, we'll print a single warning:
-  console.warn(
-    `Failed to load favicon for document_id=${associatedDoc?.document_id}. Replacing with fallback icon.`
-  );
-};
-
 export const MemoizedAnchor = memo(
   ({
     docs,
-    iconMap,
     updatePresentingDocument,
     children,
   }: {
     docs?: OnyxDocument[] | null;
-    iconMap?: Record<string, string>;
     updatePresentingDocument: (doc: OnyxDocument) => void;
     children: React.ReactNode;
   }) => {
@@ -48,26 +25,9 @@ export const MemoizedAnchor = memo(
           return <>{children}</>;
         }
 
-        const effectiveIconURL =
-          iconMap?.[associatedDoc.document_id] || "globe.svg";
-
         let icon: React.ReactNode = null;
         if (associatedDoc.source_type === "web") {
           icon = <WebResultIcon url={associatedDoc.link} />;
-          // (
-          //   <img
-          //     className="!m-0 !p-0 rounded-full"
-          //     height={18}
-          //     onError={(e) => {
-          //       handleError(e, associatedDoc);
-          //     }}
-          //     width={18}
-          //     src={`https://www.google.com/s2/favicons?domain=${
-          //       new URL(associatedDoc.link).hostname
-          //     }`}
-          //     alt="Favicon"
-          //   />
-          // );
         } else {
           icon = (
             <SourceIcon sourceType={associatedDoc.source_type} iconSize={18} />

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

@@ -72,7 +72,6 @@ import SourceCard, {
 import remarkMath from "remark-math";
 import rehypeKatex from "rehype-katex";
 import "katex/dist/katex.min.css";
-import { useFaviconPreloader } from "@/hooks/useFaviconPreloader";
 
 const TOOLS_WITH_CUSTOM_HANDLING = [
   SEARCH_TOOL_NAME,
@@ -315,20 +314,16 @@ export const AIMessage = ({
     []
   );
 
-  const { iconMap, handleError } = useFaviconPreloader(docs || []);
-
   const anchorCallback = useCallback(
     (props: any) => (
       <MemoizedAnchor
-        handleError={handleError}
         updatePresentingDocument={setPresentingDocument!}
         docs={docs}
-        iconMap={iconMap}
       >
         {props.children}
       </MemoizedAnchor>
     ),
-    [docs, iconMap]
+    [docs]
   );
 
   const currentMessageInd = messageId