[document intelligence] Fix copy operation polling (#40105)

* fix copy operation polling

* spelling

* remove import

* release prep

* pylint

* date

---------

Co-authored-by: catalinaperalta <[email protected]>

Author: catalinaperalta

sdk/documentintelligence/azure-ai-documentintelligence/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Release History
 
+## 1.0.2 (2025-03-26)
+
+### Bugs Fixed
+
+- Fix polling for `begin_copy_classifier_to()` to stop on success response from the "Operation-Location" endpoint and correctly parse the result.
+
 ## 1.0.1 (2025-03-13)
 
 ### Bugs Fixed

sdk/documentintelligence/azure-ai-documentintelligence/azure/ai/documentintelligence/_operations/_patch.py

@@ -14,7 +14,7 @@
 
 from azure.core.pipeline import PipelineResponse
 from azure.core.polling import LROPoller, NoPolling, PollingMethod
-from azure.core.polling.base_polling import LROBasePolling, OperationResourcePolling
+from azure.core.polling.base_polling import LROBasePolling
 from azure.core.rest import HttpRequest, HttpResponse
 from azure.core.tracing.decorator import distributed_trace
 from azure.core.utils import case_insensitive_dict
@@ -41,17 +41,6 @@ def _parse_operation_id(operation_location_header):
     return re.match(regex, operation_location_header).group(1)
 
 
-class DocumentModelAdministrationPolling(OperationResourcePolling):
-    """Polling method overrides for administration endpoints."""
-
-    def get_final_get_url(self, pipeline_response: Any) -> None:
-        """If a final GET is needed, returns the URL.
-
-        :param any pipeline_response: The pipeline response to get the final url.
-        :rtype: None
-        """
-        return None
-
 class AnalyzeDocumentLROPoller(LROPoller[PollingReturnType_co]):
     @property
     def details(self) -> Mapping[str, Any]:
@@ -295,7 +284,7 @@ def get_long_running_output(pipeline_response):
                 "str", response.headers.get("Operation-Location")
             )
 
-            deserialized = _deserialize(_models.DocumentModelDetails, response.json()["result"])
+            deserialized = _deserialize(_models.DocumentModelDetails, response.json().get("result"))
             if cls:
                 return cls(pipeline_response, deserialized, response_headers)  # type: ignore
             return deserialized
@@ -306,7 +295,13 @@ def get_long_running_output(pipeline_response):
 
         if polling is True:
             polling_method: PollingMethod = cast(
-                PollingMethod, LROBasePolling(lro_delay, path_format_arguments=path_format_arguments, lro_algorithms=[DocumentModelAdministrationPolling()], **kwargs)
+                PollingMethod,
+                LROBasePolling(
+                    lro_delay,
+                    path_format_arguments=path_format_arguments,
+                    lro_options={"final-state-via": "operation-location"},
+                    **kwargs,
+                ),
             )
         elif polling is False:
             polling_method = cast(PollingMethod, NoPolling())
@@ -359,7 +354,7 @@ def get_long_running_output(pipeline_response):
                 "str", response.headers.get("Operation-Location")
             )
 
-            deserialized = _deserialize(_models.DocumentClassifierDetails, response.json())
+            deserialized = _deserialize(_models.DocumentClassifierDetails, response.json().get("result"))
             if cls:
                 return cls(pipeline_response, deserialized, response_headers)  # type: ignore
             return deserialized
@@ -370,7 +365,7 @@ def get_long_running_output(pipeline_response):
 
         if polling is True:
             polling_method: PollingMethod = cast(
-                PollingMethod, LROBasePolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs)
+                PollingMethod, LROBasePolling(lro_delay, path_format_arguments=path_format_arguments, lro_options={"final-state-via": "operation-location"}, **kwargs)
             )
         elif polling is False:
             polling_method = cast(PollingMethod, NoPolling())

sdk/documentintelligence/azure-ai-documentintelligence/azure/ai/documentintelligence/_version.py

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "1.0.1"
+VERSION = "1.0.2"

sdk/documentintelligence/azure-ai-documentintelligence/azure/ai/documentintelligence/aio/_operations/_patch.py

@@ -1,3 +1,4 @@
+# pylint: disable=line-too-long,useless-suppression
 # ------------------------------------
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
@@ -23,7 +24,7 @@
 )
 from ... import models as _models
 from ..._model_base import _deserialize
-from ..._operations._patch import PollingReturnType_co, _parse_operation_id, DocumentModelAdministrationPolling
+from ..._operations._patch import PollingReturnType_co, _parse_operation_id
 
 if sys.version_info >= (3, 9):
     from collections.abc import MutableMapping
@@ -265,7 +266,7 @@ def get_long_running_output(pipeline_response):
                 "str", response.headers.get("Operation-Location")
             )
 
-            deserialized = _deserialize(_models.DocumentModelDetails, response.json()["result"])
+            deserialized = _deserialize(_models.DocumentModelDetails, response.json().get("result"))
             if cls:
                 return cls(pipeline_response, deserialized, response_headers)  # type: ignore
             return deserialized
@@ -280,7 +281,7 @@ def get_long_running_output(pipeline_response):
                 AsyncLROBasePolling(
                     lro_delay,
                     path_format_arguments=path_format_arguments,
-                    lro_algorithms=[DocumentModelAdministrationPolling()],
+                    lro_options={"final-state-via": "operation-location"},
                     **kwargs
                 ),
             )
@@ -332,7 +333,7 @@ def get_long_running_output(pipeline_response):
                 "str", response.headers.get("Operation-Location")
             )
 
-            deserialized = _deserialize(_models.DocumentClassifierDetails, response.json())
+            deserialized = _deserialize(_models.DocumentClassifierDetails, response.json().get("result"))
             if cls:
                 return cls(pipeline_response, deserialized, response_headers)  # type: ignore
             return deserialized
@@ -344,7 +345,7 @@ def get_long_running_output(pipeline_response):
         if polling is True:
             polling_method: AsyncPollingMethod = cast(
                 AsyncPollingMethod,
-                AsyncLROBasePolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs),
+                AsyncLROBasePolling(lro_delay, path_format_arguments=path_format_arguments, lro_options={"final-state-via": "operation-location"}, **kwargs),
             )
         elif polling is False:
             polling_method = cast(AsyncPollingMethod, AsyncNoPolling())