refresh tasks on entering page, make failed files persist
This commit is contained in:
parent
1fdb251a47
commit
946d3edc89
2 changed files with 8 additions and 2 deletions
|
|
@ -7,6 +7,7 @@ import { useRouter } from "next/navigation";
|
|||
import {
|
||||
type ChangeEvent,
|
||||
useCallback,
|
||||
useEffect,
|
||||
useRef,
|
||||
useState,
|
||||
} from "react";
|
||||
|
|
@ -51,7 +52,7 @@ function getSourceIcon(connectorType?: string) {
|
|||
|
||||
function SearchPage() {
|
||||
const router = useRouter();
|
||||
const { isMenuOpen, files: taskFiles } = useTask();
|
||||
const { isMenuOpen, files: taskFiles, refreshTasks } = useTask();
|
||||
const { selectedFilter, setSelectedFilter, parsedFilterData, isPanelOpen } =
|
||||
useKnowledgeFilter();
|
||||
const [selectedRows, setSelectedRows] = useState<File[]>([]);
|
||||
|
|
@ -59,6 +60,10 @@ function SearchPage() {
|
|||
|
||||
const deleteDocumentMutation = useDeleteDocument();
|
||||
|
||||
useEffect(() => {
|
||||
refreshTasks();
|
||||
}, [refreshTasks]);
|
||||
|
||||
const { data: searchData = [], isFetching } = useGetSearchQuery(
|
||||
parsedFilterData?.query || "*",
|
||||
parsedFilterData,
|
||||
|
|
|
|||
|
|
@ -226,7 +226,7 @@ export function TaskProvider({ children }: { children: React.ReactNode }) {
|
|||
setTimeout(() => {
|
||||
refetchSearch();
|
||||
setFiles((prevFiles) =>
|
||||
prevFiles.filter((file) => file.task_id !== currentTask.task_id && file.status !== "failed"),
|
||||
prevFiles.filter((file) => file.task_id !== currentTask.task_id || file.status === "failed"),
|
||||
);
|
||||
}, 500);
|
||||
} else if (
|
||||
|
|
@ -258,6 +258,7 @@ export function TaskProvider({ children }: { children: React.ReactNode }) {
|
|||
);
|
||||
|
||||
const refreshTasks = useCallback(async () => {
|
||||
setFiles([]);
|
||||
await refetchTasks();
|
||||
}, [refetchTasks]);
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue