Update __init__.py

regenerate requires public key not private key
serengil · Aug 3, 2024 · e8ef9c1 · e8ef9c1
1 parent c8bd4ca
commit e8ef9c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightphe/__init__.py b/lightphe/__init__.py
@@ -229,8 +229,8 @@ def regenerate_ciphertext(self, ciphertext: Ciphertext) -> Ciphertext:
         Returns:
             ciphertext (from lightphe.models.Ciphertext import Ciphertext): encrypted message
         """
-        if self.cs.keys.get("private_key") is None:
-            raise ValueError("You must have private key to perform decryption")
+        if self.cs.keys.get("public_key") is None:
+            raise ValueError("You must have public key to perform decryption")
 
         ciphertext_new = self.cs.reencrypt(ciphertext=ciphertext.value)
         return Ciphertext(