From b1e42e498d78bf31aed0026378d5bf8ae11c36b0 Mon Sep 17 00:00:00 2001
From: thedragonsinn <98635854+thedragonsinn@users.noreply.github.com>
Date: Wed, 7 Aug 2024 15:02:37 +0530
Subject: [PATCH] fix the new decorator by removing 'async'

---
 app/plugins/ai/media_query.py | 5 +++--
 app/plugins/ai/models.py      | 3 ++-
 app/plugins/ai/text_query.py  | 5 +++--
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/app/plugins/ai/media_query.py b/app/plugins/ai/media_query.py
index 1def923..ff09619 100644
--- a/app/plugins/ai/media_query.py
+++ b/app/plugins/ai/media_query.py
@@ -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",
diff --git a/app/plugins/ai/models.py b/app/plugins/ai/models.py
index d0eb4b7..46690cc 100644
--- a/app/plugins/ai/models.py
+++ b/app/plugins/ai/models.py
@@ -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):
diff --git a/app/plugins/ai/text_query.py b/app/plugins/ai/text_query.py
index 1dd37c1..31da9f8 100644
--- a/app/plugins/ai/text_query.py
+++ b/app/plugins/ai/text_query.py
@@ -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] = {}