Fix retry tests by setting env once

This commit is contained in:
aibrahim-oai
2025-07-11 13:23:00 -07:00
parent e92f464830
commit 77da2843a9
2 changed files with 7 additions and 12 deletions
+7
View File
@@ -517,6 +517,13 @@ mod tests {
.create_async()
.await;
// Ensure we control the retry behaviour for the remainder of this test
// suite. This test is executed first alphabetically so it sets the
// value before any use of the flag.
unsafe {
std::env::set_var("OPENAI_REQUEST_MAX_RETRIES", "1");
}
let config = sample_config(&server, WireApi::Chat);
let provider = config.model_provider.clone();
-12
View File
@@ -455,10 +455,6 @@ data: {{\"type\":\"response.completed\",\"response\":{{\"id\":\"{id}\",\"output\
.create_async()
.await;
unsafe {
std::env::set_var("OPENAI_REQUEST_MAX_RETRIES", "1");
}
let config = sample_config(&server);
let provider = config.model_provider.clone();
let client = ModelClient::new(
@@ -503,10 +499,6 @@ data: {{\"type\":\"response.completed\",\"response\":{{\"id\":\"{id}\",\"output\
.create_async()
.await;
unsafe {
std::env::set_var("OPENAI_REQUEST_MAX_RETRIES", "1");
}
let config = sample_config(&server);
let provider = config.model_provider.clone();
let client = ModelClient::new(
@@ -532,10 +524,6 @@ data: {{\"type\":\"response.completed\",\"response\":{{\"id\":\"{id}\",\"output\
.create_async()
.await;
unsafe {
std::env::set_var("OPENAI_REQUEST_MAX_RETRIES", "0");
}
let config = sample_config(&server);
let provider = config.model_provider.clone();
let client = ModelClient::new(