increase default GPT_MAX_TOKENS to 500

phelps-sg · Nov 13, 2023 · 8108e3f · 8108e3f
1 parent d69ccc5
commit 8108e3f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -114,7 +114,7 @@ To override default parameters use the following shell environment variables:
 ~~~bash
 export GPT_MODEL=gpt-3.5-turbo
 export GPT_TEMPERATURE=0.2
-export GPT_MAX_TOKENS=100
+export GPT_MAX_TOKENS=500
 export GPT_MAX_RETRIES=5
 export GPT_RETRY_EXPONENT_SECONDS=2
 export GPT_RETRY_BASE_SECONDS=20

diff --git a/src/openai_pygenerator/openai_pygenerator.py b/src/openai_pygenerator/openai_pygenerator.py
@@ -56,7 +56,7 @@ def var(name: str, to_type: Callable[[str], T], default: T) -> T:
 
 GPT_MODEL = var("GPT_MODEL", str, "gpt-3.5-turbo")
 GPT_TEMPERATURE = var("GPT_TEMPERATURE", float, 0.2)
-GPT_MAX_TOKENS = var("GPT_MAX_TOKENS", int, 100)
+GPT_MAX_TOKENS = var("GPT_MAX_TOKENS", int, 500)
 GPT_MAX_RETRIES = var("GPT_MAX_RETRIES", int, 5)
 GPT_RETRY_EXPONENT_SECONDS = Seconds(var("GPT_RETRY_EXPONENT_SECONDS", int, 2))
 GPT_RETRY_BASE_SECONDS = Seconds(var("GPT_RETRY_BASE_SECONDS", int, 20))