57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
测试AI服务配置
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import asyncio
|
|||
|
|
from ai_service import AIService, get_ai_config
|
|||
|
|
|
|||
|
|
async def test_ai_connection():
|
|||
|
|
"""测试AI连接"""
|
|||
|
|
print("测试AI服务配置...")
|
|||
|
|
|
|||
|
|
# 获取配置
|
|||
|
|
config = get_ai_config()
|
|||
|
|
print(f"AI提供商: {config.provider}")
|
|||
|
|
print(f"模型: {config.model}")
|
|||
|
|
print(f"Base URL: {config.base_url}")
|
|||
|
|
print(f"API Key: {config.api_key[:10]}..." if config.api_key else "未配置")
|
|||
|
|
|
|||
|
|
# 测试连接
|
|||
|
|
try:
|
|||
|
|
async with AIService(config) as ai_service:
|
|||
|
|
# 简单的测试提示
|
|||
|
|
test_prompt = "请回答：1+1等于几？"
|
|||
|
|
|
|||
|
|
if config.provider == "qwen":
|
|||
|
|
response = await ai_service._call_qwen(test_prompt)
|
|||
|
|
elif config.provider == "openai":
|
|||
|
|
response = await ai_service._call_openai(test_prompt)
|
|||
|
|
else:
|
|||
|
|
response = "暂不支持该提供商的测试"
|
|||
|
|
|
|||
|
|
print(f"\nAI响应: {response}")
|
|||
|
|
print("\n✅ AI服务配置测试成功！")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n❌ 测试失败: {str(e)}")
|
|||
|
|
print("\n请检查：")
|
|||
|
|
print("1. API Key是否正确")
|
|||
|
|
print("2. 网络连接是否正常")
|
|||
|
|
print("3. Base URL是否正确")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 设置环境变量（如果未设置）
|
|||
|
|
if not os.getenv("AI_PROVIDER"):
|
|||
|
|
os.environ["AI_PROVIDER"] = "qwen"
|
|||
|
|
os.environ["QWEN_API_KEY"] = "sk-9bf015da696048eea231085615892b68"
|
|||
|
|
os.environ["QWEN_BASE_URL"] = "https://dashscope.aliyuncs.com/compatible-mode/v1"
|
|||
|
|
|
|||
|
|
# 加载.env文件（如果存在）
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
load_dotenv()
|
|||
|
|
|
|||
|
|
asyncio.run(test_ai_connection())
|