60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
简单测试：验证 Baosi API 是否正常工作
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
# 添加项目路径
|
|||
|
|
project_root = Path(__file__).resolve().parent.parent
|
|||
|
|
sys.path.insert(0, str(project_root))
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
|
|||
|
|
# 加载环境变量
|
|||
|
|
load_dotenv(project_root / ".env")
|
|||
|
|
|
|||
|
|
from backend.app.model_services.chat_services import BaosiChatProvider
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def test_baosi_provider():
|
|||
|
|
"""测试 Baosi API Provider"""
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("测试 Baosi API Provider")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 创建 provider
|
|||
|
|
provider = BaosiChatProvider()
|
|||
|
|
|
|||
|
|
# 检查是否可用
|
|||
|
|
print(f"\n检查是否可用: {provider.is_available()}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取 LLM
|
|||
|
|
llm = provider.get_service()
|
|||
|
|
print(f"\n✓ LLM 获取成功: {type(llm)}")
|
|||
|
|
|
|||
|
|
# 测试简单调用
|
|||
|
|
print(f"\n测试简单调用...")
|
|||
|
|
from langchain_core.messages import HumanMessage
|
|||
|
|
response = await llm.ainvoke([
|
|||
|
|
HumanMessage(content="你好，请简单介绍一下你自己")
|
|||
|
|
])
|
|||
|
|
print(f"\n✓ 响应成功:")
|
|||
|
|
print(f" 响应类型: {type(response)}")
|
|||
|
|
print(f" 响应内容:\n{response.content}")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n✗ 测试失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
print(f"堆栈:\n{traceback.format_exc()}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(test_baosi_provider())
|