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

split responses when longer than 2000 bytes

pull/3/head
Michael Yang 2023-08-04 18:02:03 -07:00
parent f32a4b09b3
commit cef45352bb
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
discollama.py
View File

@ -88,7 +88,6 @@ async def on_message(message):
if raw_content.strip() == '': if raw_content.strip() == '':
raw_content = 'Tell me about yourself.' raw_content = 'Tell me about yourself.'
# TODO: discord has a 2000 character limit, so we need to split the response
response = None response = None
response_content = '' response_content = ''
async with message.channel.typing(): async with message.channel.typing():
@ -100,11 +99,17 @@ async def on_message(message):
save_session(response, chunk) save_session(response, chunk)
break break
if response: if not response:
await response.edit(content=response_content + '...')
else:
response = await message.reply(response_content) response = await message.reply(response_content)
await message.remove_reaction('🤔', client.user) await message.remove_reaction('🤔', client.user)
continue
if len(response_content) + 3 >= 2000:
response = await response.reply(buffer)
response_content = buffer
continue
await response.edit(content=response_content + '...')
await response.edit(content=response_content) await response.edit(content=response_content)