refactor

check_system.py

@@ -0,0 +1,160 @@
+#!/usr/bin/env python3
+"""
+系统完整性检查
+"""
+import sys
+import os
+sys.path.append(os.path.dirname(__file__))
+
+def check_imports():
+    """检查所有关键模块是否能正常导入"""
+    print("🔍 检查模块导入...")
+
+    try:
+        # 检查Function Call系统
+        from function_calls.spec import FunctionSpec, FunctionResult
+        from function_calls.registry import register_function, list_functions
+        from function_calls.router import FunctionCallRouter
+        from function_calls.llm import FunctionCallLLM
+        import function_calls.init_handlers
+        print("✅ Function Call核心模块导入成功")
+
+        # 检查处理器
+        from function_calls.handlers import (
+            handle_weather, handle_news, handle_help,
+            handle_reminder_set, handle_reminder_list, handle_reminder_delete,
+            handle_perplexity_search, handle_summary, handle_clear_messages
+        )
+        print("✅ 所有处理器导入成功")
+
+        # 检查参数模型
+        from function_calls.models import (
+            WeatherArgs, NewsArgs, HelpArgs, ReminderArgs,
+            PerplexityArgs, SummaryArgs, ClearMessagesArgs
+        )
+        print("✅ 参数模型导入成功")
+
+        return True
+    except Exception as e:
+        print(f"❌ 导入失败: {e}")
+        return False
+
+def check_function_registration():
+    """检查函数注册是否正常"""
+    print("\n🔍 检查函数注册...")
+
+    try:
+        from function_calls.registry import list_functions
+        functions = list_functions()
+
+        expected_count = 10
+        if len(functions) != expected_count:
+            print(f"⚠️ 函数数量异常: 期望{expected_count}个，实际{len(functions)}个")
+            return False
+
+        required_functions = [
+            'weather_query', 'news_query', 'help', 'summary',
+            'reminder_set', 'reminder_list', 'reminder_delete',
+            'perplexity_search', 'clear_messages', 'insult'
+        ]
+
+        missing_functions = []
+        for func_name in required_functions:
+            if func_name not in functions:
+                missing_functions.append(func_name)
+
+        if missing_functions:
+            print(f"❌ 缺少函数: {missing_functions}")
+            return False
+
+        print("✅ 所有必需函数都已正确注册")
+        return True
+
+    except Exception as e:
+        print(f"❌ 函数注册检查失败: {e}")
+        return False
+
+def check_router_initialization():
+    """检查路由器初始化"""
+    print("\n🔍 检查路由器初始化...")
+
+    try:
+        from function_calls.router import FunctionCallRouter
+        router = FunctionCallRouter()
+        print("✅ FunctionCallRouter初始化成功")
+
+        # 测试直接命令匹配
+        class MockCtx:
+            def __init__(self, text):
+                self.text = text
+
+        test_cases = [
+            ("help", "help"),
+            ("新闻", "news_query"),
+            ("天气 北京", "weather_query")
+        ]
+
+        for input_text, expected in test_cases:
+            ctx = MockCtx(input_text)
+            result = router._try_direct_command_match(ctx)
+            if result != expected:
+                print(f"❌ 直接匹配失败: '{input_text}' -> {result} (期望: {expected})")
+                return False
+
+        print("✅ 直接命令匹配正常")
+        return True
+
+    except Exception as e:
+        print(f"❌ 路由器初始化失败: {e}")
+        return False
+
+def check_config_compatibility():
+    """检查配置兼容性"""
+    print("\n🔍 检查配置文件...")
+
+    try:
+        # 检查模板文件是否包含新配置
+        with open('config.yaml.template', 'r', encoding='utf-8') as f:
+            content = f.read()
+
+        if 'function_call_router:' not in content:
+            print("❌ config.yaml.template缺少function_call_router配置")
+            return False
+
+        print("✅ 配置文件包含Function Call配置")
+        return True
+
+    except Exception as e:
+        print(f"❌ 配置检查失败: {e}")
+        return False
+
+def main():
+    print("🚀 Function Call系统完整性检查\n")
+
+    checks = [
+        ("模块导入", check_imports),
+        ("函数注册", check_function_registration),
+        ("路由器初始化", check_router_initialization),
+        ("配置兼容性", check_config_compatibility)
+    ]
+
+    passed = 0
+    total = len(checks)
+
+    for name, check_func in checks:
+        if check_func():
+            passed += 1
+        else:
+            print(f"❌ {name}检查失败")
+
+    print(f"\n📊 检查结果: {passed}/{total} 通过")
+
+    if passed == total:
+        print("🎉 Function Call系统完整性检查全部通过！系统已准备就绪。")
+        return 0
+    else:
+        print("⚠️ 部分检查失败，请检查上述错误信息。")
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main())