diff --git a/README.md b/README.md index e3a1cae..a833246 100644 --- 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 index 70ef010..c220c8f 100644 --- 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))