Просмотр исходного кода

Add chat feature to discord_bot example (#643)

Dev Khant 1 год назад
Родитель
Сommit
8e0f05055e
2 измененных файлов с 16 добавлено и 1 удалено
  1. 4 0
      docs/examples/discord_bot.mdx
  2. 12 1
      examples/discord_bot/discord_bot.py

+ 4 - 0
docs/examples/discord_bot.mdx

@@ -53,6 +53,10 @@ python -m embedchain.bots.discord --include-question
 ```text
 /query <question>
 ```
+- You can chat with the bot using the slash command:
+```text
+/chat <question>
+```
 📝 Note: To use the bot privately, you can message the bot directly by right clicking the bot and selecting `Message`.
 
 🎉 Happy Chatting! 🎉

+ 12 - 1
examples/discord_bot/discord_bot.py

@@ -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)