pass priority

darnautov · darnautov · commit 2d04feeab682 · 2022-11-08T15:06:29.000+01:00
diff --git a/x-pack/plugins/ml/public/application/services/ml_api_service/trained_models.ts b/x-pack/plugins/ml/public/application/services/ml_api_service/trained_models.ts
@@ -126,7 +126,11 @@ export function trainedModelsApiProvider(httpService: HttpService) {
 
     startModelAllocation(
       modelId: string,
-      queryParams?: { number_of_allocations: number; threads_per_allocation: number }
+      queryParams?: {
+        number_of_allocations: number;
+        threads_per_allocation: number;
+        priority: 'low' | 'normal';
+      }
     ) {
       return httpService.http<{ acknowledge: boolean }>({
         path: `${apiBasePath}/trained_models/${modelId}/deployment/_start`,
diff --git a/x-pack/plugins/ml/public/application/trained_models/models_management/model_actions.tsx b/x-pack/plugins/ml/public/application/trained_models/models_management/model_actions.tsx
@@ -167,6 +167,7 @@ export function useModelActions({
             await trainedModelsApiService.startModelAllocation(item.model_id, {
               number_of_allocations: threadingParams.numOfAllocations,
               threads_per_allocation: threadingParams.threadsPerAllocations!,
+              priority: threadingParams.priority!,
             });
             displaySuccessToast(
               i18n.translate('xpack.ml.trainedModels.modelsList.startSuccess', {
diff --git a/x-pack/plugins/ml/server/routes/schemas/inference_schema.ts b/x-pack/plugins/ml/server/routes/schemas/inference_schema.ts
@@ -18,6 +18,7 @@ export const threadingParamsSchema = schema.maybe(
   schema.object({
     number_of_allocations: schema.number(),
     threads_per_allocation: schema.number(),
+    priority: schema.oneOf([schema.literal('low'), schema.literal('normal')]),
   })
 );
 
