fix: fetch all files

devflowinc · Dec 21, 2024 · 7f47e8e · 7f47e8e
1 parent f95ef2f
commit 7f47e8e
Showing 1 changed file with 25 additions and 10 deletions.
diff --git a/clients/search-component/src/utils/hooks/file-context.tsx b/clients/search-component/src/utils/hooks/file-context.tsx
@@ -8,21 +8,36 @@ const FileContext = createContext<{
 
 export const FileContextProvider = (props: { children: ReactNode }) => {
   const [files, setFiles] = useState<Record<string, string>>({});
-  const [page] = useState(1);
   const state = useModalState();
 
   useEffect(() => {
     const getFiles = async () => {
-      const files = state.trieveSDK.trieve.fetch(
-        "/api/dataset/files/{dataset_id}/{page}",
-        "get",
-        {
-          page: 1,
-          datasetId: state.props.datasetId,
-        },
-      );
+      const page = 1;
+      let done = false;
+      const fileMapResult: Record<string, string> = {};
+      while (!done) {
+        const files = await state.trieveSDK.trieve.fetch(
+          "/api/dataset/files/{dataset_id}/{page}",
+          "get",
+          {
+            page,
+            datasetId: state.props.datasetId,
+          },
+        );
+
+        if (files.length) {
+          files.reduce((acc, file) => {
+            acc[file.file_name] = file.id;
+            return acc;
+          }, fileMapResult);
+        } else {
+          done = true;
+        }
+      }
+
+      setFiles(fileMapResult);
     };
-    getFiles();
+    void getFiles();
   }, []);
 
   return (