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

chore(ui): clean up modelSlice

Browse Source
This commit is contained in:
psychedelicious 2023-05-16 09:45:49 +10:00
parent 84b801d88f
commit 9f4d39aa42
1 changed files with 3 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
invokeai/frontend/web/src/features/system/store/modelSlice.ts
View File

@ -13,26 +13,15 @@ export const modelsAdapter = createEntityAdapter<Model>({
sortComparer: (a, b) => a.name.localeCompare(b.name), sortComparer: (a, b) => a.name.localeCompare(b.name),
}); });
type AdditionalModelsState = { export const initialModelsState = modelsAdapter.getInitialState();
selectedModelName: string;
};
export const initialModelsState =
modelsAdapter.getInitialState<AdditionalModelsState>({
selectedModelName: '',
});
export type ModelsState = typeof initialModelsState; export type ModelsState = typeof initialModelsState;
export const modelsSlice = createSlice({ export const modelsSlice = createSlice({
name: 'models', name: 'models',
initialState: modelsAdapter.getInitialState(), initialState: initialModelsState,
// initialState: initialModelsState,
reducers: { reducers: {
modelAdded: modelsAdapter.upsertOne, modelAdded: modelsAdapter.upsertOne,
// modelSelected: (state, action: PayloadAction<string>) => {
// state.selectedModelName = action.payload;
// },
}, },
extraReducers(builder) { extraReducers(builder) {
/** /**
@ -41,32 +30,10 @@ export const modelsSlice = createSlice({
builder.addCase(receivedModels.fulfilled, (state, action) => { builder.addCase(receivedModels.fulfilled, (state, action) => {
const models = action.payload; const models = action.payload;
modelsAdapter.setAll(state, models); modelsAdapter.setAll(state, models);
// If the current selected model is `''` or isn't actually in the list of models,
// choose a random model
// if (
// !state.selectedModelName ||
// !keys(models).includes(state.selectedModelName)
// ) {
// const randomModel = sample(models);
// if (randomModel) {
// state.selectedModelName = randomModel.name;
// } else {
// state.selectedModelName = '';
// }
// }
}); });
}, },
}); });
// export const selectedModelSelector = (state: RootState) => {
// const { selectedModelName } = state.models;
// const selectedModel = selectModelsById(state, selectedModelName);
// return selectedModel ?? null;
// };
export const { export const {
selectAll: selectModelsAll, selectAll: selectModelsAll,
selectById: selectModelsById, selectById: selectModelsById,
@ -75,9 +42,6 @@ export const {
selectTotal: selectModelsTotal, selectTotal: selectModelsTotal,
} = modelsAdapter.getSelectors<RootState>((state) => state.models); } = modelsAdapter.getSelectors<RootState>((state) => state.models);
export const { export const { modelAdded } = modelsSlice.actions;
modelAdded,
// modelSelected
} = modelsSlice.actions;
export default modelsSlice.reducer; export default modelsSlice.reducer;