xz-rerank 0.1.1

检索结果重排序 — 本地多信号融合 + 远程 Rerank API 适配
Documentation 
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

use async_trait::async_trait;

use crate::error::RerankError;
use crate::traits::SignalPlugin;
use crate::types::RerankCandidate;

/// 内容质量启发式信号
#[derive(Debug)]
pub struct ContentQualitySignal;

#[async_trait]
impl SignalPlugin for ContentQualitySignal {
    fn name(&self) -> &str {
        "content_quality"
    }

    fn weight_key(&self) -> &'static str {
        "content_quality"
    }

    async fn score(&self, _query: &str, candidate: &RerankCandidate) -> Result<f32, RerankError> {
        let content = &candidate.content;
        let len = content.len() as f32;

        let optimal_min = 200.0f32;
        let optimal_max = 2000.0f32;

        if len < optimal_min {
            Ok(len / optimal_min)
        } else if len <= optimal_max {
            Ok(1.0)
        } else {
            Ok((optimal_max / len).max(0.8))
        }
    }
}