From 770ebad9c2af2cd280b39207404169f46fd14743 Mon Sep 17 00:00:00 2001
From: jatZama <josephandre.turk@zama.ai>
Date: Sat, 14 Dec 2024 16:16:21 +0100
Subject: [PATCH] feat: add allowTransient for ebytes64 and ebytes128

---
 codegen/templates.ts | 8 ++++++++
 lib/TFHE.sol         | 8 ++++++++
 2 files changed, 16 insertions(+)

diff --git a/codegen/templates.ts b/codegen/templates.ts
index c1e82f06..44245e5f 100644
--- a/codegen/templates.ts
+++ b/codegen/templates.ts
@@ -877,6 +877,14 @@ function tfheAclMethods(supportedBits: number[]): string {
       Impl.allowTransient(eaddress.unwrap(value), account);
     }
 
+    function allowTransient(ebytes64 value, address account) internal {
+      Impl.allowTransient(ebytes64.unwrap(value), account);
+    }
+
+    function allowTransient(ebytes128 value, address account) internal {
+      Impl.allowTransient(ebytes128.unwrap(value), account);
+    }
+
     function allowTransient(ebytes256 value, address account) internal {
       Impl.allowTransient(ebytes256.unwrap(value), account);
     }
diff --git a/lib/TFHE.sol b/lib/TFHE.sol
index b8303680..ac1b9f6f 100644
--- a/lib/TFHE.sol
+++ b/lib/TFHE.sol
@@ -11033,6 +11033,14 @@ library TFHE {
         Impl.allowTransient(eaddress.unwrap(value), account);
     }
 
+    function allowTransient(ebytes64 value, address account) internal {
+        Impl.allowTransient(ebytes64.unwrap(value), account);
+    }
+
+    function allowTransient(ebytes128 value, address account) internal {
+        Impl.allowTransient(ebytes128.unwrap(value), account);
+    }
+
     function allowTransient(ebytes256 value, address account) internal {
         Impl.allowTransient(ebytes256.unwrap(value), account);
     }