@agentunion/kite 1.3.2 → 1.5.0

package/extensions/services/proxy/agentcp/samples/filewriter/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):
+    """创建智能体能力、权限描述"""
+    data = {
+        "publisherInfo": publisherInfo,
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "文件读写",
+        "description": "读写agent目录下的文件",
+        "avaUrl": "https://t8.baidu.com/it/u=655275816,1648044724&fm=193",
+        "capabilities": {
+            "core": ["读取文件", "写入文件", "查询文件列表"],
+            "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/filewriter/main.py

@@ -0,0 +1,289 @@
+# 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 os
+import re
+import os
+import json
+
+class FileOperator:
+    """文件操作类，负责安全地读取文本文件内容"""
+
+    def __init__(self, sandbox_root: str = None):
+        self.sandbox_root = sandbox_root or os.getcwd()  # 默认使用当前工作目录
+
+    def is_text_file(self, file_path: str) -> bool:
+        """
+        通过检查文件内容判断是否为文本文件
+        
+        Args:
+            file_path: 文件路径
+            
+        Returns:
+            bool: 是否为文本文件
+        """
+        try:
+            # 尝试以文本模式打开文件
+            with open(file_path, 'r', encoding='utf-8') as f:
+                # 读取前1024个字符
+                f.read(1024)
+            return True
+        except UnicodeDecodeError:
+            # 如果解码失败，说明不是文本文件
+            return False
+        except Exception:
+            # 处理其他异常，如文件不存在等，默认不是文本文件
+            return False
+
+    def extract_file_path(self, text: str) -> str:
+        """从文本中提取文件名(不需要完整路径)"""
+        match = re.search(r"(?:读取|打开|查看)文件\s*([^\s\/\\]+)", text)
+        return match.group(1) if match else None
+
+    def sanitize_path(self, filename: str) -> str:
+        """遍历sandbox目录，找到这个文件"""
+        def recursive_search(root_dir):
+            try:
+                for root, dirs, files in os.walk(root_dir):
+                    if filename in files:
+                        full_path = os.path.join(root, filename)
+                        normalized_path = os.path.normpath(full_path)
+                        return normalized_path
+            except:
+                return None
+            return None
+
+        return recursive_search(self.sandbox_root)
+
+    def walk_directory(self):
+        """
+        遍历目录并打印所有文件和文件夹
+        """
+        filenamess = []
+        for root, dirs, files in os.walk(self.sandbox_root):
+            for file in files:
+                filenamess.append(file)
+        return filenamess
+
+    def exist_file(self, filename):
+        """
+        检查文件是否存在
+        """
+        safe_path = self.sanitize_path(filename)
+        return os.path.exists(safe_path)
+
+    def read_file(self, file_path: str) -> str:
+        """
+        安全地读取文本文件内容
+        
+        Args:
+            file_path: 文件路径
+            
+        Returns:
+            文件内容字符串
+            
+        Raises:
+            ValueError: 如果文件不是文本格式
+        """
+
+        safe_path = self.sanitize_path(file_path)
+        if not safe_path:
+            raise ValueError("文件不存在")
+        if not self.is_text_file(safe_path):
+            raise ValueError("只能读取文本文件")
+        with open(safe_path, "r") as file:
+            return file.read()
+
+def parse_command(text):
+    if not text:
+        return False
+    if text.find("查询") >=0 or text.find("列表") >=0:
+        return 'list'
+    if text.find("写入") >=0 or text.find("保存") >=0:
+        return 'write'
+    if text.find("读取") >= 0 or text.find("查看") >= 0:
+        return "read"
+
+class FileAgent:
+    def __init__(self, endpoint: str, name: str):
+        """
+        初始化文件Agent
+        """
+        self.acp = AgentCP("./", seed_password="888777")
+        self.endpoint = endpoint
+        self.name = name
+        self.aid = None
+        self.last_command = ''
+        self.last_data = ''
+    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'
+        }
+        """
+        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", "")
+            command = parse_command(text)
+            print(f"收到消息: {content}")
+
+            if self.last_command:
+                if self.last_command == "write" and self.last_data:
+                    ret = await self.write_file(msg, text, self.last_data)
+                    if not ret:
+                        await self._send_reply(msg, "文件已存在请重新输入名字")
+                        return
+                    self.last_data = ''
+                    self.last_command = ""
+                    return 
+                elif self.last_command == "write":
+                    self.last_data = text
+                    await self._send_reply(msg, "请输入你要保存的文件名")
+                    return True
+                elif self.last_command == "read":
+                    self.last_command = ''
+                    return await self.read_file(msg, text)  # 调用 read_file 方法处理读取文件的逻辑
+
+            elif command == 'list':
+                files = self.file_operator.walk_directory()
+                if not files:
+                    await self._send_reply(msg, "当前目录下没有文件")
+                    return True
+                to = "文件列表<br>" + "<br>".join(files)
+                await self._send_reply(msg, to)
+                return True
+            elif command == 'write':
+                self.last_command = "write"
+                await self._send_reply(msg, "请输入你要写入的内容")
+                return True
+            elif command == "read":
+                self.last_command = "read"
+                files = self.file_operator.walk_directory()
+                if not files:
+                    await self._send_reply(msg, "当前目录下没有文件")
+                    return True
+                to = "读取哪一个文件？<br>" + "<br>".join(files)
+                print(f"send message: {to}")
+                await self._send_reply(msg, to)
+                return True
+            else:
+                await self._send_reply(
+                    msg, "你可以对我说: <br>查询文件<br>读取文件<br>写入文件"
+                )
+        except Exception as e:
+            print(f"处理消息出错: {str(e)}")
+            await self._send_reply(msg, f"处理文件时出错: {str(e)}")
+            return False
+
+    async def write_file(self, msg, filename, text):
+        file_path = f"{self.aid.get_agent_public_path()}/{filename}"
+        # 检查文件是否已存在
+        if os.path.exists(file_path):
+            return False
+        try:
+            # 以写入模式打开文件，并写入内容
+            with open(file_path, 'w', encoding='utf-8') as file:
+                file.write(text)
+            await self._send_reply(msg, f"成功写入文件: {file_path}")
+            return True
+        except Exception as e:
+            print(f"写入文件时出错: {str(e)}")
+            await self._send_reply(msg, f"写入文件 {file_path} 时出错: {str(e)}")
+            return False
+
+    async def read_file(self, msg, text):
+        try:
+            filename = self.file_operator.extract_file_path(text)  # 现在只获取文件名
+            if not filename:
+                try:
+                    print(f"未提供文件名尝试读取: {text}")
+                    file_content = self.file_operator.read_file(text)  # 直接传入文件名
+                    await self._send_reply(msg, f"文件内容:<br>{file_content}")
+                    return
+                except Exception as e:
+                    print(f"处理消息出错: {str(e)}")
+            try:
+                file_content = self.file_operator.read_file(filename)  # 直接传入文件名
+                await self._send_reply(msg, f"文件内容:<br>{file_content}")
+            except PermissionError:
+                await self._send_reply(msg, "访问文件被拒绝")
+            except FileNotFoundError:
+                await self._send_reply(msg, f"文件不存在: {filename}")
+            except ValueError as e:
+                await self._send_reply(msg, f"{str(e)}")
+            return True
+        except Exception as e:
+            print(f"处理消息出错: {str(e)}")
+            await self._send_reply(msg, f"处理文件时出错: {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 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()
+            self.file_operator = FileOperator(self.aid.get_agent_public_path())
+            print("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 = "w212" #guest 是一个临时aid会被定时清理，实际业务中请使用真实的aid
+
+    # 创建并运行Agent
+    agent = FileAgent(
+        ENDPOINT,
+        AGENT_NAME,
+    )
+    # agent.test()
+    asyncio.run(agent.run())

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

@@ -0,0 +1,85 @@
+```markdown:d:/github_agentcp/samples/hcp/README.md
+# 智能天气助手（HCP）
+
+基于 agentcp 框架开发的智能天气助手，支持自然语言交互和上下文记忆，能够将用户作为外部工具回调获取必要信息。
+
+## 🚀 使用指南
+
+1. **创建agent身份(aid)**
+   - 参考文档：创建身份&读写公私有数据
+
+2. **配置Agent参数**
+   ```python:hcp.py
+   # 修改hcp.py中的agent_id配置项
+   agent_id = "your_agent_id"  # 替换为实际创建的aid
+   ```
+
+3. **配置环境变量**
+   ```bash:.env
+   OPENAI_API_KEY=your_api_key
+   BASE_URL=https://api.openai.com/v1
+   MODEL=gpt-3.5-turbo
+   ```
+
+4. **安装依赖**
+   ```bash
+   pip install openai agentcp python-dotenv
+   ```
+
+5. **启动服务**
+   ```bash
+   python hcp.py
+   ```
+
+## ✨ 核心功能
+
+- 🌦️ 多维度天气查询（当前天气/历史天气/天气预报）
+- 🗣️ 上下文感知的对话管理
+- 🔧 外部工具回调机制（用于获取缺失的位置/时间信息）
+- ⏰ 自动时间推理（支持绝对时间、相对时间和年号纪年）
+
+## 🧩 核心类说明：MCPClient
+
+### 初始化方法
+```python
+def __init__(self):
+    # 初始化 OpenAI 客户端
+    self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+    # 创建 AgentCP 实例
+    self.acp = agentcp.AgentCP(app_path)
+    # 消息存储器（按 session_id 隔离）
+    self.message_store = dict()
+```
+
+### 消息管理机制
+```python
+def record_message(self, session_id: str, role: str, content: str):
+    # 记录对话历史（支持 user/assistant/system 三种角色）
+
+def get_messages_for_llm(self, session_id: str, max_messages=20, system_message=None):
+    # 构造符合 OpenAI 格式的对话历史（自动维护最近20条消息）
+```
+
+## 🔄 交互流程
+
+```mermaid
+graph TD
+    A[用户提问] --> B{是否天气相关?}
+    B -->|是| C[检查位置/时间信息]
+    B -->|否| D[简短回答并引导]
+    C --> E{信息完整?}
+    E -->|是| F[生成天气报告]
+    E -->|否| G[调用用户工具获取信息]
+```
+
+## 💬 示例对话
+
+> **用户**：明天天气怎么样  
+> **Agent**：你想查询哪里的天气呢  
+> **用户**：上海  
+> **Agent**：上海的天气晴  
+> **用户**：上海  
+> **Agent**：你是想查询上海什么时候的天气呢
+```
+
+        

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

@@ -0,0 +1,125 @@
+from datetime import datetime, timezone
+import agentcp
+from pathlib import Path  # 新增导入
+import json
+def create_financial_analyzer_json(publisherInfo):
+    """创建智能体能力、权限描述"""
+    profile_json_data = {
+        "publisherInfo": publisherInfo,
+        "avaUrl": "https://img1.baidu.com/it/u=4196730940,2454447876&fm=253&fmt=auto&app=120&f=JPEG?w=933&h=800",
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "HCP 天气问答智能体",
+        "description": "智能查询天气信息，提供准确的天气数据。",
+        "capabilities": {
+            "core": ["智能查询天气信息，提供准确的天气数据。"],
+            "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": 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时打开日志查看")