52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""
|
||
简单测试 LLM 模型服务的提供商检测功能
|
||
"""
|
||
import sys
|
||
import os
|
||
|
||
# 添加backend目录到路径
|
||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'backend'))
|
||
|
||
# 直接导入服务模块，避免导入__init__.py
|
||
from services.llm_model_service import LLMModelService
|
||
|
||
|
||
def test_detect_provider():
|
||
"""测试提供商检测功能"""
|
||
print("=" * 60)
|
||
print("测试提供商检测功能")
|
||
print("=" * 60)
|
||
|
||
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"),
|
||
("https://custom-api.example.com/v1", "openai"),
|
||
]
|
||
|
||
all_passed = True
|
||
for url, expected in test_cases:
|
||
result = LLMModelService.detect_provider(url)
|
||
passed = result == expected
|
||
status = "✓" if passed else "✗"
|
||
print(f" {status} {url}")
|
||
print(f" 结果: {result}, 期望: {expected}")
|
||
if not passed:
|
||
all_passed = False
|
||
|
||
print("=" * 60)
|
||
if all_passed:
|
||
print("所有测试通过！✓")
|
||
else:
|
||
print("部分测试失败！✗")
|
||
print("=" * 60)
|
||
|
||
return all_passed
|
||
|
||
|
||
if __name__ == "__main__":
|
||
test_detect_provider()
|