[ML] DF Analytics creation: update schema definition for create route (…

…#56979) (#57131) * fix df analytics create route schema definition * add all optional params to schema definiton
elastic · Feb 7, 2020 · bf32ca1 · bf32ca1
1 parent 29b4933
commit bf32ca1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/x-pack/legacy/plugins/ml/server/new_platform/data_analytics_schema.ts b/x-pack/legacy/plugins/ml/server/new_platform/data_analytics_schema.ts
@@ -13,8 +13,16 @@ export const dataAnalyticsJobConfigSchema = {
     results_field: schema.maybe(schema.string()),
   }),
   source: schema.object({
-    index: schema.string(),
+    index: schema.oneOf([schema.string(), schema.arrayOf(schema.string())]),
+    query: schema.maybe(schema.any()),
+    _source: schema.maybe(
+      schema.object({
+        includes: schema.maybe(schema.arrayOf(schema.maybe(schema.string()))),
+        excludes: schema.maybe(schema.arrayOf(schema.maybe(schema.string()))),
+      })
+    ),
   }),
+  allow_lazy_start: schema.maybe(schema.boolean()),
   analysis: schema.any(),
   analyzed_fields: schema.any(),
   model_memory_limit: schema.string(),

diff --git a/x-pack/legacy/plugins/ml/server/routes/data_frame_analytics.ts b/x-pack/legacy/plugins/ml/server/routes/data_frame_analytics.ts
@@ -156,7 +156,7 @@ export function dataFrameAnalyticsRoutes({ xpackMainPlugin, router }: RouteIniti
         params: schema.object({
           analyticsId: schema.string(),
         }),
-        body: schema.object({ ...dataAnalyticsJobConfigSchema }),
+        body: schema.object(dataAnalyticsJobConfigSchema),
       },
     },
     licensePreRoutingFactory(xpackMainPlugin, async (context, request, response) => {