diff --git a/wheel/generate_type_stubs.py b/wheel/generate_type_stubs.py index 0672be01a..188071eb0 100644 --- a/wheel/generate_type_stubs.py +++ b/wheel/generate_type_stubs.py @@ -340,6 +340,8 @@ def get_flags_for_height_and_constants( NO_UNKNOWN_CONDS: int = ... STRICT_ARGS_COUNT: int = ... LIMIT_HEAP: int = ... +ENABLE_KECCAK: int = ... +ENABLE_KECCAK_OPS_OUTSIDE_GUARD: int = ... MEMPOOL_MODE: int = ... ALLOW_BACKREFS: int = ... DONT_VALIDATE_SIGNATURE: int = ... diff --git a/wheel/python/chia_rs/chia_rs.pyi b/wheel/python/chia_rs/chia_rs.pyi index 7513d7582..34fa2efdd 100644 --- a/wheel/python/chia_rs/chia_rs.pyi +++ b/wheel/python/chia_rs/chia_rs.pyi @@ -69,6 +69,8 @@ def get_flags_for_height_and_constants( NO_UNKNOWN_CONDS: int = ... STRICT_ARGS_COUNT: int = ... LIMIT_HEAP: int = ... +ENABLE_KECCAK: int = ... +ENABLE_KECCAK_OPS_OUTSIDE_GUARD: int = ... MEMPOOL_MODE: int = ... ALLOW_BACKREFS: int = ... DONT_VALIDATE_SIGNATURE: int = ... diff --git a/wheel/src/api.rs b/wheel/src/api.rs index ebbaa7b94..e0a149f54 100644 --- a/wheel/src/api.rs +++ b/wheel/src/api.rs @@ -46,6 +46,7 @@ use chia_protocol::{ }; use chia_traits::ChiaToPython; use clvm_utils::tree_hash_from_bytes; +use clvmr::chia_dialect::{ENABLE_KECCAK, ENABLE_KECCAK_OPS_OUTSIDE_GUARD}; use clvmr::{LIMIT_HEAP, NO_UNKNOWN_OPS}; use pyo3::buffer::PyBuffer; use pyo3::exceptions::{PyRuntimeError, PyTypeError, PyValueError}; @@ -621,6 +622,11 @@ pub fn chia_rs(_py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> { m.add_function(wrap_pyfunction!(run_chia_program, m)?)?; m.add("NO_UNKNOWN_OPS", NO_UNKNOWN_OPS)?; m.add("LIMIT_HEAP", LIMIT_HEAP)?; + m.add("ENABLE_KECCAK", ENABLE_KECCAK)?; + m.add( + "ENABLE_KECCAK_OPS_OUTSIDE_GUARD", + ENABLE_KECCAK_OPS_OUTSIDE_GUARD, + )?; m.add_function(wrap_pyfunction!(serialized_length, m)?)?; m.add_function(wrap_pyfunction!(compute_merkle_set_root, m)?)?;