Update functions.py

functions.py

@@ -22,6 +22,11 @@ import os
 import requests
 from dotenv import load_dotenv
 
+try:
+    from aiohttp import ClientSession as aiohttp_client
+except ImportError:
+    aiohttp_client = None
+
 load_dotenv()
 HUGGING_TOKEN = os.environ["HUGGING_TOKEN"]
 SOURCE_ALPHA_URL = os.environ["SOURCE_ALPHA_URL"]
@@ -32,3 +37,32 @@ def ryuzaki_ai_text(text):
     headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
     response = requests.post(API_URL, headers=headers, json={"inputs": text})
     return response.json()
+
+
+async def async_searcher(
+    url: str,
+    post: bool = False,
+    head: bool = False,
+    headers: dict = None,
+    evaluate=None,
+    object: bool = False,
+    re_json: bool = False,
+    re_content: bool = False,
+    *args,
+    **kwargs,
+):
+    if aiohttp_client:
+        async with aiohttp_client(headers=headers) as client:
+            method = client.head if head else (client.post if post else client.get)
+            data = await method(url, *args, **kwargs)
+            if evaluate:
+                return await evaluate(data)
+            if re_json:
+                return await data.json()
+            if re_content:
+                return await data.read()
+            if head or object:
+                return data
+            return await data.text()
+    else:
+        return "Error"