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

fix(test): add env_lock to is_openai_reasoning_model tests (#7917)

Browse Source 
Signed-off-by: Adrian Cole <adrian@tetrate.io>
This commit is contained in:
Adrian Cole
2026-03-16 13:05:52 +08:00
committed by GitHub
parent 5265c52b74
commit 10d597de3b
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/model.rs
+12
View File
@@ -468,8 +468,17 @@ mod tests {
mod is_openai_reasoning_model {
use super::*;
const ENV_LOCK_KEYS: [(&str, Option<&str>); 5] = [
("GOOSE_MAX_TOKENS", None),
("GOOSE_TEMPERATURE", None),
("GOOSE_CONTEXT_LIMIT", None),
("GOOSE_TOOLSHIM", None),
("GOOSE_TOOLSHIM_OLLAMA_MODEL", None),
];
#[test]
fn bare_reasoning_models() {
let _guard = env_lock::lock_env(ENV_LOCK_KEYS);
assert!(ModelConfig::new_or_fail("o1").is_openai_reasoning_model());
assert!(ModelConfig::new_or_fail("o1-preview").is_openai_reasoning_model());
assert!(ModelConfig::new_or_fail("o3").is_openai_reasoning_model());
@@ -481,6 +490,7 @@ mod tests {
#[test]
fn goose_prefixed_reasoning_models() {
let _guard = env_lock::lock_env(ENV_LOCK_KEYS);
assert!(ModelConfig::new_or_fail("goose-o3-mini").is_openai_reasoning_model());
assert!(ModelConfig::new_or_fail("goose-o4-mini").is_openai_reasoning_model());
assert!(ModelConfig::new_or_fail("goose-gpt-5").is_openai_reasoning_model());
@@ -488,6 +498,7 @@ mod tests {
#[test]
fn databricks_prefixed_reasoning_models() {
let _guard = env_lock::lock_env(ENV_LOCK_KEYS);
assert!(ModelConfig::new_or_fail("databricks-o3-mini").is_openai_reasoning_model());
assert!(ModelConfig::new_or_fail("databricks-o4-mini").is_openai_reasoning_model());
assert!(ModelConfig::new_or_fail("databricks-gpt-5").is_openai_reasoning_model());
@@ -495,6 +506,7 @@ mod tests {
#[test]
fn non_reasoning_models() {
let _guard = env_lock::lock_env(ENV_LOCK_KEYS);
assert!(!ModelConfig::new_or_fail("claude-sonnet-4").is_openai_reasoning_model());
assert!(!ModelConfig::new_or_fail("gpt-4o").is_openai_reasoning_model());
assert!(