From 8299b5ff67dc5de79435ae739b5c8b425986d6c0 Mon Sep 17 00:00:00 2001
From: ZhidaLiu <zhili@microsoft.com>
Date: Wed, 16 Oct 2024 14:49:34 -0700
Subject: [PATCH] [AML]Update capabilityHost schema (#30957)

* Update capabilityHost

* Update CapabilityHost schema

* Fix typo
---
 .../preview/2024-10-01-preview/mfe.json       | 47 ++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/preview/2024-10-01-preview/mfe.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/preview/2024-10-01-preview/mfe.json
index a927c6ee0853..3f21c8c0b78e 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/preview/2024-10-01-preview/mfe.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/preview/2024-10-01-preview/mfe.json
@@ -15010,8 +15010,21 @@
         }
       ],
       "properties": {
+        "aiServicesConnections": {
+          "description": "List of AI services connections.",
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "x-nullable": true
+        },
+        "capabilityHostKind": {
+          "description": "Kind of this capability host.",
+          "default": "Agents",
+          "$ref": "#/definitions/CapabilityHostKind"
+        },
         "customerSubnet": {
-          "description": "Represents a request to provision a resource.",
+          "description": "Customer subnet info to help set up this capability host.",
           "type": "string",
           "x-nullable": true
         },
@@ -15022,11 +15035,43 @@
           "x-ms-mutability": [
             "read"
           ]
+        },
+        "storageConnections": {
+          "description": "List of Storage connections.",
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "x-nullable": true
+        },
+        "vectorStoreConnections": {
+          "description": "List of VectorStore connections.",
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "x-nullable": true
         }
       },
       "x-ms-client-name": "CapabilityHostProperties",
       "additionalProperties": false
     },
+    "CapabilityHostKind": {
+      "enum": [
+        "Agents"
+      ],
+      "type": "string",
+      "x-ms-enum": {
+        "name": "CapabilityHostKind",
+        "modelAsString": true,
+        "values": [
+          {
+            "value": "Agents"
+          }
+        ]
+      },
+      "additionalProperties": false
+    },
     "CapabilityHostProvisioningState": {
       "description": "Provisioning state of capability host.",
       "enum": [