Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): model list refreshes after changes

Browse Source 
When consolidating all the model queries I messed up the query tags. Fixed now, so that when a model is installed, removed, or changed, the list refreshes.
This commit is contained in:
psychedelicious 2024-03-19 22:31:40 +11:00
parent 07c9c0b0ab
commit 3f61c51c3a
3 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketModelInstall.ts
View File

@ -1,5 +1,5 @@
import type { AppStartListening } from 'app/store/middleware/listenerMiddleware';
import { api } from 'services/api';
import { api, LIST_TAG } from 'services/api';
import { modelsApi } from 'services/api/endpoints/models';
import {
socketModelInstallCancelled,
@ -42,7 +42,7 @@ export const addModelInstallEventListener = (startAppListening: AppStartListenin
return draft;
})
);
dispatch(api.util.invalidateTags(['Model']));
dispatch(api.util.invalidateTags([{ type: 'ModelConfig', id: LIST_TAG }]));
},
});

16
invokeai/frontend/web/src/services/api/endpoints/models.ts
View File

@ -118,7 +118,7 @@ export const modelsApi = api.injectEndpoints({
body: formData,
};
},
invalidatesTags: ['Model'],
invalidatesTags: [{ type: 'ModelConfig', id: LIST_TAG }],
}),
installModel: build.mutation<InstallModelResponse, InstallModelArg>({
query: ({ source, inplace = true }) => {
@ -128,7 +128,7 @@ export const modelsApi = api.injectEndpoints({
method: 'POST',
};
},
invalidatesTags: ['Model', 'ModelInstalls'],
invalidatesTags: ['ModelInstalls'],
}),
deleteModels: build.mutation<DeleteModelResponse, DeleteModelArg>({
query: ({ key }) => {
@ -137,7 +137,7 @@ export const modelsApi = api.injectEndpoints({
method: 'DELETE',
};
},
invalidatesTags: ['Model'],
invalidatesTags: [{ type: 'ModelConfig', id: LIST_TAG }],
}),
deleteModelImage: build.mutation<DeleteModelImageResponse, string>({
query: (key) => {
@ -146,7 +146,7 @@ export const modelsApi = api.injectEndpoints({
method: 'DELETE',
};
},
invalidatesTags: ['Model'],
invalidatesTags: [{ type: 'ModelConfig', id: LIST_TAG }],
}),
getModelImage: build.query<string, string>({
query: (key) => buildModelsUrl(`i/${key}/image`),
@ -158,12 +158,12 @@ export const modelsApi = api.injectEndpoints({
method: 'PUT',
};
},
invalidatesTags: ['ModelConfig'],
invalidatesTags: [{ type: 'ModelConfig', id: LIST_TAG }],
}),
getModelConfig: build.query<GetModelConfigResponse, string>({
query: (key) => buildModelsUrl(`i/${key}`),
providesTags: (result) => {
const tags: ApiTagDescription[] = ['Model'];
const tags: ApiTagDescription[] = [];
if (result) {
tags.push({ type: 'ModelConfig', id: result.key });
@ -175,7 +175,7 @@ export const modelsApi = api.injectEndpoints({
getModelConfigByAttrs: build.query<AnyModelConfig, GetByAttrsArg>({
query: (arg) => buildModelsUrl(`get_by_attrs?${queryString.stringify(arg)}`),
providesTags: (result) => {
const tags: ApiTagDescription[] = ['Model'];
const tags: ApiTagDescription[] = [];
if (result) {
tags.push({ type: 'ModelConfig', id: result.key });
@ -192,7 +192,7 @@ export const modelsApi = api.injectEndpoints({
method: 'PATCH',
};
},
invalidatesTags: ['Model'],
invalidatesTags: [{ type: 'ModelConfig', id: LIST_TAG }],
}),
scanFolder: build.query<ScanFolderResponse, ScanFolderArg>({
query: (arg) => {

1
invokeai/frontend/web/src/services/api/index.ts
View File

@ -26,7 +26,6 @@ export const tagTypes = [
'NextSessionQueueItem',
'BatchStatus',
'InvocationCacheStatus',
'Model',
'ModelConfig',
'ModelInstalls',
'T2IAdapterModel',