use serde::{Deserialize, Serialize};
use url::Url;
use super::Config;
pub(crate) const DEFAULT_OLLAMA_BASE_URL: &str = rho_providers::model::registry::OLLAMA_API_BASE;
#[derive(Clone, Debug, PartialEq, Eq)]
pub(crate) struct ProviderConfigs {
pub(crate) ollama: OllamaProviderConfig,
}
#[derive(Clone, Debug, PartialEq, Eq)]
pub(crate) struct OllamaProviderConfig {
pub(crate) base_url: Url,
}
impl Default for ProviderConfigs {
fn default() -> Self {
Self {
ollama: OllamaProviderConfig {
base_url: Url::parse(DEFAULT_OLLAMA_BASE_URL)
.expect("the default Ollama API base must be a valid URL"),
},
}
}
}
impl ProviderConfigs {
fn endpoint(&self, provider: &str) -> Option<&Url> {
match provider {
"ollama" => Some(&self.ollama.base_url),
_ => None,
}
}
pub(super) fn apply(&mut self, partial: PartialProviderConfigs) -> anyhow::Result<()> {
let Some(ollama) = partial.ollama else {
return Ok(());
};
let Some(base_url) = ollama.base_url else {
return Ok(());
};
let parsed = Url::parse(&base_url)
.map_err(|error| anyhow::anyhow!("invalid providers.ollama.base_url: {error}"))?;
if !matches!(parsed.scheme(), "http" | "https") {
anyhow::bail!("providers.ollama.base_url must use http or https");
}
if !parsed.username().is_empty() || parsed.password().is_some() {
anyhow::bail!("providers.ollama.base_url must not contain credentials");
}
if parsed.query().is_some() || parsed.fragment().is_some() {
anyhow::bail!("providers.ollama.base_url must not contain a query or fragment");
}
self.ollama.base_url = parsed;
Ok(())
}
}
impl Config {
pub(crate) fn normalize_provider_profiles(&mut self) -> anyhow::Result<()> {
let profile = rho_providers::provider::resolve_profile(&self.provider, &self.auth)?;
self.provider = profile.name.into();
self.auth = profile.auth.into();
self.model = profile.canonicalize_model_id(&self.model);
for (id, selection) in &mut self.internal_agents {
let profile =
rho_providers::provider::resolve_profile(&selection.provider, &selection.auth)
.map_err(|error| anyhow::anyhow!("internal agent '{id}': {error}"))?;
selection.provider = profile.name.into();
selection.auth = profile.auth.into();
selection.model = profile.canonicalize_model_id(&selection.model);
}
Ok(())
}
pub(crate) fn resolved_provider_endpoint(&self, provider: &str) -> Option<Url> {
self.providers.endpoint(provider).cloned().or_else(|| {
match rho_providers::model::registry::provider_runtime(provider) {
Some(rho_providers::model::registry::ProviderRuntime::OpenAiCompatible {
default_api_base,
..
}) => Some(
Url::parse(default_api_base)
.expect("built-in provider API bases must be valid URLs"),
),
_ => None,
}
})
}
}
#[derive(Serialize)]
pub(super) struct PersistedProviderConfigs<'a> {
ollama: PersistedOllamaProviderConfig<'a>,
}
#[derive(Serialize)]
struct PersistedOllamaProviderConfig<'a> {
base_url: &'a str,
}
impl<'a> From<&'a ProviderConfigs> for PersistedProviderConfigs<'a> {
fn from(config: &'a ProviderConfigs) -> Self {
Self {
ollama: PersistedOllamaProviderConfig {
base_url: config.ollama.base_url.as_str(),
},
}
}
}
#[derive(Deserialize)]
pub(super) struct PartialProviderConfigs {
pub(super) ollama: Option<PartialOllamaProviderConfig>,
}
#[derive(Deserialize)]
pub(super) struct PartialOllamaProviderConfig {
pub(super) base_url: Option<String>,
}
#[cfg(test)]
#[path = "provider_config_tests.rs"]
mod tests;