cdnutter
/
discollama
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use reaction to communicate thinking state

pull/3/head
Michael Yang 2023-08-04 00:31:17 -07:00
parent 29040d6dfc
commit 2a39686d43
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
discollama.py
View File

@ -76,7 +76,8 @@ async def on_message(message):
await message.channel.send('What can I do for you?', reference=message) await message.channel.send('What can I do for you?', reference=message)
return return
response = await message.reply(':thinking:') await message.add_reaction('🤔')
response = None
# TODO: discord has a 2000 character limit, so we need to split the response # TODO: discord has a 2000 character limit, so we need to split the response
buffer = '' buffer = ''
@ -92,7 +93,12 @@ async def on_message(message):
if len(buffer) >= args.buffer_size: if len(buffer) >= args.buffer_size:
# buffer the edit so as to not call Discord API too often # buffer the edit so as to not call Discord API too often
response_content += buffer response_content += buffer
await response.edit(content=response_content + '...')
if response:
await response.edit(content=response_content + '...')
else:
response = await message.reply(response_content)
await message.remove_reaction('🤔', client.user)
buffer = '' buffer = ''