vecboost 0.3.0-rc.1

High-performance embedding vector service written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

// Copyright (c) 2025-2026 Kirky.X🌠
// SPDX-License-Identifier: Apache-2.0

pub mod domain;
// endpoint 依赖 axum，仅在 http feature 下编译
#[cfg(feature = "http")]
pub mod endpoint;
pub mod inference;
pub mod performance;
// prometheus_exporter 依赖 prometheus crate（仅 http feature 引入），
// library 模式下不需要 Prometheus 指标收集
#[cfg(feature = "http")]
pub mod prometheus_exporter;

#[cfg(feature = "http")]
pub use endpoint::metrics_endpoint;
#[cfg(feature = "http")]
pub use endpoint::metrics_middleware;
pub use inference::InferenceCollector;
#[cfg(feature = "http")]
pub use prometheus_exporter::PrometheusCollector;

/// 批内去重率：`1 - unique/total`；空批为 0。
/// 无重复时为 0；n 条全重复（unique=1）时为 `(n-1)/n`。
pub fn inbatch_dedup_ratio(total: usize, unique: usize) -> f64 {
    if total == 0 || unique >= total {
        return 0.0;
    }
    1.0 - (unique as f64 / total as f64)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_dedup_ratio_no_dup_is_zero() {
        assert_eq!(inbatch_dedup_ratio(4, 4), 0.0);
        assert_eq!(inbatch_dedup_ratio(1, 1), 0.0);
    }

    #[test]
    fn test_dedup_ratio_all_dup() {
        let r = inbatch_dedup_ratio(4, 1);
        assert!((r - 0.75).abs() < 1e-12, "4 条全重复应为 (4-1)/4=0.75");
    }

    #[test]
    fn test_dedup_ratio_empty_is_zero() {
        assert_eq!(inbatch_dedup_ratio(0, 0), 0.0);
    }
}