缓存user_prompts以提高性能

src/voice_dialogue/config/user_config.py

@@ -1,10 +1,13 @@
 """用户配置管理模块"""
 import json
-from typing import Dict
+from typing import Dict, Optional
 
-from voice_dialogue.utils.logger import logger
 from .llm_config import CHINESE_SYSTEM_PROMPT, ENGLISH_SYSTEM_PROMPT
 from .paths import USER_PROMPTS_PATH
+from ..utils.logger import logger
+
+# 内存缓存，避免重复读取文件
+_user_prompts_cache: Optional[Dict[str, str]] = None
 
 
 def get_user_prompts() -> Dict[str, str]:
@@ -12,25 +15,32 @@ def get_user_prompts() -> Dict[str, str]:
     加载用户自定义的 prompts
     
     Returns:
-        Dict[str, str]: 用户自定义的 prompts，如果文件不存在或解析失败则返回空字典
+        Dict[str, str]: 用户自定义的 prompts。
     """
+    global _user_prompts_cache
+    if _user_prompts_cache is not None:
+        return _user_prompts_cache
+
     if not USER_PROMPTS_PATH.exists():
-        logger.info(f"用户配置文件不存在: {USER_PROMPTS_PATH}")
-        return {}
+        logger.info(f"用户配置文件不存在，使用空配置: {USER_PROMPTS_PATH}")
+        _user_prompts_cache = {}
+        return _user_prompts_cache
 
     try:
         with open(USER_PROMPTS_PATH, 'r', encoding='utf-8') as f:
             user_prompts = json.load(f)
-            logger.info("成功加载用户自定义 prompts")
-            return user_prompts
+            logger.info("成功从文件加载用户自定义 prompts 到缓存")
+            _user_prompts_cache = user_prompts
+            return _user_prompts_cache
     except (json.JSONDecodeError, IOError) as e:
-        logger.error(f"无法加载用户 prompt 配置文件: {e}")
-        return {}
+        logger.error(f"无法加载用户 prompt 配置文件，使用空配置: {e}")
+        _user_prompts_cache = {}
+        return _user_prompts_cache
 
 
 def save_user_prompts(prompts: Dict[str, str]) -> bool:
     """
-    保存用户自定义的 prompts 到 JSON 文件
+    保存用户自定义的 prompts 到 JSON 文件，并更新缓存。
     
     Args:
         prompts: 要保存的 prompts 字典
@@ -38,6 +48,7 @@ def save_user_prompts(prompts: Dict[str, str]) -> bool:
     Returns:
         bool: 保存是否成功
     """
+    global _user_prompts_cache
     try:
         # 确保目录存在
         if not USER_PROMPTS_PATH.parent.exists():
@@ -46,6 +57,7 @@ def save_user_prompts(prompts: Dict[str, str]) -> bool:
         with open(USER_PROMPTS_PATH, 'w', encoding='utf-8') as f:
             json.dump(prompts, f, ensure_ascii=False, indent=4)
         logger.info(f"用户 prompts 已保存到: {USER_PROMPTS_PATH}")
+        _user_prompts_cache = prompts  # 更新缓存
         return True
     except IOError as e:
         logger.error(f"无法保存用户 prompt 配置文件: {e}")
@@ -99,15 +111,17 @@ def get_raw_prompt(language: str) -> str:
 
 def reset_prompts_to_default() -> bool:
     """
-    重置 prompts 为默认值
+    重置 prompts 为默认值，并清空缓存。
     
     Returns:
         bool: 重置是否成功
     """
+    global _user_prompts_cache
     try:
         if USER_PROMPTS_PATH.exists():
             USER_PROMPTS_PATH.unlink()
             logger.info("用户自定义 prompts 已重置为默认值")
+        _user_prompts_cache = {}  # 重置缓存为空字典
         return True
     except IOError as e:
         logger.error(f"重置 prompts 失败: {e}")