chore: fixed model background refetch

hotosm · Nov 25, 2024 · dc4e544 · dc4e544
1 parent a603e6d
commit dc4e544
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 10 deletions.
diff --git a/frontend/src/app/providers/models-provider.tsx b/frontend/src/app/providers/models-provider.tsx
@@ -222,8 +222,8 @@ const ModelsContext = createContext<{
   validateEditMode: boolean;
 }>({
   formData: initialFormState,
-  setFormData: () => {},
-  handleChange: () => {},
+  setFormData: () => { },
+  handleChange: () => { },
   createNewTrainingDatasetMutation: {} as UseMutationResult<
     TTrainingDataset,
     Error,
@@ -238,13 +238,13 @@ const ModelsContext = createContext<{
   >,
   hasLabeledTrainingAreas: false,
   hasAOIsWithGeometry: false,
-  resetState: () => {},
+  resetState: () => { },
   isEditMode: false,
   modelId: "",
   getFullPath: () => "",
-  handleModelCreationAndUpdate: () => {},
+  handleModelCreationAndUpdate: () => { },
   trainingDatasetCreationInProgress: false,
-  handleTrainingDatasetCreation: () => {},
+  handleTrainingDatasetCreation: () => { },
   validateEditMode: false,
 });
 

diff --git a/frontend/src/app/routes/models/model-details.tsx b/frontend/src/app/routes/models/model-details.tsx
@@ -32,7 +32,7 @@ export const ModelDetailsPage = () => {
     openDialog: openModelFilesDialog,
   } = useDialog();
   const navigate = useNavigate();
-  const { data, isPending, isError, error } = useModelDetails(id as string);
+  const { data, isPending, isError, error } = useModelDetails(id as string, id !== undefined, 10000);
   const { user } = useAuth();
 
   useEffect(() => {

diff --git a/frontend/src/app/routes/start-mapping/start-mapping.tsx b/frontend/src/app/routes/start-mapping/start-mapping.tsx
@@ -56,6 +56,7 @@ export const StartMappingPage = () => {
   const { modelId } = useParams();
   const { isError, isPending, data, error } = useModelDetails(
     modelId as string,
+    modelId !== undefined
   );
   const navigate = useNavigate();
   const { currentZoom } = useMap();

diff --git a/frontend/src/features/models/api/factory.ts b/frontend/src/features/models/api/factory.ts
@@ -46,11 +46,12 @@ export const getModelsQueryOptions = ({
   });
 };
 
-export const getModelDetailsQueryOptions = (id: string) => {
+export const getModelDetailsQueryOptions = (id: string, refetchInterval: boolean | number) => {
   return queryOptions({
     queryKey: [queryKeys.MODEL_DETAILS(id)],
     queryFn: () => getModelDetails(id),
-    refetchInterval: 10000,
+    //@ts-expect-error bad type definition
+    refetchInterval: refetchInterval
   });
 };
 

diff --git a/frontend/src/features/models/hooks/use-models.ts b/frontend/src/features/models/hooks/use-models.ts
@@ -39,9 +39,9 @@ export const useModels = ({
   });
 };
 
-export const useModelDetails = (id: string, enabled: boolean = true) => {
+export const useModelDetails = (id: string, enabled: boolean = true, refetchInterval: boolean | number = false) => {
   return useQuery({
-    ...getModelDetailsQueryOptions(id),
+    ...getModelDetailsQueryOptions(id, refetchInterval),
     //@ts-expect-error bad type definition
     throwOnError: (error) => error.response?.status >= 500,
     retry: (_, error) => {