完成世界书、骰子、apiconfig页面处理

test_llm_model_service.py

@@ -0,0 +1,66 @@
+"""
+测试 LLM 模型服务
+"""
+import sys
+sys.path.insert(0, 'backend')
+
+from backend.services.llm_model_service import LLMModelService
+
+
+def test_detect_provider():
+    """测试提供商检测功能"""
+    print("测试提供商检测...")
+    
+    test_cases = [
+        ("https://api.openai.com/v1", "openai"),
+        ("https://api.anthropic.com/v1", "anthropic"),
+        ("http://localhost:11434", "ollama"),
+        ("http://127.0.0.1:11434", "ollama"),
+        ("https://api.siliconflow.cn/v1", "openai"),
+        ("https://api.deepseek.com/v1", "openai"),
+    ]
+    
+    for url, expected in test_cases:
+        result = LLMModelService.detect_provider(url)
+        status = "✓" if result == expected else "✗"
+        print(f"  {status} {url} -> {result} (期望: {expected})")
+
+
+def test_get_openai_models():
+    """测试获取 OpenAI 模型列表（需要有效的 API Key）"""
+    print("\n测试 OpenAI 模型列表获取...")
+    
+    # 这里需要一个有效的 API Key 才能测试
+    api_key = "sk-test-key"  # 替换为实际的 API Key
+    try:
+        models = LLMModelService.get_openai_models(api_key)
+        print(f"  ✓ 成功获取 {len(models)} 个模型")
+        print(f"  前5个模型: {models[:5]}")
+    except Exception as e:
+        print(f"  ✗ 失败: {str(e)}")
+
+
+def test_get_ollama_models():
+    """测试获取 Ollama 模型列表（需要本地运行 Ollama）"""
+    print("\n测试 Ollama 模型列表获取...")
+    
+    try:
+        models = LLMModelService.get_ollama_models()
+        print(f"  ✓ 成功获取 {len(models)} 个模型")
+        print(f"  模型列表: {models}")
+    except Exception as e:
+        print(f"  ✗ 失败: {str(e)}")
+
+
+if __name__ == "__main__":
+    print("=" * 60)
+    print("LLM 模型服务测试")
+    print("=" * 60)
+    
+    test_detect_provider()
+    # test_get_openai_models()  # 需要有效的 API Key
+    # test_get_ollama_models()  # 需要本地运行 Ollama
+    
+    print("\n" + "=" * 60)
+    print("测试完成")
+    print("=" * 60)