feat(host): assert image_id.is_none

taikoxyz · Jan 6, 2025 · 4031f19 · 4031f19
1 parent 7655373
commit 4031f19
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/host/src/server/api/v2/proof/mod.rs b/host/src/server/api/v2/proof/mod.rs
@@ -48,6 +48,11 @@ async fn proof_handler(
     let mut config = prover_state.request_config();
     config.merge(&req)?;
 
+    // TODO: remove this assert after we support custom image_id for RISC0/SP1 proof type
+    assert!(
+        config.image_id.is_none(),
+        "currently we don't support custom image_id for RISC0/SP1 proof type"
+    );
     ensure_proof_request_image_id(&mut config)?;
 
     // Construct the actual proof request from the available configs.

diff --git a/host/src/server/api/v3/proof/mod.rs b/host/src/server/api/v3/proof/mod.rs
@@ -44,6 +44,12 @@ async fn proof_handler(
     inc_current_req();
 
     ensure_not_paused(&prover_state)?;
+
+    // TODO: remove this assert after we support custom image_id for RISC0/SP1 proof type
+    assert!(
+        aggregation_request.image_id.is_none(),
+        "currently we don't support custom image_id for RISC0/SP1 proof type"
+    );
     ensure_aggregation_request_image_id(&mut aggregation_request)?;
 
     // Override the existing proof request config from the config file and command line