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

add option to disable quoting in private chats

Browse Source
This commit is contained in:
ned
2023-04-03 19:50:08 +02:00
parent dbae807477
commit ae9243a6c4
3 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
bot/telegram_bot.py
View File

@@ -187,7 +187,7 @@ class ChatGPTTelegramBot:
image_url, image_size = await self.openai.generate_image(prompt=image_query)
await context.bot.send_photo(
chat_id=chat_id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
photo=image_url
)
# add image request to users usage tracker
@@ -201,7 +201,7 @@ class ChatGPTTelegramBot:
logging.exception(e)
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
text=f'Failed to generate image: {str(e)}',
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -232,7 +232,7 @@ class ChatGPTTelegramBot:
logging.exception(e)
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
text=f'Failed to download audio file: {str(e)}. Make sure the file is not too large. (max 20MB)',
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -249,7 +249,7 @@ class ChatGPTTelegramBot:
logging.exception(e)
await context.bot.send_message(
chat_id=update.effective_chat.id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
text='Unsupported file type'
)
if os.path.exists(filename):
@@ -284,7 +284,7 @@ class ChatGPTTelegramBot:
for index, transcript_chunk in enumerate(chunks):
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id if index == 0 else None,
reply_to_message_id=self.get_reply_to_message_id(update) if index == 0 else None,
text=transcript_chunk,
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -305,7 +305,7 @@ class ChatGPTTelegramBot:
for index, transcript_chunk in enumerate(chunks):
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id if index == 0 else None,
reply_to_message_id=self.get_reply_to_message_id(update) if index == 0 else None,
text=transcript_chunk,
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -314,7 +314,7 @@ class ChatGPTTelegramBot:
logging.exception(e)
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
text=f'Failed to transcribe text: {str(e)}',
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -400,7 +400,7 @@ class ChatGPTTelegramBot:
message_id=sent_message.message_id)
sent_message = await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
text=content
)
except:
@@ -445,7 +445,7 @@ class ChatGPTTelegramBot:
try:
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id if index == 0 else None,
reply_to_message_id=self.get_reply_to_message_id(update) if index == 0 else None,
text=chunk,
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -453,7 +453,7 @@ class ChatGPTTelegramBot:
try:
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id if index == 0 else None,
reply_to_message_id=self.get_reply_to_message_id(update) if index == 0 else None,
text=chunk
)
except Exception as e:
@@ -475,7 +475,7 @@ class ChatGPTTelegramBot:
logging.exception(e)
await context.bot.send_message(
chat_id=chat_id,
reply_to_message_id=update.message.message_id,
reply_to_message_id=self.get_reply_to_message_id(update),
text=f'Failed to get response: {str(e)}',
parse_mode=constants.ParseMode.MARKDOWN
)
@@ -733,6 +733,16 @@ class ChatGPTTelegramBot:
return True
def get_reply_to_message_id(self, update: Update):
"""
Returns the message id of the message to reply to
:param update: Telegram update object
:return: Message id of the message to reply to, or None if quoting is disabled
"""
if self.config['enable_quoting'] or self.is_group_chat(update):
return update.message.message_id
return None
def split_into_chunks(self, text: str, chunk_size: int = 4096) -> list[str]:
"""
Splits a string into chunks of a given size.