Jasper
39 lines
1.4 KiB
Rust
39 lines
1.4 KiB
Rust
use goose::providers::litellm::LiteLLMProvider;
|
|
|
|
#[tokio::test]
|
|
async fn litellm_host_follows_the_configuration_contract() {
|
|
let temp_dir = tempfile::tempdir().expect("tempdir");
|
|
let root = temp_dir.path().to_string_lossy().to_string();
|
|
let _guard = env_lock::lock_env([
|
|
("GOOSE_PATH_ROOT", Some(root.as_str())),
|
|
("GOOSE_DISABLE_KEYRING", Some("1")),
|
|
("GOOSE_ADDITIONAL_CONFIG_FILES", None::<&str>),
|
|
("LITELLM_API_KEY", Some("test-key")),
|
|
("LITELLM_HOST", None::<&str>),
|
|
("LITELLM_CUSTOM_HEADERS", None::<&str>),
|
|
]);
|
|
|
|
let provider = LiteLLMProvider::from_env(None)
|
|
.await
|
|
.expect("provider should use its default host");
|
|
let debug = format!("{provider:?}");
|
|
assert!(
|
|
debug.contains("http://localhost:4000"),
|
|
"missing host should stay on the advertised local proxy: {debug}"
|
|
);
|
|
assert!(
|
|
!debug.contains("api.litellm.ai"),
|
|
"missing host should not select a remote service: {debug}"
|
|
);
|
|
|
|
std::env::set_var("LITELLM_HOST", "https://proxy.example.test");
|
|
let provider = LiteLLMProvider::from_env(None)
|
|
.await
|
|
.expect("provider should accept an explicit host");
|
|
let debug = format!("{provider:?}");
|
|
assert!(
|
|
debug.contains("https://proxy.example.test"),
|
|
"explicit host should remain unchanged: {debug}"
|
|
);
|
|
}
|