mirror of
https://github.com/katanemo/plano.git
synced 2026-06-20 15:28:07 +02:00
feat(providers): add Qianfan support
This commit is contained in:
parent
938f9c4bdf
commit
e3911a2f43
9 changed files with 147 additions and 0 deletions
|
|
@ -372,6 +372,8 @@ pub enum LlmProviderType {
|
|||
OpenAI,
|
||||
#[serde(rename = "xiaomi")]
|
||||
Xiaomi,
|
||||
#[serde(rename = "qianfan")]
|
||||
Qianfan,
|
||||
#[serde(rename = "gemini")]
|
||||
Gemini,
|
||||
#[serde(rename = "xai")]
|
||||
|
|
@ -412,6 +414,7 @@ impl Display for LlmProviderType {
|
|||
LlmProviderType::Mistral => write!(f, "mistral"),
|
||||
LlmProviderType::OpenAI => write!(f, "openai"),
|
||||
LlmProviderType::Xiaomi => write!(f, "xiaomi"),
|
||||
LlmProviderType::Qianfan => write!(f, "qianfan"),
|
||||
LlmProviderType::XAI => write!(f, "xai"),
|
||||
LlmProviderType::TogetherAI => write!(f, "together_ai"),
|
||||
LlmProviderType::AzureOpenAI => write!(f, "azure_openai"),
|
||||
|
|
@ -783,6 +786,15 @@ mod test {
|
|||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_llm_provider_type_qianfan_roundtrip() {
|
||||
let parsed: LlmProviderType =
|
||||
serde_yaml::from_str("qianfan").expect("variant should deserialize");
|
||||
assert_eq!(parsed, LlmProviderType::Qianfan);
|
||||
assert_eq!(parsed.to_string(), "qianfan");
|
||||
assert_eq!(parsed.to_provider_id(), hermesllm::ProviderId::Qianfan);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_overrides_disable_signals_default_none() {
|
||||
let overrides = super::Overrides::default();
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue