From e2b79d4a9f149e5d7db200aab1530c87493d5691 Mon Sep 17 00:00:00 2001
From: Melissa Alvarez <melissa.alvarez@elastic.co>
Date: Fri, 7 Feb 2020 12:27:18 -0500
Subject: [PATCH] [ML] DF Analytics creation: update schema definition for
 create route (#56979)

* fix df analytics create route schema definition

* add all optional params to schema definiton
---
 .../ml/server/new_platform/data_analytics_schema.ts    | 10 +++++++++-
 .../plugins/ml/server/routes/data_frame_analytics.ts   |  2 +-
 2 files changed, 10 insertions(+), 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
index f5d72c51dc070..21454fa884b82 100644
--- 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
index 7b855e5f87cbf..6946094198b1e 100644
--- a/x-pack/legacy/plugins/ml/server/routes/data_frame_analytics.ts
+++ b/x-pack/legacy/plugins/ml/server/routes/data_frame_analytics.ts
@@ -108,7 +108,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) => {