make logic more straight forward

This commit is contained in:
Mary Hipp
2025-05-21 09:57:08 -04:00
parent 323da2d66d
commit 080c05f09b
3 changed files with 8 additions and 9 deletions

View File

@@ -11,7 +11,7 @@ export const DisabledModelWarning = () => {
const model = useAppSelector(selectModel);
const accountSettingsLink = useStore($accountSettingsLink);
const isChatGPT4oHighModelDisabled = useIsModelDisabled('chatGPT4oHigh');
const { isChatGPT4oHighModelDisabled } = useIsModelDisabled();
if (!model || !isChatGPT4oHighModelDisabled(model)) {
return null;

View File

@@ -1,17 +1,16 @@
import type { AppFeature } from 'app/types/invokeai';
import type { ParameterModel } from 'features/parameters/types/parameterSchemas';
import { useFeatureStatus } from 'features/system/hooks/useFeatureStatus';
import { useCallback } from 'react';
export const useIsModelDisabled = (feature: AppFeature) => {
const isEnabled = useFeatureStatus(feature);
export const useIsModelDisabled = () => {
const isChatGPT4oHighEnabled = useFeatureStatus('chatGPT4oHigh');
const isModelDisabled = useCallback(
const isChatGPT4oHighModelDisabled = useCallback(
(model: ParameterModel) => {
return model?.base === 'chatgpt-4o' && model.name.toLowerCase().includes('high') && !isEnabled;
return model?.base === 'chatgpt-4o' && model.name.toLowerCase().includes('high') && !isChatGPT4oHighEnabled;
},
[isEnabled]
[isChatGPT4oHighEnabled]
);
return isModelDisabled;
return { isChatGPT4oHighModelDisabled };
};

View File

@@ -153,7 +153,7 @@ export const useReadinessWatcher = () => {
const canvasIsSelectingObject = useStore(canvasManager?.stateApi.$isSegmenting ?? $true);
const canvasIsCompositing = useStore(canvasManager?.compositor.$isBusy ?? $true);
const isInPublishFlow = useStore($isInPublishFlow);
const isChatGPT4oHighModelDisabled = useIsModelDisabled('chatGPT4oHigh');
const { isChatGPT4oHighModelDisabled } = useIsModelDisabled();
useEffect(() => {
debouncedUpdateReasons(