fix: change max token for gpt-3.5 and davinci to 4096

1 year ago · 021366ca67
parent 8c2380d1f8
commit 021366ca67
1 changed files with 3 additions and 1 deletions
--- a/Seanium_Brain.py
+++ b/Seanium_Brain.py
@ -151,7 +151,9 @@ with body:
        st_tool.rebuild_brain(chunk_size)
    if not query == '':
        if models.question_model == 'text-davinci-003' or 'text-davinci-003' in models.other_models:
-            max_model_token = 4000
+            max_model_token = 4096
+        elif models.question_model == 'gpt-3.5-turbo' or 'gpt-3.5-turbo' in models.other_models:
+            max_model_token = 4096
        else:
            max_model_token = 2048