diff --git a/pkg/openai/session.py b/pkg/openai/session.py index aa0f6ba7..3aff1895 100644 --- a/pkg/openai/session.py +++ b/pkg/openai/session.py @@ -185,7 +185,7 @@ class Session: if checked_rounds > max_rounds: break - if len(result_temp) > max_tokens: + if int((len(result_temp.encode('utf-8')) - len(result_temp)) / 2 + len(result_temp)) > max_tokens: break result = result_temp