@autorest/python 6.38.2 → 6.39.0

package/generator/build/lib/pygen/codegen/models/parameter_list.py

@@ -7,8 +7,6 @@ import logging
 from typing import (
     Any,
     Callable,
-    Dict,
-    List,
     Optional,
     TYPE_CHECKING,
     Union,
@@ -48,7 +46,7 @@ class ParameterImplementation(Enum):
 _LOGGER = logging.getLogger(__name__)
 
 
-def method_signature_helper(positional: List[str], keyword_only: Optional[List[str]], kwarg_params: List[str]):
+def method_signature_helper(positional: list[str], keyword_only: Optional[list[str]], kwarg_params: list[str]):
     keyword_only = keyword_only or []
     return positional + keyword_only + kwarg_params
 
@@ -64,9 +62,9 @@ class _ParameterListBase(
 
     def __init__(
         self,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
-        parameters: List[ParameterType],
+        parameters: list[ParameterType],
         body_parameter: Optional[BodyParameterType] = None,
     ) -> None:
         self.yaml_data = yaml_data
@@ -97,18 +95,18 @@ class _ParameterListBase(
 
     @staticmethod
     @abstractmethod
-    def parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], ParameterType]:
+    def parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], ParameterType]:
         """Callable for creating parameters"""
 
     @staticmethod
     @abstractmethod
-    def body_parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], BodyParameterType]:
+    def body_parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], BodyParameterType]:
         """Callable for creating body parameters"""
 
     @property
-    def grouped(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def grouped(self) -> list[Union[ParameterType, BodyParameterType]]:
         """All parameters that are inside a parameter group"""
-        params: List[Union[ParameterType, BodyParameterType]] = [p for p in self.parameters if p.grouped_by]
+        params: list[Union[ParameterType, BodyParameterType]] = [p for p in self.parameters if p.grouped_by]
         if self.has_body and self.body_parameter.grouped_by:
             params.append(self.body_parameter)
         return params
@@ -123,41 +121,41 @@ class _ParameterListBase(
         return bool(self._body_parameter)
 
     @property
-    def path(self) -> List[ParameterType]:
+    def path(self) -> list[ParameterType]:
         """All path parameters"""
         return [p for p in self.parameters if p.location in (ParameterLocation.PATH, ParameterLocation.ENDPOINT_PATH)]
 
     @property
-    def query(self) -> List[ParameterType]:
+    def query(self) -> list[ParameterType]:
         """All query parameters"""
         return [p for p in self.parameters if p.location == ParameterLocation.QUERY]
 
     @property
-    def headers(self) -> List[ParameterType]:
+    def headers(self) -> list[ParameterType]:
         """All header parameters"""
         return [p for p in self.parameters if p.location == ParameterLocation.HEADER]
 
     @property
-    def constant(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def constant(self) -> list[Union[ParameterType, BodyParameterType]]:
         """All constant parameters"""
         return [p for p in self.parameters if p.constant]
 
     @property
-    def positional(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def positional(self) -> list[Union[ParameterType, BodyParameterType]]:
         """All positional parameters"""
         return _sort(
             [p for p in self.unsorted_method_params if p.method_location == ParameterMethodLocation.POSITIONAL]
         )
 
     @property
-    def keyword_only(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def keyword_only(self) -> list[Union[ParameterType, BodyParameterType]]:
         """All keyword only parameters"""
         return _sort(
             [p for p in self.unsorted_method_params if p.method_location == ParameterMethodLocation.KEYWORD_ONLY]
         )
 
     @property
-    def kwarg(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def kwarg(self) -> list[Union[ParameterType, BodyParameterType]]:
         """All kwargs"""
         return _sort([p for p in self.unsorted_method_params if p.method_location == ParameterMethodLocation.KWARG])
 
@@ -174,9 +172,9 @@ class _ParameterListBase(
         """Whether this is a client or a method parameter"""
 
     @property
-    def unsorted_method_params(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def unsorted_method_params(self) -> list[Union[ParameterType, BodyParameterType]]:
         """Method params before sorting"""
-        method_params: List[Union[ParameterType, BodyParameterType]] = [
+        method_params: list[Union[ParameterType, BodyParameterType]] = [
             p
             for p in self.parameters
             if p.in_method_signature
@@ -196,11 +194,11 @@ class _ParameterListBase(
         return method_params
 
     @property
-    def method(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def method(self) -> list[Union[ParameterType, BodyParameterType]]:
         """Sorted method params. First positional, then keyword only, then kwarg"""
         return self.positional + self.keyword_only + self.kwarg
 
-    def method_signature(self, async_mode: bool, **kwargs: Any) -> List[str]:
+    def method_signature(self, async_mode: bool, **kwargs: Any) -> list[str]:
         """Method signature for this parameter list."""
         return method_signature_helper(
             positional=self.method_signature_positional(async_mode, **kwargs),
@@ -208,11 +206,11 @@ class _ParameterListBase(
             kwarg_params=self.method_signature_kwargs,
         )
 
-    def method_signature_positional(self, async_mode: bool, **kwargs: Any) -> List[str]:
+    def method_signature_positional(self, async_mode: bool, **kwargs: Any) -> list[str]:
         """Signature for positional parameters"""
         return [parameter.method_signature(async_mode, **kwargs) for parameter in self.positional]
 
-    def method_signature_keyword_only(self, async_mode: bool, **kwargs: Any) -> List[str]:
+    def method_signature_keyword_only(self, async_mode: bool, **kwargs: Any) -> list[str]:
         """Signature for keyword only parameters"""
         result = [
             parameter.method_signature(async_mode, **kwargs)
@@ -222,19 +220,19 @@ class _ParameterListBase(
         return ["*,"] + result if result else []
 
     @property
-    def method_signature_kwargs(self) -> List[str]:
+    def method_signature_kwargs(self) -> list[str]:
         """Signature for kwargs"""
         return ["**kwargs: Any"]
 
     @property
-    def kwargs_to_pop(self) -> List[Union[ParameterType, BodyParameterType]]:
+    def kwargs_to_pop(self) -> list[Union[ParameterType, BodyParameterType]]:
         """Method kwargs we want to pop"""
         # don't want to pop bodies unless it's a constant
         kwargs_to_pop = self.kwarg
         return [k for k in kwargs_to_pop if k.location != ParameterLocation.BODY or k.constant]
 
     @property
-    def call(self) -> List[str]:
+    def call(self) -> list[str]:
         """How to pass in parameters to call the operation"""
         retval = [p.client_name for p in self.method if p.method_location == ParameterMethodLocation.POSITIONAL]
         retval.extend(
@@ -248,7 +246,7 @@ class _ParameterListBase(
         return retval
 
     @classmethod
-    def from_yaml(cls, yaml_data: Dict[str, Any], code_model: "CodeModel"):
+    def from_yaml(cls, yaml_data: dict[str, Any], code_model: "CodeModel"):
         parameters = [cls.parameter_creator()(parameter, code_model) for parameter in yaml_data["parameters"]]
         body_parameter = None
         if yaml_data.get("bodyParameter"):
@@ -264,11 +262,11 @@ class _ParameterListBase(
 class _ParameterList(_ParameterListBase[Parameter, BodyParameter]):
 
     @staticmethod
-    def parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], Parameter]:
+    def parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], Parameter]:
         return Parameter.from_yaml
 
     @staticmethod
-    def body_parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], BodyParameter]:
+    def body_parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], BodyParameter]:
         return BodyParameter.from_yaml
 
     @property
@@ -276,7 +274,7 @@ class _ParameterList(_ParameterListBase[Parameter, BodyParameter]):
         return "Method"
 
     @property
-    def path(self) -> List[Parameter]:
+    def path(self) -> list[Parameter]:
         return [k for k in super().path if k.location == ParameterLocation.ENDPOINT_PATH]
 
 
@@ -288,11 +286,11 @@ class _RequestBuilderParameterList(_ParameterListBase[RequestBuilderParameter, R
     """_RequestBuilderParameterList is base parameter list for RequestBuilder classes"""
 
     @staticmethod
-    def parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], RequestBuilderParameter]:
+    def parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], RequestBuilderParameter]:
         return RequestBuilderParameter.from_yaml
 
     @staticmethod
-    def body_parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], RequestBuilderBodyParameter]:
+    def body_parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], RequestBuilderBodyParameter]:
         return RequestBuilderBodyParameter.from_yaml
 
     @property
@@ -302,7 +300,7 @@ class _RequestBuilderParameterList(_ParameterListBase[RequestBuilderParameter, R
     @property
     def unsorted_method_params(
         self,
-    ) -> List[Union[RequestBuilderParameter, RequestBuilderBodyParameter]]:
+    ) -> list[Union[RequestBuilderParameter, RequestBuilderBodyParameter]]:
         # don't have access to client params in request builder
         retval = [
             p
@@ -313,13 +311,13 @@ class _RequestBuilderParameterList(_ParameterListBase[RequestBuilderParameter, R
         return retval
 
     @property
-    def path(self) -> List[RequestBuilderParameter]:
+    def path(self) -> list[RequestBuilderParameter]:
         return [p for p in super().path if p.location != ParameterLocation.ENDPOINT_PATH]
 
     @property
     def constant(
         self,
-    ) -> List[Union[RequestBuilderParameter, RequestBuilderBodyParameter]]:
+    ) -> list[Union[RequestBuilderParameter, RequestBuilderBodyParameter]]:
         """All constant parameters"""
         return [p for p in super().constant if p.location != ParameterLocation.ENDPOINT_PATH]
 
@@ -336,7 +334,7 @@ class _ClientGlobalParameterList(_ParameterListBase[ParameterType, BodyParameter
     """Base parameter list for client and config classes"""
 
     @staticmethod
-    def body_parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], BodyParameter]:
+    def body_parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], BodyParameter]:
         return BodyParameter.from_yaml
 
     @property
@@ -351,7 +349,7 @@ class _ClientGlobalParameterList(_ParameterListBase[ParameterType, BodyParameter
             return None
 
     @property
-    def path(self) -> List[ParameterType]:
+    def path(self) -> list[ParameterType]:
         return [p for p in super().path if p.location == ParameterLocation.ENDPOINT_PATH]
 
 
@@ -359,11 +357,11 @@ class ClientGlobalParameterList(_ClientGlobalParameterList[ClientParameter]):
     """Parameter list for Client class"""
 
     @staticmethod
-    def parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], ClientParameter]:
+    def parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], ClientParameter]:
         return ClientParameter.from_yaml
 
     @property
-    def path(self) -> List[ClientParameter]:
+    def path(self) -> list[ClientParameter]:
         return [p for p in super().path if not p.is_host]
 
     @property
@@ -379,7 +377,7 @@ class ConfigGlobalParameterList(_ClientGlobalParameterList[ConfigParameter]):
     """Parameter list for config"""
 
     @staticmethod
-    def parameter_creator() -> Callable[[Dict[str, Any], "CodeModel"], ConfigParameter]:
+    def parameter_creator() -> Callable[[dict[str, Any], "CodeModel"], ConfigParameter]:
         return ConfigParameter.from_yaml
 
     @property

package/generator/build/lib/pygen/codegen/models/primitive_types.py

@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------
 import datetime
 import decimal
-from typing import Any, Dict, List, Optional, Union, TYPE_CHECKING
+from typing import Any, Optional, Union, TYPE_CHECKING
 
 from .base import BaseType
 from .imports import FileImport, ImportType, TypingSection
@@ -61,7 +61,7 @@ class BooleanType(PrimitiveType):
 
 
 class BinaryType(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.type = "IO"
 
@@ -176,7 +176,7 @@ class AnyObjectType(PrimitiveType):
 
 
 class NumberType(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.precision: Optional[int] = yaml_data.get("precision")
         self.multiple: Optional[int] = yaml_data.get("multipleOf")
@@ -186,7 +186,7 @@ class NumberType(PrimitiveType):
         self.exclusive_minimum: Optional[int] = yaml_data.get("exclusiveMinimum")
 
     @property
-    def serialization_constraints(self) -> List[str]:
+    def serialization_constraints(self) -> list[str]:
         validation_constraints = [
             (f"maximum_ex={self.maximum}" if self.maximum is not None and self.exclusive_maximum else None),
             (f"maximum={self.maximum}" if self.maximum is not None and not self.exclusive_maximum else None),
@@ -197,8 +197,8 @@ class NumberType(PrimitiveType):
         return [x for x in validation_constraints if x is not None]
 
     @property
-    def validation(self) -> Optional[Dict[str, Union[bool, int, str]]]:
-        validation: Dict[str, Union[bool, int, str]] = {}
+    def validation(self) -> Optional[dict[str, Union[bool, int, str]]]:
+        validation: dict[str, Union[bool, int, str]] = {}
         if self.maximum is not None:
             if self.exclusive_maximum:
                 validation["maximum_ex"] = self.maximum
@@ -221,7 +221,7 @@ class NumberType(PrimitiveType):
 
 class IntegerType(NumberType):
 
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         if yaml_data.get("encode") == "string":
             self.encode = "str"
@@ -294,7 +294,7 @@ class DecimalType(NumberType):
 
 
 class StringType(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.max_length: Optional[int] = yaml_data.get("maxLength")
         self.min_length: Optional[int] = (
@@ -303,7 +303,7 @@ class StringType(PrimitiveType):
         self.pattern: Optional[str] = yaml_data.get("pattern")
 
     @property
-    def serialization_constraints(self) -> List[str]:
+    def serialization_constraints(self) -> list[str]:
         validation_constraints = [
             f"max_length={self.max_length}" if self.max_length is not None else None,
             f"min_length={self.min_length}" if self.min_length is not None else None,
@@ -312,8 +312,8 @@ class StringType(PrimitiveType):
         return [x for x in validation_constraints if x is not None]
 
     @property
-    def validation(self) -> Optional[Dict[str, Union[bool, int, str]]]:
-        validation: Dict[str, Union[bool, int, str]] = {}
+    def validation(self) -> Optional[dict[str, Union[bool, int, str]]]:
+        validation: dict[str, Union[bool, int, str]] = {}
         if self.max_length is not None:
             validation["max_length"] = self.max_length
         if self.min_length is not None:
@@ -338,7 +338,7 @@ class StringType(PrimitiveType):
 
 
 class DatetimeType(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.encode = (
             "rfc3339"
@@ -564,7 +564,7 @@ class DurationType(PrimitiveType):
 
 
 class ByteArraySchema(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.encode = yaml_data.get("encode", "base64")
 
@@ -585,20 +585,26 @@ class ByteArraySchema(PrimitiveType):
 
 
 class SdkCoreType(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.name = yaml_data.get("name", "")
         self.submodule = yaml_data.get("submodule", "")
+        self.is_typing_only = yaml_data.get("isTypingOnly", False)
 
     def docstring_type(self, **kwargs: Any) -> str:
-        return f"~{self.code_model.core_library}.{self.type_annotation(**kwargs)}"
+        return f"~{self.code_model.core_library}.{self.name}"
 
     def type_annotation(self, **kwargs: Any) -> str:
-        return self.name
+        return f'"{self.name}"' if self.is_typing_only else self.name
 
     def imports(self, **kwargs: Any) -> FileImport:
         file_import = super().imports(**kwargs)
-        file_import.add_submodule_import(self.submodule, self.name, ImportType.SDKCORE)
+        file_import.add_submodule_import(
+            self.submodule,
+            self.name,
+            ImportType.SDKCORE,
+            typing_section=TypingSection.TYPING if self.is_typing_only else TypingSection.REGULAR,
+        )
         return file_import
 
     @property
@@ -610,7 +616,7 @@ class SdkCoreType(PrimitiveType):
 
 
 class MultiPartFileType(PrimitiveType):
-    def __init__(self, yaml_data: Dict[str, Any], code_model: "CodeModel") -> None:
+    def __init__(self, yaml_data: dict[str, Any], code_model: "CodeModel") -> None:
         super().__init__(yaml_data=yaml_data, code_model=code_model)
         self.name = "FileType"
 

package/generator/build/lib/pygen/codegen/models/property.py

@@ -3,7 +3,7 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from typing import Any, Dict, Optional, TYPE_CHECKING, List, cast, Union
+from typing import Any, Optional, TYPE_CHECKING, cast, Union
 
 from .base import BaseModel
 from .constant_type import ConstantType
@@ -20,7 +20,7 @@ if TYPE_CHECKING:
 class Property(BaseModel):  # pylint: disable=too-many-instance-attributes
     def __init__(
         self,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
         type: BaseType,
     ) -> None:
@@ -30,13 +30,13 @@ class Property(BaseModel):  # pylint: disable=too-many-instance-attributes
         self.type = type
         self.optional: bool = self.yaml_data["optional"]
         self.readonly: bool = self.yaml_data.get("readonly", False)
-        self.visibility: List[str] = self.yaml_data.get("visibility", [])
+        self.visibility: list[str] = self.yaml_data.get("visibility", [])
         self.is_polymorphic: bool = self.yaml_data.get("isPolymorphic", False)
         self.is_discriminator: bool = yaml_data.get("isDiscriminator", False)
         self.client_default_value = yaml_data.get("clientDefaultValue", None)
         if self.client_default_value is None:
             self.client_default_value = self.type.client_default_value
-        self.flattened_names: List[str] = yaml_data.get("flattenedNames", [])
+        self.flattened_names: list[str] = yaml_data.get("flattenedNames", [])
         self.is_multipart_file_input: bool = yaml_data.get("isMultipartFileInput", False)
         self.flatten = self.yaml_data.get("flatten", False) and not getattr(self.type, "flattened_property", False)
 
@@ -101,7 +101,7 @@ class Property(BaseModel):  # pylint: disable=too-many-instance-attributes
         return self.is_discriminator and self.is_polymorphic and cast(ConstantType, self.type).value is None
 
     @property
-    def xml_metadata(self) -> Optional[Dict[str, Union[str, bool]]]:
+    def xml_metadata(self) -> Optional[dict[str, Union[str, bool]]]:
         return self.yaml_data.get("xmlMetadata")
 
     def type_annotation(self, *, is_operation_file: bool = False, **kwargs: Any) -> str:
@@ -127,22 +127,22 @@ class Property(BaseModel):  # pylint: disable=too-many-instance-attributes
             client_default_value_declaration=client_default_value_declaration,
         )
 
-    def get_polymorphic_subtypes(self, polymorphic_subtypes: List["ModelType"]) -> None:
+    def get_polymorphic_subtypes(self, polymorphic_subtypes: list["ModelType"]) -> None:
         from .model_type import ModelType
 
         if isinstance(self.type, ModelType):
             self.type.get_polymorphic_subtypes(polymorphic_subtypes)
 
     @property
-    def validation(self) -> Optional[Dict[str, Any]]:
-        retval: Dict[str, Any] = {}
+    def validation(self) -> Optional[dict[str, Any]]:
+        retval: dict[str, Any] = {}
         if not self.optional:
             retval["required"] = True
         if self.readonly:
             retval["readonly"] = True
         if self.constant:
             retval["constant"] = True
-        retval.update(self.type.validation or {})
+        retval |= self.type.validation or {}
         return retval or None
 
     def imports(self, **kwargs) -> FileImport:
@@ -164,7 +164,7 @@ class Property(BaseModel):  # pylint: disable=too-many-instance-attributes
     @classmethod
     def from_yaml(
         cls,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
     ) -> "Property":
         from . import build_type  # pylint: disable=import-outside-toplevel

package/generator/build/lib/pygen/codegen/models/request_builder.py

@@ -6,7 +6,6 @@
 from typing import (
     Any,
     Callable,
-    Dict,
     TypeVar,
     TYPE_CHECKING,
     Union,
@@ -38,7 +37,7 @@ ParameterListType = TypeVar(
 class RequestBuilderBase(BaseBuilder[ParameterListType, Sequence["RequestBuilder"]]):
     def __init__(
         self,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
         client: "Client",
         name: str,
@@ -116,13 +115,13 @@ class RequestBuilderBase(BaseBuilder[ParameterListType, Sequence["RequestBuilder
 
     @staticmethod
     @abstractmethod
-    def parameter_list_type() -> Callable[[Dict[str, Any], "CodeModel"], ParameterListType]: ...
+    def parameter_list_type() -> Callable[[dict[str, Any], "CodeModel"], ParameterListType]: ...
 
     @classmethod
     def get_name(
         cls,
         name: str,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
         client: "Client",
     ) -> str:
@@ -140,7 +139,7 @@ class RequestBuilderBase(BaseBuilder[ParameterListType, Sequence["RequestBuilder
     @classmethod
     def from_yaml(
         cls,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
         client: "Client",
     ):
@@ -165,18 +164,18 @@ class RequestBuilderBase(BaseBuilder[ParameterListType, Sequence["RequestBuilder
 
 class RequestBuilder(RequestBuilderBase[RequestBuilderParameterList]):
     @staticmethod
-    def parameter_list_type() -> Callable[[Dict[str, Any], "CodeModel"], RequestBuilderParameterList]:
+    def parameter_list_type() -> Callable[[dict[str, Any], "CodeModel"], RequestBuilderParameterList]:
         return RequestBuilderParameterList.from_yaml
 
 
 class OverloadedRequestBuilder(RequestBuilderBase[OverloadedRequestBuilderParameterList]):
     @staticmethod
-    def parameter_list_type() -> Callable[[Dict[str, Any], "CodeModel"], OverloadedRequestBuilderParameterList]:
+    def parameter_list_type() -> Callable[[dict[str, Any], "CodeModel"], OverloadedRequestBuilderParameterList]:
         return OverloadedRequestBuilderParameterList.from_yaml
 
 
 def get_request_builder(
-    yaml_data: Dict[str, Any], code_model: "CodeModel", client: "Client"
+    yaml_data: dict[str, Any], code_model: "CodeModel", client: "Client"
 ) -> Union[RequestBuilder, OverloadedRequestBuilder]:
     if yaml_data.get("overloads"):
         return OverloadedRequestBuilder.from_yaml(yaml_data, code_model, client)

package/generator/build/lib/pygen/codegen/models/request_builder_parameter.py

@@ -3,7 +3,7 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from typing import TYPE_CHECKING, Any, Dict
+from typing import TYPE_CHECKING, Any
 from .parameter import (
     ParameterLocation,
     ParameterMethodLocation,
@@ -52,7 +52,7 @@ class RequestBuilderBodyParameter(BodyParameter):
         )
 
     @classmethod
-    def from_yaml(cls, yaml_data: Dict[str, Any], code_model: "CodeModel") -> "RequestBuilderBodyParameter":
+    def from_yaml(cls, yaml_data: dict[str, Any], code_model: "CodeModel") -> "RequestBuilderBodyParameter":
         return super().from_yaml(yaml_data, code_model)  # type: ignore
 
     @property
@@ -67,7 +67,7 @@ class RequestBuilderParameter(Parameter):
 
     def __init__(
         self,
-        yaml_data: Dict[str, Any],
+        yaml_data: dict[str, Any],
         code_model: "CodeModel",
         type: BaseType,
     ) -> None: