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

minor improvements

Browse Source
This commit is contained in:
ned
2023-03-27 17:14:50 +02:00
parent bf6de710af
commit f8f3c229c3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bot/telegram_bot.py
View File

@@ -15,18 +15,18 @@ from openai_helper import OpenAIHelper
from usage_tracker import UsageTracker from usage_tracker import UsageTracker
def message_text(message: Message) -> str|None: def message_text(message: Message) -> str:
""" """
Returns the text of a message, excluding any bot commands. Returns the text of a message, excluding any bot commands.
""" """
message_text = message.text message_text = message.text
if message_text is None: if message_text is None:
return message_text return ''
for _, text in sorted(message.parse_entities([MessageEntity.BOT_COMMAND]).items(), key=(lambda item: item[0].offset)): for _, text in sorted(message.parse_entities([MessageEntity.BOT_COMMAND]).items(), key=(lambda item: item[0].offset)):
message_text = message_text.replace(text, '').strip() message_text = message_text.replace(text, '').strip()
return message_text if len(message_text) > 0 else None return message_text if len(message_text) > 0 else ''
class ChatGPTTelegramBot: class ChatGPTTelegramBot: