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(