mirror of
https://github.com/openai/codex.git
synced 2026-10-21 05:22:21 +00:00
Fix retry tests by setting env once
This commit is contained in:
@@ -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();
|
||||
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user