CodeGen from PR 32103 in Azure/azure-rest-api-specs

Merge 1f152afbf84c6febe1f4447d284750f03c7f188f into fc2f4c6f6f8453c2eb58b6fb5ad8952d4c507971

Author: SDKAuto

sdk/ai/azure-ai-inference/_meta.json

@@ -0,0 +1,6 @@
+{
+  "commit": "023d3cfb7915c51b09b19a7bf9e30d1643cbd8b8",
+  "repository_url": "https://github.com/Azure/azure-rest-api-specs",
+  "typespec_src": "specification/ai/ModelClient",
+  "@azure-tools/typespec-python": "0.38.0"
+}

sdk/ai/azure-ai-inference/azure/ai/inference/__init__.py

@@ -5,24 +5,32 @@
 # Code generated by Microsoft (R) Python Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
+# pylint: disable=wrong-import-position
 
-from ._patch import ChatCompletionsClient
-from ._patch import EmbeddingsClient
-from ._patch import ImageEmbeddingsClient
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ._patch import *  # pylint: disable=unused-wildcard-import
+
+from ._client import ChatCompletionsClient  # type: ignore
+from ._client import EmbeddingsClient  # type: ignore
+from ._client import ImageEmbeddingsClient  # type: ignore
 from ._version import VERSION
 
 __version__ = VERSION
 
-
-from ._patch import load_client
+try:
+    from ._patch import __all__ as _patch_all
+    from ._patch import *
+except ImportError:
+    _patch_all = []
 from ._patch import patch_sdk as _patch_sdk
 
 __all__ = [
-    "load_client",
     "ChatCompletionsClient",
     "EmbeddingsClient",
     "ImageEmbeddingsClient",
 ]
-
+__all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
 
 _patch_sdk()

sdk/ai/azure-ai-inference/azure/ai/inference/_client.py

@@ -28,17 +28,16 @@
 from ._serialization import Deserializer, Serializer
 
 if TYPE_CHECKING:
-    # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
 
-class ChatCompletionsClient(ChatCompletionsClientOperationsMixin):  # pylint: disable=client-accepts-api-version-keyword
+class ChatCompletionsClient(ChatCompletionsClientOperationsMixin):
     """ChatCompletionsClient.
 
     :param endpoint: Service host. Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Is one of the
-     following types: AzureKeyCredential, AzureKeyCredential, TokenCredential Required.
+     following types: AzureKeyCredential, AzureKeyCredential, token credential Required.
     :type credential: ~azure.core.credentials.AzureKeyCredential or
      ~azure.core.credentials.AzureKeyCredential or ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is
@@ -110,13 +109,13 @@ def __exit__(self, *exc_details: Any) -> None:
         self._client.__exit__(*exc_details)
 
 
-class EmbeddingsClient(EmbeddingsClientOperationsMixin):  # pylint: disable=client-accepts-api-version-keyword
+class EmbeddingsClient(EmbeddingsClientOperationsMixin):
     """EmbeddingsClient.
 
     :param endpoint: Service host. Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Is one of the
-     following types: AzureKeyCredential, AzureKeyCredential, TokenCredential Required.
+     following types: AzureKeyCredential, AzureKeyCredential, token credential Required.
     :type credential: ~azure.core.credentials.AzureKeyCredential or
      ~azure.core.credentials.AzureKeyCredential or ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is
@@ -188,13 +187,13 @@ def __exit__(self, *exc_details: Any) -> None:
         self._client.__exit__(*exc_details)
 
 
-class ImageEmbeddingsClient(ImageEmbeddingsClientOperationsMixin):  # pylint: disable=client-accepts-api-version-keyword
+class ImageEmbeddingsClient(ImageEmbeddingsClientOperationsMixin):
     """ImageEmbeddingsClient.
 
     :param endpoint: Service host. Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Is one of the
-     following types: AzureKeyCredential, AzureKeyCredential, TokenCredential Required.
+     following types: AzureKeyCredential, AzureKeyCredential, token credential Required.
     :type credential: ~azure.core.credentials.AzureKeyCredential or
      ~azure.core.credentials.AzureKeyCredential or ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is

sdk/ai/azure-ai-inference/azure/ai/inference/_configuration.py

@@ -14,11 +14,10 @@
 from ._version import VERSION
 
 if TYPE_CHECKING:
-    # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
 
-class ChatCompletionsClientConfiguration:  # pylint: disable=too-many-instance-attributes,name-too-long
+class ChatCompletionsClientConfiguration:  # pylint: disable=too-many-instance-attributes
     """Configuration for ChatCompletionsClient.
 
     Note that all parameters used to create this instance are saved as instance
@@ -27,7 +26,7 @@ class ChatCompletionsClientConfiguration:  # pylint: disable=too-many-instance-a
     :param endpoint: Service host. Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Is one of the
-     following types: AzureKeyCredential, AzureKeyCredential, TokenCredential Required.
+     following types: AzureKeyCredential, AzureKeyCredential, token credential Required.
     :type credential: ~azure.core.credentials.AzureKeyCredential or
      ~azure.core.credentials.AzureKeyCredential or ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is
@@ -75,7 +74,7 @@ def _configure(self, **kwargs: Any) -> None:
             self.authentication_policy = self._infer_policy(**kwargs)
 
 
-class EmbeddingsClientConfiguration:  # pylint: disable=too-many-instance-attributes,name-too-long
+class EmbeddingsClientConfiguration:  # pylint: disable=too-many-instance-attributes
     """Configuration for EmbeddingsClient.
 
     Note that all parameters used to create this instance are saved as instance
@@ -84,7 +83,7 @@ class EmbeddingsClientConfiguration:  # pylint: disable=too-many-instance-attrib
     :param endpoint: Service host. Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Is one of the
-     following types: AzureKeyCredential, AzureKeyCredential, TokenCredential Required.
+     following types: AzureKeyCredential, AzureKeyCredential, token credential Required.
     :type credential: ~azure.core.credentials.AzureKeyCredential or
      ~azure.core.credentials.AzureKeyCredential or ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is
@@ -132,7 +131,7 @@ def _configure(self, **kwargs: Any) -> None:
             self.authentication_policy = self._infer_policy(**kwargs)
 
 
-class ImageEmbeddingsClientConfiguration:  # pylint: disable=too-many-instance-attributes,name-too-long
+class ImageEmbeddingsClientConfiguration:  # pylint: disable=too-many-instance-attributes
     """Configuration for ImageEmbeddingsClient.
 
     Note that all parameters used to create this instance are saved as instance
@@ -141,7 +140,7 @@ class ImageEmbeddingsClientConfiguration:  # pylint: disable=too-many-instance-a
     :param endpoint: Service host. Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Is one of the
-     following types: AzureKeyCredential, AzureKeyCredential, TokenCredential Required.
+     following types: AzureKeyCredential, AzureKeyCredential, token credential Required.
     :type credential: ~azure.core.credentials.AzureKeyCredential or
      ~azure.core.credentials.AzureKeyCredential or ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is