321 lines
11 KiB
Python
321 lines
11 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
创建「企业复杂编排」示例 Agent（单 DAG、无环）：
|
|||
|
|
|
|||
|
|
Start → Condition(用户 query 是否含触发标记)
|
|||
|
|
├─ True: Code(预检/标注) → LLM(多内置工具)
|
|||
|
|
└─ False: LLM(轻量直连)
|
|||
|
|
→ Merge → End
|
|||
|
|
|
|||
|
|
触发深度链路：在用户问题中任意位置包含 **[[深度]]**（不含引号），例如：
|
|||
|
|
「[[深度]]请用 http_request 拉取 https://httpbin.org/get 的 JSON 并摘要」
|
|||
|
|
未包含标记则走轻量分支（仍可调 LLM，但默认关闭工具）。
|
|||
|
|
|
|||
|
|
用法:
|
|||
|
|
cd backend && .\\venv\\Scripts\\python.exe scripts/create_complex_enterprise_agent.py
|
|||
|
|
USE_TESTCLIENT=1 # 不经 TCP，直接内存请求 FastAPI（推荐本机多实例抢端口时）
|
|||
|
|
|
|||
|
|
环境变量:
|
|||
|
|
PLATFORM_BASE_URL, PLATFORM_USERNAME, PLATFORM_PASSWORD
|
|||
|
|
COMPLEX_AGENT_NAME（默认 企业复杂编排_深度分流）
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
from typing import Any, Dict, List
|
|||
|
|
|
|||
|
|
DEEP_MARKER = "[[深度]]"
|
|||
|
|
DEFAULT_NAME = os.getenv("COMPLEX_AGENT_NAME", "企业复杂编排_深度分流")
|
|||
|
|
|
|||
|
|
TOOLS_DEEP = [
|
|||
|
|
"http_request",
|
|||
|
|
"text_analyze",
|
|||
|
|
"json_process",
|
|||
|
|
"datetime",
|
|||
|
|
"math_calculate",
|
|||
|
|
"file_read",
|
|||
|
|
"system_info",
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
PROMPT_DEEP = f"""你是企业级「深度分析」助手。用户消息中可能含有触发标记「{DEEP_MARKER}」，回答时请忽略该标记本身，专注实质需求。
|
|||
|
|
可调用工具完成：HTTP 请求、文本分析、JSON 处理、时间、数学、读文件、系统信息。按需调用，勿编造工具结果。
|
|||
|
|
最后用简洁中文总结结论。"""
|
|||
|
|
|
|||
|
|
PROMPT_FAST = """你是企业助手「快速通道」。用户未走深度编排；请直接、简洁地回答，避免冗长。若信息不足先追问一句。"""
|
|||
|
|
|
|||
|
|
CODE_PREFLIGHT = """
|
|||
|
|
d = input_data if isinstance(input_data, dict) else {}
|
|||
|
|
q = str(d.get("query", "") or "")
|
|||
|
|
result = dict(d)
|
|||
|
|
result["_deep_preflight"] = True
|
|||
|
|
result["_query_chars"] = len(q)
|
|||
|
|
result["_marker_detected"] = "[[深度]]" in q
|
|||
|
|
""".strip()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _sanitize_edges(edges: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
|
|||
|
|
seen: set = set()
|
|||
|
|
out: List[Dict[str, Any]] = []
|
|||
|
|
for e in edges or []:
|
|||
|
|
s, t = e.get("source"), e.get("target")
|
|||
|
|
if not s or not t or s == t:
|
|||
|
|
continue
|
|||
|
|
key = (s, t, e.get("sourceHandle") or "")
|
|||
|
|
if key in seen:
|
|||
|
|
continue
|
|||
|
|
seen.add(key)
|
|||
|
|
ne = dict(e)
|
|||
|
|
if not ne.get("targetHandle"):
|
|||
|
|
ne["targetHandle"] = "left"
|
|||
|
|
if not ne.get("id"):
|
|||
|
|
sh = ne.get("sourceHandle") or "r"
|
|||
|
|
ne["id"] = f"e_{s}_{t}_{sh}"
|
|||
|
|
out.append(ne)
|
|||
|
|
return out
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_complex_workflow() -> Dict[str, Any]:
|
|||
|
|
"""Condition 表达式与引擎一致：{{query}} contains \"...\" """
|
|||
|
|
cond_expr = f'{{query}} contains "{DEEP_MARKER}"'
|
|||
|
|
nodes: List[Dict[str, Any]] = [
|
|||
|
|
{"id": "start-1", "type": "start", "position": {"x": 40, "y": 260}, "data": {"label": "开始"}},
|
|||
|
|
{
|
|||
|
|
"id": "cond-1",
|
|||
|
|
"type": "condition",
|
|||
|
|
"position": {"x": 280, "y": 260},
|
|||
|
|
"data": {"label": "深度分流", "condition": cond_expr},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": "code-pf",
|
|||
|
|
"type": "code",
|
|||
|
|
"position": {"x": 520, "y": 120},
|
|||
|
|
"data": {
|
|||
|
|
"label": "预检/标注",
|
|||
|
|
"language": "python",
|
|||
|
|
"code": CODE_PREFLIGHT,
|
|||
|
|
"timeout": 15,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": "llm-deep",
|
|||
|
|
"type": "llm",
|
|||
|
|
"position": {"x": 760, "y": 120},
|
|||
|
|
"data": {
|
|||
|
|
"label": "深度 LLM+工具",
|
|||
|
|
"prompt": PROMPT_DEEP,
|
|||
|
|
"temperature": 0.25,
|
|||
|
|
"enable_tools": True,
|
|||
|
|
"tools": TOOLS_DEEP,
|
|||
|
|
"selected_tools": TOOLS_DEEP,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": "llm-fast",
|
|||
|
|
"type": "llm",
|
|||
|
|
"position": {"x": 520, "y": 400},
|
|||
|
|
"data": {
|
|||
|
|
"label": "快速 LLM",
|
|||
|
|
"prompt": PROMPT_FAST,
|
|||
|
|
"temperature": 0.35,
|
|||
|
|
"enable_tools": False,
|
|||
|
|
"tools": [],
|
|||
|
|
"selected_tools": [],
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": "merge-1",
|
|||
|
|
"type": "merge",
|
|||
|
|
"position": {"x": 1000, "y": 260},
|
|||
|
|
"data": {"label": "合并输出", "mode": "merge_all", "strategy": "object"},
|
|||
|
|
},
|
|||
|
|
{"id": "end-1", "type": "end", "position": {"x": 1240, "y": 260}, "data": {"label": "结束"}},
|
|||
|
|
]
|
|||
|
|
edges = _sanitize_edges(
|
|||
|
|
[
|
|||
|
|
{"source": "start-1", "target": "cond-1", "sourceHandle": "right", "targetHandle": "left"},
|
|||
|
|
{"source": "cond-1", "target": "code-pf", "sourceHandle": "true", "targetHandle": "left"},
|
|||
|
|
{"source": "cond-1", "target": "llm-fast", "sourceHandle": "false", "targetHandle": "left"},
|
|||
|
|
{"source": "code-pf", "target": "llm-deep", "sourceHandle": "right", "targetHandle": "left"},
|
|||
|
|
{"source": "llm-deep", "target": "merge-1", "sourceHandle": "right", "targetHandle": "left"},
|
|||
|
|
{"source": "llm-fast", "target": "merge-1", "sourceHandle": "right", "targetHandle": "left"},
|
|||
|
|
{"source": "merge-1", "target": "end-1", "sourceHandle": "right", "targetHandle": "left"},
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
return {"nodes": nodes, "edges": edges}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _validate_local(wf: Dict[str, Any]) -> None:
|
|||
|
|
from app.services.workflow_validator import validate_workflow
|
|||
|
|
|
|||
|
|
r = validate_workflow(wf.get("nodes") or [], wf.get("edges") or [])
|
|||
|
|
if not r.get("valid"):
|
|||
|
|
errs = r.get("errors") or []
|
|||
|
|
raise ValueError("工作流校验失败: " + "; ".join(errs))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _via_requests() -> int:
|
|||
|
|
import requests
|
|||
|
|
|
|||
|
|
base = os.getenv("PLATFORM_BASE_URL", "http://127.0.0.1:8037").rstrip("/")
|
|||
|
|
user = os.getenv("PLATFORM_USERNAME", "admin")
|
|||
|
|
pwd = os.getenv("PLATFORM_PASSWORD", "123456")
|
|||
|
|
wf = build_complex_workflow()
|
|||
|
|
_validate_local(wf)
|
|||
|
|
|
|||
|
|
lr = requests.post(
|
|||
|
|
f"{base}/api/v1/auth/login",
|
|||
|
|
data={"username": user, "password": pwd},
|
|||
|
|
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
|||
|
|
timeout=20,
|
|||
|
|
)
|
|||
|
|
if lr.status_code != 200:
|
|||
|
|
print("登录失败:", lr.status_code, lr.text[:500], file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
token = lr.json().get("access_token")
|
|||
|
|
if not token:
|
|||
|
|
print("无 access_token", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
h = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
|||
|
|
|
|||
|
|
def find_id() -> str | None:
|
|||
|
|
rr = requests.get(f"{base}/api/v1/agents", params={"search": DEFAULT_NAME, "limit": 50}, headers=h, timeout=30)
|
|||
|
|
if rr.status_code != 200:
|
|||
|
|
return None
|
|||
|
|
for a in rr.json() or []:
|
|||
|
|
if a.get("name") == DEFAULT_NAME:
|
|||
|
|
return a.get("id")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
desc = (
|
|||
|
|
f"复杂编排示例：条件分流（query 含「{DEEP_MARKER}」→ 代码预检 + 多工具 LLM），否则快速 LLM；Merge 汇总后结束。"
|
|||
|
|
" 适合演示画布、引擎分支与预算。"
|
|||
|
|
)
|
|||
|
|
budget = {"max_steps": 120, "max_llm_invocations": 6, "max_tool_calls": 48}
|
|||
|
|
existing = find_id()
|
|||
|
|
if existing:
|
|||
|
|
ur = requests.put(
|
|||
|
|
f"{base}/api/v1/agents/{existing}",
|
|||
|
|
headers=h,
|
|||
|
|
json={"description": desc, "workflow_config": wf, "budget_config": budget},
|
|||
|
|
timeout=120,
|
|||
|
|
)
|
|||
|
|
if ur.status_code != 200:
|
|||
|
|
print("更新失败:", ur.status_code, ur.text[:800], file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
aid = existing
|
|||
|
|
print("已更新:", DEFAULT_NAME, aid)
|
|||
|
|
else:
|
|||
|
|
cr = requests.post(
|
|||
|
|
f"{base}/api/v1/agents",
|
|||
|
|
headers=h,
|
|||
|
|
json={
|
|||
|
|
"name": DEFAULT_NAME,
|
|||
|
|
"description": desc,
|
|||
|
|
"workflow_config": wf,
|
|||
|
|
"budget_config": budget,
|
|||
|
|
},
|
|||
|
|
timeout=120,
|
|||
|
|
)
|
|||
|
|
if cr.status_code != 201:
|
|||
|
|
print("创建失败:", cr.status_code, cr.text[:800], file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
aid = cr.json()["id"]
|
|||
|
|
print("已创建:", DEFAULT_NAME, aid)
|
|||
|
|
|
|||
|
|
print(
|
|||
|
|
json.dumps(
|
|||
|
|
{
|
|||
|
|
"id": aid,
|
|||
|
|
"name": DEFAULT_NAME,
|
|||
|
|
"deep_marker": DEEP_MARKER,
|
|||
|
|
"hint": f"执行时在 query 中加入 {DEEP_MARKER} 可走深度分支",
|
|||
|
|
},
|
|||
|
|
ensure_ascii=False,
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _via_testclient() -> int:
|
|||
|
|
from fastapi.testclient import TestClient
|
|||
|
|
|
|||
|
|
from app.main import app
|
|||
|
|
|
|||
|
|
wf = build_complex_workflow()
|
|||
|
|
_validate_local(wf)
|
|||
|
|
|
|||
|
|
c = TestClient(app)
|
|||
|
|
lr = c.post(
|
|||
|
|
"/api/v1/auth/login",
|
|||
|
|
data={
|
|||
|
|
"username": os.getenv("PLATFORM_USERNAME", "admin"),
|
|||
|
|
"password": os.getenv("PLATFORM_PASSWORD", "123456"),
|
|||
|
|
},
|
|||
|
|
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
|||
|
|
)
|
|||
|
|
if lr.status_code != 200:
|
|||
|
|
print("login:", lr.status_code, lr.text[:400], file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
token = lr.json().get("access_token")
|
|||
|
|
h = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
|||
|
|
|
|||
|
|
gr = c.get("/api/v1/agents", params={"search": DEFAULT_NAME, "limit": 50}, headers=h)
|
|||
|
|
existing = None
|
|||
|
|
if gr.status_code == 200:
|
|||
|
|
for a in gr.json() or []:
|
|||
|
|
if a.get("name") == DEFAULT_NAME:
|
|||
|
|
existing = a.get("id")
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
desc = (
|
|||
|
|
f"复杂编排示例：条件分流（query 含「{DEEP_MARKER}」→ 代码预检 + 多工具 LLM），否则快速 LLM；Merge 汇总后结束。"
|
|||
|
|
)
|
|||
|
|
budget = {"max_steps": 120, "max_llm_invocations": 6, "max_tool_calls": 48}
|
|||
|
|
if existing:
|
|||
|
|
ur = c.put(
|
|||
|
|
f"/api/v1/agents/{existing}",
|
|||
|
|
headers=h,
|
|||
|
|
json={"description": desc, "workflow_config": wf, "budget_config": budget},
|
|||
|
|
)
|
|||
|
|
if ur.status_code != 200:
|
|||
|
|
print("put:", ur.status_code, ur.text[:800], file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
aid = existing
|
|||
|
|
print("已更新(TestClient):", DEFAULT_NAME, aid)
|
|||
|
|
else:
|
|||
|
|
cr = c.post(
|
|||
|
|
"/api/v1/agents",
|
|||
|
|
headers=h,
|
|||
|
|
json={
|
|||
|
|
"name": DEFAULT_NAME,
|
|||
|
|
"description": desc,
|
|||
|
|
"workflow_config": wf,
|
|||
|
|
"budget_config": budget,
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
if cr.status_code != 201:
|
|||
|
|
print("post:", cr.status_code, cr.text[:800], file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
aid = cr.json()["id"]
|
|||
|
|
print("已创建(TestClient):", DEFAULT_NAME, aid)
|
|||
|
|
|
|||
|
|
print(
|
|||
|
|
json.dumps(
|
|||
|
|
{"id": aid, "name": DEFAULT_NAME, "deep_marker": DEEP_MARKER},
|
|||
|
|
ensure_ascii=False,
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
if os.getenv("USE_TESTCLIENT", "").strip() in ("1", "true", "yes"):
|
|||
|
|
return _via_testclient()
|
|||
|
|
return _via_requests()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
raise SystemExit(main())
|