use serde::{Deserialize, Serialize};
use crate::providers::ProviderName;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
#[serde(rename_all = "snake_case")]
pub enum TriggerPolicy {
#[default]
HasRetrieval,
Always,
Manual,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct QualityConfig {
#[serde(default)]
pub self_check: bool,
#[serde(default)]
pub proposer_provider: ProviderName,
#[serde(default)]
pub checker_provider: ProviderName,
#[serde(default)]
pub trigger: TriggerPolicy,
#[serde(default = "default_min_evidence")]
pub min_evidence: f32,
#[serde(default)]
pub async_run: bool,
#[serde(default = "default_latency_budget_ms")]
pub latency_budget_ms: u64,
#[serde(default = "default_per_call_timeout_ms")]
pub per_call_timeout_ms: u64,
#[serde(default = "default_max_assertions")]
pub max_assertions: usize,
#[serde(default = "default_max_response_chars")]
pub max_response_chars: usize,
#[serde(default = "default_cache_disabled_for_checker")]
pub cache_disabled_for_checker: bool,
#[serde(default = "default_flag_marker")]
pub flag_marker: String,
}
fn default_min_evidence() -> f32 {
0.6
}
fn default_latency_budget_ms() -> u64 {
4_000
}
fn default_per_call_timeout_ms() -> u64 {
2_000
}
fn default_max_assertions() -> usize {
12
}
fn default_max_response_chars() -> usize {
8_000
}
fn default_cache_disabled_for_checker() -> bool {
true
}
fn default_flag_marker() -> String {
"[verify]".into()
}
impl Default for QualityConfig {
fn default() -> Self {
Self {
self_check: false,
proposer_provider: ProviderName::default(),
checker_provider: ProviderName::default(),
trigger: TriggerPolicy::default(),
min_evidence: default_min_evidence(),
async_run: false,
latency_budget_ms: default_latency_budget_ms(),
per_call_timeout_ms: default_per_call_timeout_ms(),
max_assertions: default_max_assertions(),
max_response_chars: default_max_response_chars(),
cache_disabled_for_checker: default_cache_disabled_for_checker(),
flag_marker: default_flag_marker(),
}
}
}