@agentunion/kite 1.4.0 → 1.5.0

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

@@ -0,0 +1,126 @@
+from datetime import datetime, timezone
+import agentcp
+import os
+from pathlib import Path  # 新增导入
+import json
+def create_financial_analyzer_json(publisherInfo):
+    """创建智能体能力、权限描述"""
+    profile_json_data = {
+        "publisherInfo": publisherInfo,
+        "avaUrl": "https://tse3-mm.cn.bing.net/th/id/OIP-C.Hux_MNYiHtMI5EnBzSIubAAAAA?rs=1&pid=ImgDetMain",
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "deepseek",
+        "description": "SSE模式接入dify的工作流:输入输出都是字符串(其中输入字符串应该包含:旅行目的地、旅行天数、旅行预算信息)",
+        "capabilities": {
+            "core": ["dify工作流智能体"],
+            "extended": []
+        },
+        "llm":{
+            "model":"qwen-plus", #模型名称，或使用aid
+            "num_parameters":"",  #模型参数量（如"7B"表示70亿参数）
+            "quantization_bits":"",  #量化位数（如Q4表示4位量化）
+            "context_length":"",  #上下文长度（如"4096"表示4096个token）
+        },
+        "references": {
+            "knowledgeBases": [""],
+            "tools": [""],
+            "companyInfo": [""],
+            "productInfo": [""]
+        },
+        "authorization": {
+            "modes": ["free"],
+            "fee": {},
+            "description": "当前智能体免费使用，无费用",
+            "sla": {}
+        },
+        "input": {
+            "types": ["content"], # 目前支持"content", "search", "reasoning_content", "error", 'file',后续会支持语音视频流
+            "formats": ["json"], #  详细类型
+            "examples": {
+                "type": "content",
+                "format": "text",
+                "content": "搜索智能体：xxx"
+            },
+            "semantics": [""],
+            "compatibleAids": ["*"]
+        },
+        "output": {
+            "types": ["content"],
+            "formats": ["markdown"],
+            "examples": {
+                "type": "content",
+                "format": "markdown",
+                "content": ""
+            },
+            "semantics": [""],
+            "compatibleAids": [""]
+        },
+        "supportStream": False, # 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(os.path.pardir ,seed_password="123456",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/dify_workflow/dify_workflow.py

@@ -0,0 +1,46 @@
+# -*- coding:utf-8 -*-
+"""
+基础配置:
+SSE模式接入dify的工作流:输入输出都是字符串(其中输入字符串应该包含:旅行目的地、旅行天数、旅行预算信息)
+https://tse3-mm.cn.bing.net/th/id/OIP-C.Hux_MNYiHtMI5EnBzSIubAAAAA?rs=1&pid=ImgDetMain
+"""
+import agentcp
+import os
+import json
+import requests
+from dotenv import load_dotenv, find_dotenv
+
+# 加载并读取环境变量
+load_dotenv(find_dotenv())
+base_url = os.getenv('BASE_URL')
+api_key = os.getenv('API_KEY')
+
+def dify_workflow_client(query: str)-> str:
+    """ dify工作流客户端"""
+    headers = { 'Authorization': f'Bearer {api_key}', 'Content-Type': 'application/json; charset=utf-8'}
+    data = json.dumps({'inputs': {'query':query}, 'conversation_id': '', 'user': os.getenv('AID')}, ensure_ascii=False).encode('utf-8')
+    response = requests.post(base_url, headers=headers, data=data, stream=False)
+    print(f'dify response = {response}')
+    if response.status_code != 200:
+        return f'请求失败：{response.text}'
+    result = json.loads(response.text)['data']['outputs']['text']
+    print(f'difyworkflow回复[query = {query}]：response = {result}')
+    return result
+
+if __name__ == "__main__":
+    acp = agentcp.AgentCP(os.path.pardir, seed_password='123456', debug=True)
+    print(f"当前acp访问路径:{acp.app_path}\n开始:agentcp版本:{agentcp.__version__},{__file__}")
+    aid = acp.load_aid(os.getenv('AID'))
+
+    @aid.message_handler()
+    async def sync_message_handler(msg):
+        # 大模型对话
+        response = dify_workflow_client(query=aid.get_content_from_message(msg))
+        # 消息回复
+        aid.send_message_content(aid.get_session_id_from_message(msg), [aid.get_sender_from_message(msg)], response)
+        return True
+
+    # agent上线
+    aid.online()
+    # 开启永久监听
+    acp.serve_forever()

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

@@ -0,0 +1,44 @@
+# 代码解释器Agent
+
+这是一个基于ACP协议的代码解释器Agent，可以python的代码执行。
+
+```
+
+# 安装依赖
+pip install agentcp RestrictedPython
+```
+
+## 使用说明
+
+1. 启动Agent:
+
+```bash
+# 创建身份
+python create_profile.py
+
+# 修改main.py里的 AGENT_NAME 为你创建的身份信息
+python main.py
+```
+
+## 功能特性
+
+- ✅ 安全的python代码执行
+
+```python
+def add(a, b):
+    return a + b
+result = add(1, 2)
+```
+
+## 调用说明
+
+1. 传入参数 `code` 应为合法的Python代码字符串。
+2. 执行代码时，会使用 `RestrictedPython` 进行安全限制，防止执行危险操作。
+3. 代码中的 `print` 输出会被 `PrintCollector` 捕获，但当前方法未返回该输出。
+4. 最终结果需要存储在名为 `result` 的变量中，方法会尝试从执行环境中获取该变量的值并返回。
+5. 代码中可以使用 `import` 进行模块导入，但由于安全性考虑，建议谨慎使用。
+
+## 注意事项
+
+- 复杂代码执行可能失败
+- 某些代码可能因为没有模块缺失失败

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

@@ -0,0 +1,89 @@
+{
+  "publisherInfo": {
+    "publisherAid": "exec1.agentunion.cn",
+    "organization": "agentunion.cn",
+    "certificationSignature": "agentunion.cn"
+  },
+  "version": "1.0.0",
+  "lastUpdated": "2025-05-08T09:44:33Z",
+  "name": "代码执行器",
+  "description": "Python代码执行器，支持Python代码的执行和调试。",
+  "avaUrl": "https://imgapi.imooc.com/66ebdca0091376a110241024.jpg",
+  "capabilities": {
+    "core": [
+      "Python代码执行"
+    ],
+    "extended": []
+  },
+  "llm": {
+    "model": "qwen-plus",
+    "num_parameters": "",
+    "quantization_bits": "",
+    "context_length": ""
+  },
+  "references": {
+    "knowledgeBases": [
+      ""
+    ],
+    "tools": [
+      ""
+    ],
+    "companyInfo": [
+      ""
+    ],
+    "productInfo": [
+      ""
+    ]
+  },
+  "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/executor/create_profile.py

@@ -0,0 +1,139 @@
+# 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):
+    """创建智能体能力、权限描述"""
+    data = {
+        "publisherInfo": publisherInfo,
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "代码执行器",
+        "description": "Python代码执行器，支持Python代码的执行和调试。",
+        "avaUrl": "https://imgapi.imooc.com/66ebdca0091376a110241024.jpg",
+        "capabilities": {
+            "core": ["Python代码执行"],
+            "extended": [],
+        },
+        "llm": {
+            "model": "qwen-plus",  # 模型名称，或使用aid
+            "num_parameters": "",  # 模型参数量（如"7B"表示70亿参数）
+            "quantization_bits": "",  # 量化位数（如Q4表示4位量化）
+            "context_length": "",  # 上下文长度（如"4096"表示4096个token）
+        },
+        "references": {
+            "knowledgeBases": [""],
+            "tools": [""],
+            "companyInfo": [""],
+            "productInfo": [""],
+        },
+        "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,  # False代表当前智能体不支持流式输出
+        "supportAsync": True,
+        "permission": ["*"],
+    }
+    return 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/executor/main.py

@@ -0,0 +1,160 @@
+# 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 agentcp import AgentCP
+import asyncio
+import json
+from RestrictedPython import compile_restricted
+from RestrictedPython.Guards import safe_builtins, guarded_unpack_sequence
+
+class PrintCollector:
+    """Collect written text, and return it when called."""
+
+    def __init__(self, _getattr_=None):
+        self.txt = []
+        self._getattr_ = _getattr_
+
+    def write(self, text):
+        self.txt.append(text)
+
+    def __call__(self, *args, **kwargs):
+        return self
+
+    def _call_print(self, *objects, **kwargs):
+        if kwargs.get("file", None) is None:
+            kwargs["file"] = self
+        else:
+            self._getattr_(kwargs["file"], "write")
+
+        print(*objects, **kwargs)
+
+
+class PythonExecutorAgent:
+    def __init__(self, endpoint: str, name: str):
+        self.acp = AgentCP("./", seed_password="888777")
+        self.endpoint = endpoint
+        self.name = name
+        self.aid = None
+
+    async def message_handler(self, msg):
+        """
+        消息处理器 - 执行Python代码并返回结果
+        """
+        try:
+            ref_msg_id = msg.get("ref_msg_id")
+            content = msg.get("message", '"{}"')
+            content = json.loads(content)[0]["content"]
+            content = json.loads(content)
+            text = content.get("text", "")
+            if not text.strip():
+                await self._send_reply(msg, "请输入要执行的Python代码")
+                return True
+
+            try:
+                # 安全地执行代码
+                result = await self.safe_exec(text)
+                print(f"执行结果: {result}")
+                await self._send_reply(msg, f"执行结果:<br>{result}")
+            except Exception as e:
+                print(f"执行代码出错: {str(e)}")
+                await self._send_reply(msg, f"执行出错:<br> {str(e)}")
+
+            return True
+
+        except Exception as e:
+            print(f"处理消息出错: {str(e)}")
+            await self._send_reply(msg, f"处理代码时出错:<br> {str(e)}")
+            return False
+
+    async def _send_reply(self, original_msg, content: str):
+        """
+        发送回复消息
+        """
+        try:
+            self.aid.send_message_content(
+                to_aid_list=[original_msg.get("sender")],
+                session_id=original_msg.get("session_id"),
+                llm_content=content,
+            )
+        except Exception as e:
+            print(f"发送回复消息出错: {str(e)}")
+    async def safe_exec(self, code: str):
+        """
+        安全地执行Python代码
+
+        调用说明：
+        1. 传入参数 `code` 应为合法的Python代码字符串。
+        2. 执行代码时，会使用 `RestrictedPython` 进行安全限制，防止执行危险操作。
+        3. 代码中的 `print` 输出会被 `PrintCollector` 捕获，但当前方法未返回该输出。
+        4. 最终结果需要存储在名为 `result` 的变量中，方法会尝试从执行环境中获取该变量的值并返回。
+        5. 代码中可以使用 `import` 进行模块导入，但由于安全性考虑，建议谨慎使用。
+
+        参数:
+        code (str): 要安全执行的Python代码字符串。
+
+        返回:
+        Any: 执行代码中 `result` 变量的值，如果不存在则返回 `None`。
+        """
+        safe_builtins.update(
+            {
+                "__import__": __import__,
+            }
+        )
+
+        policy_globals = {
+            "__builtins__": safe_builtins,
+            "_print_": PrintCollector(),  # 用于捕获print输出
+            "_getattr_": getattr,
+            "_getiter_": iter,
+            "_getitem_": lambda obj, index: obj[index],
+            "_iter_unpack_sequence_": guarded_unpack_sequence,  # 支持多变量赋值
+        }
+
+        byte_code = compile_restricted(code, filename="<inline>", mode="exec")
+        exec(byte_code, policy_globals)
+        # 查看print输出
+        # output = policy_globals["_print_"]()
+        # print(f"沙箱输出结果：\n{output.txt}")
+        # 获取结果, 最终结果返回必须是变量result
+        result = policy_globals.get("result")
+        print("Captured result:", result)
+        return result
+
+    async def run(self):
+        """运行Agent"""
+        try:
+            self.aid = self.acp.create_aid(self.endpoint, self.name)
+            self.aid.add_message_handler(self.message_handler)
+            self.aid.online()
+            print("Python执行器Agent已上线，等待代码执行指令...")
+
+            while True:
+                await asyncio.sleep(1)
+
+        except Exception as e:
+            print(f"发生错误: {str(e)}")
+        finally:
+            if self.aid:
+                self.aid.offline()
+                print("Agent已下线")
+
+
+if __name__ == "__main__":
+    # 配置参数
+    ENDPOINT = "agentunion.cn"
+    AGENT_NAME = "exec1"  # 请使用真实的aid
+
+    # 创建并运行Agent
+    agent = PythonExecutorAgent(ENDPOINT, AGENT_NAME)
+    asyncio.run(agent.run())

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

@@ -0,0 +1,45 @@
+# 文件操作Agent
+
+这是一个基于ACP协议的文件操作Agent，可以实现文件读取和查询功能。
+
+```
+
+# 安装依赖
+pip install agentcp
+
+```
+
+## 功能特性
+
+- ✅ 安全的文件读取
+- 🔄 支持文本文件内容检查
+- 📂 支持遍历目录查看文件列表
+- 🤖 交互式消息处理
+
+## 使用说明
+
+1. 启动Agent:
+
+```bash
+# 创建身份
+python create_profile.py
+
+# 修改main.py里的 AGENT_NAME 为你创建的身份信息
+python main.py
+```
+
+
+## 交互说明
+
+发送指令格式:
+
+- 查询文件列表: "查询文件" 或 "列表"
+- 读取文件内容: "读取文件" 或 "查看文件"  (会进入交互模式, 只需输入文件名即可)
+
+
+
+## 注意事项
+
+- 文件操作限制在沙箱目录内（AgentID的公有数据目录）
+- 只能操作文本文件
+