初步重构llmrequest

debug_config.py

@@ -0,0 +1,111 @@
+#!/usr/bin/env python3
+"""
+调试配置加载问题，查看API provider的配置是否正确传递
+"""
+
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+def debug_config_loading():
+    try:
+        # 临时配置API key
+        import toml
+        config_path = "config/model_config.toml"
+        
+        with open(config_path, 'r', encoding='utf-8') as f:
+            config = toml.load(f)
+        
+        original_keys = {}
+        for provider in config['api_providers']:
+            original_keys[provider['name']] = provider['api_key']
+            provider['api_key'] = f"sk-test-key-for-{provider['name'].lower()}-12345"
+        
+        with open(config_path, 'w', encoding='utf-8') as f:
+            toml.dump(config, f)
+        
+        print("✅ 配置了测试API key")
+        
+        try:
+            # 清空缓存
+            modules_to_remove = [
+                'src.config.config',
+                'src.config.api_ada_configs',
+                'src.llm_models.model_manager',
+                'src.llm_models.model_client',
+                'src.llm_models.utils_model'
+            ]
+            for module in modules_to_remove:
+                if module in sys.modules:
+                    del sys.modules[module]
+            
+            # 导入配置
+            from src.config.config import model_config
+            print("\n🔍 调试配置加载:")
+            print(f"model_config类型: {type(model_config)}")
+            
+            # 检查API providers
+            if hasattr(model_config, 'api_providers'):
+                print(f"API providers数量: {len(model_config.api_providers)}")
+                for name, provider in model_config.api_providers.items():
+                    print(f"  - {name}: {provider.base_url}")
+                    print(f"    API key: {provider.api_key[:10]}...{provider.api_key[-5:] if len(provider.api_key) > 15 else provider.api_key}")
+                    print(f"    Client type: {provider.client_type}")
+            
+            # 检查模型配置
+            if hasattr(model_config, 'models'):
+                print(f"模型数量: {len(model_config.models)}")
+                for name, model in model_config.models.items():
+                    print(f"  - {name}: {model.model_identifier} (提供商: {model.api_provider})")
+            
+            # 检查任务配置
+            if hasattr(model_config, 'task_model_arg_map'):
+                print(f"任务配置数量: {len(model_config.task_model_arg_map)}")
+                for task_name, task_config in model_config.task_model_arg_map.items():
+                    print(f"  - {task_name}: {task_config}")
+                    
+            # 尝试初始化ModelManager
+            print("\n🔍 调试ModelManager初始化:")
+            from src.llm_models.model_manager import ModelManager
+            
+            try:
+                model_manager = ModelManager(model_config)
+                print("✅ ModelManager初始化成功")
+                
+                # 检查API客户端映射
+                print(f"API客户端数量: {len(model_manager.api_client_map)}")
+                for name, client in model_manager.api_client_map.items():
+                    print(f"  - {name}: {type(client).__name__}")
+                    if hasattr(client, 'client') and hasattr(client.client, 'api_key'):
+                        api_key = client.client.api_key
+                        print(f"    Client API key: {api_key[:10]}...{api_key[-5:] if len(api_key) > 15 else api_key}")
+                
+                # 尝试获取任务处理器
+                try:
+                    handler = model_manager["llm_normal"]
+                    print("✅ 成功获取llm_normal任务处理器")
+                    print(f"任务处理器类型: {type(handler).__name__}")
+                except Exception as e:
+                    print(f"❌ 获取任务处理器失败: {e}")
+                
+            except Exception as e:
+                print(f"❌ ModelManager初始化失败: {e}")
+                import traceback
+                traceback.print_exc()
+                
+        finally:
+            # 恢复配置
+            for provider in config['api_providers']:
+                provider['api_key'] = original_keys[provider['name']]
+            
+            with open(config_path, 'w', encoding='utf-8') as f:
+                toml.dump(config, f)
+            print("\n✅ 配置已恢复")
+        
+    except Exception as e:
+        print(f"❌ 调试失败: {e}")
+        import traceback
+        traceback.print_exc()
+
+if __name__ == "__main__":
+    debug_config_loading()