klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add chat feature to discord_bot example (#643)

Browse Source
This commit is contained in:
Dev Khant
2023-09-24 22:19:59 +05:30
committed by GitHub
parent 4768bacf1c
commit 8e0f05055e
2 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
examples/discord_bot/discord_bot.py
View File

@@ -48,13 +48,24 @@ async def add(ctx, data_type: str, *, url_or_text: str):
async def query(ctx, *, question: str):
print(f"User: {ctx.author.name}, Query: {question}")
try:
response = chat_bot.chat(question)
response = chat_bot.query(question)
await send_response(ctx, response)
except Exception as e:
await send_response(ctx, "An error occurred. Please try again!")
print("Error occurred during 'query' command:", e)
@bot.command()
async def chat(ctx, *, question: str):
print(f"User: {ctx.author.name}, Query: {question}")
try:
response = chat_bot.chat(question)
await send_response(ctx, response)
except Exception as e:
await send_response(ctx, "An error occurred. Please try again!")
print("Error occurred during 'chat' command:", e)
async def send_response(ctx, message):
if ctx.guild is None:
await ctx.send(message)