@lobehub/chat/src/features/FileManager/FileList/useCheckTaskStatus.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { useEffect } from 'react';

import { useFileStore } from '@/store/file';
import { AsyncTaskStatus } from '@/types/asyncTask';
import { FileListItem } from '@/types/files';

export const useCheckTaskStatus = (data: FileListItem[] | undefined) => {
  const [refreshFileList] = useFileStore((s) => [s.refreshFileList]);
  const hasProcessingChunkTask = data?.some(
    (item) => item.chunkingStatus === AsyncTaskStatus.Processing,
  );
  const hasProcessingEmbeddingTask = data?.some(
    (item) => item.embeddingStatus === AsyncTaskStatus.Processing,
  );

  const isProcessing = hasProcessingChunkTask || hasProcessingEmbeddingTask;

  // every 3s to check if the chunking status is changed
  useEffect(() => {
    if (!isProcessing) return;

    const interval = setInterval(refreshFileList, 5000);
    return () => {
      clearInterval(interval);
    };
  }, [isProcessing]);
};