Refactor: Add support for additional keys in GetKeys function

Lightprotocol · Oct 12, 2024 · 633b76e · 633b76e
1 parent 9fce7eb
commit 633b76e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/light-prover/prover/circuit_utils.go b/light-prover/prover/circuit_utils.go
@@ -228,6 +228,7 @@ func GetKeys(keysDir string, circuitTypes []CircuitType, isTestMode bool) []stri
 	if IsCircuitEnabled(circuitTypes, Inclusion) {
 		if isTestMode {
 			keys = append(keys, keysDir+"inclusion_26_1.key")
+			keys = append(keys, keysDir+"inclusion_26_2.key")
 		} else {
 			keys = append(keys, keysDir+"inclusion_26_1.key")
 			keys = append(keys, keysDir+"inclusion_26_2.key")
@@ -239,6 +240,7 @@ func GetKeys(keysDir string, circuitTypes []CircuitType, isTestMode bool) []stri
 	if IsCircuitEnabled(circuitTypes, NonInclusion) {
 		if isTestMode {
 			keys = append(keys, keysDir+"non-inclusion_26_1.key")
+			keys = append(keys, keysDir+"non-inclusion_26_2.key")
 		} else {
 			keys = append(keys, keysDir+"non-inclusion_26_1.key")
 			keys = append(keys, keysDir+"non-inclusion_26_2.key")

diff --git a/light-prover/scripts/download_keys.sh b/light-prover/scripts/download_keys.sh
@@ -16,8 +16,12 @@ APPEND_BUCKET="bafybeiatsnqj4ksuddtzixv5si5cfoy7bah723zigvuuut55vkbvkivig4"
 LIGHTWEIGHT_FILES=(
   "inclusion_26_1.key"
   "inclusion_26_1.vkey"
+  "inclusion_26_2.key"
+  "inclusion_26_2.vkey"
   "non-inclusion_26_1.key"
   "non-inclusion_26_1.vkey"
+  "non-inclusion_26_2.key"
+  "non-inclusion_26_2.vkey"
   "combined_26_1_1.key"
   "combined_26_1_1.vkey"
   "append_10_10.key"