summaryrefslogtreecommitdiff
path: root/py/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'py/utils.py')
-rw-r--r--py/utils.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/py/utils.py b/py/utils.py
index 09dfaa3..8539347 100644
--- a/py/utils.py
+++ b/py/utils.py
@@ -27,9 +27,10 @@ def load_api_key():
return api_key.strip()
def make_openai_options(options):
+ max_tokens = int(options['max_tokens'])
return {
'model': options['model'],
- 'max_tokens': int(options['max_tokens']),
+ 'max_tokens': max_tokens if max_tokens > 0 else None,
'temperature': float(options['temperature']),
}