@agentunion/kite 1.3.2 → 1.5.0

package/extensions/services/proxy/agentcp/msg/wss_binary_message.py

@@ -0,0 +1,137 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 AgentUnion Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import struct
+import zlib
+import json
+from dataclasses import dataclass
+
+@dataclass
+class WssBinaryMessage:
+    """
+    用于存储 WSS 二进制消息中从 magic_byte1 到 payload 的各个字段的数据类。
+    """
+    magic_byte1: int = 77  # ord('M')
+    magic_byte2: int = 85  # ord('U')
+    version: int = 0x101
+    flags: int = 0
+    msg_type: int = 1
+    msg_seq: int = 0
+    content_type: int = 1
+    compressed: int = 0
+    reserved: int = 0
+    crc32: int = 0
+    payload_length: int = 0
+    payload: bytes = b''
+
+def encode_wss_binary_message(json_data):
+    magic_byte1 = ord('M')
+    magic_byte2 = ord('U')
+    version = 0x101
+    flags = 0
+    msg_type = 1
+    # 假设 msg_seq 初始为 0，可根据实际情况调整
+    msg_seq = 0
+    payload = json_data.encode()
+    if len(payload) < 512:
+        content_type = 1
+        compressed = 0
+    else:
+        content_type = 1
+        compressed = 1
+        payload = zlib.compress(payload)
+    reserved = 0
+    crc32 = zlib.crc32(payload)
+    payload_length = len(payload)
+
+
+    header = struct.pack('>BBHIHIBBIII', magic_byte1, magic_byte2, version, flags, msg_type, msg_seq, content_type, compressed, reserved, crc32, payload_length)
+    #print(f"Header: {header}")
+    #print(f"Payload: {payload}")
+    return header + payload
+
+
+def decode_wss_binary_message(data):
+    try:
+        magic_byte1, magic_byte2, version, flags, msg_type, msg_seq, content_type, compressed, reserved, crc32, payload_length = struct.unpack('>BBHIHIBBIII', data[:28])
+        if magic_byte1 != ord('M') or magic_byte2 != ord('U'):
+            return ""
+        payload = data[28:]
+        if len(payload) != payload_length:
+            return ""
+        if zlib.crc32(payload) != crc32:
+            return ""
+        if compressed != 0 and compressed != 1:
+            return ""
+        if compressed == 1:
+            payload = zlib.decompress(payload)
+        return payload.decode()
+    except (struct.error, zlib.error):
+        return ""
+    except Exception:
+        return ""
+
+
+def encode_wss_binary_buffer(payload, msg_header: WssBinaryMessage):
+    magic_byte1 = ord('M')
+    magic_byte2 = ord('U')
+    version = msg_header.version
+    flags = msg_header.flags
+    msg_type = msg_header.msg_type
+    # 假设 msg_seq 初始为 0，可根据实际情况调整
+    msg_seq = msg_header.msg_seq
+    content_type = msg_header.content_type
+    compressed = msg_header.compressed
+    reserved = msg_header.reserved
+    crc32 = zlib.crc32(payload)
+    payload_length = len(payload)
+
+    header = struct.pack('>BBHIHIBBIII', magic_byte1, magic_byte2, version, flags, msg_type, msg_seq, content_type, compressed, reserved, crc32, payload_length)
+    #print(f"Header: {header}")
+    #print(f"Payload: {payload}")
+    return header + payload
+
+
+def decode_wss_binary_buffer(data):
+    try:
+        magic_byte1, magic_byte2, version, flags, msg_type, msg_seq, content_type, compressed, reserved, crc32, payload_length = struct.unpack('>BBHIHIBBIII', data[:28])
+        if magic_byte1 != ord('M') or magic_byte2 != ord('U'):
+            return None
+        payload = data[28:]
+        if len(payload) != payload_length:
+            return None
+        if zlib.crc32(payload) != crc32:
+            return None
+        if compressed != 0 and compressed != 1:
+            return None
+        if compressed == 1:
+            payload = zlib.decompress(payload)
+
+        wss_msg = WssBinaryMessage()
+        wss_msg.version = version
+        wss_msg.flags = flags
+        wss_msg.msg_type = msg_type
+        wss_msg.msg_seq = msg_seq
+        wss_msg.content_type = content_type
+        wss_msg.compressed = compressed
+        wss_msg.reserved = reserved
+        wss_msg.crc32 = crc32
+        wss_msg.payload_length = len(payload)
+        wss_msg.payload = payload    
+         
+        return wss_msg
+    except (struct.error, zlib.error):
+        return None
+    except Exception:
+        return None

package/extensions/services/proxy/agentcp/requirements.txt

@@ -0,0 +1,7 @@
+cryptography>=3.4.7
+requests>=2.26.0
+websocket-client>=1.2.1
+python-dotenv>=0.19.0
+asyncio>=3.4.3
+typing-extensions>=4.0.1
+openai>=0.27.2

package/extensions/services/proxy/agentcp/samples/agent_graph/README.md

@@ -0,0 +1,37 @@
+# 文件操作Agent
+
+这是一个基于ACP协议的Agent，可以根据会话和群组成员生成关系图
+
+```
+
+# 安装依赖
+pip install agentcp
+
+```
+
+## 功能特性
+
+- ✅ 消息问答
+- ✅ 生成调用关系图
+
+## 使用说明
+
+1. 启动Agent:
+
+```bash
+# 创建身份
+python create_profile.py
+
+# 修改main.py里的 AGENT_NAME 为你创建的身份信息
+python main.py
+```
+
+## 交互说明
+
+发送指令格式:
+
+- 生成调用关系图: "查看关系图"
+
+
+## 注意事项
+

package/extensions/services/proxy/agentcp/samples/agent_graph/agentprofile.json

@@ -0,0 +1,89 @@
+{
+  "publisherInfo": {
+    "publisherAid": "gggg12.agentunion.cn",
+    "organization": "agentunion.cn",
+    "certificationSignature": "agentunion.cn"
+  },
+  "version": "1.0.0",
+  "lastUpdated": "2025-05-08T09:40:38Z",
+  "name": "Agent调用关系图",
+  "description": "Agent调用关系图绘制",
+  "avaUrl": "https://t9.baidu.com/it/u=2472616479,557015018&fm=193",
+  "capabilities": {
+    "core": [
+      "调用关系图绘制"
+    ],
+    "extended": []
+  },
+  "references": {
+    "knowledgeBases": [
+      ""
+    ],
+    "tools": [
+      ""
+    ],
+    "companyInfo": [
+      ""
+    ],
+    "productInfo": [
+      ""
+    ]
+  },
+  "llm": {
+    "model": "qwen-plus",
+    "num_parameters": "",
+    "quantization_bits": "",
+    "context_length": ""
+  },
+  "authorization": {
+    "modes": [
+      "free"
+    ],
+    "fee": {},
+    "description": "",
+    "sla": {}
+  },
+  "input": {
+    "types": [
+      "content"
+    ],
+    "formats": [
+      "json"
+    ],
+    "examples": {
+      "type": "content",
+      "format": "text",
+      "content": "查看关系图"
+    },
+    "semantics": [
+      ""
+    ],
+    "compatibleAids": [
+      "*"
+    ]
+  },
+  "output": {
+    "types": [
+      "content"
+    ],
+    "formats": [
+      "markdown"
+    ],
+    "examples": {
+      "type": "content",
+      "format": "markdown",
+      "content": ""
+    },
+    "semantics": [
+      ""
+    ],
+    "compatibleAids": [
+      ""
+    ]
+  },
+  "supportStream": true,
+  "supportAsync": true,
+  "permission": [
+    "*"
+  ]
+}

package/extensions/services/proxy/agentcp/samples/agent_graph/create_profile.py

@@ -0,0 +1,138 @@
+# Copyright 2025 ModelUnion Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from datetime import datetime, timezone
+import agentcp
+from pathlib import Path  # 新增导入
+import json
+
+
+def create_financial_analyzer_json(publisherInfo):
+    """创建智能体能力、权限描述"""
+    profile_json_data = {
+        "publisherInfo": publisherInfo,
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "Agent调用关系图",
+        "description": "Agent调用关系图绘制",
+        "avaUrl": "https://t9.baidu.com/it/u=2472616479,557015018&fm=193",
+        "capabilities": {"core": ["调用关系图绘制"], "extended": []},
+        "references": {
+            "knowledgeBases": [""],
+            "tools": [""],
+            "companyInfo": [""],
+            "productInfo": [""],
+        },
+        "llm": {
+            "model": "qwen-plus",  # 模型名称，或使用aid
+            "num_parameters": "",  # 模型参数量（如"7B"表示70亿参数）
+            "quantization_bits": "",  # 量化位数（如Q4表示4位量化）
+            "context_length": "",  # 上下文长度（如"4096"表示4096个token）
+        },
+        "authorization": {
+            "modes": ["free"],
+            "fee": {},
+            "description": "",
+            "sla": {},
+        },
+        "input": {
+            "types": [
+                "content"
+            ],  # 目前支持"content", "search", "reasoning_content", "error", 'file',后续会支持语音视频流
+            "formats": ["json"],  #  详细类型
+            "examples": {
+                "type": "content",
+                "format": "text",
+                "content": "查看关系图",
+            },
+            "semantics": [""],
+            "compatibleAids": ["*"],
+        },
+        "output": {
+            "types": ["content"],
+            "formats": ["markdown"],
+            "examples": {"type": "content", "format": "markdown", "content": ""},
+            "semantics": [""],
+            "compatibleAids": [""],
+        },
+        "supportStream": True,  # False代表当前智能体不支持流式输出
+        "supportAsync": True,
+        "permission": ["*"],
+    }
+    return profile_json_data
+
+
+def write_agent_profile_json(json_data):
+    try:
+        import os
+
+        json_path = Path(__file__).resolve()
+        json_dir = json_path.parent
+        json_file = os.path.join(json_dir, "agentprofile.json")
+        with open(json_file, "w", encoding="utf-8") as f:
+            json.dump(json_data, f, ensure_ascii=False, indent=2)
+            print("智能体描述文件已保存至当前目录下agentprofile.json")
+    except Exception as e:
+        print(f"文件写入失败: {str(e)}")
+        exit(1)
+
+
+if __name__ == "__main__":
+    # 创建JSON数据
+    # 将加密种子修改为自己的加密种子，可以是随机字符串，也可以是固定字符串，只要保证一致即可。
+    acp = agentcp.AgentCP(".", seed_password="888777", debug=False)
+    agentid_list = acp.get_aid_list()
+    agentid: agentcp.AgentID = None
+    while agentid is None:
+        print("请选择一个身份（aid）:")
+        for i, agentid in enumerate(agentid_list):
+            print(f"{i+1}. {agentid}")
+        print(f"{len(agentid_list)+1}. 创建一个新的身份（aid）")
+        choice = input("请输入数字选择一个身份（aid）: ")
+        try:
+            choice = int(choice) - 1
+            if choice < 0 or choice > len(agentid_list):
+                raise ValueError
+            if choice == len(agentid_list):
+                aid = input("请输入名称: ")
+                agentid = acp.create_aid("agentunion.cn", aid)
+                if agentid is None:
+                    print("创建身份（aid）失败，请打开日志查看原因")
+                    exit(1)
+                agentid_list = acp.get_aid_list()
+            else:
+                agentid = acp.load_aid(agentid_list[choice])
+                if agentid is None:
+                    print("加载身份（aid）失败,请打开日志查看原因")
+                    exit(1)
+        except ValueError:
+            print("无效的选择，请重新输入。")
+    print(f"当前选择的身份（aid）是: {str(agentid)}")
+    agentid.init_ap_client()
+    json_data = create_financial_analyzer_json(agentid.get_publisher_info())
+    write_agent_profile_json(json_data)
+    select_result = input("是否将文件拷贝到agent公有数据目录下（Y/N）: ")
+    if select_result.upper() != "Y":
+        print("程序运行结束")
+        exit(1)
+    agentid.create_agent_profile(json_data)
+    select_result = input("拷贝成功，是否同步到接入服务器（Y/N）: ")
+    if select_result.upper() != "Y":
+        print("程序运行结束")
+        exit(1)
+    result = agentid.sync_public_files()
+    if result:
+        print("文件同步成功！")
+    else:
+        print("文件同步失败,请初始化ACP时打开日志查看")

package/extensions/services/proxy/agentcp/samples/agent_graph/main.py

@@ -0,0 +1,164 @@
+# Copyright 2025 ModelUnion Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import json
+from agentcp import AgentCP
+import networkx as nx
+from pyvis.network import Network
+
+
+class Graph(nx.DiGraph):
+
+    def add(self, source, target, info=""):
+        if self.has_edge(source, target):
+            # 更新 info 信息
+            self[source][target]["info"] = info
+        else:
+            # 添加新边
+            self.add_edge(source, target, info=info)
+
+    def draw(self, name="agent_call_graph.html"):
+        net = Network(
+            height="600px",
+            width="100%",
+            directed=True,
+            notebook=False,
+            cdn_resources="in_line",
+        )
+        net.from_nx(self)
+        root_nodes = [node for node in self.nodes if self.in_degree(node) == 0]
+
+        # 创建 pyvis 网络图
+        net = Network(
+            height="600px",
+            width="100%",
+            directed=True,
+            notebook=False,
+            cdn_resources="in_line",
+        )
+
+        # 从 NetworkX 图导入结构
+        net.from_nx(self)
+
+        # 放大节点
+        for node in net.nodes:
+            node["size"] = 15  # 默认 25 左右，40 会明显更大
+            node["color"] = "#ADD8E6"  # 可选：设置更醒目的颜色
+            node["font"] = {"size": 20}  # 放大标签字体
+            if self.in_degree(node["id"]) == 0:
+                node["color"] = "#CFD8E6"  # 可选：设置更醒目的颜色
+
+        # 更新每条边的 title（hover 时显示调用频率）
+        for edge in net.edges:
+            src = edge["from"]
+            tgt = edge["to"]
+            info = self.get_edge_data(src, tgt).get("info", "")
+            edge["title"] = info  # 设置 hover 提示
+            edge["arrows"] = "to"  # 确保箭头方向正确
+
+        # 添加物理布局和交互设置
+        net.set_options(
+            """
+        var options = {
+          "physics": {
+            "enabled": false,
+            "stabilization": {
+              "iterations": 300
+            }
+          },
+          "interaction": {
+            "hover": true,
+            "navigationButtons": true,
+            "zoomView": true,
+            "dragNodes": true
+          }
+        }
+        """
+        )
+
+        # 生成 HTML 并在浏览器中打开
+        net.write_html(name, local=False, notebook=False, open_browser=True)
+
+
+class AgentGraph:
+
+    def __init__(self, name, endpoint="agentunion.cn"):
+        self.name = name
+        self.endpoint = endpoint
+        self.acp = AgentCP("./", seed_password="888777")
+        self.graph = Graph()
+        self.aid = self.acp.create_aid(self.endpoint, self.name)
+        self.call_count = {}
+        self.id = f'{self.name}.{self.endpoint}'
+        self.aid.add_message_handler(self.message_handler)
+        self.graph_file = f"{self.aid.get_agent_public_path()}/agent_call_graph.html"
+
+    async def message_handler(self, msg):
+        """
+        消息处理器 - 根据消息内容安全地读取文件
+        {
+            'session_id': '1831173476580327424', 
+            'request_id': '', 'message_id': '9', 
+            'ref_msg_id': '', 
+            'sender': 'samplesdeveloper.agentunion.cn', 
+            'receiver': 'guest_1831158907166261248.agentunion.cn', 
+            'message': '[{"type": "text", "status": "success", "timestamp": 1746343146261, 
+            "content": "{\\"text\\":\\"\\u8bfb\\u53d6\\u6587\\u4ef6agentprofile.json\\",\\"files\\":[],\\"links\\":[],\\"search\\":false,\\"think\\":false}",
+            "stream": false, "prompt": null, "extra": null, "artifact": null}]',
+            'timestamp': '1746343146265'
+        }
+        """
+
+        sender = msg.get("sender")
+        self.call_count[sender] = self.call_count.get(sender, 0) + 1
+        print(f"收到来自 {sender} 的消息")
+        self.graph.add(sender, self.id, f"调用次数: {self.call_count[sender]}")
+        message = json.loads(msg.get("message"))[0]
+        message = json.loads(message.get('content', '{}'))
+        message = message.get('text', '')
+        # self.graph.add(self.id, sender, f'调用次数: {self.call_count[sender]}')
+        members = self.aid.get_session_member_list(msg.get("session_id"))
+        print(f"群成员: {members}")
+        for member in members:
+            print(f"群组成员: {member}")
+            # self.graph.add(self.id, member, f'调用次数: {self.call_count[member]}'
+        if message.find('关系图') != -1:
+            print(f"生成关系图: {self.graph_file}")
+            try:
+                self.graph.draw(name=self.graph_file)
+                self.aid.sync_public_files()
+                url = f'https://{self.aid.id}/{self.graph_file.split("/")[-1]}'
+                self.aid.reply_message(msg, f"关系图已生成可访问: {url} 查看")
+                self.graph.add(self.id, sender, f"调用次数: {self.call_count[sender]}")
+            except Exception as e:
+                print(f"生成关系图失败: {e}")
+        else:
+            self.graph.add(self.id, sender, f"调用次数: {self.call_count[sender]}")
+            self.aid.reply_message(msg, "收到消息")
+
+    def online(self):
+        self.aid.online()
+        self.acp.register_signal_handler()
+        self.acp.serve_forever()
+
+    def offline(self):
+        self.aid.offline()
+        self.graph.draw()
+
+
+if __name__ == "__main__":
+    ENDPOINT = "agentunion.cn"
+    AGENT_NAME = "gggg12"
+    agent = AgentGraph(AGENT_NAME, ENDPOINT)
+    agent.online()