jwcrypto: type kty parameter and add TODO note

Author: jkrejcha

stubs/jwcrypto/jwcrypto/jwk.pyi

@@ -79,6 +79,7 @@ IANANamedInformationHashAlgorithmRegistry: dict[
     | hashes.BLAKE2b
     | None,
 ]
+JWKKeyTypeSupported = Literal["oct", "RSA", "EC", "OKP"]
 
 class InvalidJWKType(JWException):
     value: str | None
@@ -98,9 +99,12 @@ class InvalidJWKValue(JWException): ...
 
 class JWK(dict[str, Any]):
     def __init__(self, **kwargs) -> None: ...
+    # `kty` and the other keyword arguments are passed as `params` to the called generator
+    # function. The possible arguments depend on the value of `kty`.
+    # TODO: Add overloads for the individual `kty` values.
     @classmethod
-    def generate(cls, **kwargs: Any) -> Self: ...
-    def generate_key(self, **params) -> None: ...
+    def generate(cls, kty: JWKKeyTypeSupported, **kwargs) -> Self: ...
+    def generate_key(self, kty: JWKKeyTypeSupported, **kwargs) -> None: ...
     def import_key(self, **kwargs) -> None: ...
     @classmethod
     def from_json(cls, key) -> Self: ...