linguasteg-core 0.2.0

Core domain contracts and pipeline abstractions for LinguaSteg
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

use crate::{CoreResult, LanguageTag, StyleProfileDescriptor, StyleProfileId, StyleStrength};

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct StyleSelection {
    pub profile_id: StyleProfileId,
    pub strength_override: Option<StyleStrength>,
}

#[derive(Debug, Clone, PartialEq)]
pub struct StyleCandidate {
    pub text: String,
    pub quality_score: Option<f32>,
    pub style_score: Option<f32>,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct StyleRerankRequest {
    pub language: LanguageTag,
    pub profile: Option<StyleSelection>,
    pub candidates: Vec<String>,
}

pub trait StyleScorer: Send + Sync {
    fn score_candidate(
        &self,
        language: &LanguageTag,
        profile: &StyleProfileDescriptor,
        candidate: &str,
    ) -> CoreResult<f32>;
}

pub trait StyleReranker: Send + Sync {
    fn rerank_candidates(&self, request: &StyleRerankRequest) -> CoreResult<Vec<StyleCandidate>>;
}