prover type

sdk/src/lib.rs

@@ -41,6 +41,7 @@ use strum_macros::EnumString;
 pub struct ProverClient {
     /// The underlying prover implementation.
     pub prover: Box<dyn Prover>,
+    pub prover_type: ProverType,
 }
 
 /// The type of prover used by the [ProverClient].
@@ -96,12 +97,15 @@ impl ProverClient {
         {
             "mock" => Self {
                 prover: Box::new(MockProver::new()),
+                prover_type: ProverType::Mock,
             },
             "local" => Self {
                 prover: Box::new(LocalProver::new()),
+                prover_type: ProverType::Local,
             },
             "network" => Self {
                 prover: Box::new(NetworkProver::new()),
+                prover_type: ProverType::Network,
             },
             _ => panic!(
                 "invalid value for SP1_PROVER enviroment variable: expected 'local', 'mock', or 'remote'"
@@ -124,6 +128,7 @@ impl ProverClient {
     pub fn mock() -> Self {
         Self {
             prover: Box::new(MockProver::new()),
+            prover_type: ProverType::Mock,
         }
     }
 
@@ -142,6 +147,7 @@ impl ProverClient {
     pub fn local() -> Self {
         Self {
             prover: Box::new(LocalProver::new()),
+            prover_type: ProverType::Local,
         }
     }
 
@@ -159,30 +165,7 @@ impl ProverClient {
     pub fn remote() -> Self {
         Self {
             prover: Box::new(NetworkProver::new()),
-        }
-    }
-
-    /// Returns the type of prover used by the [ProverClient].
-    ///
-    /// ### Examples
-    ///
-    /// ```no_run
-    /// use sp1_sdk::ProverClient;
-    ///
-    /// let client = ProverClient::local();
-    /// let prover_type = client.prover_type();
-    /// assert_eq!(prover_type, ProverType::Local);
-    /// ```
-    pub fn prover_type(&self) -> ProverType {
-        let prover_type_id = (*self.prover).type_id();
-        if prover_type_id == TypeId::of::<LocalProver>() {
-            ProverType::Local
-        } else if prover_type_id == TypeId::of::<MockProver>() {
-            ProverType::Mock
-        } else if prover_type_id == TypeId::of::<NetworkProver>() {
-            ProverType::Network
-        } else {
-            panic!("invalid prover type")
+            prover_type: ProverType::Network,
         }
     }