starknet_os_runner: fix log`s prefix (#12529)

Author: AvivYossef-starkware

crates/starknet_os_runner/src/runner.rs

@@ -23,7 +23,8 @@ use starknet_api::transaction::{
 use starknet_os::commitment_infos::CommitmentInfo;
 use starknet_os::io::os_input::{OsBlockInput, OsHints, OsHintsConfig, StarknetOsInput};
 use starknet_os::runner::run_virtual_os;
-use tracing::info;
+use tracing::field::display;
+use tracing::{info, Span};
 use url::Url;
 
 use crate::classes_provider::ClassesProvider;
@@ -279,11 +280,10 @@ where
                 let tx_hash = tx
                     .calculate_transaction_hash(&chain_id, &version)
                     .map_err(|e| RunnerError::TransactionHashError(e.to_string()))?;
-                info!(
-                    block_id = ?block_id,
-                    tx_hash = %tx_hash,
-                    "Starting transaction proving"
-                );
+                // Record tx_hash on the parent span (`prove_transaction`) so all
+                // subsequent logs carry it as a prefix.
+                Span::current().record("tx_hash", display(&tx_hash));
+                info!(transaction = ?tx, "Starting transaction proving");
                 Ok((tx, tx_hash))
             })
             .collect::<Result<Vec<_>, RunnerError>>()?;

crates/starknet_os_runner/src/virtual_snos_prover.rs

@@ -118,7 +118,7 @@ impl<R: VirtualSnosRunner> VirtualSnosProver<R> {
     /// 2. Calculates the transaction hash.
     /// 3. Runs the Starknet OS.
     /// 4. Generates a proof.
-    #[instrument(skip(self), fields(block_id, tx_hash))]
+    #[instrument(skip(self, transaction), fields(block_id = ?block_id, tx_hash))]
     pub async fn prove_transaction(
         &self,
         block_id: BlockId,