feat(provider): add Wanjie Ark support
This commit is contained in:
@@ -87,6 +87,16 @@ impl Default for ModelRegistry {
|
||||
supports_tools: true,
|
||||
supports_reasoning: false,
|
||||
},
|
||||
ModelInfo {
|
||||
id: "deepseek-reasoner".to_string(),
|
||||
provider: ProviderKind::WanjieArk,
|
||||
aliases: vec![
|
||||
"wanjie-deepseek-reasoner".to_string(),
|
||||
"ark-wanjie-deepseek-reasoner".to_string(),
|
||||
],
|
||||
supports_tools: true,
|
||||
supports_reasoning: true,
|
||||
},
|
||||
ModelInfo {
|
||||
id: "deepseek/deepseek-v4-pro".to_string(),
|
||||
provider: ProviderKind::Openrouter,
|
||||
@@ -361,6 +371,16 @@ mod tests {
|
||||
assert_eq!(resolved.resolved.id, "deepseek/deepseek-v4-pro");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn wanjie_ark_default_uses_reasoner_model_id() {
|
||||
let registry = ModelRegistry::default();
|
||||
let resolved = registry.resolve(None, Some(ProviderKind::WanjieArk));
|
||||
|
||||
assert_eq!(resolved.resolved.provider, ProviderKind::WanjieArk);
|
||||
assert_eq!(resolved.resolved.id, "deepseek-reasoner");
|
||||
assert!(resolved.resolved.supports_reasoning);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn novita_default_uses_namespaced_model_id() {
|
||||
let registry = ModelRegistry::default();
|
||||
|
||||
Reference in New Issue
Block a user