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

Escape html text in openai chatbot

Browse source 
Thanks to mas bagus yg sudah ingetin 🤣

Signed-off-by: Yasir Aris M <git@yasirdev.my.id>
This commit is contained in:
Yasir Aris M 2023-12-14 07:20:40 +07:00 committed by GitHub
parent fc58079ee9
commit 593527f5a7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
misskaty/plugins/chatbot_ai.py
View file

@ -7,7 +7,7 @@ import html
import random
from openai import AsyncOpenAI, APIConnectionError, RateLimitError, APIStatusError
from pyrogram import filters
from pyrogram import filters, enums
from pyrogram.errors import MessageTooLong
from pyrogram.types import Message
@ -34,7 +34,7 @@ async def bard_chatbot(_, ctx: Message, strings):
)
random_choice = random.choice(req.json().get("choices"))
await msg.edit_msg(
random_choice["content"][0]
html.escape(random_choice["content"][0])
if req.json().get("content") != ""
else "Failed getting data from Bard"
)
@ -50,7 +50,7 @@ async def openai_chatbot(_, ctx: Message, strings):
strings("no_question").format(cmd=ctx.command[0]), quote=True, del_in=5
)
if not OPENAI_API:
return await ctx.reply_msg("OPENAI_API env is mising!!!")
return await ctx.reply_msg("OPENAI_API env is missing!!!")
uid = ctx.from_user.id if ctx.from_user else ctx.sender_chat.id
is_in_gap, _ = await check_time_gap(uid)
if is_in_gap and (uid not in SUDO):
@ -73,10 +73,10 @@ async def openai_chatbot(_, ctx: Message, strings):
num += 1
answer += chunk.choices[0].delta.content
if num == 30:
await msg.edit_msg(answer)
await msg.edit_msg(html.escape(answer), parse_mode=enums.ParseMode.HTML)
await asyncio.sleep(1.5)
num = 0
await msg.edit_msg(answer)
await msg.edit_msg(html.escape(answer), parse_mode=enums.ParseMode.HTML)
except MessageTooLong:
answerlink = await post_to_telegraph(
False, "MissKaty ChatBot ", html.escape(f"<code>{answer}</code>")
@ -91,3 +91,5 @@ async def openai_chatbot(_, ctx: Message, strings):
await msg.edit_msg("A 429 status code was received; we should back off a bit.")
except APIStatusError as e:
await msg.edit_msg(f"Another {e.status_code} status code was received with response {e.response}")
except Exception as e:
await msg.edit_msg(f"ERROR: {e}")