fix: use rawPredict for non-streaming Claude requests

2026-06-09 23:05:31 -06:00
parent bca25404ab
commit 33c6f2c4e3
3 changed files with 9 additions and 5 deletions
@@ -16,8 +16,8 @@ use parking_lot::RwLock;
 use serde::{Deserialize, Serialize};
 use serde_json::json;
 use std::{
-    collections::HashMap, env, fmt, fmt::Debug, fs, hash::Hash, path::Path, sync::Arc,
-    time::Duration,
+    cmp::Ordering, collections::HashMap, env, fmt, fmt::Debug, fs, hash::Hash, path::Path,
+    sync::Arc, time::Duration,
 };
 use tokio::time::sleep;

@@ -1196,7 +1196,7 @@ fn reciprocal_rank_fusion(
        }
    }
    let mut sorted_items: Vec<(DocumentId, f32)> = map.into_iter().collect();
-    sorted_items.sort_by(|a, b| b.1.partial_cmp(&a.1).unwrap());
+    sorted_items.sort_by(|a, b| b.1.partial_cmp(&a.1).unwrap_or(Ordering::Equal));

    sorted_items
        .into_iter()