Public inputs refactor

exercises/message/src/cairo/air.rs

@@ -485,8 +485,7 @@ fn add_pub_memory_in_public_input_section(
     let mut v_aux = values.to_owned();
 
     let public_input_section = addresses.len() - public_input.public_memory.len();
-    let output_range = public_input.memory_segments.get(&MemorySegment::Output);
-    let pub_memory_addrs = get_pub_memory_addrs(output_range, public_input);
+    let pub_memory_addrs = public_input.public_memory.clone().into_keys().collect();
 
     a_aux.splice(public_input_section.., pub_memory_addrs);
     for i in public_input_section..a_aux.len() {

provers/cairo/Cargo.toml

@@ -23,7 +23,7 @@ thiserror = "1.0.38"
 log = "0.4.17"
 bincode = { version = "2.0.0-rc.2", tag = "v2.0.0-rc.2", git = "https://github.com/bincode-org/bincode.git", features= ['serde'] }
 # NOTE: For cairo 1 compatibility, add the `cairo-1-hints` feature.
-cairo-vm = { git = "https://github.com/lambdaclass/cairo-vm", rev = "e763cef", default-features = false }
+cairo-vm = { git = "https://github.com/lambdaclass/cairo-vm", rev = "e61ae177edb94e29470fed23bdda43329b16c057", default-features = false }
 sha3 = "0.10.6"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"