puantum 1.1.1__tar.gz → 3.3.3__tar.gz

puantum-3.3.3/.github/workflows/publish.yml

@@ -0,0 +1,36 @@
+name: Build and Publish to PyPI
+
+on:
+  push:
+    branches:
+      - main  # Trigger on every push to the main branch
+
+jobs:
+  build-and-publish:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.13'  # Adjust if needed
+
+      - name: Install Rust
+        uses: dtolnay/rust-toolchain@stable
+
+      - name: Install maturin
+        run: pip install maturin
+
+      - name: Build the wheel and sdist
+        run: maturin build --sdist --release
+
+      - name: Publish to PyPI
+        env:
+          TWINE_USERNAME: __token__
+          TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}
+        run: |
+          pip install twine
+          twine upload target/wheels/*

{puantum-1.1.1 → puantum-3.3.3}/Cargo.toml

@@ -2,6 +2,7 @@
 name = "puantum"
 version = "1.0.0"
 edition = "2024"
+license = "Unlicense"
 
 [lib]
 crate-type = ["cdylib"]

puantum-3.3.3/Makefile

@@ -0,0 +1,13 @@
+all: release
+
+clean:
+	cargo clean
+
+release: clean
+	cargo build --release
+
+python: clean
+	maturin build --release
+
+pypi: clean
+	maturin build --release --sdist

{puantum-1.1.1 → puantum-3.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: puantum
-Version: 1.1.1
+Version: 3.3.3
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Rust
 Classifier: Operating System :: OS Independent
@@ -19,7 +19,7 @@ Project-URL: x, https://x.com/1xfakebit
 
 A blazing-fast cryptography library for Python, built on Rust.
 
-Puantum supports an extensive set of **post-quantum** key encapsulation mechanisms (KEM) and digital signature algorithms (DSA), and will soon support **classic cryptography** as well.
+Pryptography supports an extensive set of **post-quantum** digital signature algorithms (DSA) and key encapsulation mechanisms (KEM).
 
 ---
 ## ⚡ Features
@@ -48,15 +48,49 @@ Puantum supports an extensive set of **post-quantum** key encapsulation mechanis
 - #### Sphincs
 - #### Uov
 
-### 🚧 Coming Soon
-- #### AES, ChaCha20, XChaCha20
-- #### RSA, EC
-- #### Argon2, Bcrypt
+### ❔ Examples
+
+#### DSA Example
+```python
+# IMPORT
+from puantum.quantum.dsa import Algorithm, KeyPair
+
+# MAIN
+alicesk, alicepk = KeyPair(Algorithm.MLDSA.MLDSA87)
+message = "Hello".encode()
+
+signature = alicesk.sign(message=message)
+valid = alicepk.verify(signature=signature, message=message)
+
+assert valid, "Signature verification failed!"
+
+print(f"Message:   [{message.decode()}]")
+print(f"Signature: [{signature.hex()[:len(message)]}]")
+```
+
+#### KEM Example
+```python
+# IMPORT
+from puantum.quantum.kem import Algorithm, KeyPair
+
+# MAIN
+alicesk, alicepk = KeyPair(Algorithm.MLKEM.MLKEM1024)
+_bobsk, _bobpk = KeyPair(Algorithm.MLKEM.MLKEM1024)
+
+bobss, bobct = alicepk.encapsulate()
+alicess = alicesk.decapsulate(bobct)
+
+assert alicess == bobss, "Shared secrets do not match!"
+
+print(f"Alice's Shared Secret: [{alicess.hex()}]")
+print(f"Bob's   Shared Secret: [{bobss.hex()}]")
+```
 
 ### 📦 Install
 ```shell
 pip install puantum
 ```
+
 #### or from source:
 ```shell
 make python

puantum-3.3.3/puantum/classic/__init__.pyi

@@ -0,0 +1 @@
+# EMPTY

puantum-3.3.3/puantum/quantum/__init__.py

@@ -0,0 +1 @@
+# EMPTY

puantum-3.3.3/puantum/quantum/__init__.pyi

@@ -0,0 +1 @@
+# EMPTY

puantum-3.3.3/puantum/quantum/dsa/__init__.pyi

@@ -0,0 +1,5 @@
+# IMPORT
+from puantum.quantum.dsa.__internal__ import Algorithm, KeyPair
+
+# MAIN
+__all__: list[str] = ["Algorithm", "KeyPair"]

{puantum-1.1.1 → puantum-3.3.3}/puantum/quantum/dsa/__internal__.py

@@ -75,93 +75,6 @@ class Algorithm:
 
 
 
-class Signature:
-    def __init__(self, signature: bytes) -> None:
-        if not isinstance(signature, bytes):
-            raise TypeError("Signature Not Valid")
-        #
-        self.signature = signature
-        #
-        return None
-
-
-
-class CROSSSignature(Signature):
-    def __init__(self, name: Algorithm.CROSS, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.CROSS):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-
-class DILITHIUMSignature(Signature):
-    def __init__(self, name: Algorithm.DILITHIUM, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.DILITHIUM):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-
-class FALCONSignature(Signature):
-    def __init__(self, name: Algorithm.FALCON, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.FALCON):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-
-class MAYOSignature(Signature):
-    def __init__(self, name: Algorithm.MAYO, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.MAYO):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-
-class MLDSASignature(Signature):
-    def __init__(self, name: Algorithm.MLDSA, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.MLDSA):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-
-class SPHINCSSignature(Signature):
-    def __init__(self, name: Algorithm.SPHINCS, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.SPHINCS):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-class UOVSignature(Signature):
-    def __init__(self, name: Algorithm.UOV, signature: bytes) -> None:
-        if not isinstance(name, Algorithm.UOV):
-            raise ValueError(f"Unsupported algorithm: {name}")
-        else:
-            super().__init__(signature)
-        #
-        return None
-
-
-
 class PublicKey:
     def __init__(self, publickey: bytes) -> None:
         if not isinstance(publickey, bytes):
@@ -187,8 +100,8 @@ class CROSSPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: CROSSSignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -204,8 +117,8 @@ class DILITHIUMPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: DILITHIUMSignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -221,8 +134,8 @@ class FALCONPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: FALCONSignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -238,8 +151,8 @@ class MAYOPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: MAYOSignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -255,8 +168,8 @@ class MLDSAPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: MLDSASignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -272,8 +185,8 @@ class SPHINCSPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: SPHINCSSignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -289,8 +202,8 @@ class UOVPublicKey(PublicKey):
         #
         return None
     #
-    def verify(self, signature: UOVSignature, message: bytes) -> bool:
-        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature.signature, message) # type: ignore
+    def verify(self, signature: bytes, message: bytes) -> bool:
+        result: bool = _internal.sigverify(self._algorithm.value, self.publickey, signature, message) # type: ignore
         return result # type: ignore
 
 
@@ -320,9 +233,9 @@ class CROSSSecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> CROSSSignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return CROSSSignature(self._algorithm, signature) # type: ignore
+        return signature # type: ignore
 
 
 
@@ -337,9 +250,9 @@ class DILITHIUMSecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> DILITHIUMSignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return DILITHIUMSignature(self._algorithm, signature) # type: ignore
+        return signature # type: ignore
 
 
 
@@ -354,9 +267,9 @@ class FALCONSecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> FALCONSignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return FALCONSignature(self._algorithm, signature) # type: ignore
+        return signature # type: ignore
 
 
 
@@ -371,9 +284,9 @@ class MAYOSecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> MAYOSignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return MAYOSignature(self._algorithm, signature) # type: ignore
+        return signature # type: ignore
 
 
 
@@ -388,9 +301,9 @@ class MLDSASecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> MLDSASignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return MLDSASignature(self._algorithm, signature) # type: ignore
+        return signature # type: ignore
 
 
 
@@ -405,9 +318,9 @@ class SPHINCSSecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> SPHINCSSignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return SPHINCSSignature(self._algorithm, signature) # type: ignore
+        return signature # type: ignore
 
 
 
@@ -422,77 +335,216 @@ class UOVSecretKey(SecretKey):
         #
         return None
     #
-    def sign(self, message: bytes) -> UOVSignature:
+    def sign(self, message: bytes) -> bytes:
         signature: bytes = _internal.sigsign(self._algorithm.value, self.secretkey, message) # type: ignore
-        return UOVSignature(self._algorithm, signature) # type: ignore
-
-
-
-Algorithms = Algorithm | Algorithm.CROSS | Algorithm.DILITHIUM | Algorithm.FALCON | Algorithm.MAYO | Algorithm.MLDSA | Algorithm.SPHINCS | Algorithm.UOV
+        return signature # type: ignore
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm,
+    *,
+    secretkey: bytes, publickey: bytes
+) -> tuple[SecretKey, PublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm, *, secretkey: bytes) -> SecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm, *, publickey: bytes) -> PublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm) -> tuple[SecretKey, PublicKey]: ...
 
 
+
+@_typing.overload
+def KeyPair(
+    name: Algorithm.CROSS,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[CROSSSecretKey, CROSSPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.CROSS, *, secretkey: bytes) -> CROSSSecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.CROSS, *, publickey: bytes) -> CROSSPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.CROSS) -> tuple[CROSSSecretKey, CROSSPublicKey]: ...
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm.DILITHIUM,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[DILITHIUMSecretKey, DILITHIUMPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.DILITHIUM, *, secretkey: bytes) -> DILITHIUMSecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.DILITHIUM, *, publickey: bytes) -> DILITHIUMPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.DILITHIUM) -> tuple[DILITHIUMSecretKey, DILITHIUMPublicKey]: ...
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm.FALCON,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[FALCONSecretKey, FALCONPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.FALCON, *, secretkey: bytes) -> FALCONSecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.FALCON, *, publickey: bytes) -> FALCONPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.FALCON) -> tuple[FALCONSecretKey, FALCONPublicKey]: ...
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm.MAYO,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[MAYOSecretKey, MAYOPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.MAYO, *, secretkey: bytes) -> MAYOSecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.MAYO, *, publickey: bytes) -> MAYOPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.MAYO) -> tuple[MAYOSecretKey, MAYOPublicKey]: ...
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm.MLDSA,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[MLDSASecretKey, MLDSAPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.MLDSA, *, secretkey: bytes) -> MLDSASecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.MLDSA, *, publickey: bytes) -> MLDSAPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.MLDSA) -> tuple[MLDSASecretKey, MLDSAPublicKey]: ...
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm.SPHINCS,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[SPHINCSSecretKey, SPHINCSPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.SPHINCS, *, secretkey: bytes) -> SPHINCSSecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.SPHINCS, *, publickey: bytes) -> SPHINCSPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.SPHINCS) -> tuple[SPHINCSSecretKey, SPHINCSPublicKey]: ...
 
 
 
+@_typing.overload
+def KeyPair(
+    name: Algorithm.UOV,
+    *,
+    secretkey: bytes,
+    publickey: bytes
+) -> tuple[UOVSecretKey, UOVPublicKey]: ...
+@_typing.overload
+def KeyPair(name: Algorithm.UOV, *, secretkey: bytes) -> UOVSecretKey: ...
+@_typing.overload
+def KeyPair(name: Algorithm.UOV, *, publickey: bytes) -> UOVPublicKey: ...
 @_typing.overload
 def KeyPair(name: Algorithm.UOV) -> tuple[UOVSecretKey, UOVPublicKey]: ...
 
 
 
-def KeyPair(name: Algorithms) -> tuple[SecretKey, PublicKey]:
+Algorithms = Algorithm.CROSS | Algorithm.DILITHIUM | Algorithm.FALCON | Algorithm.MAYO | Algorithm.MLDSA | Algorithm.SPHINCS | Algorithm.UOV
+def KeyPair(
+        name: Algorithm | Algorithms,
+        *,
+        secretkey: bytes | None = None,
+        publickey: bytes | None = None
+    ) -> tuple[SecretKey, PublicKey] | SecretKey | PublicKey:
     algorithm = name
     if isinstance(algorithm, Algorithm.CROSS):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return CROSSSecretKey(algorithm, secretkey), CROSSPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return CROSSSecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return CROSSPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return CROSSSecretKey(algorithm, secretkey), CROSSPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return CROSSSecretKey(algorithm, secretkey), CROSSPublicKey(algorithm, publickey) # type: ignore
     elif isinstance(algorithm, Algorithm.DILITHIUM):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return DILITHIUMSecretKey(algorithm, secretkey), DILITHIUMPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return DILITHIUMSecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return DILITHIUMPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return DILITHIUMSecretKey(algorithm, secretkey), DILITHIUMPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return DILITHIUMSecretKey(algorithm, secretkey), DILITHIUMPublicKey(algorithm, publickey) # type: ignore
     elif isinstance(algorithm, Algorithm.FALCON):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return FALCONSecretKey(algorithm, secretkey), FALCONPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return FALCONSecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return FALCONPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return FALCONSecretKey(algorithm, secretkey), FALCONPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return FALCONSecretKey(algorithm, secretkey), FALCONPublicKey(algorithm, publickey) # type: ignore
     elif isinstance(algorithm, Algorithm.MAYO):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return MAYOSecretKey(algorithm, secretkey), MAYOPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return MAYOSecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return MAYOPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return MAYOSecretKey(algorithm, secretkey), MAYOPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return MAYOSecretKey(algorithm, secretkey), MAYOPublicKey(algorithm, publickey) # type: ignore
     elif isinstance(algorithm, Algorithm.MLDSA):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return MLDSASecretKey(algorithm, secretkey), MLDSAPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return MLDSASecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return MLDSAPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return MLDSASecretKey(algorithm, secretkey), MLDSAPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return MLDSASecretKey(algorithm, secretkey), MLDSAPublicKey(algorithm, publickey) # type: ignore
     elif isinstance(algorithm, Algorithm.SPHINCS):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return SPHINCSSecretKey(algorithm, secretkey), SPHINCSPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return SPHINCSSecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return SPHINCSPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return SPHINCSSecretKey(algorithm, secretkey), SPHINCSPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return SPHINCSSecretKey(algorithm, secretkey), SPHINCSPublicKey(algorithm, publickey) # type: ignore
     elif isinstance(algorithm, Algorithm.UOV):
-        secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
-        return UOVSecretKey(algorithm, secretkey), UOVPublicKey(algorithm, publickey) # type: ignore
+        if secretkey is not None and publickey is None:
+            return UOVSecretKey(algorithm, secretkey)
+        elif publickey is not None and secretkey is None:
+            return UOVPublicKey(algorithm, publickey)
+        elif secretkey is not None and publickey is not None:
+            return UOVSecretKey(algorithm, secretkey), UOVPublicKey(algorithm, publickey)
+        else:
+            secretkey, publickey = _internal.sigkeygen(algorithm.value) # type: ignore
+            return UOVSecretKey(algorithm, secretkey), UOVPublicKey(algorithm, publickey) # type: ignore
     else:
         raise ValueError(f"Unsupported algorithm: {name}")