fix tsc errors

invoke-ai · Jun 27, 2024 · 514225b · 514225b
1 parent c1868aa
commit 514225b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/imageUploaded.ts b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/imageUploaded.ts
@@ -11,6 +11,7 @@ import {
   ipaLayerImageChanged,
   rgLayerIPAdapterImageChanged,
 } from 'features/controlLayers/store/controlLayersSlice';
+import { selectListBoardsQueryArgs } from 'features/gallery/store/gallerySelectors';
 import { fieldImageValueChanged } from 'features/nodes/store/nodesSlice';
 import { selectOptimalDimension } from 'features/parameters/store/generationSlice';
 import { toast } from 'features/toast/toast';
@@ -62,7 +63,8 @@ export const addImageUploadedFulfilledListener = (startAppListening: AppStartLis
           );
 
           // Attempt to get the board's name for the toast
-          const { data } = boardsApi.endpoints.listAllBoards.select()(state);
+          const queryArgs = selectListBoardsQueryArgs(state);
+          const { data } = boardsApi.endpoints.listAllBoards.select(queryArgs)(state);
 
           // Fall back to just the board id if we can't find the board for some reason
           const board = data?.find((b) => b.board_id === autoAddBoardId);

diff --git a/invokeai/frontend/web/src/services/api/hooks/accessChecks.ts b/invokeai/frontend/web/src/services/api/hooks/accessChecks.ts
@@ -1,4 +1,5 @@
 import { getStore } from 'app/store/nanostores/store';
+import { selectListBoardsQueryArgs } from 'features/gallery/store/gallerySelectors';
 import { boardsApi } from 'services/api/endpoints/boards';
 import { imagesApi } from 'services/api/endpoints/images';
 import { modelsApi } from 'services/api/endpoints/models';
@@ -43,9 +44,10 @@ export const checkImageAccess = async (name: string): Promise<boolean> => {
  * @returns A promise that resolves to true if the client has access, else false.
  */
 export const checkBoardAccess = async (id: string): Promise<boolean> => {
-  const { dispatch } = getStore();
+  const { dispatch, getState } = getStore();
   try {
-    const req = dispatch(boardsApi.endpoints.listAllBoards.initiate());
+    const queryArgs = selectListBoardsQueryArgs(getState());
+    const req = dispatch(boardsApi.endpoints.listAllBoards.initiate(queryArgs));
     req.unsubscribe();
     const result = await req.unwrap();
     return result.some((b) => b.board_id === id);