use crate::event::{EventType, EvidenceStrength};
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize)]
pub struct ClassifyInput {
pub text: String,
pub author_hint: String,
pub recent_tasks: Vec<TaskContext>,
}
#[derive(Debug, Clone, Serialize)]
pub struct TaskContext {
pub task_id: String,
pub title: String,
pub last_events: Vec<String>,
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct ClassifyOutput {
pub event_type: EventType,
pub task_id_guess: Option<String>,
pub confidence: f64,
pub evidence_strength: Option<EvidenceStrength>,
pub suggested_text: String,
}
pub trait Classifier: Send + Sync {
fn classify(&self, input: &ClassifyInput) -> anyhow::Result<ClassifyOutput>;
}
use crate::event::EventStatus;
pub const CONFIDENCE_THRESHOLD: f64 = 0.85;
pub fn decide_status(confidence: f64) -> EventStatus {
if confidence >= CONFIDENCE_THRESHOLD {
EventStatus::Confirmed
} else {
EventStatus::Suggested
}
}
pub mod cli;
pub mod http;
pub mod mock;
pub mod prompt;
pub mod telemetry;
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn classify_input_serializes() {
let i = ClassifyInput {
text: "Adopted Rust for the journal".into(),
author_hint: "assistant".into(),
recent_tasks: vec![],
};
let s = serde_json::to_string(&i).unwrap();
assert!(s.contains("Adopted Rust"));
}
#[test]
fn decide_status_high_confidence_is_confirmed() {
assert_eq!(decide_status(0.95), EventStatus::Confirmed);
assert_eq!(decide_status(0.85), EventStatus::Confirmed);
}
#[test]
fn decide_status_low_confidence_is_suggested() {
assert_eq!(decide_status(0.84), EventStatus::Suggested);
assert_eq!(decide_status(0.0), EventStatus::Suggested);
}
}