feat(zkvms_host_io): Add public_inputs and private_inputs attributes to RunWith

1 files changed, 7 insertions, 1 deletions

diff --git a/zkvms_host_io/src/lib.rs b/zkvms_host_io/src/lib.rs
index d63cb56..2624352 100644
--- a/zkvms_host_io/src/lib.rs
+++ b/zkvms_host_io/src/lib.rs
@@ -31,6 +31,8 @@ pub enum RunType {
 pub struct RunWith {
     pub run_type: RunType,
     pub input: Input,
+    pub public_input: PublicInput,
+    pub private_input: PrivateInput,
     pub default_env: HashMap<String, String>,
 }
 
@@ -66,13 +68,17 @@ pub fn read_args() -> RunWith {
         } else {
             DEFAULT_PRIVATE_INPUT.to_string()
         };
-    let input: Input = toml::from_str(&(public_contents + &private_contents)).unwrap();
+    let input: Input = toml::from_str(&(public_contents.clone() + &private_contents)).unwrap();
+    let public_input: PublicInput = toml::from_str(&public_contents).unwrap();
+    let private_input: PrivateInput = toml::from_str(&private_contents).unwrap();
 
     let default_env = read_str(DEFAULT_ENV).unwrap();
 
     RunWith {
         run_type: cli.run_type,
         input,
+        public_input,
+        private_input,
         default_env,
     }
 }