91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
"""
|
||
Coze 智能体 API 调用测试脚本
|
||
运行前请确保已配置 .env 文件
|
||
"""
|
||
from coze_client import CozeClient
|
||
|
||
|
||
def test_basic_chat():
|
||
"""基础对话测试"""
|
||
print("=" * 50)
|
||
print("🤖 大沃智能助手 - API 调用测试")
|
||
print("=" * 50)
|
||
|
||
try:
|
||
client = CozeClient()
|
||
except ValueError as e:
|
||
print(f"\n❌ 初始化失败: {e}")
|
||
print("请先配置 .env 文件（参考 .env.example）")
|
||
return
|
||
|
||
# 测试用例
|
||
test_messages = [
|
||
{
|
||
"message": "帮我写一段招商文案",
|
||
"variables": {"zishu": "200字以内", "fengge": "专业正式"},
|
||
"description": "测试招商Agent",
|
||
},
|
||
{
|
||
"message": "早上好",
|
||
"variables": {"zishu": "100字以内", "fengge": "轻松活泼"},
|
||
"description": "测试问候Agent",
|
||
},
|
||
{
|
||
"message": "介绍一下公司",
|
||
"variables": {"zishu": "300字以内", "fengge": "专业正式"},
|
||
"description": "测试公司介绍Agent",
|
||
},
|
||
]
|
||
|
||
for i, test in enumerate(test_messages, 1):
|
||
print(f"\n--- 测试 {i}: {test['description']} ---")
|
||
print(f"📝 输入: {test['message']}")
|
||
print(f"⚙️ 参数: 字数={test['variables']['zishu']}, 风格={test['variables']['fengge']}")
|
||
print("⏳ 等待回复中...")
|
||
|
||
reply = client.chat_and_poll(
|
||
message=test["message"],
|
||
custom_variables=test["variables"],
|
||
)
|
||
|
||
print(f"💬 回复:\n{reply}")
|
||
print()
|
||
|
||
|
||
def test_interactive():
|
||
"""交互式对话测试"""
|
||
print("=" * 50)
|
||
print("🤖 大沃智能助手 - 交互模式")
|
||
print("输入 'quit' 退出")
|
||
print("=" * 50)
|
||
|
||
try:
|
||
client = CozeClient()
|
||
except ValueError as e:
|
||
print(f"\n❌ 初始化失败: {e}")
|
||
return
|
||
|
||
while True:
|
||
user_input = input("\n你: ").strip()
|
||
if user_input.lower() in ("quit", "exit", "q"):
|
||
print("再见！")
|
||
break
|
||
if not user_input:
|
||
continue
|
||
|
||
print("⏳ 思考中...")
|
||
reply = client.chat_and_poll(
|
||
message=user_input,
|
||
custom_variables={"zishu": "200字以内", "fengge": "轻松活泼"},
|
||
)
|
||
print(f"\n助手: {reply}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import sys
|
||
|
||
if len(sys.argv) > 1 and sys.argv[1] == "--interactive":
|
||
test_interactive()
|
||
else:
|
||
test_basic_chat()
|