codetether-rlm 0.1.0

Recursive Language Model processing for CodeTether
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

//! Explicit model selection from [`crate::RlmConfig`].

use crate::config::RlmConfig;

use super::{RlmModelPurpose, RlmModelSource};

pub(super) fn configured(
    config: &RlmConfig,
    purpose: RlmModelPurpose,
) -> Option<(&str, RlmModelSource)> {
    let root = clean(config.root_model.as_deref());
    let sub = clean(config.subcall_model.as_deref());
    match purpose {
        RlmModelPurpose::Subcall => sub
            .map(|m| (m, RlmModelSource::SubcallConfig))
            .or_else(|| root.map(|m| (m, RlmModelSource::RootConfig))),
        RlmModelPurpose::Root | RlmModelPurpose::Compression | RlmModelPurpose::Background => {
            root.map(|m| (m, RlmModelSource::RootConfig))
        }
    }
}

fn clean(model: Option<&str>) -> Option<&str> {
    model.filter(|m| !m.trim().is_empty())
}