From 55a2eee77cad9de57aa3d76e804b8c453275be1a Mon Sep 17 00:00:00 2001
From: yasirarism <55983182+yasirarism@users.noreply.github.com>
Date: Sat, 10 Jun 2023 11:54:03 +0700
Subject: [PATCH] Fix bard

---
 misskaty/plugins/chatbot_ai.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/misskaty/plugins/chatbot_ai.py b/misskaty/plugins/chatbot_ai.py
index 4b4f8a04..8a6d254a 100644
--- a/misskaty/plugins/chatbot_ai.py
+++ b/misskaty/plugins/chatbot_ai.py
@@ -17,13 +17,17 @@ openai.api_key = OPENAI_API
 
 # This only for testing things, since maybe in future it will got blocked
 @app.on_message(filters.command("bard", COMMAND_HANDLER))
+@use_chat_lang()
 async def bard_chatbot(self: Client, ctx: Message, strings):
     if len(ctx.command) == 1:
         return await ctx.reply_msg(strings("no_question").format(cmd=ctx.command[0]), quote=True, del_in=5)
     msg = await ctx.reply_msg(strings("find_answers_str"), quote=True)
     data = {'message': ctx.input, 'session_id':'XAjzKUFvf_nQtNg4bt0pG54rCLnaWeJFE1_FXuQnVjNyfmjDhkKZyoqXqW5cgBmmnf8Eqg.'}
-    req = await http.post("https://bard-api-rho.vercel.app/ask", json=data)
-    await msg.edit_msg(req.get("content"))
+    try:
+        req = await http.post("https://bard-api-rho.vercel.app/ask", json=data)
+        await msg.edit_msg(req.get("content"))
+    except Exception as e:
+        await msg.edit_msg(str(e))
 
 @app.on_message(filters.command("ask", COMMAND_HANDLER))
 @ratelimiter