Merge pull request #1 from zchk0/zchk-gptbot-custom

zchk0 gptbot custom

.gitignore

@@ -5,3 +5,4 @@ __pycache__
 /usage_logs
 venv
 /.cache
+translations.json

bot/main.py

@@ -53,7 +53,7 @@ def main():
         'bot_language': os.environ.get('BOT_LANGUAGE', 'en'),
         'show_plugins_used': os.environ.get('SHOW_PLUGINS_USED', 'false').lower() == 'true',
         'whisper_prompt': os.environ.get('WHISPER_PROMPT', ''),
-        'vision_model': os.environ.get('VISION_MODEL', 'gpt-4-vision-preview'),
+        'vision_model': os.environ.get('VISION_MODEL', 'gpt-4o'),
         'enable_vision_follow_up_questions': os.environ.get('ENABLE_VISION_FOLLOW_UP_QUESTIONS', 'true').lower() == 'true',
         'vision_prompt': os.environ.get('VISION_PROMPT', 'What is in this image'),
         'vision_detail': os.environ.get('VISION_DETAIL', 'auto'),

bot/openai_helper.py

@@ -26,9 +26,9 @@
 GPT_3_16K_MODELS = ("gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-1106", "gpt-3.5-turbo-0125")
 GPT_4_MODELS = ("gpt-4", "gpt-4-0314", "gpt-4-0613", "gpt-4-turbo-preview")
 GPT_4_32K_MODELS = ("gpt-4-32k", "gpt-4-32k-0314", "gpt-4-32k-0613")
-GPT_4_VISION_MODELS = ("gpt-4-vision-preview",)
+GPT_4_VISION_MODELS = ("gpt-4-vision-preview", "gpt-4o")
 GPT_4_128K_MODELS = ("gpt-4-1106-preview","gpt-4-0125-preview","gpt-4-turbo-preview", "gpt-4-turbo", "gpt-4-turbo-2024-04-09")
-GPT_4O_MODELS = ("gpt-4o",)
+GPT_4O_MODELS = ("gpt-4o","gpt-4o-mini")
 GPT_ALL_MODELS = GPT_3_MODELS + GPT_3_16K_MODELS + GPT_4_MODELS + GPT_4_32K_MODELS + GPT_4_VISION_MODELS + GPT_4_128K_MODELS + GPT_4O_MODELS
 
 def default_max_tokens(model: str) -> int:
@@ -652,8 +652,8 @@ def __count_tokens(self, messages) -> int:
         model = self.config['model']
         try:
             encoding = tiktoken.encoding_for_model(model)
-        except KeyError:
-            encoding = tiktoken.get_encoding("gpt-3.5-turbo")
+        except KeyError: # было gpt-3.5-turbo фикс на cl100k_base или p50k_base https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb
+            encoding = tiktoken.get_encoding("cl100k_base")
 
         if model in GPT_3_MODELS + GPT_3_16K_MODELS:
             tokens_per_message = 4  # every message follows <|start|>{role/name}\n{content}<|end|>\n

requirements.txt

@@ -1,7 +1,7 @@
 python-dotenv~=1.0.0
 pydub~=0.25.1
-tiktoken==0.7.0
-openai==1.29.0
+tiktoken==0.8.0
+openai==1.45.0
 python-telegram-bot==21.1.1
 requests~=2.31.0
 tenacity==8.3.0

translations.json

@@ -644,9 +644,9 @@
         "stats_description":"Получить статистику использования",
         "resend_description":"Повторная отправка последнего сообщения",
         "chat_description":"Общайся с ботом!",
-        "disallowed":"Извини, тебе запрещено использовать этого бота. Исходный код можно найти здесь https://github.com/n3d1117/chatgpt-telegram-bot",
+        "disallowed":"хе-хе-хе тебе никто не разрешал использовать меня :)",
         "budget_limit":"Извини, ты достиг предела использования",
-        "help_text":["Я бот ChatGPT, поговори со мной!", "Пришли мне голосовое сообщение или файл, и я сделаю тебе расшифровку", "Открытый исходный код на https://github.com/n3d1117/chatgpt-telegram-bot"],
+        "help_text":["Я бот ChatGPT, поговори со мной!", "Пришли мне голосовое сообщение или файл, и я сделаю тебе расшифровку", "*тык*"],
         "stats_conversation":["Текущий разговор", "сообщения в истории", "токены чата в истории"],
         "usage_today":"Использование сегодня",
         "usage_month":"Использование в этом месяце",