Adding restriciton on AI Playground

yansavitski · Mar 5, 2024 · 55edc9c · 55edc9c
1 parent 88c2b02
commit 55edc9c
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 16 deletions.
diff --git a/x-pack/plugins/enterprise_search/public/applications/enterprise_search_content/index.tsx b/x-pack/plugins/enterprise_search/public/applications/enterprise_search_content/index.tsx
@@ -75,6 +75,9 @@ export const EnterpriseSearchContent: React.FC<InitialAppData> = (props) => {
 };
 
 export const EnterpriseSearchContentConfigured: React.FC<Required<InitialAppData>> = () => {
+  const {
+    productFeatures: { showAIPlayground },
+  } = useValues(KibanaLogic);
   return (
     <Routes>
       <Redirect exact from={ROOT_PATH} to={SEARCH_INDICES_PATH} />
@@ -93,9 +96,11 @@ export const EnterpriseSearchContentConfigured: React.FC<Required<InitialAppData
       <Route path={SETTINGS_PATH}>
         <Settings />
       </Route>
-      <Route path={AI_PLAYGROUND_PATH}>
-        <AIPlayground />
-      </Route>
+      {showAIPlayground && (
+        <Route path={AI_PLAYGROUND_PATH}>
+          <AIPlayground />
+        </Route>
+      )}
       <Route>
         <NotFound />
       </Route>

diff --git a/x-pack/plugins/enterprise_search/public/applications/shared/layout/nav.tsx b/x-pack/plugins/enterprise_search/public/applications/shared/layout/nav.tsx
@@ -109,17 +109,21 @@ export const useEnterpriseSearchNav = () => {
               },
             ]
           : []),
-        {
-          id: 'ai-playground',
-          name: i18n.translate('xpack.enterpriseSearch.nav.aiPlaygroundTitle', {
-            defaultMessage: 'AI Playground',
-          }),
-          ...generateNavLink({
-            shouldNotCreateHref: true,
-            shouldShowActiveForSubroutes: true,
-            to: ENTERPRISE_SEARCH_CONTENT_PLUGIN.URL + AI_PLAYGROUND_PATH,
-          }),
-        },
+        ...(productFeatures.showAIPlayground
+          ? [
+              {
+                id: 'ai-playground',
+                name: i18n.translate('xpack.enterpriseSearch.nav.aiPlaygroundTitle', {
+                  defaultMessage: 'AI Playground',
+                }),
+                ...generateNavLink({
+                  shouldNotCreateHref: true,
+                  shouldShowActiveForSubroutes: true,
+                  to: ENTERPRISE_SEARCH_CONTENT_PLUGIN.URL + AI_PLAYGROUND_PATH,
+                }),
+              },
+            ]
+          : []),
       ],
       name: i18n.translate('xpack.enterpriseSearch.nav.contentTitle', {
         defaultMessage: 'Content',

diff --git a/x-pack/plugins/enterprise_search/server/routes/enterprise_search/ai_playground.ts b/x-pack/plugins/enterprise_search/server/routes/enterprise_search/ai_playground.ts
@@ -15,9 +15,12 @@ import { streamFactory } from '@kbn/ml-response-stream/server';
 
 import { RouteDependencies } from '../../plugin';
 import { elasticsearchErrorHandler } from '../../utils/elasticsearch_error_handler';
-import { createApiKey } from '@kbn/enterprise-search-plugin/server/lib/analytics/create_api_key';
 
-export function registerAIPlaygroundRoutes({ log, router }: RouteDependencies) {
+export function registerAIPlaygroundRoutes({ log, router, config }: RouteDependencies) {
+  if (!config.showAIPlayground) {
+    return;
+  }
+
   router.post(
     {
       path: '/internal/enterprise_search/ai_playground/query_source_fields',