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

do not fail if total_tokens is not set

Browse Source
This commit is contained in:
ned
2023-03-26 12:28:38 +02:00
parent 26322b7bd9
commit 1841bb58fc
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
bot/telegram_bot.py
View File

@@ -384,12 +384,15 @@ class ChatGPT3TelegramBot:
parse_mode=constants.ParseMode.MARKDOWN
)
# add chat request to users usage tracker
self.usage[user_id].add_chat_tokens(total_tokens, self.config['token_price'])
# add guest chat request to guest usage tracker
allowed_user_ids = self.config['allowed_user_ids'].split(',')
if str(user_id) not in allowed_user_ids and 'guests' in self.usage:
self.usage["guests"].add_chat_tokens(total_tokens, self.config['token_price'])
try:
# add chat request to users usage tracker
self.usage[user_id].add_chat_tokens(total_tokens, self.config['token_price'])
# add guest chat request to guest usage tracker
allowed_user_ids = self.config['allowed_user_ids'].split(',')
if str(user_id) not in allowed_user_ids and 'guests' in self.usage:
self.usage["guests"].add_chat_tokens(total_tokens, self.config['token_price'])
except:
pass
except Exception as e:
logging.exception(e)