use mnem_ingest::NerConfig;
use serde::Deserialize;
#[derive(Debug, Deserialize)]
struct NerOnlyConfig {
ner: Option<NerConfig>,
}
pub(crate) fn resolve_ner_cfg(repo_path: &std::path::Path) -> NerConfig {
if let Ok(p) = std::env::var("MNEM_NER_PROVIDER") {
return match p.to_ascii_lowercase().as_str() {
"none" => NerConfig::None,
_ => NerConfig::Rule,
};
}
let cfg_path = repo_path.join("config.toml");
if let Ok(s) = std::fs::read_to_string(&cfg_path)
&& let Ok(parsed) = toml::from_str::<NerOnlyConfig>(&s)
&& let Some(ner) = parsed.ner
{
return ner;
}
NerConfig::Rule
}