fix the new decorator by removing 'async'

This commit is contained in:
thedragonsinn
2024-08-07 15:02:37 +05:30
parent 5fb4e5aaf9
commit b1e42e498d
3 changed files with 8 additions and 5 deletions

View File

@@ -7,11 +7,12 @@ import time
from io import BytesIO
import google.generativeai as genai
from app import BOT, Message, bot
from app.plugins.ai.models import get_response_text, run_basic_check
from google.ai import generativelanguage as glm
from ub_core.utils import run_shell_cmd
from app import BOT, Message, bot
from app.plugins.ai.models import get_response_text, run_basic_check
CODE_EXTS = {
".txt",
".java",

View File

@@ -1,6 +1,7 @@
from functools import wraps
import google.generativeai as genai
from app import BOT, Message, extra_config
@@ -26,7 +27,7 @@ MODEL = genai.GenerativeModel(
)
async def run_basic_check(func):
def run_basic_check(func):
@wraps(func)
async def wrapper(bot: BOT, message: Message):

View File

@@ -1,11 +1,12 @@
import pickle
from io import BytesIO
from app import BOT, Convo, Message, bot
from app.plugins.ai.models import get_response_text, run_basic_check
from pyrogram import filters
from pyrogram.enums import ParseMode
from app import BOT, Convo, Message, bot
from app.plugins.ai.models import get_response_text, run_basic_check
CONVO_CACHE: dict[str, Convo] = {}