yasirarism/MissKatyPyro
1
0
Fork 0
mirror of https://github.com/yasirarism/MissKatyPyro.git synced 2025-12-29 17:44:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle if prompt contain blacklist

Browse source 
Signed-off-by: Yasir Aris M <git@yasirdev.my.id>
This commit is contained in:
Yasir Aris M 2024-01-15 06:44:55 +07:00 committed by GitHub
parent 18a21ef093
commit 1358f23c2d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
misskaty/plugins/chatbot_ai.py
View file

@ -55,7 +55,9 @@ async def gemini_chatbot(_, ctx: Message, strings):
json=json_data,
timeout=20.0,
)
await msg.edit_msg(response.json()["candidates"][0]["content"]["parts"][0]["text"])
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.")
await msg.edit_msg(html.escape(response.json()["candidates"][0]["content"]["parts"][0]["text"]))
except Exception as e:
await msg.edit_msg(str(e))