fix flaky id in polling (#19921)

Azure · Jul 23, 2021 · ee646d7 · ee646d7
1 parent f0f09cf
commit ee646d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_polling.py b/sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_polling.py
@@ -94,7 +94,7 @@ def _current_body(self):
 
     def _get_id_from_headers(self):
         # type: () -> str
-        return self._pipeline_response.http_response.headers[
+        return self._initial_response.http_response.headers[
             "Operation-Location"
         ].split("/batches/")[1]
 

diff --git a/...slation/azure-ai-translation-document/azure/ai/translation/document/aio/_async_polling.py b/...slation/azure-ai-translation-document/azure/ai/translation/document/aio/_async_polling.py
@@ -76,7 +76,7 @@ def _current_body(self) -> _TranslationStatus:
         return _TranslationStatus.deserialize(self._pipeline_response)
 
     def _get_id_from_headers(self) -> str:
-        return self._pipeline_response.http_response.headers[
+        return self._initial_response.http_response.headers[
             "Operation-Location"
         ].split("/batches/")[1]