From 531a1f646144d195fc58c2756d0109d793e0ae96 Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Sun, 11 Jun 2023 12:53:56 +0200
Subject: optional max_tokens, fixes #42

---
 py/utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'py')

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']),
     }
 
-- 
cgit v1.2.3