pre implement cache

nonebot_plugin_marshoai/decos.py

@@ -0,0 +1,15 @@
+from .instances import cache
+
+
+def from_cache(key):
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            cached = cache.get(key)
+            if cached:
+                return cached
+            else:
+                result = func(*args, **kwargs)
+                cache.set(key, result)
+                return result
+
+        return wrapper

nonebot_plugin_marshoai/instances.py

@@ -3,7 +3,7 @@ from nonebot import get_driver
 from openai import AsyncOpenAI
 
 from .config import config
-from .models import MarshoContext, MarshoTools
+from .models import Cache, MarshoContext, MarshoTools
 
 driver = get_driver()
 
@@ -11,6 +11,7 @@ command_start = driver.config.command_start
 model_name = config.marshoai_default_model
 context = MarshoContext()
 tools = MarshoTools()
+cache = Cache()
 token = config.marshoai_token
 endpoint = config.marshoai_azure_endpoint
 # client = ChatCompletionsClient(endpoint=endpoint, credential=AzureKeyCredential(token))

nonebot_plugin_marshoai/models.py

@@ -11,6 +11,25 @@ from nonebot import logger
 from .config import config
 
 
+class Cache:
+    """
+    缓存类
+    """
+
+    def __init__(self):
+        self.cache = {}
+
+    def get(self, key):
+        if key in self.cache:
+            return self.cache[key]
+        else:
+            self.cache[key] = None
+            return None
+
+    def set(self, key, value):
+        self.cache[key] = value
+
+
 class MarshoContext:
     """
     Marsho 的上下文类

nonebot_plugin_marshoai/util.py

@@ -23,6 +23,7 @@ from ._types import DeveloperMessage
 from .config import config
 from .constants import CODE_BLOCK_PATTERN, IMG_LATEX_PATTERN, OPENAI_NEW_MODELS
 from .deal_latex import ConvertLatex
+from .instances import cache
 
 nickname_json = None  # 记录昵称
 praises_json = None  # 记录夸赞名单