tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Fix leaky env variable causing flaky test (#3761)

Browse Source
This commit is contained in:
David Katz
2025-08-05 04:45:04 -04:00
committed by GitHub
parent 66f8ef123b
commit 97e508b3a1
1 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/model.rs
+19 -10
View File
@@ -249,19 +249,28 @@ mod tests {
#[test] #[test]
#[serial] #[serial]
fn test_model_config_context_limits() { fn test_model_config_context_limits() {
let config = ModelConfig::new("claude-3-opus") // Clear all GOOSE environment variables to ensure clean test environment
.unwrap() with_var("GOOSE_TEMPERATURE", None::<&str>, || {
.with_context_limit(Some(150_000)); with_var("GOOSE_CONTEXT_LIMIT", None::<&str>, || {
assert_eq!(config.context_limit(), 150_000); with_var("GOOSE_TOOLSHIM", None::<&str>, || {
with_var("GOOSE_TOOLSHIM_OLLAMA_MODEL", None::<&str>, || {
let config = ModelConfig::new("claude-3-opus")
.unwrap()
.with_context_limit(Some(150_000));
assert_eq!(config.context_limit(), 150_000);
let config = ModelConfig::new("claude-3-opus").unwrap(); let config = ModelConfig::new("claude-3-opus").unwrap();
assert_eq!(config.context_limit(), 200_000); assert_eq!(config.context_limit(), 200_000);
let config = ModelConfig::new("gpt-4-turbo").unwrap(); let config = ModelConfig::new("gpt-4-turbo").unwrap();
assert_eq!(config.context_limit(), 128_000); assert_eq!(config.context_limit(), 128_000);
let config = ModelConfig::new("unknown-model").unwrap(); let config = ModelConfig::new("unknown-model").unwrap();
assert_eq!(config.context_limit(), DEFAULT_CONTEXT_LIMIT); assert_eq!(config.context_limit(), DEFAULT_CONTEXT_LIMIT);
});
});
});
});
} }
#[test] #[test]