From 4062c554a30df02c4ddd0d82bcc80c9b131875f8 Mon Sep 17 00:00:00 2001
From: lightsing <light.tsing@gmail.com>
Date: Wed, 26 Nov 2025 13:58:12 +0800
Subject: [PATCH] add logging

---
 Cargo.lock                            | 1 +
 crates/prover-bin/Cargo.toml          | 1 +
 crates/prover-bin/src/main.rs         | 3 +++
 crates/prover-bin/src/prover/axiom.rs | 9 ++++++++-
 4 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/Cargo.lock b/Cargo.lock
index 0ae7f3a7c..88093dab4 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -6859,6 +6859,7 @@ dependencies = [
  "sled",
  "tiny-keccak",
  "tokio",
+ "tracing",
  "url",
 ]
 
diff --git a/crates/prover-bin/Cargo.toml b/crates/prover-bin/Cargo.toml
index 6b4f22595..bdc05a987 100644
--- a/crates/prover-bin/Cargo.toml
+++ b/crates/prover-bin/Cargo.toml
@@ -17,6 +17,7 @@ once_cell.workspace =true
 base64.workspace = true
 tiny-keccak = { workspace = true, features = ["sha3", "keccak"] }
 eyre.workspace = true
+tracing.workspace = true
 
 futures = "0.3.30"
 futures-util = "0.3"
diff --git a/crates/prover-bin/src/main.rs b/crates/prover-bin/src/main.rs
index 3737c03b6..bd132e106 100644
--- a/crates/prover-bin/src/main.rs
+++ b/crates/prover-bin/src/main.rs
@@ -1,3 +1,6 @@
+#[macro_use]
+extern crate tracing;
+
 mod prover;
 mod types;
 mod zk_circuits_handler;
diff --git a/crates/prover-bin/src/prover/axiom.rs b/crates/prover-bin/src/prover/axiom.rs
index 2dbf018e2..af77a59a8 100644
--- a/crates/prover-bin/src/prover/axiom.rs
+++ b/crates/prover-bin/src/prover/axiom.rs
@@ -35,7 +35,7 @@ pub struct AxiomProverConfig {
     pub programs: HashMap<String, AxiomProgram>,
 }
 
-#[derive(Clone, Serialize, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct AxiomProgram {
     pub program_id: String,
     pub config_id: String,
@@ -121,8 +121,10 @@ impl AxiomProver {
         .flatten()
     }
 
+    #[instrument(skip_all, ret, err)]
     fn get_program(&self, vk: &[u8]) -> eyre::Result<AxiomProgram> {
         let vk = hex::encode(vk);
+        info!(vk = %vk, "looking up axiom program for vk");
         self.config
             .programs
             .get(vk.as_str())
@@ -130,6 +132,7 @@ impl AxiomProver {
             .ok_or_else(|| eyre::eyre!("no axiom program configured for vk: {vk}"))
     }
 
+    #[instrument(skip_all, fields(proof_type = ?req.proof_type), err)]
     async fn prove_inner(&mut self, req: ProveRequest) -> eyre::Result<ProveResponse> {
         let prover_task = UniversalHandler::get_task_from_input(&req.input)?;
         if prover_task.use_openvm_13 {
@@ -165,10 +168,12 @@ impl AxiomProver {
                 })
             })
             .await?;
+        info!(task_id = %response.task_id, "submitted axiom proving task");
 
         Ok(response)
     }
 
+    #[instrument(skip_all, fields(task_id = %req.task_id), err)]
     async fn query_task_inner(&mut self, req: QueryTaskRequest) -> eyre::Result<QueryTaskResponse> {
         let mut response = QueryTaskResponse {
             task_id: req.task_id.clone(),
@@ -180,6 +185,7 @@ impl AxiomProver {
         let (status, proof_type, proof) = self
             .make_axiom_request(None, move |sdk| {
                 let status = sdk.get_proof_status(&task_id)?;
+                debug!(status = ?status, "fetched axiom task status");
 
                 let program_status = sdk.get_build_status(&status.program_uuid)?;
                 let proof_type = match program_status.name.as_str() {
@@ -219,6 +225,7 @@ impl AxiomProver {
                 return Err(eyre::eyre!("unrecognized axiom task status: {other}"));
             }
         };
+        info!(status = ?response.status, "mapped axiom task status");
 
         response.proof_type = proof_type;