azure-quantum 1.1.1__py3-none-any.whl → 1.1.2.dev0__py3-none-any.whl

azure/quantum/_client/operations/_operations.py

@@ -1,4 +1,4 @@
-# pylint: disable=too-many-lines
+# pylint: disable=too-many-lines,too-many-statements
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
@@ -6,7 +6,7 @@
 # Code generated by Microsoft (R) AutoRest Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
-import sys
+from io import IOBase
 from typing import Any, Callable, Dict, IO, Iterable, List, Optional, TypeVar, Union, overload
 import urllib.parse
 
@@ -20,19 +20,13 @@ from azure.core.exceptions import (
 )
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
-from azure.core.pipeline.transport import HttpResponse
-from azure.core.rest import HttpRequest
+from azure.core.rest import HttpRequest, HttpResponse
 from azure.core.tracing.decorator import distributed_trace
 from azure.core.utils import case_insensitive_dict
 
 from .. import models as _models
 from .._serialization import Serializer
-from .._vendor import _format_url_section
 
-if sys.version_info >= (3, 8):
-    from typing import Literal  # pylint: disable=no-name-in-module, ungrouped-imports
-else:
-    from typing_extensions import Literal  # type: ignore  # pylint: disable=ungrouped-imports
 T = TypeVar("T")
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]]
 
@@ -46,9 +40,7 @@ def build_jobs_list_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -59,7 +51,7 @@ def build_jobs_list_request(
         "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -76,9 +68,7 @@ def build_jobs_get_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -96,7 +86,7 @@ def build_jobs_get_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -114,9 +104,7 @@ def build_jobs_create_request(
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
     content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -134,7 +122,7 @@ def build_jobs_create_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -153,9 +141,7 @@ def build_jobs_cancel_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -173,7 +159,7 @@ def build_jobs_cancel_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -191,9 +177,7 @@ def build_jobs_patch_request(
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
     content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -211,7 +195,7 @@ def build_jobs_patch_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -230,9 +214,7 @@ def build_providers_get_status_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -243,7 +225,7 @@ def build_providers_get_status_request(
         "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -261,9 +243,7 @@ def build_storage_sas_uri_request(
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
     content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -274,7 +254,7 @@ def build_storage_sas_uri_request(
         "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -293,9 +273,7 @@ def build_quotas_list_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -306,7 +284,7 @@ def build_quotas_list_request(
         "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -323,9 +301,7 @@ def build_sessions_list_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -336,7 +312,7 @@ def build_sessions_list_request(
         "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -353,9 +329,7 @@ def build_sessions_get_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -373,7 +347,7 @@ def build_sessions_get_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -391,9 +365,7 @@ def build_sessions_open_request(
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
     content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -411,7 +383,7 @@ def build_sessions_open_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -430,9 +402,7 @@ def build_sessions_close_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -450,7 +420,7 @@ def build_sessions_close_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -467,9 +437,7 @@ def build_sessions_jobs_list_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -487,7 +455,7 @@ def build_sessions_jobs_list_request(
         ),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -504,9 +472,7 @@ def build_top_level_items_list_request(
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-    api_version: Literal["2022-09-12-preview"] = kwargs.pop(
-        "api_version", _params.pop("api-version", "2022-09-12-preview")
-    )
+    api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-11-13-preview"))
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
@@ -517,7 +483,7 @@ def build_top_level_items_list_request(
         "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str")
@@ -571,7 +537,7 @@ class JobsOperations:
         def prepare_request(next_link=None):
             if not next_link:
 
-                request = build_jobs_list_request(
+                _request = build_jobs_list_request(
                     subscription_id=self._config.subscription_id,
                     resource_group_name=self._config.resource_group_name,
                     workspace_name=self._config.workspace_name,
@@ -579,7 +545,12 @@ class JobsOperations:
                     headers=_headers,
                     params=_params,
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
             else:
                 # make call to next link with the client's api-version
@@ -591,12 +562,17 @@ class JobsOperations:
                     }
                 )
                 _next_request_params["api-version"] = self._config.api_version
-                request = HttpRequest(
+                _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
-            return request
+            return _request
 
         def extract_data(pipeline_response):
             deserialized = self._deserialize(
@@ -608,15 +584,17 @@ class JobsOperations:
             return deserialized.next_link or None, iter(list_of_elem)
 
         def get_next(next_link=None):
-            request = prepare_request(next_link)
+            _request = prepare_request(next_link)
 
             _stream = False
             pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
             response = pipeline_response.http_response
 
             if response.status_code not in [200]:
+                if _stream:
+                    response.read()  # Load the body in memory and close the socket
                 map_error(status_code=response.status_code, response=response, error_map=error_map)
                 error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
                 raise HttpResponseError(response=response, model=error)
@@ -648,7 +626,7 @@ class JobsOperations:
 
         cls: ClsType[_models.JobDetails] = kwargs.pop("cls", None)
 
-        request = build_jobs_get_request(
+        _request = build_jobs_get_request(
             job_id=job_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -657,16 +635,23 @@ class JobsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -674,9 +659,9 @@ class JobsOperations:
         deserialized = self._deserialize("JobDetails", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def create(
@@ -698,14 +683,14 @@ class JobsOperations:
 
     @overload
     def create(
-        self, job_id: str, job: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, job_id: str, job: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.JobDetails:
         """Create a job.
 
         :param job_id: Id of the job. Required.
         :type job_id: str
         :param job: The complete metadata of the job to submit. Required.
-        :type job: IO
+        :type job: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -715,14 +700,14 @@ class JobsOperations:
         """
 
     @distributed_trace
-    def create(self, job_id: str, job: Union[_models.JobDetails, IO], **kwargs: Any) -> _models.JobDetails:
+    def create(self, job_id: str, job: Union[_models.JobDetails, IO[bytes]], **kwargs: Any) -> _models.JobDetails:
         """Create a job.
 
         :param job_id: Id of the job. Required.
         :type job_id: str
-        :param job: The complete metadata of the job to submit. Is either a JobDetails type or a IO
-         type. Required.
-        :type job: ~azure.quantum._client.models.JobDetails or IO
+        :param job: The complete metadata of the job to submit. Is either a JobDetails type or a
+         IO[bytes] type. Required.
+        :type job: ~azure.quantum._client.models.JobDetails or IO[bytes]
         :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
          Default value is None.
         :paramtype content_type: str
@@ -747,12 +732,12 @@ class JobsOperations:
         content_type = content_type or "application/json"
         _json = None
         _content = None
-        if isinstance(job, (IO, bytes)):
+        if isinstance(job, (IOBase, bytes)):
             _content = job
         else:
             _json = self._serialize.body(job, "JobDetails")
 
-        request = build_jobs_create_request(
+        _request = build_jobs_create_request(
             job_id=job_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -764,16 +749,23 @@ class JobsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 201]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -812,7 +804,7 @@ class JobsOperations:
 
         cls: ClsType[None] = kwargs.pop("cls", None)
 
-        request = build_jobs_cancel_request(
+        _request = build_jobs_cancel_request(
             job_id=job_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -821,22 +813,29 @@ class JobsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [204]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
 
         if cls:
-            return cls(pipeline_response, None, {})
+            return cls(pipeline_response, None, {})  # type: ignore
 
     @overload
     def patch(
@@ -863,14 +862,14 @@ class JobsOperations:
 
     @overload
     def patch(
-        self, job_id: str, patch_job: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, job_id: str, patch_job: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> Optional[_models.JobDetails]:
         """Patch a job.
 
         :param job_id: Id of the job. Required.
         :type job_id: str
         :param patch_job: The json patch document containing the patch operations. Required.
-        :type patch_job: IO
+        :type patch_job: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -881,15 +880,15 @@ class JobsOperations:
 
     @distributed_trace
     def patch(
-        self, job_id: str, patch_job: Union[List[_models.JsonPatchDocument], IO], **kwargs: Any
+        self, job_id: str, patch_job: Union[List[_models.JsonPatchDocument], IO[bytes]], **kwargs: Any
     ) -> Optional[_models.JobDetails]:
         """Patch a job.
 
         :param job_id: Id of the job. Required.
         :type job_id: str
         :param patch_job: The json patch document containing the patch operations. Is either a
-         [JsonPatchDocument] type or a IO type. Required.
-        :type patch_job: list[~azure.quantum._client.models.JsonPatchDocument] or IO
+         [JsonPatchDocument] type or a IO[bytes] type. Required.
+        :type patch_job: list[~azure.quantum._client.models.JsonPatchDocument] or IO[bytes]
         :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
          Default value is None.
         :paramtype content_type: str
@@ -914,12 +913,12 @@ class JobsOperations:
         content_type = content_type or "application/json"
         _json = None
         _content = None
-        if isinstance(patch_job, (IO, bytes)):
+        if isinstance(patch_job, (IOBase, bytes)):
             _content = patch_job
         else:
             _json = self._serialize.body(patch_job, "[JsonPatchDocument]")
 
-        request = build_jobs_patch_request(
+        _request = build_jobs_patch_request(
             job_id=job_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -931,16 +930,23 @@ class JobsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 204]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -950,9 +956,9 @@ class JobsOperations:
             deserialized = self._deserialize("JobDetails", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
 
 class ProvidersOperations:
@@ -998,7 +1004,7 @@ class ProvidersOperations:
         def prepare_request(next_link=None):
             if not next_link:
 
-                request = build_providers_get_status_request(
+                _request = build_providers_get_status_request(
                     subscription_id=self._config.subscription_id,
                     resource_group_name=self._config.resource_group_name,
                     workspace_name=self._config.workspace_name,
@@ -1006,7 +1012,12 @@ class ProvidersOperations:
                     headers=_headers,
                     params=_params,
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
             else:
                 # make call to next link with the client's api-version
@@ -1018,12 +1029,17 @@ class ProvidersOperations:
                     }
                 )
                 _next_request_params["api-version"] = self._config.api_version
-                request = HttpRequest(
+                _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
-            return request
+            return _request
 
         def extract_data(pipeline_response):
             deserialized = self._deserialize(
@@ -1035,15 +1051,17 @@ class ProvidersOperations:
             return deserialized.next_link or None, iter(list_of_elem)
 
         def get_next(next_link=None):
-            request = prepare_request(next_link)
+            _request = prepare_request(next_link)
 
             _stream = False
             pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
             response = pipeline_response.http_response
 
             if response.status_code not in [200]:
+                if _stream:
+                    response.read()  # Load the body in memory and close the socket
                 map_error(status_code=response.status_code, response=response, error_map=error_map)
                 error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
                 raise HttpResponseError(response=response, model=error)
@@ -1092,14 +1110,14 @@ class StorageOperations:
 
     @overload
     def sas_uri(
-        self, blob_details: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, blob_details: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.SasUriResponse:
         """Gets a URL with SAS token for a container/blob in the storage account associated with the
         workspace. The SAS URL can be used to upload job input and/or download job output.
 
         :param blob_details: The details (name and container) of the blob to store or download data.
          Required.
-        :type blob_details: IO
+        :type blob_details: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -1109,13 +1127,13 @@ class StorageOperations:
         """
 
     @distributed_trace
-    def sas_uri(self, blob_details: Union[_models.BlobDetails, IO], **kwargs: Any) -> _models.SasUriResponse:
+    def sas_uri(self, blob_details: Union[_models.BlobDetails, IO[bytes]], **kwargs: Any) -> _models.SasUriResponse:
         """Gets a URL with SAS token for a container/blob in the storage account associated with the
         workspace. The SAS URL can be used to upload job input and/or download job output.
 
         :param blob_details: The details (name and container) of the blob to store or download data. Is
-         either a BlobDetails type or a IO type. Required.
-        :type blob_details: ~azure.quantum._client.models.BlobDetails or IO
+         either a BlobDetails type or a IO[bytes] type. Required.
+        :type blob_details: ~azure.quantum._client.models.BlobDetails or IO[bytes]
         :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
          Default value is None.
         :paramtype content_type: str
@@ -1140,12 +1158,12 @@ class StorageOperations:
         content_type = content_type or "application/json"
         _json = None
         _content = None
-        if isinstance(blob_details, (IO, bytes)):
+        if isinstance(blob_details, (IOBase, bytes)):
             _content = blob_details
         else:
             _json = self._serialize.body(blob_details, "BlobDetails")
 
-        request = build_storage_sas_uri_request(
+        _request = build_storage_sas_uri_request(
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
             workspace_name=self._config.workspace_name,
@@ -1156,16 +1174,23 @@ class StorageOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -1173,9 +1198,9 @@ class StorageOperations:
         deserialized = self._deserialize("SasUriResponse", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
 
 class QuotasOperations:
@@ -1221,7 +1246,7 @@ class QuotasOperations:
         def prepare_request(next_link=None):
             if not next_link:
 
-                request = build_quotas_list_request(
+                _request = build_quotas_list_request(
                     subscription_id=self._config.subscription_id,
                     resource_group_name=self._config.resource_group_name,
                     workspace_name=self._config.workspace_name,
@@ -1229,7 +1254,12 @@ class QuotasOperations:
                     headers=_headers,
                     params=_params,
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
             else:
                 # make call to next link with the client's api-version
@@ -1241,12 +1271,17 @@ class QuotasOperations:
                     }
                 )
                 _next_request_params["api-version"] = self._config.api_version
-                request = HttpRequest(
+                _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
-            return request
+            return _request
 
         def extract_data(pipeline_response):
             deserialized = self._deserialize(
@@ -1258,15 +1293,17 @@ class QuotasOperations:
             return deserialized.next_link or None, iter(list_of_elem)
 
         def get_next(next_link=None):
-            request = prepare_request(next_link)
+            _request = prepare_request(next_link)
 
             _stream = False
             pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
             response = pipeline_response.http_response
 
             if response.status_code not in [200]:
+                if _stream:
+                    response.read()  # Load the body in memory and close the socket
                 map_error(status_code=response.status_code, response=response, error_map=error_map)
                 error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
                 raise HttpResponseError(response=response, model=error)
@@ -1319,7 +1356,7 @@ class SessionsOperations:
         def prepare_request(next_link=None):
             if not next_link:
 
-                request = build_sessions_list_request(
+                _request = build_sessions_list_request(
                     subscription_id=self._config.subscription_id,
                     resource_group_name=self._config.resource_group_name,
                     workspace_name=self._config.workspace_name,
@@ -1327,7 +1364,12 @@ class SessionsOperations:
                     headers=_headers,
                     params=_params,
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
             else:
                 # make call to next link with the client's api-version
@@ -1339,12 +1381,17 @@ class SessionsOperations:
                     }
                 )
                 _next_request_params["api-version"] = self._config.api_version
-                request = HttpRequest(
+                _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
-            return request
+            return _request
 
         def extract_data(pipeline_response):
             deserialized = self._deserialize(
@@ -1356,15 +1403,17 @@ class SessionsOperations:
             return deserialized.next_link or None, iter(list_of_elem)
 
         def get_next(next_link=None):
-            request = prepare_request(next_link)
+            _request = prepare_request(next_link)
 
             _stream = False
             pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
             response = pipeline_response.http_response
 
             if response.status_code not in [200]:
+                if _stream:
+                    response.read()  # Load the body in memory and close the socket
                 map_error(status_code=response.status_code, response=response, error_map=error_map)
                 error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
                 raise HttpResponseError(response=response, model=error)
@@ -1396,7 +1445,7 @@ class SessionsOperations:
 
         cls: ClsType[_models.SessionDetails] = kwargs.pop("cls", None)
 
-        request = build_sessions_get_request(
+        _request = build_sessions_get_request(
             session_id=session_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -1405,16 +1454,23 @@ class SessionsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -1422,9 +1478,9 @@ class SessionsOperations:
         deserialized = self._deserialize("SessionDetails", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def open(
@@ -1446,14 +1502,14 @@ class SessionsOperations:
 
     @overload
     def open(
-        self, session_id: str, session: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, session_id: str, session: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.SessionDetails:
         """Open a session.
 
         :param session_id: Id of the session. Required.
         :type session_id: str
         :param session: The complete metadata of the session to be opened. Required.
-        :type session: IO
+        :type session: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -1464,15 +1520,15 @@ class SessionsOperations:
 
     @distributed_trace
     def open(
-        self, session_id: str, session: Union[_models.SessionDetails, IO], **kwargs: Any
+        self, session_id: str, session: Union[_models.SessionDetails, IO[bytes]], **kwargs: Any
     ) -> _models.SessionDetails:
         """Open a session.
 
         :param session_id: Id of the session. Required.
         :type session_id: str
         :param session: The complete metadata of the session to be opened. Is either a SessionDetails
-         type or a IO type. Required.
-        :type session: ~azure.quantum._client.models.SessionDetails or IO
+         type or a IO[bytes] type. Required.
+        :type session: ~azure.quantum._client.models.SessionDetails or IO[bytes]
         :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
          Default value is None.
         :paramtype content_type: str
@@ -1497,12 +1553,12 @@ class SessionsOperations:
         content_type = content_type or "application/json"
         _json = None
         _content = None
-        if isinstance(session, (IO, bytes)):
+        if isinstance(session, (IOBase, bytes)):
             _content = session
         else:
             _json = self._serialize.body(session, "SessionDetails")
 
-        request = build_sessions_open_request(
+        _request = build_sessions_open_request(
             session_id=session_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -1514,16 +1570,23 @@ class SessionsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 201]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -1562,7 +1625,7 @@ class SessionsOperations:
 
         cls: ClsType[_models.SessionDetails] = kwargs.pop("cls", None)
 
-        request = build_sessions_close_request(
+        _request = build_sessions_close_request(
             session_id=session_id,
             subscription_id=self._config.subscription_id,
             resource_group_name=self._config.resource_group_name,
@@ -1571,16 +1634,23 @@ class SessionsOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        path_format_arguments = {
+            "azureRegion": self._serialize.url(
+                "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+            ),
+        }
+        _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
         _stream = False
         pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            request, stream=_stream, **kwargs
+            _request, stream=_stream, **kwargs
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
             raise HttpResponseError(response=response, model=error)
@@ -1588,9 +1658,9 @@ class SessionsOperations:
         deserialized = self._deserialize("SessionDetails", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def jobs_list(self, session_id: str, **kwargs: Any) -> Iterable["_models.JobDetails"]:
@@ -1618,7 +1688,7 @@ class SessionsOperations:
         def prepare_request(next_link=None):
             if not next_link:
 
-                request = build_sessions_jobs_list_request(
+                _request = build_sessions_jobs_list_request(
                     session_id=session_id,
                     subscription_id=self._config.subscription_id,
                     resource_group_name=self._config.resource_group_name,
@@ -1627,7 +1697,12 @@ class SessionsOperations:
                     headers=_headers,
                     params=_params,
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
             else:
                 # make call to next link with the client's api-version
@@ -1639,12 +1714,17 @@ class SessionsOperations:
                     }
                 )
                 _next_request_params["api-version"] = self._config.api_version
-                request = HttpRequest(
+                _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
-            return request
+            return _request
 
         def extract_data(pipeline_response):
             deserialized = self._deserialize(
@@ -1656,15 +1736,17 @@ class SessionsOperations:
             return deserialized.next_link or None, iter(list_of_elem)
 
         def get_next(next_link=None):
-            request = prepare_request(next_link)
+            _request = prepare_request(next_link)
 
             _stream = False
             pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
             response = pipeline_response.http_response
 
             if response.status_code not in [200]:
+                if _stream:
+                    response.read()  # Load the body in memory and close the socket
                 map_error(status_code=response.status_code, response=response, error_map=error_map)
                 error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
                 raise HttpResponseError(response=response, model=error)
@@ -1717,7 +1799,7 @@ class TopLevelItemsOperations:
         def prepare_request(next_link=None):
             if not next_link:
 
-                request = build_top_level_items_list_request(
+                _request = build_top_level_items_list_request(
                     subscription_id=self._config.subscription_id,
                     resource_group_name=self._config.resource_group_name,
                     workspace_name=self._config.workspace_name,
@@ -1725,7 +1807,12 @@ class TopLevelItemsOperations:
                     headers=_headers,
                     params=_params,
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
             else:
                 # make call to next link with the client's api-version
@@ -1737,12 +1824,17 @@ class TopLevelItemsOperations:
                     }
                 )
                 _next_request_params["api-version"] = self._config.api_version
-                request = HttpRequest(
+                _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                request.url = self._client.format_url(request.url)
+                path_format_arguments = {
+                    "azureRegion": self._serialize.url(
+                        "self._config.azure_region", self._config.azure_region, "str", skip_quote=True
+                    ),
+                }
+                _request.url = self._client.format_url(_request.url, **path_format_arguments)
 
-            return request
+            return _request
 
         def extract_data(pipeline_response):
             deserialized = self._deserialize(
@@ -1754,15 +1846,17 @@ class TopLevelItemsOperations:
             return deserialized.next_link or None, iter(list_of_elem)
 
         def get_next(next_link=None):
-            request = prepare_request(next_link)
+            _request = prepare_request(next_link)
 
             _stream = False
             pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
             response = pipeline_response.http_response
 
             if response.status_code not in [200]:
+                if _stream:
+                    response.read()  # Load the body in memory and close the socket
                 map_error(status_code=response.status_code, response=response, error_map=error_map)
                 error = self._deserialize.failsafe_deserialize(_models.RestError, pipeline_response)
                 raise HttpResponseError(response=response, model=error)