CodeGen from PR 3537 in test-repo-billy/azure-rest-api-specs

Merge 6296b8e5c14516850f6b39a7fee10b2f0db170a5 into 6c6e1713f03acaf37e5c89652b023dd2c9eae16e

Author: SDKAuto

sdk/translation/azure-ai-translation-document/_meta.json

@@ -0,0 +1,6 @@
+{
+  "commit": "c29452397f8d29456182b951987a765d1ae01bb1",
+  "repository_url": "https://github.com/test-repo-billy/azure-rest-api-specs",
+  "typespec_src": "specification/translation/Azure.AI.DocumentTranslation",
+  "@azure-tools/typespec-python": "0.31.1"
+}

sdk/translation/azure-ai-translation-document/azure/ai/translation/document/__init__.py

@@ -6,39 +6,23 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from ._patch import DocumentTranslationClient
+from ._client import DocumentTranslationClient
 from ._client import SingleDocumentTranslationClient
 from ._version import VERSION
 
 __version__ = VERSION
 
-
-from ._patch import DocumentTranslationApiVersion
-from ._patch import DocumentTranslationLROPoller
-from ._patch import TranslationGlossary
-from ._patch import TranslationTarget
-from ._patch import DocumentTranslationInput
-from ._patch import TranslationStatus
-from ._patch import DocumentStatus
-from ._patch import DocumentTranslationError
-from ._patch import DocumentTranslationFileFormat
-from ._patch import StorageInputType
+try:
+    from ._patch import __all__ as _patch_all
+    from ._patch import *  # pylint: disable=unused-wildcard-import
+except ImportError:
+    _patch_all = []
 from ._patch import patch_sdk as _patch_sdk
 
 __all__ = [
-    "DocumentTranslationApiVersion",
-    "DocumentTranslationLROPoller",
-    "TranslationGlossary",
-    "TranslationTarget",
-    "DocumentTranslationInput",
-    "TranslationStatus",
-    "DocumentStatus",
-    "DocumentTranslationError",
-    "DocumentTranslationFileFormat",
-    "StorageInputType",
     "DocumentTranslationClient",
     "SingleDocumentTranslationClient",
 ]
-
+__all__.extend([p for p in _patch_all if p not in __all__])
 
 _patch_sdk()

sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_client.py

@@ -8,6 +8,7 @@
 
 from copy import deepcopy
 from typing import Any, TYPE_CHECKING, Union
+from typing_extensions import Self
 
 from azure.core import PipelineClient
 from azure.core.credentials import AzureKeyCredential
@@ -97,7 +98,7 @@ def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs:
     def close(self) -> None:
         self._client.close()
 
-    def __enter__(self) -> "DocumentTranslationClient":
+    def __enter__(self) -> Self:
         self._client.__enter__()
         return self
 
@@ -177,7 +178,7 @@ def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs:
     def close(self) -> None:
         self._client.close()
 
-    def __enter__(self) -> "SingleDocumentTranslationClient":
+    def __enter__(self) -> Self:
         self._client.__enter__()
         return self