mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2026-01-02 10:44:50 +00:00
88 lines
3 KiB
Python
88 lines
3 KiB
Python
# * @author Yasir Aris M <yasiramunandar@gmail.com>
|
|
# * @date 2023-06-21 22:12:27
|
|
# * @projectName MissKatyPyro
|
|
# * Copyright ©YasirPedia All rights reserved
|
|
import asyncio
|
|
import html
|
|
|
|
import openai
|
|
from aiohttp import ClientSession
|
|
from pyrogram.errors import MessageTooLong
|
|
from pyrogram.types import Message
|
|
|
|
from misskaty import app
|
|
from misskaty.core.decorator.ratelimiter import ratelimiter
|
|
from misskaty.helper import check_time_gap, post_to_telegraph
|
|
from misskaty.helper.http import http
|
|
from misskaty.helper.localization import use_chat_lang
|
|
from misskaty.vars import OPENAI_API, SUDO
|
|
|
|
openai.api_key = OPENAI_API
|
|
|
|
|
|
# This only for testing things, since maybe in future it will got blocked
|
|
@app.on_cmd("bard", is_disabled=True)
|
|
@use_chat_lang()
|
|
@ratelimiter
|
|
async def bard_chatbot(_, 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": "XQjzKRYITZ7fhplF-rXa_GTynUwdctKq4aGm-lqUCCJzF98xqDulL9UKopIadNpQn0lvnA.",
|
|
}
|
|
try:
|
|
req = await http.post("https://bard-api-rho.vercel.app/ask", json=data)
|
|
await msg.edit_msg(req.json().get("content"))
|
|
except Exception as e:
|
|
await msg.edit_msg(str(e))
|
|
|
|
|
|
@app.on_cmd("ask")
|
|
@ratelimiter
|
|
@use_chat_lang()
|
|
async def openai_chatbot(_, 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
|
|
)
|
|
uid = ctx.from_user.id if ctx.from_user else ctx.sender_chat.id
|
|
is_in_gap, sleep_time = await check_time_gap(uid)
|
|
if is_in_gap and (uid not in SUDO):
|
|
return await ctx.reply_msg(strings("dont_spam"), del_in=5)
|
|
openai.aiosession.set(ClientSession())
|
|
pertanyaan = ctx.input
|
|
msg = await ctx.reply_msg(strings("find_answers_str"), quote=True)
|
|
num = 0
|
|
answer = ""
|
|
try:
|
|
response = await openai.ChatCompletion.acreate(
|
|
model="gpt-3.5-turbo-0613",
|
|
messages=[{"role": "user", "content": pertanyaan}],
|
|
temperature=0.7,
|
|
stream=True,
|
|
)
|
|
async for chunk in response:
|
|
if not chunk.choices[0].delta or chunk.choices[0].delta.get("role"):
|
|
continue
|
|
num += 1
|
|
answer += chunk.choices[0].delta.content
|
|
if num == 30:
|
|
await msg.edit_msg(answer)
|
|
await asyncio.sleep(1.5)
|
|
num = 0
|
|
await msg.edit_msg(answer)
|
|
except MessageTooLong:
|
|
answerlink = await post_to_telegraph(
|
|
False, "MissKaty ChatBot ", html.escape(answer)
|
|
)
|
|
await msg.edit_msg(
|
|
strings("answers_too_long").format(answerlink=answerlink),
|
|
disable_web_page_preview=True,
|
|
)
|
|
except Exception as err:
|
|
await msg.edit_msg(f"ERROR: {str(err)}")
|
|
await openai.aiosession.get().close()
|