Richard FeldmanRemove Qwen2 model (#18444)caaa9a02 years agoHistory
31 lines · 789 B · Rust
RawDownload
use pretty_assertions::assert_eq;use rpc::LanguageModelProvider; use crate::llm::db::LlmDatabase;use crate::test_llm_db; test_llm_db!(    test_initialize_providers,    test_initialize_providers_postgres); async fn test_initialize_providers(db: &mut LlmDatabase) {    let initial_providers = db.list_providers().await.unwrap();    assert_eq!(initial_providers, vec![]);     db.initialize_providers().await.unwrap();     // Do it twice, to make sure the operation is idempotent.    db.initialize_providers().await.unwrap();     let providers = db.list_providers().await.unwrap();     assert_eq!(        providers,        &[            LanguageModelProvider::Anthropic,            LanguageModelProvider::Google,            LanguageModelProvider::OpenAi,        ]    )}