azure-quantum 2.5.0.dev0__py3-none-any.whl → 2.5.0.dev2__py3-none-any.whl

azure/quantum/_client/__init__.py

@@ -2,31 +2,25 @@
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
-# Code generated by Microsoft (R) Python Code Generator.
+# Code generated by Microsoft (R) AutoRest Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
-# pylint: disable=wrong-import-position
 
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from ._patch import *  # pylint: disable=unused-wildcard-import
-
-from ._client import ServicesClient  # type: ignore
+from ._client import QuantumClient
 from ._version import VERSION
 
 __version__ = VERSION
 
 try:
     from ._patch import __all__ as _patch_all
-    from ._patch import *
+    from ._patch import *  # pylint: disable=unused-wildcard-import
 except ImportError:
     _patch_all = []
 from ._patch import patch_sdk as _patch_sdk
 
 __all__ = [
-    "ServicesClient",
+    "QuantumClient",
 ]
-__all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
+__all__.extend([p for p in _patch_all if p not in __all__])
 
 _patch_sdk()

azure/quantum/_client/_client.py

@@ -2,20 +2,19 @@
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
-# Code generated by Microsoft (R) Python Code Generator.
+# Code generated by Microsoft (R) AutoRest Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
 from copy import deepcopy
-from typing import Any, TYPE_CHECKING, Union
-from typing_extensions import Self
+from typing import Any, TYPE_CHECKING
 
 from azure.core import PipelineClient
-from azure.core.credentials import AzureKeyCredential
 from azure.core.pipeline import policies
 from azure.core.rest import HttpRequest, HttpResponse
 
-from ._configuration import ServicesClientConfiguration
+from . import models as _models
+from ._configuration import QuantumClientConfiguration
 from ._serialization import Deserializer, Serializer
 from .operations import (
     JobsOperations,
@@ -27,50 +26,59 @@ from .operations import (
 )
 
 if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
 
-class ServicesClient:
-    """Azure Quantum Workspace Services.
+class QuantumClient:  # pylint: disable=client-accepts-api-version-keyword
+    """Azure Quantum REST API client.
 
     :ivar jobs: JobsOperations operations
-    :vartype jobs: azure.quantum.operations.JobsOperations
-    :ivar sessions: SessionsOperations operations
-    :vartype sessions: azure.quantum.operations.SessionsOperations
+    :vartype jobs: azure.quantum._client.operations.JobsOperations
     :ivar providers: ProvidersOperations operations
-    :vartype providers: azure.quantum.operations.ProvidersOperations
+    :vartype providers: azure.quantum._client.operations.ProvidersOperations
     :ivar storage: StorageOperations operations
-    :vartype storage: azure.quantum.operations.StorageOperations
+    :vartype storage: azure.quantum._client.operations.StorageOperations
     :ivar quotas: QuotasOperations operations
-    :vartype quotas: azure.quantum.operations.QuotasOperations
+    :vartype quotas: azure.quantum._client.operations.QuotasOperations
+    :ivar sessions: SessionsOperations operations
+    :vartype sessions: azure.quantum._client.operations.SessionsOperations
     :ivar top_level_items: TopLevelItemsOperations operations
-    :vartype top_level_items: azure.quantum.operations.TopLevelItemsOperations
-    :param region: The Azure region where the Azure Quantum Workspace is located. Required.
-    :type region: str
-    :param credential: Credential used to authenticate requests to the service. Is either a
-     TokenCredential type or a AzureKeyCredential type. Required.
-    :type credential: ~azure.core.credentials.TokenCredential or
-     ~azure.core.credentials.AzureKeyCredential
-    :keyword service_base_url: The Azure Quantum service base url. Default value is
-     "quantum.azure.com".
-    :paramtype service_base_url: str
-    :keyword api_version: The API version to use for this operation. Default value is
-     "2024-10-01-preview". Note that overriding this default value may result in unsupported
-     behavior.
+    :vartype top_level_items: azure.quantum._client.operations.TopLevelItemsOperations
+    :param azure_region: Supported Azure regions for Azure Quantum Services. For example, "eastus".
+     Required.
+    :type azure_region: str
+    :param subscription_id: The Azure subscription ID. This is a GUID-formatted string (e.g.
+     00000000-0000-0000-0000-000000000000). Required.
+    :type subscription_id: str
+    :param resource_group_name: Name of an Azure resource group. Required.
+    :type resource_group_name: str
+    :param workspace_name: Name of the workspace. Required.
+    :type workspace_name: str
+    :param credential: Credential needed for the client to connect to Azure. Required.
+    :type credential: ~azure.core.credentials.TokenCredential
+    :keyword api_version: Api Version. Default value is "2023-11-13-preview". Note that overriding
+     this default value may result in unsupported behavior.
     :paramtype api_version: str
     """
 
     def __init__(
         self,
-        region: str,
-        credential: Union["TokenCredential", AzureKeyCredential],
-        *,
-        service_base_url: str = "quantum.azure.com",
+        azure_region: str,
+        subscription_id: str,
+        resource_group_name: str,
+        workspace_name: str,
+        credential: "TokenCredential",
         **kwargs: Any
     ) -> None:
-        _endpoint = "https://{region}.{serviceBaseUrl}"
-        self._config = ServicesClientConfiguration(
-            region=region, credential=credential, service_base_url=service_base_url, **kwargs
+        _endpoint = kwargs.pop("endpoint", f"https://{azure_region}.quantum.azure.com")
+        self._config = QuantumClientConfiguration(
+            azure_region=azure_region,
+            subscription_id=subscription_id,
+            resource_group_name=resource_group_name,
+            workspace_name=workspace_name,
+            credential=credential,
+            **kwargs
         )
         _policies = kwargs.pop("policies", None)
         if _policies is None:
@@ -91,14 +99,16 @@ class ServicesClient:
             ]
         self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs)
 
-        self._serialize = Serializer()
-        self._deserialize = Deserializer()
+        client_models = {k: v for k, v in _models._models.__dict__.items() if isinstance(v, type)}
+        client_models.update({k: v for k, v in _models.__dict__.items() if isinstance(v, type)})
+        self._serialize = Serializer(client_models)
+        self._deserialize = Deserializer(client_models)
         self._serialize.client_side_validation = False
         self.jobs = JobsOperations(self._client, self._config, self._serialize, self._deserialize)
-        self.sessions = SessionsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.providers = ProvidersOperations(self._client, self._config, self._serialize, self._deserialize)
         self.storage = StorageOperations(self._client, self._config, self._serialize, self._deserialize)
         self.quotas = QuotasOperations(self._client, self._config, self._serialize, self._deserialize)
+        self.sessions = SessionsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.top_level_items = TopLevelItemsOperations(self._client, self._config, self._serialize, self._deserialize)
 
     def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse:
@@ -121,9 +131,8 @@ class ServicesClient:
 
         request_copy = deepcopy(request)
         path_format_arguments = {
-            "region": self._serialize.url("self._config.region", self._config.region, "str"),
-            "serviceBaseUrl": self._serialize.url(
-                "self._config.service_base_url", self._config.service_base_url, "str"
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
             ),
         }
 
@@ -133,7 +142,7 @@ class ServicesClient:
     def close(self) -> None:
         self._client.close()
 
-    def __enter__(self) -> Self:
+    def __enter__(self) -> "QuantumClient":
         self._client.__enter__()
         return self
 

azure/quantum/_client/_configuration.py

@@ -2,72 +2,77 @@
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
-# Code generated by Microsoft (R) Python Code Generator.
+# Code generated by Microsoft (R) AutoRest Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from typing import Any, TYPE_CHECKING, Union
+from typing import Any, TYPE_CHECKING
 
-from azure.core.credentials import AzureKeyCredential
 from azure.core.pipeline import policies
 
 from ._version import VERSION
 
 if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
 
-class ServicesClientConfiguration:  # pylint: disable=too-many-instance-attributes
-    """Configuration for ServicesClient.
+class QuantumClientConfiguration:  # pylint: disable=too-many-instance-attributes
+    """Configuration for QuantumClient.
 
     Note that all parameters used to create this instance are saved as instance
     attributes.
 
-    :param region: The Azure region where the Azure Quantum Workspace is located. Required.
-    :type region: str
-    :param credential: Credential used to authenticate requests to the service. Is either a
-     TokenCredential type or a AzureKeyCredential type. Required.
-    :type credential: ~azure.core.credentials.TokenCredential or
-     ~azure.core.credentials.AzureKeyCredential
-    :param service_base_url: The Azure Quantum service base url. Default value is
-     "quantum.azure.com".
-    :type service_base_url: str
-    :keyword api_version: The API version to use for this operation. Default value is
-     "2024-10-01-preview". Note that overriding this default value may result in unsupported
-     behavior.
+    :param azure_region: Supported Azure regions for Azure Quantum Services. For example, "eastus".
+     Required.
+    :type azure_region: str
+    :param subscription_id: The Azure subscription ID. This is a GUID-formatted string (e.g.
+     00000000-0000-0000-0000-000000000000). Required.
+    :type subscription_id: str
+    :param resource_group_name: Name of an Azure resource group. Required.
+    :type resource_group_name: str
+    :param workspace_name: Name of the workspace. Required.
+    :type workspace_name: str
+    :param credential: Credential needed for the client to connect to Azure. Required.
+    :type credential: ~azure.core.credentials.TokenCredential
+    :keyword api_version: Api Version. Default value is "2023-11-13-preview". Note that overriding
+     this default value may result in unsupported behavior.
     :paramtype api_version: str
     """
 
     def __init__(
         self,
-        region: str,
-        credential: Union["TokenCredential", AzureKeyCredential],
-        service_base_url: str = "quantum.azure.com",
-        **kwargs: Any,
+        azure_region: str,
+        subscription_id: str,
+        resource_group_name: str,
+        workspace_name: str,
+        credential: "TokenCredential",
+        **kwargs: Any
     ) -> None:
-        api_version: str = kwargs.pop("api_version", "2024-10-01-preview")
+        api_version: str = kwargs.pop("api_version", "2022-09-12-preview")
 
-        if region is None:
-            raise ValueError("Parameter 'region' must not be None.")
+        if azure_region is None:
+            raise ValueError("Parameter 'azure_region' must not be None.")
+        if subscription_id is None:
+            raise ValueError("Parameter 'subscription_id' must not be None.")
+        if resource_group_name is None:
+            raise ValueError("Parameter 'resource_group_name' must not be None.")
+        if workspace_name is None:
+            raise ValueError("Parameter 'workspace_name' must not be None.")
         if credential is None:
             raise ValueError("Parameter 'credential' must not be None.")
 
-        self.region = region
+        self.azure_region = azure_region
+        self.subscription_id = subscription_id
+        self.resource_group_name = resource_group_name
+        self.workspace_name = workspace_name
         self.credential = credential
-        self.service_base_url = service_base_url
         self.api_version = api_version
         self.credential_scopes = kwargs.pop("credential_scopes", ["https://quantum.microsoft.com/.default"])
         kwargs.setdefault("sdk_moniker", "quantum/{}".format(VERSION))
         self.polling_interval = kwargs.get("polling_interval", 30)
         self._configure(**kwargs)
 
-    def _infer_policy(self, **kwargs):
-        if hasattr(self.credential, "get_token"):
-            return policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs)
-        if isinstance(self.credential, AzureKeyCredential):
-            return policies.AzureKeyCredentialPolicy(self.credential, "x-ms-quantum-api-key", **kwargs)
-        raise TypeError(f"Unsupported credential: {self.credential}")
-
     def _configure(self, **kwargs: Any) -> None:
         self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs)
         self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs)
@@ -79,4 +84,6 @@ class ServicesClientConfiguration:  # pylint: disable=too-many-instance-attribut
         self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs)
         self.authentication_policy = kwargs.get("authentication_policy")
         if self.credential and not self.authentication_policy:
-            self.authentication_policy = self._infer_policy(**kwargs)
+            self.authentication_policy = policies.BearerTokenCredentialPolicy(
+                self.credential, *self.credential_scopes, **kwargs
+            )