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

handle streaming responses over 4096 characters

Browse Source
This commit is contained in:
ned
2023-03-27 00:46:19 +02:00
parent ff188ce1c2
commit bc6a4e48f0
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
bot/telegram_bot.py
View File

@@ -331,11 +331,29 @@ class ChatGPT3TelegramBot:
prev = '' prev = ''
sent_message = None sent_message = None
backoff = 0 backoff = 0
chunk = 0
async for content, tokens in stream_response: async for content, tokens in stream_response:
if len(content.strip()) == 0: if len(content.strip()) == 0:
continue continue
chunks = self.split_into_chunks(content)
if len(chunks) > 1:
content = chunks[-1]
if chunk != len(chunks) - 1:
chunk += 1
try:
await context.bot.edit_message_text(chunks[-2], chat_id=sent_message.chat_id,
message_id=sent_message.message_id,
parse_mode=constants.ParseMode.MARKDOWN)
sent_message = await context.bot.send_message(
chat_id=sent_message.chat_id,
text=content if len(content) > 0 else "..."
)
except:
pass
continue
if is_group_chat: if is_group_chat:
# group chats have stricter flood limits # group chats have stricter flood limits
cutoff = 180 if len(content) > 1000 else 120 if len(content) > 200 else 90 if len(content) > 50 else 50 cutoff = 180 if len(content) > 1000 else 120 if len(content) > 200 else 90 if len(content) > 50 else 50