diff --git a/misskaty/plugins/chatbot_ai.py b/misskaty/plugins/chatbot_ai.py index e78c8e79..a41c0e08 100644 --- a/misskaty/plugins/chatbot_ai.py +++ b/misskaty/plugins/chatbot_ai.py @@ -55,8 +55,8 @@ async def gemini_chatbot(_, ctx: Message, strings): json=json_data, timeout=20.0, ) - if response.json().get("promptFeedback"): - return await msg.edit_msg("⚠️ Sorry, the prompt you sent contains a forbidden word that is not permitted by AI.") + if response.json().get("candidates"): + return await msg.edit_msg("⚠️ Sorry, the prompt you sent maybe contains a forbidden word that is not permitted by AI.") await msg.edit_msg(html.escape(response.json()["candidates"][0]["content"]["parts"][0]["text"])) except Exception as e: await msg.edit_msg(str(e))