diff --git a/config.env.sample b/config.env.sample index 3f1ed593..de549b8d 100644 --- a/config.env.sample +++ b/config.env.sample @@ -12,4 +12,5 @@ SUPPORT_CHAT=YasirPediaChannel COMMAND_HANDLER= USER_SESSION= OPENAI_API= +BARD_API= CURRENCY_API= \ No newline at end of file diff --git a/misskaty/plugins/chatbot_ai.py b/misskaty/plugins/chatbot_ai.py index da3abba4..c453ac53 100644 --- a/misskaty/plugins/chatbot_ai.py +++ b/misskaty/plugins/chatbot_ai.py @@ -15,7 +15,7 @@ from pyrogram.types import Message from misskaty import app from misskaty.core import pyro_cooldown from misskaty.helper import check_time_gap, fetch, post_to_telegraph, use_chat_lang -from misskaty.vars import COMMAND_HANDLER, OPENAI_API, SUDO +from misskaty.vars import BARD_API, COMMAND_HANDLER, OPENAI_API, SUDO openai.api_key = OPENAI_API @@ -28,10 +28,12 @@ async def bard_chatbot(_, ctx: Message, strings): return await ctx.reply_msg( strings("no_question").format(cmd=ctx.command[0]), quote=True, del_in=5 ) + if not BARD_API: + return await ctx.reply_msg("BARD_API env is mising!!!") msg = await ctx.reply_msg(strings("find_answers_str"), quote=True) try: req = await fetch.get( - f"https://yasirapi.eu.org/bard?input={ctx.text.split(maxsplit=1)[1]}" + f"https://yasirapi.eu.org/bard?input={ctx.text.split(maxsplit=1)[1]}&key={BARD_API}" ) random_choice = random.choice(req.json().get("choices")) await msg.edit_msg( @@ -50,6 +52,8 @@ async def openai_chatbot(_, ctx: Message, strings): return await ctx.reply_msg( 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!!!") 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): diff --git a/misskaty/vars.py b/misskaty/vars.py index 64acb927..3186addf 100644 --- a/misskaty/vars.py +++ b/misskaty/vars.py @@ -55,6 +55,7 @@ SUDO = list( SUPPORT_CHAT = environ.get("SUPPORT_CHAT", "YasirPediaChannel") AUTO_RESTART = environ.get("AUTO_RESTART", False) OPENAI_API = environ.get("OPENAI_API") +BARD_API = environ.get("BARD_API") ## Config For AUtoForwarder # Forward From Chat ID