diff --git a/src/composables/useFeatureFlags.ts b/src/composables/useFeatureFlags.ts index 643bb103ff..de8c859124 100644 --- a/src/composables/useFeatureFlags.ts +++ b/src/composables/useFeatureFlags.ts @@ -1,5 +1,6 @@ import { computed, reactive, readonly } from 'vue' +import { remoteConfig } from '@/platform/remoteConfig/remoteConfig' import { api } from '@/scripts/api' /** @@ -9,7 +10,8 @@ export enum ServerFeatureFlag { SUPPORTS_PREVIEW_METADATA = 'supports_preview_metadata', MAX_UPLOAD_SIZE = 'max_upload_size', MANAGER_SUPPORTS_V4 = 'extension.manager.supports_v4', - MODEL_UPLOAD_BUTTON_ENABLED = 'model_upload_button_enabled' + MODEL_UPLOAD_BUTTON_ENABLED = 'model_upload_button_enabled', + ASSET_UPDATE_OPTIONS_ENABLED = 'asset_update_options_enabled' } /** @@ -27,9 +29,23 @@ export function useFeatureFlags() { return api.getServerFeature(ServerFeatureFlag.MANAGER_SUPPORTS_V4) }, get modelUploadButtonEnabled() { - return api.getServerFeature( - ServerFeatureFlag.MODEL_UPLOAD_BUTTON_ENABLED, - false + // Check remote config first (from /api/features), fall back to websocket feature flags + return ( + remoteConfig.value.model_upload_button_enabled ?? + api.getServerFeature( + ServerFeatureFlag.MODEL_UPLOAD_BUTTON_ENABLED, + false + ) + ) + }, + get assetUpdateOptionsEnabled() { + // Check remote config first (from /api/features), fall back to websocket feature flags + return ( + remoteConfig.value.asset_update_options_enabled ?? + api.getServerFeature( + ServerFeatureFlag.ASSET_UPDATE_OPTIONS_ENABLED, + false + ) ) } }) diff --git a/src/platform/remoteConfig/types.ts b/src/platform/remoteConfig/types.ts index 43bfbd70fe..8d2c400f37 100644 --- a/src/platform/remoteConfig/types.ts +++ b/src/platform/remoteConfig/types.ts @@ -1,3 +1,5 @@ +import type { TelemetryEventName } from '@/platform/telemetry/types' + /** * Server health alert configuration from the backend */ @@ -31,4 +33,7 @@ export type RemoteConfig = { comfy_api_base_url?: string comfy_platform_base_url?: string firebase_config?: FirebaseRuntimeConfig + telemetry_disabled_events?: TelemetryEventName[] + model_upload_button_enabled?: boolean + asset_update_options_enabled?: boolean }