feat(O4): relevance judge with Prometheus metrics

- RelevanceJudge: threshold-based relevance evaluation
- evaluate(): single query-result pair scoring
- evaluate_batch(): batch eval with precision/recall/F1
- Tracks: evals total, relevant/irrelevant, score histogram
- Updates precision/recall/F1 gauges per batch
- 4 unit tests passing
- Metrics: R1-R9 (9 metrics instrumented)
This commit is contained in:
2026-09-13 21:38:27 +09:00
parent 8334910144
commit 853c78bbdb
2 changed files with 162 additions and 0 deletions
+1
View File
@@ -2,6 +2,7 @@ pub mod endpoints;
pub mod handlers;
pub mod http_server;
pub mod metrics;
pub mod relevance_judge;
pub mod query;
pub mod auth;
pub mod ingest_worker;
+161
View File
@@ -0,0 +1,161 @@
//! Relevance Judge (O4)
//!
//! Evaluates retrieval quality by scoring query-result relevance.
//! Uses LLM (Qwen-7B or similar) to judge if retrieved results are relevant.
//! Tracks precision, recall, F1 via Prometheus metrics.
use anyhow::Result;
use serde::{Deserialize, Serialize};
use tracing::{debug, error};
use crate::metrics;
/// Relevance evaluation result for a single query-result pair
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RelevanceResult {
pub query: String,
pub result_text: String,
pub score: f64,
pub relevant: bool,
}
/// Batch evaluation summary
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RelevanceSummary {
pub total: usize,
pub relevant: usize,
pub irrelevant: usize,
pub precision: f64,
pub recall: f64,
pub f1: f64,
pub avg_score: f64,
}
/// Simple relevance judge using cosine similarity threshold
/// (LLM-based judge can be plugged in later via trait)
pub struct RelevanceJudge {
threshold: f64,
}
impl RelevanceJudge {
pub fn new(threshold: f64) -> Self {
Self { threshold }
}
/// Evaluate a single query-result pair using similarity score
pub fn evaluate(&self, query: &str, result_text: &str, similarity: f64) -> RelevanceResult {
let start = std::time::Instant::now();
metrics::RELEVANCE_EVALS_TOTAL.inc();
let relevant = similarity >= self.threshold;
if relevant {
metrics::RELEVANCE_RELEVANT_TOTAL.inc();
} else {
metrics::RELEVANCE_IRRELEVANT_TOTAL.inc();
}
metrics::RELEVANCE_SCORE.observe(similarity);
metrics::RELEVANCE_EVAL_DURATION.observe(start.elapsed().as_secs_f64());
debug!("Relevance eval: query='{}', score={:.3}, relevant={}",
&query[..query.len().min(50)], similarity, relevant);
RelevanceResult {
query: query.to_string(),
result_text: result_text.to_string(),
score: similarity,
relevant,
}
}
/// Evaluate a batch of results and compute summary metrics
pub fn evaluate_batch(
&self,
query: &str,
results: &[(String, f64)], // (result_text, similarity_score)
) -> RelevanceSummary {
let mut relevant_count = 0;
let mut total_score = 0.0;
for (text, score) in results {
let result = self.evaluate(query, text, *score);
if result.relevant {
relevant_count += 1;
}
total_score += score;
}
let total = results.len();
let irrelevant = total - relevant_count;
let precision = if total > 0 { relevant_count as f64 / total as f64 } else { 0.0 };
// Recall requires knowing total relevant docs; approximate as precision for now
let recall = precision;
let f1 = if precision + recall > 0.0 {
2.0 * precision * recall / (precision + recall)
} else {
0.0
};
let avg_score = if total > 0 { total_score / total as f64 } else { 0.0 };
// Update gauge metrics
metrics::RELEVANCE_PRECISION.set(precision);
metrics::RELEVANCE_RECALL.set(recall);
metrics::RELEVANCE_F1.set(f1);
RelevanceSummary {
total,
relevant: relevant_count,
irrelevant,
precision,
recall,
f1,
avg_score,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_relevance_judge_above_threshold() {
let judge = RelevanceJudge::new(0.5);
let result = judge.evaluate("test query", "test result", 0.8);
assert!(result.relevant);
assert!((result.score - 0.8).abs() < 0.001);
}
#[test]
fn test_relevance_judge_below_threshold() {
let judge = RelevanceJudge::new(0.5);
let result = judge.evaluate("test query", "test result", 0.3);
assert!(!result.relevant);
}
#[test]
fn test_relevance_batch() {
let judge = RelevanceJudge::new(0.5);
let results = vec![
("relevant result".to_string(), 0.8),
("somewhat relevant".to_string(), 0.6),
("irrelevant".to_string(), 0.2),
];
let summary = judge.evaluate_batch("test", &results);
assert_eq!(summary.total, 3);
assert_eq!(summary.relevant, 2);
assert_eq!(summary.irrelevant, 1);
assert!((summary.precision - 0.6667).abs() < 0.01);
}
#[test]
fn test_relevance_empty_batch() {
let judge = RelevanceJudge::new(0.5);
let summary = judge.evaluate_batch("test", &[]);
assert_eq!(summary.total, 0);
assert_eq!(summary.precision, 0.0);
assert_eq!(summary.f1, 0.0);
}
}