112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
模型管理器测试脚本
|
||
用于验证新模型管理功能的正确性
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
|
||
# 添加项目根目录到Python路径
|
||
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from model_manager import model_router, OpenAIProvider, OpenRouterProvider, SiliconFlowProvider
|
||
from config import settings
|
||
|
||
def test_model_router():
|
||
"""测试模型路由器功能"""
|
||
print("=== 测试模型路由器功能 ===")
|
||
|
||
# 显示所有可用模型
|
||
print("可用模型:")
|
||
for model in model_router.list_models():
|
||
print(f" - {model}")
|
||
|
||
# 显示所有提供商
|
||
print("\n可用提供商:")
|
||
for provider in model_router.list_providers():
|
||
print(f" - {provider}")
|
||
|
||
# 测试模型路由
|
||
print("\n测试模型路由:")
|
||
test_models = ["gpt-3.5-turbo", "text-embedding-ada-002"]
|
||
for model in test_models:
|
||
provider = model_router.get_provider_for_model(model)
|
||
if provider:
|
||
print(f" {model} -> {provider.name}")
|
||
else:
|
||
print(f" {model} -> 未找到提供商")
|
||
|
||
print("\n=== 测试完成 ===")
|
||
|
||
def test_openai_provider():
|
||
"""测试OpenAI提供商"""
|
||
print("\n=== 测试OpenAI提供商 ===")
|
||
|
||
if settings.OPENAI_API_KEY:
|
||
provider = OpenAIProvider(
|
||
name="openai_test",
|
||
api_key=settings.OPENAI_API_KEY,
|
||
base_url=settings.OPENAI_API_BASE or "https://api.openai.com/v1",
|
||
models=["gpt-3.5-turbo"]
|
||
)
|
||
|
||
try:
|
||
# 测试文本生成
|
||
messages = [
|
||
{"role": "user", "content": "你好，这是一个测试消息"}
|
||
]
|
||
response = provider.generate_text("gpt-3.5-turbo", messages, max_tokens=50)
|
||
print(f"文本生成测试通过: {response[:50]}...")
|
||
|
||
# 测试嵌入获取
|
||
texts = ["这是一个测试文本"]
|
||
embeddings = provider.get_embeddings("text-embedding-ada-002", texts)
|
||
print(f"嵌入获取测试通过: 维度 {len(embeddings[0])}")
|
||
|
||
except Exception as e:
|
||
print(f"OpenAI提供商测试失败: {e}")
|
||
else:
|
||
print("未配置OpenAI API密钥，跳过测试")
|
||
|
||
print("=== OpenAI提供商测试完成 ===")
|
||
|
||
def test_config_loading():
|
||
"""测试配置加载功能"""
|
||
print("\n=== 测试配置加载功能 ===")
|
||
|
||
# 检查模型配置文件是否存在
|
||
config_file = "model_config.json"
|
||
if os.path.exists(config_file):
|
||
try:
|
||
with open(config_file, 'r', encoding='utf-8') as f:
|
||
config = json.load(f)
|
||
print("模型配置文件加载成功")
|
||
print(f"配置文件包含 {len(config.get('providers', []))} 个提供商")
|
||
except Exception as e:
|
||
print(f"模型配置文件加载失败: {e}")
|
||
else:
|
||
print("模型配置文件不存在")
|
||
|
||
print("=== 配置加载测试完成 ===")
|
||
|
||
def main():
|
||
"""主测试函数"""
|
||
print("开始测试模型管理功能...")
|
||
|
||
# 测试配置加载
|
||
test_config_loading()
|
||
|
||
# 测试模型路由器
|
||
test_model_router()
|
||
|
||
# 测试OpenAI提供商
|
||
test_openai_provider()
|
||
|
||
print("\n所有测试完成!")
|
||
|
||
if __name__ == "__main__":
|
||
main() |