362 lines
13 KiB
Python
362 lines
13 KiB
Python
# create_agents.py
|
||
import requests
|
||
import urllib3
|
||
from uuid import uuid4
|
||
|
||
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
||
|
||
BASE_URL = "http://192.168.0.46:22000"
|
||
AGENT_API = f"{BASE_URL}/api/v1/agents/"
|
||
HEADERS = {
|
||
"Accept": "application/json",
|
||
"Content-Type": "application/json",
|
||
}
|
||
|
||
TIMEOUT = 10
|
||
OUTPUT_FILE = "agent_ids.txt"
|
||
|
||
AGENT_META = {
|
||
str(uuid4()): {
|
||
"icon_index": 1,
|
||
"name": "修理工程管理",
|
||
"description": "对舰船修理工程实施全过程管理，包括任务分解、进度控制、工序协调和资源调配。",
|
||
"prompt": "助理员您好，我是修理工程管理助手，我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。我可以为您提供舰船设备的标准修理内容范围、作业工序清单以及安全施工要求，请直接告诉我您需要查询的设备名称。",
|
||
"default_query": [
|
||
{"title": "查询设备标准 →", "content": "查询设备标准"},
|
||
{"title": "查看作业工序 →", "content": "查看作业工序"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 2,
|
||
"name": "器材统计与分析",
|
||
"description": "对舰船维修保障过程中涉及的器材进行统计分析与趋势研判。",
|
||
"prompt": "助理员您好，我是器材统计与分析助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。为您提供多维度的器材消耗统计与库存分析服务。请告诉我您需要统计的具体业务场景。",
|
||
"default_query": [
|
||
{"title": "统计第一季度器材消耗 →", "content": "统计第一季度器材消耗"},
|
||
{"title": "分析当前库存周转率 →", "content": "分析当前库存周转率"},
|
||
{"title": "查看特定型号历史记录 →", "content": "查看特定型号历史记录"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 3,
|
||
"name": "修理法规知识查询",
|
||
"description": "提供舰船修理相关法规、标准和制度的查询与解读。",
|
||
"prompt": "您好！我是修理法规知识查询助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我需要查询的主题。",
|
||
"default_query": [
|
||
{"title": "查询焊接工艺规程 →", "content": "查询焊接工艺规程"},
|
||
{"title": "查看设备拆装安全规程 →", "content": "查看设备拆装安全规程"},
|
||
{"title": "搜索环保作业规定 →", "content": "搜索环保作业规定"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 4,
|
||
"name": "船舰知识查询",
|
||
"description": "提供舰船结构、系统组成与功能原理等知识。",
|
||
"prompt": "您好！我是舰船知识查询助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我需要了解的主题。",
|
||
"default_query": [
|
||
{"title": "了解舰船电力系统组成 →", "content": "了解舰船电力系统组成"},
|
||
{"title": "查询舰船推进装置类型 →", "content": "查询舰船推进装置类型"},
|
||
{"title": "搜索舰船雷达系统原理 →", "content": "搜索舰船雷达系统原理"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 5,
|
||
"name": "故障排查与修理",
|
||
"description": "辅助进行舰船设备故障分析、原因定位与修理方案制定。",
|
||
"prompt": "您好！我是智能故障排查与修理助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。提供舷号、设备名称和故障现象，我可以协助您进行复杂故障的研判诊断。",
|
||
"default_query": [
|
||
{"title": "多模态输出与校验 →", "content": "多模态输出与校验"},
|
||
{"title": "诊断研判与可能性排序 →", "content": "诊断研判与可能性排序"},
|
||
{"title": "GT-25000燃气轮机的日常维护保养方案 →", "content": "GT-25000燃气轮机的日常维护保养方案"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 6,
|
||
"name": "舰船机电知识",
|
||
"description": "提供舰船动力、电气、液压等机电系统相关知识。",
|
||
"prompt": "机电长您好，我是舰船机电知识查询助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我需了解的主题。",
|
||
"default_query": [
|
||
{"title": "了解主推进系统构成 →", "content": "了解主推进系统构成"},
|
||
{"title": "查看舰艇液压机参数 →", "content": "查看舰艇液压机参数"},
|
||
{"title": "查看舰机液压系统参数 →", "content": "查看舰机液压系统参数"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 7,
|
||
"name": "现场维修作业向导",
|
||
"description": "为一线维修人员提供现场操作流程与安全指导。",
|
||
"prompt": "您好！我是现场维修作业向导。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。您可以询问具体的维修步骤、拆装工艺或查询安全操作规程。",
|
||
"default_query": [
|
||
{"title": "离心泵汽蚀的典型特征 →", "content": "离心泵汽蚀的典型特征"},
|
||
{"title": "海水泵汽蚀的维修步骤 →", "content": "海水泵汽蚀的维修步骤"},
|
||
{"title": "GT-25000燃气轮机的日常维护步骤 →", "content": "GT-25000燃气轮机的日常维护步骤"}
|
||
],
|
||
},
|
||
str(uuid4()): {
|
||
"icon_index": 8,
|
||
"name": "舰船技术知识百科",
|
||
"description": "综合性舰船技术知识库。",
|
||
"prompt": "您好，我是舰船技术知识百科助手。我已经绑定了技术资料、通用知识、故障知识、法规标准知识库。请告诉我您需要查询的技术主题。",
|
||
"default_query": [
|
||
{"title": "了解船体高强钢性能 →", "content": "了解船体高强钢性能"},
|
||
{"title": "查询焊接工艺评定标准 →", "content": "查询焊接工艺评定标准"},
|
||
{"title": "查看防腐涂装技术规范 →", "content": "查看防腐涂装技术规范"}
|
||
],
|
||
},
|
||
}
|
||
|
||
|
||
def build_agent_json(agent_id: str) -> dict:
|
||
meta = AGENT_META[agent_id]
|
||
return {
|
||
"id": agent_id,
|
||
"name": meta["name"],
|
||
"description": meta["description"],
|
||
"prompt": meta["prompt"],
|
||
"background": "string",
|
||
"access_control": {},
|
||
"default_query": meta["default_query"],
|
||
}
|
||
|
||
|
||
def create_agent(agent_id: str) -> bool:
|
||
resp = requests.post(
|
||
AGENT_API,
|
||
headers=HEADERS,
|
||
json=build_agent_json(agent_id),
|
||
verify=False,
|
||
timeout=TIMEOUT,
|
||
)
|
||
print(resp.json())
|
||
print(f"[CREATE] {agent_id} -> {resp.status_code}")
|
||
|
||
return resp.status_code in (200, 201, 409)
|
||
|
||
|
||
def main():
|
||
created_ids = []
|
||
|
||
for agent_id in AGENT_META.keys():
|
||
if create_agent(agent_id):
|
||
created_ids.append(agent_id)
|
||
|
||
with open(OUTPUT_FILE, "w", encoding="utf-8") as f:
|
||
for aid in created_ids:
|
||
f.write(aid + "\n")
|
||
|
||
print(f"\n✅ 已写入 {len(created_ids)} 个 Agent UUID 到 {OUTPUT_FILE}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
|
||
|
||
|
||
|
||
# import requests
|
||
# import urllib3
|
||
# from uuid import uuid4
|
||
# from typing import List, Tuple
|
||
|
||
# # =========================
|
||
# # 0. 关闭 HTTPS 警告（开发环境）
|
||
# # =========================
|
||
# urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
||
|
||
# # =========================
|
||
# # 1. 基础配置
|
||
# # =========================
|
||
|
||
# BASE_URL = "https://devb.zdht.hjkl01.cn:65432"
|
||
|
||
# AGENT_API = f"{BASE_URL}/api/v1/agents/"
|
||
# ROLE_BIND_API = f"{BASE_URL}/api/v1/agent-roles/association"
|
||
|
||
# HEADERS = {
|
||
# "Accept": "application/json",
|
||
# "Content-Type": "application/json",
|
||
# }
|
||
|
||
# TIMEOUT = 10
|
||
|
||
# # =========================
|
||
# # 2. 角色 ID（数据库真实存在）
|
||
# # =========================
|
||
|
||
# AGENT_ROLE_1 = "2" # 助理员助手 (1-4)
|
||
# AGENT_ROLE_2 = "3" # 机电长助手 (5-6)
|
||
# AGENT_ROLE_3 = "4" # 维修员助手 (7-8)
|
||
|
||
# ROLE_NAME_MAP = {
|
||
# AGENT_ROLE_1: "助理员助手",
|
||
# AGENT_ROLE_2: "机电长助手",
|
||
# AGENT_ROLE_3: "维修员助手",
|
||
# }
|
||
|
||
# # =========================
|
||
# # 3. Agent 元数据
|
||
# # =========================
|
||
|
||
# AGENT_META = {
|
||
# str(uuid4()): {
|
||
# "icon_index": 1,
|
||
# "name": "修理工程管理",
|
||
# "description": "对舰船修理工程实施全过程管理，包括任务分解、进度控制、工序协调和资源调配。",
|
||
# "prompt": "您好！我是一名舰船修理工程管理专家。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 2,
|
||
# "name": "器材统计与分析",
|
||
# "description": "对舰船维修保障过程中涉及的器材进行统计分析与趋势研判。",
|
||
# "prompt": "您好！我是一名舰船器材统计与分析专家。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 3,
|
||
# "name": "修理法规知识查询",
|
||
# "description": "提供舰船修理相关法规、标准和制度的查询与解读。",
|
||
# "prompt": "您好！我是一名舰船修理法规专家。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 4,
|
||
# "name": "船舰知识查询",
|
||
# "description": "提供舰船结构、系统组成与功能原理等知识。",
|
||
# "prompt": "您好！我是一名舰船专业知识顾问。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 5,
|
||
# "name": "故障排查与修理",
|
||
# "description": "辅助进行舰船设备故障分析、原因定位与修理方案制定。",
|
||
# "prompt": "您好！我是一名舰船设备故障诊断专家。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 6,
|
||
# "name": "舰船机电知识",
|
||
# "description": "提供舰船动力、电气、液压等机电系统相关知识。",
|
||
# "prompt": "您好！我是一名舰船机电系统专家。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 7,
|
||
# "name": "现场维修作业向导",
|
||
# "description": "为一线维修人员提供现场操作流程与安全指导。",
|
||
# "prompt": "您好！我是一名现场维修作业向导。",
|
||
# },
|
||
# str(uuid4()): {
|
||
# "icon_index": 8,
|
||
# "name": "舰船技术知识百科",
|
||
# "description": "综合性舰船技术知识库。",
|
||
# "prompt": "您好！我是一部舰船技术知识百科。",
|
||
# },
|
||
# }
|
||
|
||
# # =========================
|
||
# # 4. 构建 Agent JSON
|
||
# # =========================
|
||
|
||
# def build_agent_json(agent_id: str) -> dict:
|
||
# meta = AGENT_META[agent_id]
|
||
# return {
|
||
# "id": agent_id,
|
||
# "name": meta["name"],
|
||
# "description": meta["description"],
|
||
# "prompt": meta["prompt"],
|
||
# "background": "string",
|
||
# "access_control": {},
|
||
# }
|
||
|
||
# # =========================
|
||
# # 5. 创建 Agent
|
||
# # =========================
|
||
|
||
# def create_agent(agent_id: str) -> bool:
|
||
# resp = requests.post(
|
||
# AGENT_API,
|
||
# headers=HEADERS,
|
||
# json=build_agent_json(agent_id),
|
||
# verify=False,
|
||
# timeout=TIMEOUT,
|
||
# )
|
||
|
||
# print(f"[CREATE] Agent {agent_id} -> {resp.status_code}")
|
||
|
||
# if resp.status_code in (200, 201, 409):
|
||
# return True
|
||
|
||
# print(" ↳ 返回值:", resp.text)
|
||
# return False
|
||
|
||
# # =========================
|
||
# # 6. 绑定 Agent 到角色
|
||
# # =========================
|
||
|
||
# def bind_agent_to_role(agent_id: str, role_id: str) -> bool:
|
||
# payload = [
|
||
# {
|
||
# "agent_id": agent_id,
|
||
# "agent_role_id": role_id,
|
||
# "is_active": True,
|
||
# }
|
||
# ]
|
||
|
||
# resp = requests.post(
|
||
# ROLE_BIND_API,
|
||
# headers=HEADERS,
|
||
# json=payload,
|
||
# verify=False,
|
||
# timeout=TIMEOUT,
|
||
# )
|
||
|
||
# print(
|
||
# f"[BIND] Agent {agent_id} -> Role {role_id} "
|
||
# f"({ROLE_NAME_MAP.get(role_id)}) => {resp.status_code}"
|
||
# )
|
||
|
||
# if resp.status_code in (200, 201):
|
||
# return True
|
||
|
||
# print(" ↳ 返回值:", resp.text)
|
||
# return False
|
||
|
||
# # =========================
|
||
# # 7. 主流程（先创建 → 再统一绑定）
|
||
# # =========================
|
||
|
||
# def main():
|
||
# pending_bindings: List[Tuple[str, str]] = []
|
||
|
||
# print("=== 开始创建 Agent ===")
|
||
|
||
# for agent_id, meta in AGENT_META.items():
|
||
# try:
|
||
# if not create_agent(agent_id):
|
||
# continue
|
||
|
||
# idx = meta["icon_index"]
|
||
|
||
# if 1 <= idx <= 4:
|
||
# role_id = AGENT_ROLE_1
|
||
# elif 5 <= idx <= 6:
|
||
# role_id = AGENT_ROLE_2
|
||
# elif 7 <= idx <= 8:
|
||
# role_id = AGENT_ROLE_3
|
||
# else:
|
||
# continue
|
||
|
||
# pending_bindings.append((agent_id, role_id))
|
||
|
||
# except Exception as e:
|
||
# print(f"[ERROR] Agent {agent_id} -> {e}")
|
||
|
||
# print("\n=== 开始统一绑定角色 ===")
|
||
|
||
# for agent_id, role_id in pending_bindings:
|
||
# bind_agent_to_role(agent_id, role_id)
|
||
|
||
# print("\n=== 全部完成 ===")
|
||
|
||
# # =========================
|
||
# # 8. 入口
|
||
# # =========================
|
||
|
||
# if __name__ == "__main__":
|
||
# main()
|