aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2025-12-21 14:44:50 +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

2
.env.example
View File

@@ -37,7 +37,7 @@ ALLOWED_TELEGRAM_USER_IDS=USER_ID_1,USER_ID_2
# IMAGE_MODEL=dall-e-3 # IMAGE_MODEL=dall-e-3
# IMAGE_QUALITY=hd # IMAGE_QUALITY=hd
# IMAGE_STYLE=natural # IMAGE_STYLE=natural
# IMAGE_SIZE=512x512 # IMAGE_SIZE=1024x1024
# GROUP_TRIGGER_KEYWORD="" # GROUP_TRIGGER_KEYWORD=""
# IGNORE_GROUP_TRANSCRIPTIONS=true # IGNORE_GROUP_TRANSCRIPTIONS=true
# BOT_LANGUAGE=en # BOT_LANGUAGE=en

4
bot/openai_helper.py
View File

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