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