add for pro mode

Author: chienyuanchang

python/content_understanding_client.py

@@ -49,6 +49,16 @@ def _get_training_data_config(
             "kind": "blob",
             "prefix": storage_container_path_prefix,
         }
+    
+    def _get_pro_mode_reference_docs_config(
+        self, storage_container_sas_url, storage_container_path_prefix
+    ):
+        return {
+            "kind": "reference",
+            "containerUrl": storage_container_sas_url,
+            "prefix": storage_container_path_prefix,
+            "fileListPath": "sources.jsonl",
+        }
 
     def _get_headers(self, subscription_key, api_token, x_ms_useragent):
         """Returns the headers for the HTTP requests.
@@ -116,6 +126,8 @@ def begin_create_analyzer(
         analyzer_template_path: str = "",
         training_storage_container_sas_url: str = "",
         training_storage_container_path_prefix: str = "",
+        pro_mode_reference_docs_storage_container_sas_url: str = "",
+        pro_mode_reference_docs_storage_container_path_prefix: str = "",
     ):
         """
         Initiates the creation of an analyzer with the given ID and schema.
@@ -150,6 +162,15 @@ def begin_create_analyzer(
                 training_storage_container_path_prefix,
             )
 
+        if (
+            pro_mode_reference_docs_storage_container_sas_url
+            and pro_mode_reference_docs_storage_container_path_prefix
+        ):  # noqa
+            analyzer_template["knowledgeSources"] = self._get_pro_mode_reference_docs_config(
+                pro_mode_reference_docs_storage_container_sas_url,
+                pro_mode_reference_docs_storage_container_path_prefix,
+            )
+
         headers = {"Content-Type": "application/json"}
         headers.update(self._headers)