Merge branch 'dev' of https://github.com/MaiM-with-u/MaiBot into dev

Author: SengokuCola

plugins/example_plugin/plugin.py

@@ -59,14 +59,14 @@ class SmartGreetingAction(BaseAction):
     # ===== 功能定义必须项 =====
     action_parameters = {
         "username": "要问候的用户名（可选）",
-        "greeting_style": "问候风格：casual(随意)、formal(正式)、friendly(友好)，默认casual"
+        "greeting_style": "问候风格：casual(随意)、formal(正式)、friendly(友好)，默认casual",
     }
 
     action_require = [
         "用户发送包含问候词汇的消息时使用",
-        "检测到新用户加入时使用", 
+        "检测到新用户加入时使用",
         "响应友好交流需求时使用",
-        "避免在短时间内重复问候同一用户"
+        "避免在短时间内重复问候同一用户",
     ]
 
     associated_types = ["text", "emoji"]
@@ -115,12 +115,12 @@ async def _generate_template_greeting(self, template: str, username: str, style:
         style_templates = {
             "casual": "嗨{username}！很开心见到你～",
             "formal": "您好{username}，很荣幸为您服务！",
-            "friendly": "你好{username}！欢迎来到这里，希望我们能成为好朋友！😊"
+            "friendly": "你好{username}！欢迎来到这里，希望我们能成为好朋友！😊",
         }
 
         selected_template = style_templates.get(style, template)
         username_display = f" {username}" if username else ""
-        
+
         return selected_template.format(username=username_display)
 
     async def _generate_llm_greeting(self, username: str, style: str) -> str:
@@ -150,7 +150,7 @@ async def _generate_llm_greeting(self, username: str, style: str) -> str:
                 model_config=model_config,
                 request_type="plugin.greeting",
                 temperature=0.7,
-                max_tokens=100
+                max_tokens=100,
             )
 
             if success and response:
@@ -201,14 +201,14 @@ class HelpfulAction(BaseAction):
     action_parameters = {
         "help_type": "帮助类型：explanation(解释)、suggestion(建议)、guidance(指导)、tips(提示)",
         "topic": "帮助主题或用户关心的问题",
-        "complexity": "复杂度：simple(简单)、medium(中等)、advanced(高级)"
+        "complexity": "复杂度：simple(简单)、medium(中等)、advanced(高级)",
     }
 
     action_require = [
         "用户表达困惑或寻求帮助时使用",
         "检测到用户遇到技术问题时使用",
         "对话中出现知识盲点时主动提供帮助",
-        "避免过度频繁地提供帮助，要恰到好处"
+        "避免过度频繁地提供帮助，要恰到好处",
     ]
 
     associated_types = ["text", "emoji"]
@@ -246,7 +246,7 @@ async def _generate_help_message(self, help_type: str, topic: str, complexity: s
         """生成帮助消息"""
         # 获取配置
         enable_llm = self.api.get_config("help.enable_llm", False)
-        
+
         if enable_llm:
             return await self._generate_llm_help(help_type, topic, complexity)
         else:
@@ -258,11 +258,11 @@ async def _generate_template_help(self, help_type: str, topic: str, complexity:
             "explanation": f"关于{topic}，我来为你解释一下：这是一个{complexity}级别的概念...",
             "suggestion": f"针对{topic}，我建议你可以尝试以下方法...",
             "guidance": f"在{topic}方面，我可以为你提供一些指导...",
-            "tips": f"关于{topic}，这里有一些实用的小贴士..."
+            "tips": f"关于{topic}，这里有一些实用的小贴士...",
         }
 
         base_message = help_templates.get(help_type, f"关于{topic}，我很乐意为你提供帮助！")
-        
+
         # 根据复杂度调整消息
         if complexity == "advanced":
             base_message += "\n\n这个话题比较深入，需要一些基础知识。"
@@ -291,11 +291,7 @@ async def _generate_llm_help(self, help_type: str, topic: str, complexity: str)
 
             model_config = next(iter(models.values()))
             success, response, reasoning, model_name = await self.api.generate_with_model(
-                prompt=prompt,
-                model_config=model_config,
-                request_type="plugin.help",
-                temperature=0.7,
-                max_tokens=300
+                prompt=prompt, model_config=model_config, request_type="plugin.help", temperature=0.7, max_tokens=300
             )
 
             if success and response:

src/plugin_system/base/base_action.py

@@ -366,7 +366,7 @@ async def send_message_by_replyer(self, target: str = "", extra_info_block: str
     @classmethod
     def get_action_info(cls) -> "ActionInfo":
         """从类属性生成ActionInfo
-        
+
         所有信息都从类属性中读取，确保一致性和完整性。
         Action类必须定义所有必要的类属性。
 
@@ -376,7 +376,7 @@ def get_action_info(cls) -> "ActionInfo":
 
         # 从类属性读取名称，如果没有定义则使用类名自动生成
         name = getattr(cls, "action_name", cls.__name__.lower().replace("action", ""))
-        
+
         # 从类属性读取描述，如果没有定义则使用文档字符串的第一行
         description = getattr(cls, "action_description", None)
         if description is None:

src/plugin_system/base/component_types.py

@@ -36,17 +36,17 @@ class ChatMode(Enum):
 @dataclass
 class PythonDependency:
     """Python包依赖信息"""
-    
+
     package_name: str  # 包名称
     version: str = ""  # 版本要求，例如: ">=1.0.0", "==2.1.3", ""表示任意版本
     optional: bool = False  # 是否为可选依赖
     description: str = ""  # 依赖描述
     install_name: str = ""  # 安装时的包名（如果与import名不同）
-    
+
     def __post_init__(self):
         if not self.install_name:
             self.install_name = self.package_name
-    
+
     def get_pip_requirement(self) -> str:
         """获取pip安装格式的依赖字符串"""
         if self.version:
@@ -141,7 +141,7 @@ def __post_init__(self):
             self.python_dependencies = []
         if self.metadata is None:
             self.metadata = {}
-    
+
     def get_missing_packages(self) -> List[PythonDependency]:
         """检查缺失的Python包"""
         missing = []
@@ -152,7 +152,7 @@ def get_missing_packages(self) -> List[PythonDependency]:
                 if not dep.optional:
                     missing.append(dep)
         return missing
-    
+
     def get_pip_requirements(self) -> List[str]:
         """获取所有pip安装格式的依赖"""
         return [dep.get_pip_requirement() for dep in self.python_dependencies]

src/plugin_system/core/dependency_manager.py

@@ -7,8 +7,7 @@
 import subprocess
 import sys
 import importlib
-from typing import List, Dict, Tuple, Optional
-from pathlib import Path
+from typing import List, Dict, Tuple
 
 from src.common.logger import get_logger
 from src.plugin_system.base.component_types import PythonDependency
@@ -22,19 +21,21 @@ class DependencyManager:
     def __init__(self):
         self.install_log: List[str] = []
         self.failed_installs: Dict[str, str] = {}
-    
-    def check_dependencies(self, dependencies: List[PythonDependency]) -> Tuple[List[PythonDependency], List[PythonDependency]]:
+
+    def check_dependencies(
+        self, dependencies: List[PythonDependency]
+    ) -> Tuple[List[PythonDependency], List[PythonDependency]]:
         """检查依赖包状态
-        
+
         Args:
             dependencies: 依赖包列表
-            
+
         Returns:
             Tuple[List[PythonDependency], List[PythonDependency]]: (缺失的依赖, 可选缺失的依赖)
         """
         missing_required = []
         missing_optional = []
-        
+
         for dep in dependencies:
             if not self._is_package_available(dep.package_name):
                 if dep.optional:
@@ -45,71 +46,71 @@ def check_dependencies(self, dependencies: List[PythonDependency]) -> Tuple[List
                     logger.error(f"必需依赖包缺失: {dep.package_name} - {dep.description}")
             else:
                 logger.debug(f"依赖包已存在: {dep.package_name}")
-        
+
         return missing_required, missing_optional
-    
+
     def _is_package_available(self, package_name: str) -> bool:
         """检查包是否可用"""
         try:
             importlib.import_module(package_name)
             return True
         except ImportError:
             return False
-    
+
     def install_dependencies(self, dependencies: List[PythonDependency], auto_install: bool = False) -> bool:
         """安装依赖包
-        
+
         Args:
             dependencies: 需要安装的依赖包列表
             auto_install: 是否自动安装（True时不询问用户）
-            
+
         Returns:
             bool: 安装是否成功
         """
         if not dependencies:
             return True
-        
+
         logger.info(f"需要安装 {len(dependencies)} 个依赖包")
-        
+
         # 显示将要安装的包
         for dep in dependencies:
             install_cmd = dep.get_pip_requirement()
             logger.info(f"  - {install_cmd} {'(可选)' if dep.optional else '(必需)'}")
             if dep.description:
                 logger.info(f"    说明: {dep.description}")
-        
+
         if not auto_install:
             # 这里可以添加用户确认逻辑
             logger.warning("手动安装模式：请手动运行 pip install 命令安装依赖包")
             return False
-        
+
         # 执行安装
         success_count = 0
         for dep in dependencies:
             if self._install_single_package(dep):
                 success_count += 1
             else:
                 self.failed_installs[dep.package_name] = f"安装失败: {dep.get_pip_requirement()}"
-        
+
         logger.info(f"依赖安装完成: {success_count}/{len(dependencies)} 个成功")
         return success_count == len(dependencies)
-    
+
     def _install_single_package(self, dependency: PythonDependency) -> bool:
         """安装单个包"""
         pip_requirement = dependency.get_pip_requirement()
-        
+
         try:
             logger.info(f"正在安装: {pip_requirement}")
-            
+
             # 使用subprocess安装包
             cmd = [sys.executable, "-m", "pip", "install", pip_requirement]
             result = subprocess.run(
                 cmd,
                 capture_output=True,
                 text=True,
-                timeout=300  # 5分钟超时
+                timeout=300,  # 5分钟超时
             )
-            
+
             if result.returncode == 0:
                 logger.info(f"✅ 成功安装: {pip_requirement}")
                 self.install_log.append(f"成功安装: {pip_requirement}")
@@ -119,28 +120,29 @@ def _install_single_package(self, dependency: PythonDependency) -> bool:
                 logger.error(f"错误输出: {result.stderr}")
                 self.install_log.append(f"安装失败: {pip_requirement} - {result.stderr}")
                 return False
-                
+
         except subprocess.TimeoutExpired:
             logger.error(f"❌ 安装超时: {pip_requirement}")
             return False
         except Exception as e:
             logger.error(f"❌ 安装异常: {pip_requirement} - {str(e)}")
             return False
-    
-    def generate_requirements_file(self, plugins_dependencies: List[List[PythonDependency]], 
-                                 output_path: str = "plugin_requirements.txt") -> bool:
+
+    def generate_requirements_file(
+        self, plugins_dependencies: List[List[PythonDependency]], output_path: str = "plugin_requirements.txt"
+    ) -> bool:
         """生成插件依赖的requirements文件
-        
+
         Args:
             plugins_dependencies: 所有插件的依赖列表
             output_path: 输出文件路径
-            
+
         Returns:
             bool: 生成是否成功
         """
         try:
             all_deps = {}
-            
+
             # 合并所有插件的依赖
             for plugin_deps in plugins_dependencies:
                 for dep in plugin_deps:
@@ -152,39 +154,39 @@ def generate_requirements_file(self, plugins_dependencies: List[List[PythonDepen
                             logger.warning(f"依赖版本冲突: {key} ({existing.version} vs {dep.version})")
                     else:
                         all_deps[key] = dep
-            
+
             # 写入requirements文件
             with open(output_path, "w", encoding="utf-8") as f:
                 f.write("# 插件依赖包自动生成\n")
                 f.write("# Auto-generated plugin dependencies\n\n")
-                
+
                 # 按包名排序
                 sorted_deps = sorted(all_deps.values(), key=lambda x: x.install_name)
-                
+
                 for dep in sorted_deps:
                     requirement = dep.get_pip_requirement()
                     if dep.description:
                         f.write(f"# {dep.description}\n")
                     if dep.optional:
-                        f.write(f"# Optional dependency\n")
+                        f.write("# Optional dependency\n")
                     f.write(f"{requirement}\n\n")
-            
+
             logger.info(f"已生成插件依赖文件: {output_path} ({len(all_deps)} 个包)")
             return True
-            
+
         except Exception as e:
             logger.error(f"生成requirements文件失败: {str(e)}")
             return False
-    
+
     def get_install_summary(self) -> Dict[str, any]:
         """获取安装摘要"""
         return {
             "install_log": self.install_log.copy(),
             "failed_installs": self.failed_installs.copy(),
             "total_attempts": len(self.install_log),
-            "failed_count": len(self.failed_installs)
+            "failed_count": len(self.failed_installs),
         }
 
 
 # 全局依赖管理器实例
-dependency_manager = DependencyManager() 
+dependency_manager = DependencyManager()