fix: Add max_retries to configs

alexandrainst · Jun 19, 2024 · 63ffb34 · 63ffb34
1 parent 3f4632e
commit 63ffb34
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/config/generator/openai.yaml b/config/generator/openai.yaml
@@ -5,5 +5,6 @@ temperature: 0.0
 max_tokens: 256
 stream: true
 timeout: 60
+max_retries: 3
 system_prompt: ${..language.system_prompt}
 prompt: ${..language.prompt}
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -102,6 +102,7 @@ def openai_generator_params(
         temperature=0.0,
         stream=False,
         timeout=60,
+        max_retries=3,
         max_tokens=128,
         system_prompt=system_prompt,
         prompt=prompt,