add externalVectorStores feature flag to useAiAssetVectorStoresEnabled, add externalVectorStores & custom endpoints flags to PluginStoreContextProvider, update extensions.ts

Author: jharan1

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

@@ -1,7 +1,8 @@
 import { useFeatureFlag } from '@openshift/dynamic-plugin-sdk';
+import { EXTERNAL_VECTOR_STORES } from '~/odh/extensions';
 
 const useAiAssetVectorStoresEnabled = (): boolean => {
-  const [aiAssetVectorStoresEnabled] = useFeatureFlag('ai-asset-vector-stores');
+  const [aiAssetVectorStoresEnabled] = useFeatureFlag(EXTERNAL_VECTOR_STORES);
   return aiAssetVectorStoresEnabled;
 };
 

packages/gen-ai/frontend/src/odh/PluginStoreContextProvider.tsx

@@ -1,13 +1,15 @@
 import * as React from 'react';
 import { PluginStoreProvider } from '@openshift/dynamic-plugin-sdk';
 import { PluginStore } from '@odh-dashboard/plugin-core';
-import extensions from './extensions';
-
-const PLUGIN_GEN_AI = 'plugin-gen-ai';
-const MODEL_AS_SERVICE = 'model-as-service';
-const MODEL_AS_SERVICE_CAMEL = 'modelAsService';
-const GUARDRAILS = 'guardrails';
-const PROMPT_MANAGEMENT = 'promptManagement';
+import extensions, {
+  AI_ASSET_CUSTOM_ENDPOINTS,
+  EXTERNAL_VECTOR_STORES,
+  GUARDRAILS,
+  MODEL_AS_SERVICE,
+  MODEL_AS_SERVICE_CAMEL,
+  PLUGIN_GEN_AI,
+  PROMPT_MANAGEMENT,
+} from './extensions';
 
 export const PluginStoreContextProvider: React.FC<React.PropsWithChildren> = ({ children }) => {
   const store = React.useMemo(() => {
@@ -18,6 +20,8 @@ export const PluginStoreContextProvider: React.FC<React.PropsWithChildren> = ({
       [MODEL_AS_SERVICE_CAMEL]: true,
       [GUARDRAILS]: true,
       [PROMPT_MANAGEMENT]: true,
+      [EXTERNAL_VECTOR_STORES]: true,
+      [AI_ASSET_CUSTOM_ENDPOINTS]: true,
     };
 
     const params = new URLSearchParams(window.location.search);

packages/gen-ai/frontend/src/odh/extensions.ts

@@ -13,11 +13,13 @@ import {
 } from '~/app/utilities/routes';
 import type { AIAssetsTabExtension } from '~/odh/extension-points';
 
-const PLUGIN_GEN_AI = 'plugin-gen-ai';
-const GUARDRAILS = 'guardrails';
-const PROMPT_MANAGEMENT = 'promptManagement';
-const AI_ASSET_CUSTOM_ENDPOINTS = 'aiAssetCustomEndpoints';
-const AI_ASSET_VECTOR_STORES = 'ai-asset-vector-stores';
+export const PLUGIN_GEN_AI = 'plugin-gen-ai';
+export const MODEL_AS_SERVICE = 'model-as-service';
+export const MODEL_AS_SERVICE_CAMEL = 'modelAsService';
+export const GUARDRAILS = 'guardrails';
+export const PROMPT_MANAGEMENT = 'promptManagement';
+export const AI_ASSET_CUSTOM_ENDPOINTS = 'aiAssetCustomEndpoints';
+export const EXTERNAL_VECTOR_STORES = 'externalVectorStores';
 
 const extensions: (NavExtension | RouteExtension | AreaExtension | AIAssetsTabExtension)[] = [
   {
@@ -55,9 +57,9 @@ const extensions: (NavExtension | RouteExtension | AreaExtension | AIAssetsTabEx
   {
     type: 'app.area',
     properties: {
-      id: AI_ASSET_VECTOR_STORES,
+      id: EXTERNAL_VECTOR_STORES,
       reliantAreas: [PLUGIN_GEN_AI],
-      devFlags: [AI_ASSET_VECTOR_STORES],
+      featureFlags: [EXTERNAL_VECTOR_STORES],
     },
   },
   {
@@ -136,7 +138,7 @@ const extensions: (NavExtension | RouteExtension | AreaExtension | AIAssetsTabEx
   {
     type: 'gen-ai.ai-assets/tab',
     flags: {
-      required: [PLUGIN_GEN_AI, AI_ASSET_VECTOR_STORES],
+      required: [PLUGIN_GEN_AI, EXTERNAL_VECTOR_STORES],
     },
     properties: {
       id: 'vectorstores',