diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_recognizer_client.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_recognizer_client.py index 551b19709ee3..544209be1b11 100644 --- a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_recognizer_client.py +++ b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_recognizer_client.py @@ -281,7 +281,7 @@ def begin_recognize_custom_forms(self, model_id, form, **kwargs): raise ValueError("model_id cannot be None or empty.") polling_interval = kwargs.pop("polling_interval", self._client._config.polling_interval) - + continuation_token = kwargs.pop("continuation_token", None) content_type = kwargs.pop("content_type", None) if content_type == "application/json": raise TypeError("Call begin_recognize_custom_forms_from_url() to analyze a document from a URL.") @@ -301,6 +301,7 @@ def analyze_callback(raw_response, _, headers): # pylint: disable=unused-argume content_type=content_type, cls=kwargs.pop("cls", analyze_callback), polling=LROBasePolling(timeout=polling_interval, lro_algorithms=[AnalyzePolling()], **kwargs), + continuation_token=continuation_token, **kwargs ) @@ -329,7 +330,7 @@ def begin_recognize_custom_forms_from_url(self, model_id, form_url, **kwargs): raise ValueError("model_id cannot be None or empty.") polling_interval = kwargs.pop("polling_interval", self._client._config.polling_interval) - + continuation_token = kwargs.pop("continuation_token", None) include_field_elements = kwargs.pop("include_field_elements", False) def analyze_callback(raw_response, _, headers): # pylint: disable=unused-argument @@ -342,6 +343,7 @@ def analyze_callback(raw_response, _, headers): # pylint: disable=unused-argume include_text_details=include_field_elements, cls=kwargs.pop("cls", analyze_callback), polling=LROBasePolling(timeout=polling_interval, lro_algorithms=[AnalyzePolling()], **kwargs), + continuation_token=continuation_token, **kwargs ) diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py index ae1dd6b7fe0f..2fe81847989e 100644 --- a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py +++ b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py @@ -350,6 +350,7 @@ def begin_copy_model( if not model_id: raise ValueError("model_id cannot be None or empty.") polling_interval = kwargs.pop("polling_interval", self._client._config.polling_interval) + continuation_token = kwargs.pop("continuation_token", None) def _copy_callback(raw_response, _, headers): # pylint: disable=unused-argument copy_result = self._deserialize(self._generated_models.CopyOperationResult, raw_response) @@ -368,6 +369,7 @@ def _copy_callback(raw_response, _, headers): # pylint: disable=unused-argument ), cls=kwargs.pop("cls", _copy_callback), polling=LROBasePolling(timeout=polling_interval, lro_algorithms=[CopyPolling()], **kwargs), + continuation_token=continuation_token, **kwargs ) diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_recognizer_client_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_recognizer_client_async.py index da355cb0e88d..50cab0a7a2f5 100644 --- a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_recognizer_client_async.py +++ b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_recognizer_client_async.py @@ -294,6 +294,7 @@ async def begin_recognize_custom_forms( polling_interval = kwargs.pop("polling_interval", self._client._config.polling_interval) content_type = kwargs.pop("content_type", None) + continuation_token = kwargs.pop("continuation_token", None) if content_type == "application/json": raise TypeError("Call begin_recognize_custom_forms_from_url() to analyze a document from a URL.") @@ -317,6 +318,7 @@ def analyze_callback(raw_response, _, headers): # pylint: disable=unused-argume lro_algorithms=[AnalyzePolling()], **kwargs ), + continuation_token=continuation_token, **kwargs ) @@ -348,7 +350,7 @@ async def begin_recognize_custom_forms_from_url( if not model_id: raise ValueError("model_id cannot be None or empty.") polling_interval = kwargs.pop("polling_interval", self._client._config.polling_interval) - + continuation_token = kwargs.pop("continuation_token", None) include_field_elements = kwargs.pop("include_field_elements", False) def analyze_callback(raw_response, _, headers): # pylint: disable=unused-argument @@ -365,6 +367,7 @@ def analyze_callback(raw_response, _, headers): # pylint: disable=unused-argume lro_algorithms=[AnalyzePolling()], **kwargs ), + continuation_token=continuation_token, **kwargs ) diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py index 43ffeb3f8f8e..242e5504a56c 100644 --- a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py +++ b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py @@ -358,6 +358,7 @@ async def begin_copy_model( if not model_id: raise ValueError("model_id cannot be None or empty.") polling_interval = kwargs.pop("polling_interval", self._client._config.polling_interval) + continuation_token = kwargs.pop("continuation_token", None) def _copy_callback(raw_response, _, headers): # pylint: disable=unused-argument copy_result = self._deserialize(self._generated_models.CopyOperationResult, raw_response) @@ -380,6 +381,7 @@ def _copy_callback(raw_response, _, headers): # pylint: disable=unused-argument lro_algorithms=[CopyPolling()], **kwargs ), + continuation_token=continuation_token, **kwargs )