feat(mocker): add offline disagg replay (#7617)

Signed-off-by: PeaBrane <yanrpei@gmail.com>

lib/mocker/src/common/protocols.rs

@@ -139,6 +139,8 @@ impl PrefillCost {
 pub struct OutputSignal {
     pub uuid: Uuid,
     pub completed: bool,
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub handoff_delay_ms: Option<f64>,
 }
 
 /// Preemption policy for evicting decode requests under memory pressure
@@ -286,6 +288,10 @@ pub struct MockEngineArgs {
     #[builder(default = "WorkerType::Aggregated")]
     pub worker_type: WorkerType,
 
+    /// Original planner profile NPZ path used to materialize `perf_model`.
+    #[builder(default = "None")]
+    pub planner_profile_data: Option<PathBuf>,
+
     /// Performance model for timing predictions (not serialized, loaded from planner_profile_data)
     #[serde(skip)]
     #[builder(default = "Arc::new(PerfModel::default())")]
@@ -691,6 +697,7 @@ impl MockEngineArgs {
             && let Some(path_str) = path_str.as_str()
         {
             let npz_path = PathBuf::from(path_str);
+            builder = builder.planner_profile_data(Some(npz_path.clone()));
             match PerfModel::from_npz(&npz_path) {
                 Ok(model) => {
                     tracing::info!("Successfully loaded performance model from: {:?}", npz_path);

lib/mocker/src/replay/router/online.rs

@@ -138,6 +138,7 @@ impl KvReplayRouter {
             args.block_size as u32,
             selector,
             policy,
+            config.router_track_prefill_tokens,
             CancellationToken::new(),
             "replay",
             false,