mirror of
https://github.com/katanemo/plano.git
synced 2026-06-11 15:05:14 +02:00
2941392ed1
* cleaning up plano cli commands * adding support for wildcard model providers * fixing compile errors * fixing bugs related to default model provider, provider hint and duplicates in the model provider list * fixed cargo fmt issues * updating tests to always include the model id * using default for the prompt_gateway path * fixed the model name, as gpt-5-mini-2025-08-07 wasn't in the config * making sure that all aliases and models match the config * fixed the config generator to allow for base_url providers LLMs to include wildcard models * re-ran the models list utility and added a shell script to run it * updating docs to mention wildcard model providers * updated provider_models.json to yaml, added that file to our docs for reference * updating the build docs to use the new root-based build --------- Co-authored-by: Salman Paracha <salmanparacha@MacBook-Pro-342.local>
34 lines
970 B
Rust
34 lines
970 B
Rust
use std::sync::Arc;
|
|
|
|
use crate::{configuration, llm_providers::LlmProviders};
|
|
use configuration::LlmProvider;
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub enum ProviderHint {
|
|
Default,
|
|
Name(String),
|
|
}
|
|
|
|
impl From<String> for ProviderHint {
|
|
fn from(value: String) -> Self {
|
|
match value.as_str() {
|
|
"default" => ProviderHint::Default,
|
|
_ => ProviderHint::Name(value),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn get_llm_provider(
|
|
llm_providers: &LlmProviders,
|
|
provider_hint: Option<ProviderHint>,
|
|
) -> Result<Arc<LlmProvider>, String> {
|
|
match provider_hint {
|
|
Some(ProviderHint::Default) => llm_providers
|
|
.default()
|
|
.ok_or_else(|| "No default provider configured".to_string()),
|
|
Some(ProviderHint::Name(name)) => llm_providers
|
|
.get(&name)
|
|
.ok_or_else(|| format!("Model '{}' not found in configured providers", name)),
|
|
None => Err("No model specified in request".to_string()),
|
|
}
|
|
}
|