use flag imports in gen-ai feature flag hooks

Author: jharan1

packages/gen-ai/frontend/src/app/Chatbot/components/settingsPanelTabs/PromptTabContent.tsx

@@ -8,6 +8,7 @@ import PromptAssistantFormGroup from '~/app/Chatbot/components/PromptAssistantFo
 import { usePlaygroundStore } from '~/app/Chatbot/store/usePlaygroundStore';
 import { useConfirmation } from '~/app/Chatbot/hooks/useConfirmation';
 import { usePromptEdited } from '~/app/Chatbot/hooks/usePromptEdited';
+import { PROMPT_MANAGEMENT } from '~/odh/extensions';
 
 interface PromptTabContentProps {
   systemInstruction: string;
@@ -19,7 +20,7 @@ function PromptTabContent({
   onSystemInstructionChange,
 }: PromptTabContentProps): React.ReactNode {
   const { openModal } = usePlaygroundStore();
-  const [promptManagementEnabled] = useFeatureFlag('promptManagement');
+  const [promptManagementEnabled] = useFeatureFlag(PROMPT_MANAGEMENT);
   const isEdited = usePromptEdited();
   const { confirm, modal: confirmationModal } = useConfirmation(isEdited);
 

packages/gen-ai/frontend/src/app/Chatbot/hooks/useGuardrailsEnabled.ts

@@ -1,7 +1,8 @@
 import { useFeatureFlag } from '@openshift/dynamic-plugin-sdk';
+import { GUARDRAILS } from '~/odh/extensions';
 
 const useGuardrailsEnabled = (): boolean => {
-  const [guardrailsEnabled] = useFeatureFlag('guardrails');
+  const [guardrailsEnabled] = useFeatureFlag(GUARDRAILS);
   return guardrailsEnabled;
 };
 

packages/gen-ai/frontend/src/app/hooks/useAiAssetCustomEndpointsEnabled.ts

@@ -1,7 +1,8 @@
 import { useFeatureFlag } from '@openshift/dynamic-plugin-sdk';
+import { AI_ASSET_CUSTOM_ENDPOINTS } from '~/odh/extensions';
 
 const useAiAssetCustomEndpointsEnabled = (): boolean => {
-  const [aiAssetCustomEndpointsEnabled] = useFeatureFlag('aiAssetCustomEndpoints');
+  const [aiAssetCustomEndpointsEnabled] = useFeatureFlag(AI_ASSET_CUSTOM_ENDPOINTS);
   return aiAssetCustomEndpointsEnabled;
 };
 

packages/gen-ai/frontend/src/app/hooks/useAiAssetModelAsServiceEnabled.ts

@@ -1,7 +1,8 @@
 import { useFeatureFlag } from '@openshift/dynamic-plugin-sdk';
+import { MODEL_AS_SERVICE_CAMEL } from '~/odh/extensions';
 
 const useAiAssetModelAsServiceEnabled = (): boolean => {
-  const [modelAsServiceEnabled] = useFeatureFlag('modelAsService');
+  const [modelAsServiceEnabled] = useFeatureFlag(MODEL_AS_SERVICE_CAMEL);
   return modelAsServiceEnabled;
 };