Fix tests

This commit is contained in:
Mubelotix 2025-06-24 14:50:23 +02:00
parent d7721fe607
commit bc4d1530ee
3 changed files with 40 additions and 5 deletions

View file

@ -367,10 +367,12 @@ pub fn snapshot_batch(batch: &Batch) -> String {
snap.push_str(&format!("uid: {uid}, "));
snap.push_str(&format!("details: {}, ", serde_json::to_string(details).unwrap()));
snap.push_str(&format!("stats: {}, ", serde_json::to_string(&stats).unwrap()));
snap.push_str(&format!(
"embedder_stats: {}, ",
serde_json::to_string(&embedder_stats).unwrap()
));
if !embedder_stats.skip_serializing() {
snap.push_str(&format!(
"embedder stats: {}, ",
serde_json::to_string(&embedder_stats).unwrap()
));
}
snap.push_str(&format!("stop reason: {}, ", serde_json::to_string(&stop_reason).unwrap()));
snap.push('}');
snap