67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
"""
|
||
测试 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)
|