fix tag invalidation

Author: Mary Hipp

invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/imageUploaded.ts

@@ -11,6 +11,7 @@ import { omit } from 'lodash-es';
 import { boardsApi } from 'services/api/endpoints/boards';
 import { imagesApi } from 'services/api/endpoints/images';
 import type { ImageDTO } from 'services/api/types';
+import { getCategories, getListImagesUrl } from 'services/api/util';
 const log = logger('gallery');
 
 /**
@@ -59,6 +60,29 @@ export const addImageUploadedFulfilledListener = (startAppListening: AppStartLis
         return;
       }
 
+      if (imageUploadedClientSide.match(action)) {
+        const categories = getCategories(imageDTO);
+        const boardId = imageDTO.board_id ?? 'none';
+        dispatch(
+          imagesApi.util.invalidateTags([
+            {
+              type: 'ImageList',
+              id: getListImagesUrl({
+                board_id: boardId,
+                categories,
+              }),
+            },
+            {
+              type: 'Board',
+              id: boardId,
+            },
+            {
+              type: 'BoardImagesTotal',
+              id: boardId,
+            },
+          ])
+        );
+      }
       const state = getState();
 
       log.debug({ imageDTO }, 'Image uploaded');

invokeai/frontend/web/src/services/api/endpoints/images.ts

@@ -318,37 +318,11 @@ export const imagesApi = api.injectEndpoints({
       },
     }),
     createImageUploadEntry: build.mutation<ImageUploadEntryResponse, ImageUploadEntryRequest>({
-      query: ({ width, height }) => ({
+      query: ({ width, height, board_id }) => ({
         url: buildImagesUrl(),
         method: 'POST',
-        body: { width, height },
+        body: { width, height, board_id },
       }),
-      invalidatesTags: (result) => {
-        if (!result) {
-          // Don't add it to anything
-          return [];
-        }
-        const categories = getCategories(result.image_dto);
-        const boardId = result.image_dto.board_id ?? 'none';
-
-        return [
-          {
-            type: 'ImageList',
-            id: getListImagesUrl({
-              board_id: boardId,
-              categories,
-            }),
-          },
-          {
-            type: 'Board',
-            id: boardId,
-          },
-          {
-            type: 'BoardImagesTotal',
-            id: boardId,
-          },
-        ];
-      },
     }),
     deleteBoard: build.mutation<DeleteBoardResult, string>({
       query: (board_id) => ({ url: buildBoardsUrl(board_id), method: 'DELETE' }),