[Enterprise Search] Support for starting ELSER model deployment

x-pack/plugins/enterprise_search/public/applications/enterprise_search_content/api/ml_models/text_expansion/start_text_expansion_model_api_logic.ts

@@ -0,0 +1,33 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+import { ELSER_MODEL_ID } from '../../../../../../common/ml_inference_pipeline';
+import { Actions, createApiLogic } from '../../../../shared/api_logic/create_api_logic';
+import { HttpLogic } from '../../../../shared/http';
+
+export type StartTextExpansionModelArgs = undefined;
+
+export interface StartTextExpansionModelResponse {
+ deploymentState: string;
+ modelId: string;
+}
+
+export const startTextExpansionModel = async (): Promise<StartTextExpansionModelResponse> => {
+ const route = `/internal/enterprise_search/ml/models/${ELSER_MODEL_ID}/deploy`;
+ return await HttpLogic.values.http.post<StartTextExpansionModelResponse>(route, {
+ body: undefined,
+ });
+};
+
+export const StartTextExpansionModelApiLogic = createApiLogic(
+ ['start_text_expansion_model_api_logic'],
+ startTextExpansionModel
+);
+
+export type StartTextExpansionModelApiLogicActions = Actions<
+ StartTextExpansionModelArgs,
+ StartTextExpansionModelResponse
+>;

x-pack/plugins/enterprise_search/public/applications/enterprise_search_content/components/search_index/pipelines/inference_pipeline_card.tsx

@@ -167,6 +167,12 @@ export const InferencePipelineCard: React.FC<InferencePipeline> = (pipeline) =>
  )}
  >
  <EuiButtonIcon
+ aria-label={i18n.translate(
+ 'xpack.enterpriseSearch.inferencePipelineCard.modelState.notDeployed.fixLink',
+ {
+ defaultMessage: 'Fix issue in Trained Models',
+ }
+ )}
  data-telemetry-id={`entSearchContent-${ingestionMethod}-pipelines-inferencePipeline-fixIssueInTrainedModels`}
  href={http.basePath.prepend(ML_MANAGE_TRAINED_MODELS_PATH)}
  display="base"

x-pack/plugins/enterprise_search/public/applications/enterprise_search_content/components/search_index/pipelines/ml_inference/text_expansion_callout.test.tsx

@@ -19,13 +19,15 @@ import {
  ModelDeploymentInProgress,
  ModelDeployed,
  TextExpansionDismissButton,
+ ModelStarted,
 } from './text_expansion_callout';
 
 jest.mock('./text_expansion_callout_data', () => ({
  useTextExpansionCallOutData: jest.fn(() => ({
  dismiss: jest.fn(),
  isCreateButtonDisabled: false,
  isDismissable: false,
+ isStartButtonDisabled: false,
  show: true,
  })),
 }));
@@ -34,6 +36,8 @@ const DEFAULT_VALUES = {
  isCreateButtonDisabled: false,
  isModelDownloadInProgress: false,
  isModelDownloaded: false,
+ isModelStarted: false,
+ isStartButtonDisabled: false,
 };
 
 describe('TextExpansionCallOut', () => {
@@ -63,6 +67,15 @@ describe('TextExpansionCallOut', () => {
  const wrapper = shallow(<TextExpansionCallOut />);
  expect(wrapper.find(ModelDeployed).length).toBe(1);
  });
+ it('renders panel with deployment in progress status if the model has been started', () => {
+ setMockValues({
+ ...DEFAULT_VALUES,
+ isModelStarted: true,
+ });
+
+ const wrapper = shallow(<TextExpansionCallOut />);
+ expect(wrapper.find(ModelStarted).length).toBe(1);
+ });
 
  describe('DeployModel', () => {
  it('renders deploy button', () => {
@@ -109,12 +122,43 @@ describe('TextExpansionCallOut', () => {
  });
 
  describe('ModelDeployed', () => {
+ it('renders start button', () => {
+ const wrapper = shallow(
+ <ModelDeployed dismiss={() => {}} isDismissable={false} isStartButtonDisabled={false} />
+ );
+ expect(wrapper.find(EuiButton).length).toBe(1);
+ const button = wrapper.find(EuiButton);
+ expect(button.prop('disabled')).toBe(false);
+ });
+ it('renders disabled start button if it is set to disabled', () => {
+ const wrapper = shallow(
+ <ModelDeployed dismiss={() => {}} isDismissable={false} isStartButtonDisabled />
+ );
+ expect(wrapper.find(EuiButton).length).toBe(1);
+ const button = wrapper.find(EuiButton);
+ expect(button.prop('disabled')).toBe(true);
+ });
+ it('renders dismiss button if it is set to dismissable', () => {
+ const wrapper = shallow(
+ <ModelDeployed dismiss={() => {}} isDismissable isStartButtonDisabled={false} />
+ );
+ expect(wrapper.find(TextExpansionDismissButton).length).toBe(1);
+ });
+ it('does not render dismiss button if it is set to non-dismissable', () => {
+ const wrapper = shallow(
+ <ModelDeployed dismiss={() => {}} isDismissable={false} isStartButtonDisabled={false} />
+ );
+ expect(wrapper.find(TextExpansionDismissButton).length).toBe(0);
+ });
+ });
+
+ describe('ModelStarted', () => {
  it('renders dismiss button if it is set to dismissable', () => {
- const wrapper = shallow(<ModelDeployed dismiss={() => {}} isDismissable />);
+ const wrapper = shallow(<ModelStarted dismiss={() => {}} isDismissable />);
  expect(wrapper.find(TextExpansionDismissButton).length).toBe(1);
  });
  it('does not render dismiss button if it is set to non-dismissable', () => {
- const wrapper = shallow(<ModelDeployed dismiss={() => {}} isDismissable={false} />);
+ const wrapper = shallow(<ModelStarted dismiss={() => {}} isDismissable={false} />);
  expect(wrapper.find(TextExpansionDismissButton).length).toBe(0);
  });
  });