Foundry change (Azure#40753)

Adding sample and description for target storage

Author: achauhan-scc

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_workspace/_ai_workspaces/capability_host.py

@@ -36,6 +36,8 @@ class CapabilityHost(Resource):
     :param storage_connections: A list of storage connections. Default storage connection value is
         projectname/workspaceblobstore for project workspace.
     :type storage_connections: Optional[List[str]]
+    :param thread_storage_connections: A list of cosmos db connections, which will be used for thread storage.
+    :type thread_storage_connections: Optional[List[str]]
     :param capability_host_kind: The kind of capability host, either as a string or CapabilityHostKind enum.
         Default is AGENTS.
     :type capability_host_kind: Union[str, CapabilityHostKind]

sdk/ml/azure-ai-ml/samples/ml_samples_capability_host.py

@@ -30,10 +30,8 @@ def ml_capability_host_config(self):
         # [END load_capability_host]
 
         # [START capability_host_object_create]
-        from azure.ai.ml.entities._workspace._ai_workspaces.capability_host import (
-            CapabilityHost,
-        )
         from azure.ai.ml.constants._workspace import CapabilityHostKind
+        from azure.ai.ml.entities._workspace._ai_workspaces.capability_host import CapabilityHost
 
         # CapabilityHost in Hub workspace. For Hub workspace, only name and description are required.
         capability_host = CapabilityHost(
@@ -50,16 +48,15 @@ def ml_capability_host_config(self):
             ai_services_connections=["connection1"],
             storage_connections=["projectname/workspaceblobstore"],
             vector_store_connections=["connection1"],
+            thread_storage_connections=["connection1"],
         )
         # [END capability_host_object_create]
 
         # [START capability_host_begin_create_or_update_operation]
         from azure.ai.ml import MLClient
-        from azure.identity import DefaultAzureCredential
-        from azure.ai.ml.entities._workspace._ai_workspaces.capability_host import (
-            CapabilityHost,
-        )
         from azure.ai.ml.constants._workspace import CapabilityHostKind
+        from azure.ai.ml.entities._workspace._ai_workspaces.capability_host import CapabilityHost
+        from azure.identity import DefaultAzureCredential
 
         subscription_id = os.environ["AZURE_SUBSCRIPTION_ID"]
         resource_group = os.environ["RESOURCE_GROUP_NAME"]