83 lines
3.5 KiB
Python
83 lines
3.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""系统提示词分层装配 - 端到端测试"""
|
|||
|
|
import sys, os, io, json, requests
|
|||
|
|
|
|||
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", errors="replace")
|
|||
|
|
|
|||
|
|
BASE = "http://localhost:8038/api/v1"
|
|||
|
|
|
|||
|
|
def login():
|
|||
|
|
r = requests.post(f"{BASE}/auth/login", data={"username": "admin", "password": "123456"})
|
|||
|
|
r.raise_for_status()
|
|||
|
|
return r.json()["access_token"]
|
|||
|
|
|
|||
|
|
def test_prompt_sections():
|
|||
|
|
token = login()
|
|||
|
|
headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
|||
|
|
print("=== 系统提示词分层装配测试 ===\n")
|
|||
|
|
|
|||
|
|
# Test 1: Chat with prompt sections enabled (default)
|
|||
|
|
print("1. 对话（分层装配 开启）…")
|
|||
|
|
r1 = requests.post(f"{BASE}/agent-chat/bare", headers=headers,
|
|||
|
|
json={"message": "你好，请用简短回答介绍一下自己",
|
|||
|
|
"prompt_sections_enabled": True}, timeout=120)
|
|||
|
|
r1.raise_for_status()
|
|||
|
|
d1 = r1.json()
|
|||
|
|
print(f" 会话 ID: {d1['session_id']}")
|
|||
|
|
print(f" 回复: {d1['content'][:200]}…")
|
|||
|
|
print(f" 迭代: {d1['iterations_used']}, 工具调用: {d1['tool_calls_made']}")
|
|||
|
|
ok1 = d1["iterations_used"] >= 1
|
|||
|
|
print(f" {'✓ 通过' if ok1 else '✗ 失败'}\n")
|
|||
|
|
|
|||
|
|
# Test 2: Chat with prompt sections disabled
|
|||
|
|
print("2. 对话（分层装配 关闭）…")
|
|||
|
|
r2 = requests.post(f"{BASE}/agent-chat/bare", headers=headers,
|
|||
|
|
json={"message": "你好，请用简短回答介绍一下自己",
|
|||
|
|
"prompt_sections_enabled": False}, timeout=120)
|
|||
|
|
r2.raise_for_status()
|
|||
|
|
d2 = r2.json()
|
|||
|
|
print(f" 会话 ID: {d2['session_id']}")
|
|||
|
|
print(f" 回复: {d2['content'][:200]}…")
|
|||
|
|
print(f" 迭代: {d2['iterations_used']}, 工具调用: {d2['tool_calls_made']}")
|
|||
|
|
ok2 = d2["iterations_used"] >= 1
|
|||
|
|
print(f" {'✓ 通过' if ok2 else '✗ 失败'}\n")
|
|||
|
|
|
|||
|
|
# Test 3: Agent chat with prompt sections
|
|||
|
|
print("3. Agent 对话（分层装配 开启）…")
|
|||
|
|
# First find an agent
|
|||
|
|
r_agents = requests.get(f"{BASE}/agents", headers=headers, params={"limit": 5}, timeout=10)
|
|||
|
|
r_agents.raise_for_status()
|
|||
|
|
agents_data = r_agents.json()
|
|||
|
|
agents = agents_data if isinstance(agents_data, list) else agents_data.get("items", [])
|
|||
|
|
if agents:
|
|||
|
|
agent_id = agents[0]["id"]
|
|||
|
|
r3 = requests.post(f"{BASE}/agent-chat/{agent_id}", headers=headers,
|
|||
|
|
json={"message": "你好", "prompt_sections_enabled": True}, timeout=120)
|
|||
|
|
r3.raise_for_status()
|
|||
|
|
d3 = r3.json()
|
|||
|
|
print(f" Agent: {agents[0].get('name', 'N/A')}")
|
|||
|
|
print(f" 回复: {d3['content'][:200]}…")
|
|||
|
|
ok3 = d3["iterations_used"] >= 1
|
|||
|
|
print(f" {'✓ 通过' if ok3 else '✗ 失败'}")
|
|||
|
|
else:
|
|||
|
|
print(" ⚠ 跳过（无可用 Agent）")
|
|||
|
|
ok3 = True
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# Test 4: Verify system prompt composition via health endpoint
|
|||
|
|
print("4. 验证分层装配模块加载…")
|
|||
|
|
r4 = requests.get(f"http://localhost:8038/health", timeout=10)
|
|||
|
|
r4.raise_for_status()
|
|||
|
|
d4 = r4.json()
|
|||
|
|
print(f" 状态: {d4['status']}")
|
|||
|
|
print(f" 工具数: {d4.get('builtin_tools', {}).get('count', 'N/A')}")
|
|||
|
|
ok4 = d4["status"] == "healthy"
|
|||
|
|
print(f" {'✓ 通过' if ok4 else '✗ 失败'}\n")
|
|||
|
|
|
|||
|
|
all_ok = ok1 and ok2 and ok3 and ok4
|
|||
|
|
print(f"=== {'全部测试通过' if all_ok else '部分测试失败'} ===")
|
|||
|
|
return all_ok
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_prompt_sections()
|