diff --git a/surfsense_web/components/chat/SourceDetailSheet.tsx b/surfsense_web/components/chat/SourceDetailSheet.tsx index e028d6a8e..0e48043b5 100644 --- a/surfsense_web/components/chat/SourceDetailSheet.tsx +++ b/surfsense_web/components/chat/SourceDetailSheet.tsx @@ -18,7 +18,6 @@ import { SheetTitle, } from "@/components/ui/sheet"; import { getConnectorIcon } from "@/contracts/enums/connectorIcons"; -import { useDocumentByChunk } from "@/hooks/use-document-by-chunk"; import { cn } from "@/lib/utils"; interface SourceDetailSheetProps { @@ -49,13 +48,12 @@ export function SourceDetailSheet({ url, children, }: SourceDetailSheetProps) { - const { document, loading, error, fetchDocumentByChunk, clearDocument } = useDocumentByChunk(); const chunksContainerRef = useRef(null); const highlightedChunkRef = useRef(null); const [summaryOpen, setSummaryOpen] = useState(false); // Add useQuery to fetch document by chunk - const { data: documentFromChunkQuery, isLoading: isLoadingDocumentFromChunk, error: documentFetchError } = useQuery({ + const { data: document, isLoading: isDocumentByChunkFetching, error: documentByChunkFetchingError } = useQuery({ queryKey: cacheKeys.documents.byChunk(chunkId.toString()), queryFn: () => documentsApiService.getDocumentByChunk({ chunk_id: chunkId }), enabled: !!chunkId && open, @@ -69,14 +67,6 @@ export function SourceDetailSheet({ sourceType === "SEARXNG_API" || sourceType === "BAIDU_SEARCH_API"; - useEffect(() => { - if (open && chunkId && !isDirectRenderSource) { - fetchDocumentByChunk(chunkId); - } else if (!open && !isDirectRenderSource) { - clearDocument(); - } - }, [open, chunkId, isDirectRenderSource, fetchDocumentByChunk, clearDocument]); - useEffect(() => { // Scroll to highlighted chunk when document loads if (document && highlightedChunkRef.current && chunksContainerRef.current) { @@ -111,15 +101,15 @@ export function SourceDetailSheet({ - {!isDirectRenderSource && loading && ( + {!isDirectRenderSource && isDocumentByChunkFetching && (
)} - {!isDirectRenderSource && error && ( + {!isDirectRenderSource && documentByChunkFetchingError && (
-

{error}

+

{documentByChunkFetchingError.message || "Failed to load document"}

)}