58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
"""
|
|
测试脚本 - 获取Agent状态
|
|
"""
|
|
import requests
|
|
import json
|
|
import sys
|
|
|
|
# 配置
|
|
BASE_URL = "http://localhost:8000"
|
|
|
|
|
|
def test_get_agent_status():
|
|
"""测试获取Agent状态"""
|
|
print("=" * 50)
|
|
print("测试: 获取AI Agent状态")
|
|
print("=" * 50)
|
|
|
|
# 测试的Agent名称列表
|
|
agent_names = ["test-echo-1", "test-chat-1", "test-code-1"]
|
|
|
|
for agent_name in agent_names:
|
|
print(f"\n查询Agent: {agent_name}")
|
|
|
|
try:
|
|
response = requests.get(f"{BASE_URL}/agents/{agent_name}/status")
|
|
|
|
print(f"状态码: {response.status_code}")
|
|
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
print("✅ 查询成功!")
|
|
print(f"响应: {json.dumps(result, indent=2, ensure_ascii=False)}")
|
|
elif response.status_code == 404:
|
|
print(f"⚠️ Agent不存在")
|
|
print(f"错误: {response.text}")
|
|
else:
|
|
print(f"❌ 查询失败!")
|
|
print(f"错误: {response.text}")
|
|
|
|
except Exception as e:
|
|
print(f"❌ 请求失败: {str(e)}")
|
|
|
|
print("-" * 50)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
# 先检查服务是否可用
|
|
print("检查服务状态...")
|
|
response = requests.get(f"{BASE_URL}/")
|
|
print(f"服务状态: {response.json()}\n")
|
|
|
|
test_get_agent_status()
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
print("❌ 无法连接到服务。请确保服务正在运行: python app.py")
|
|
sys.exit(1)
|