From ca5a4ee59d0b07dad97967d692bf476500b4e7e2 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 24 Jul 2024 16:15:54 +1000
Subject: [PATCH] fix(ui): few cases where board totals don't updated when
 moving

---
 .../web/src/services/api/endpoints/images.ts         | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/invokeai/frontend/web/src/services/api/endpoints/images.ts b/invokeai/frontend/web/src/services/api/endpoints/images.ts
index 6f36866dce6..9672acc1490 100644
--- a/invokeai/frontend/web/src/services/api/endpoints/images.ts
+++ b/invokeai/frontend/web/src/services/api/endpoints/images.ts
@@ -382,6 +382,10 @@ export const imagesApi = api.injectEndpoints({
             type: 'BoardImagesTotal',
             id: imageDTO.board_id ?? 'none',
           },
+          {
+            type: 'BoardImagesTotal',
+            id: board_id,
+          },
         ];
       },
     }),
@@ -454,6 +458,10 @@ export const imagesApi = api.injectEndpoints({
             }),
           });
           tags.push({ type: 'Board', id: imageDTOs[0].board_id ?? 'none' });
+          tags.push({
+            type: 'BoardImagesTotal',
+            id: imageDTOs[0].board_id ?? 'none',
+          });
         }
         for (const imageDTO of imageDTOs) {
           tags.push({ type: 'Image', id: imageDTO.image_name });
@@ -498,6 +506,10 @@ export const imagesApi = api.injectEndpoints({
               categories: getCategories(imageDTOs[0]),
             }),
           });
+          tags.push({
+            type: 'BoardImagesTotal',
+            id: 'none',
+          });
         }
 
         result?.removed_image_names.forEach((image_name) => {