diff --git a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_base_client.py b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_base_client.py index ee6c6d3c0ffa..da878fc5928a 100644 --- a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_base_client.py +++ b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_base_client.py @@ -70,6 +70,7 @@ def __init__( "$top", "$skip", "opinionMining", + "api-version" } ) try: diff --git a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_policies.py b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_policies.py index 1d3ee9d1f0b7..04e1c1efa9a6 100644 --- a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_policies.py +++ b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/_policies.py @@ -28,18 +28,18 @@ def on_response(self, request, response): data = ContentDecodePolicy.deserialize_from_http_generics( response.http_response ) - if self._is_lro and (not data or data.get("status") not in _FINISHED): + if self._is_lro and (not data or data.get("status", "").lower() not in _FINISHED): + return + if response.http_response.status_code == 429: return if data: inner = data.get("results", data) # language API compat statistics = inner.get("statistics", None) model_version = inner.get("modelVersion", None) - - if statistics or model_version: - batch_statistics = TextDocumentBatchStatistics._from_generated( # pylint: disable=protected-access - statistics - ) - response.statistics = batch_statistics - response.model_version = model_version + batch_statistics = TextDocumentBatchStatistics._from_generated( # pylint: disable=protected-access + statistics + ) + response.statistics = batch_statistics + response.model_version = model_version response.raw_response = data self._response_callback(response) diff --git a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/aio/_base_client_async.py b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/aio/_base_client_async.py index 9fb4ea73df88..de1d78136bda 100644 --- a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/aio/_base_client_async.py +++ b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/aio/_base_client_async.py @@ -57,6 +57,7 @@ def __init__( "$top", "$skip", "opinionMining", + "api-version" } ) try: