aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2025-12-20 22:24:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed max response tokens for new models

Browse Source
This commit is contained in:
AlexHTW
2023-11-07 14:53:25 +01:00
parent 1f470f4e4a
commit cd287c8fe0
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/openai_helper.py
View File

@@ -34,6 +34,8 @@ def default_max_tokens(model: str) -> int:
"""
base = 1200
if model in GPT_3_MODELS:
if model == "gpt-3.5-turbo-0301":
return 4096
return base
elif model in GPT_4_MODELS:
return base * 2
@@ -42,7 +44,7 @@ def default_max_tokens(model: str) -> int:
elif model in GPT_4_32K_MODELS:
return base * 8
elif model in GPT_4_128K_MODELS:
return base * 31
return 4096
def are_functions_available(model: str) -> bool: