@agentunion/kite 1.3.2 → 1.5.0

package/extensions/services/proxy/agentcp/samples/llm_agent/qwen_agent.py

@@ -0,0 +1,136 @@
+import os
+import json
+from openai import OpenAI
+import time
+import agentcp
+class QwenClient:
+    def __init__(self):
+        self.openai_api_key = None
+        self.base_url = None
+        self.model = None
+        self.client = None
+        self.acp = agentcp.AgentCP(".",seed_password="888777",debug=True)
+        self.agentid:agentcp.AgentID = None
+        
+    def init_ai_client(self,json_data):
+        # 从环境变量中获取 API Key 和 Base URL
+        self.openai_api_key = json_data.get("OPENAI_API_KEY","")
+        self.base_url = json_data.get("BASE_URL","")
+        self.model = json_data.get("MODEL","")
+        self.client = OpenAI(api_key=self.openai_api_key, base_url=self.base_url)
+
+    async def async_message_handler(self, message_data):
+        try:
+            receiver = message_data.get("receiver")
+            sender  = message_data.get("sender","")  # 获取工具信息
+            if self.agentid.id not in receiver:
+                print("不是发给我的消息，不处理")
+                return
+            message_array = self.agentid.get_content_array_from_message(message_data)
+            if len(message_array) == 0: 
+                print("消息内容为空，不处理")
+                return
+            llm_content = self.agentid.get_content_from_message(message_data)
+            stream = message_array[0].get("stream",False)  # 获取stream信息
+            tools  = message_array[0].get("tools",[])  # 获取工具信息
+            rolesetting = message_array[0].get("prompt","")  # 获取工具信息
+            if rolesetting!="" and rolesetting!=None:
+                messages = [{"role": "system", "content": rolesetting},{"role": "user", "content": llm_content}]
+            else:
+                messages = [{"role": "user", "content": llm_content}]
+            print(f"\n[处理消息: {sender} : {llm_content}]\n") 
+            await self.stream_process_query(message_data,messages,sender,stream,tools)  #添加await关键字
+        except Exception as e:
+            import traceback
+            print(f"处理消息时发生错误: {e}\n完整堆栈跟踪:\n{traceback.format_exc()}")
+    
+    def send_message_tools_call(self, session_id,sender,llm_content: str,funcallback):
+        to_aid_list = [sender]
+        msg_block = {
+            "type": "tool_call",
+            "status": "success",
+            "timestamp": int(time.time() * 1000),
+            "content": llm_content,
+        }
+        self.agentid.add_message_handler(funcallback,session_id)
+        self.agentid.send_message(session_id,to_aid_list, msg_block)
+          
+    async def stream_process_query(self, message_data:dict, messages: list, sender:str,stream: bool,user_tools:list):
+        if user_tools is None:
+            user_tools = []  # 确保tools是一个列表，即使它为空
+        if len(user_tools) > 0:
+            response = self.client.chat.completions.create(
+                model=self.model,
+                messages=messages,
+                stream=stream,
+                tools = user_tools
+            )   
+        else:
+            response = self.client.chat.completions.create(
+                model=self.model,
+                messages=messages,
+                stream=stream
+            )   
+        session_id = message_data.get("session_id","")  # 获取session_id
+        content = response.choices[0]
+        if content.finish_reason == "tool_calls":
+            tool_call = content.message.tool_calls[0]
+            tool_name = tool_call.function.name
+            tool_args = json.loads(tool_call.function.arguments)
+            print(f"\n[Calling tool {tool_name} with args {tool_args}]\n")
+            async def funcallback(result_content):
+                self.agentid.remove_message_handler(funcallback,session_id)
+                messages.append(content.message.model_dump())
+                messages.append({
+                    "role": "tool",
+                    "content": self.agentid.get_content_from_message(result_content),
+                    "tool_call_id": tool_call.id,
+                })
+                await self.stream_process_query(message_data,messages,sender,stream,user_tools)
+            tool_content = {
+                'tool_name':tool_name,
+                'tool_args':tool_args,
+            }
+            self.send_message_tools_call(session_id,sender,json.dumps(tool_content),funcallback)
+            return
+        if stream:
+            await self.agentid.send_stream_message(message_data.get("session_id"),[sender] , response)  # 确保正确调用
+        else:
+            return self.agentid.reply_message(message_data, content.message.content)
+        
+def main():
+    client = QwenClient()
+    print("欢迎使用阿里千问 AGENT！")    
+    try:
+        client.agentid = client.acp.load_aid("your_aid")
+        @client.agentid.message_handler()
+        async def sync_message_handler(msg):
+            await client.async_message_handler(msg)  # 添加await关键字
+            return True
+        print("开始在线...")
+        env_path = os.path.join(client.agentid.get_agent_private_path(), "data", "env.json")
+        # 新增文件读取逻辑
+        try:
+            with open(env_path, 'r', encoding='utf-8') as f:
+                env_data = json.load(f)
+                client.init_ai_client(env_data)
+                print(f"成功加载配置文件: {env_path}")
+        except FileNotFoundError:
+            print(f"配置文件 {env_path} 未找到")
+            exit(1)
+        except json.JSONDecodeError:
+            print(f"配置文件 {env_path} 格式错误")
+            exit(1)
+        except Exception as e:
+            print(f"加载配置失败: {str(e)}")
+            exit(1)
+        client.agentid.online()  # 确保self.agentid不为None
+        private_path = client.agentid.get_agent_private_path()  # 获取私钥路径
+        print("开始监听消息...")
+        client.acp.serve_forever()
+    except Exception as e:
+        import traceback
+        print(f"\n⚠️ 发生错误: {traceback.format_exc()}")  # 添加堆栈信息打印
+
+if __name__ == "__main__":
+    main()

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

@@ -0,0 +1,90 @@
+## 使用指南
+### 1、环境要求
+- Python 3.8+
+- [AgentCP SDK](https://pypi.org/project/agentcp/)
+- [Ollama](https://ollama.com/)
+
+### 2、环境准备
+安装Ollama并下载模型:
+#### [Windows](https://ollama.com/download/OllamaSetup.exe)
+#### [macOS](https://ollama.com/download/Ollama-darwin.zip)
+#### Linux
+```bash
+curl -fsSL https://ollama.com/install.sh | sh
+```
+
+### 3、下载模型:
+```bash
+ollama pull qwen3:0.6b
+```
+
+### 4、安装依赖
+
+安装必要的Python库:
+```bash
+pip install agentcp requests
+```
+### 5、创建身份ID
+```bash
+python create_profile.py
+```
+### 6、修改main.py文件
+1. 将seed_password、agent_id修改为上一步创建的身份信息
+2. 将model_name修改为你本地使用的model
+
+### 7、执行main.py代码
+```bash
+python main.py
+```
+## 功能简介
+该Agent基于`agentcp`库构建，实现与本地大模型交互，主要功能包括：
+
+- 接收并处理用户的消息请求
+- 转发消息到本地大模型
+- 处理本地大模型的响应并返回给原始请求方
+
+## 完整示例代码
+```python
+import json
+import requests
+import agentcp
+model_name = 'your_local_model'
+def chatWithLLM(prompt):
+    try:
+        response = requests.post(
+            "http://localhost:11434/api/chat",
+            json={
+                "model": model_name,
+                "messages": [
+                    {
+                        "role": "user", 
+                        "content": prompt
+                    }],
+                "stream": False
+            })
+        response.encoding = 'utf-8'
+        response.raise_for_status()
+        result = response.json()
+        result = result.get('message')
+        result = result.get('content')
+        return result
+    except requests.exceptions.RequestException as e:
+        return "请求失败"
+    except json.JSONDecodeError:
+        return "响应解析失败"
+
+if __name__ == '__main__':
+    agent_id = 'your_agent_id'
+    acp = agentcp.AgentCP('.', seed_password='')
+    aid = acp.load_aid(agent_id)
+    @aid.message_handler()
+    async def sync_message_handler(msg):
+        content = aid.get_content_from_message(msg)
+        session_id = aid.get_session_id_from_message(msg)
+        sender = aid.get_sender_from_message(msg)
+        res = chatWithLLM(content)
+        aid.send_message_content(session_id, [sender], res)
+        return True
+    aid.online()
+    acp.serve_forever()
+```

package/extensions/services/proxy/agentcp/samples/local_llm_agent/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://img0.baidu.com/it/u=727206602,4114969606&fm=253&fmt=auto&app=138&f=JPEG?w=285&h=285",
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "本地大模型",
+        "description": "本地大模型交互Agent",
+        "capabilities": {
+            "core": ["chat"],
+            "extended": []
+        },
+        "llm":{
+            "model":"", #模型名称，或使用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='')
+    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/local_llm_agent/main.py

@@ -0,0 +1,49 @@
+import json
+import requests
+import agentcp
+model_name = 'your_model_name'
+def chatWithLLM(prompt):
+    try:
+        response = requests.post(
+            "http://localhost:11434/api/chat",
+            json={
+                "model": model_name,
+                "messages": [
+                    {
+                        "role": "system", 
+                        "content": '使用中文输出内容'
+                    },
+                    {
+                        "role": "user", 
+                        "content": prompt + '/no_think'
+                    }],
+                "stream": False
+            })
+        response.encoding = 'utf-8'
+        response.raise_for_status()
+        result = response.json()
+        result = result.get('message')
+        result = result.get('content')
+        return result
+    except requests.exceptions.RequestException as e:
+        import traceback
+        print(f"请求失败: {traceback.format_exc()}")
+        return "请求失败"
+    except json.JSONDecodeError:
+        return "响应解析失败"
+
+if __name__ == '__main__':
+    agent_id = 'your_agent_id_from_profile'
+    acp = agentcp.AgentCP('.', seed_password='')
+    aid = acp.load_aid(agent_id)
+    @aid.message_handler()
+    async def sync_message_handler(msg):
+        content = aid.get_content_from_message(msg)
+        session_id = aid.get_session_id_from_message(msg)
+        sender = aid.get_sender_from_message(msg)
+        res = chatWithLLM(content)
+        aid.send_message_content(session_id, [sender], res)
+        return True
+    aid.online()
+    acp.serve_forever()
+        

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

@@ -0,0 +1,55 @@
+## 使用指南
+### 1、环境要求
+- Python 3.8+
+- [AgentCP SDK](https://pypi.org/project/agentcp/)
+
+### 2、安装依赖
+
+安装必要的Python库:
+```bash
+pip install agentcp
+```
+### 3、创建身份ID
+```bash
+python create_profile.py
+```
+### 4、修改main.py文件
+1. 将seed_password、agent_id修改为上一步创建的身份信息
+2. 将 llm_agent_id修改为你想要调用的[agent_id](https://www.agentunion.cn/)
+
+### 5、执行main.py代码
+```bash
+python main.py
+```
+## 功能简介
+该Agent基于`agentcp`库构建，作为一个大语言模型的中转代理，负责：
+
+- 接收并处理用户的消息请求
+- 转发请求到目标大模型Agent（如 lwj001.agentunion.cn）
+- 异步接收大模型Agent的响应，并返回给原始请求方
+
+## 完整示例代码
+```python
+import agentcp
+if __name__ == "__main__":
+    llm_agent_id = "your_llm_agent_id_from_mu"
+    agent_id = 'your_agent_id_from_profile'
+    acp = agentcp.AgentCP('.', seed_password='')
+    aid = acp.load_aid(agent_id)
+    async def reply_message_handler(reply_msg, sender, session_id):
+        reply_text = aid.get_content_from_message(reply_msg)
+        aid.send_message_content(to_aid_list=[sender], session_id=session_id, llm_content=reply_text)
+
+    @aid.message_handler()
+    async def sync_message_handler(msg):
+        receiver = aid.get_receiver_from_message(msg)
+        if aid.id not in receiver:
+            return
+        session_id = aid.get_session_id_from_message(msg)
+        sender = aid.get_sender_from_message(msg)
+        sender_content = aid.get_content_from_message(msg)
+        aid.quick_send_messsage_content(llm_agent_id, sender_content, lambda reply_msg: reply_message_handler(reply_msg, sender, session_id))
+        return True
+    aid.online()
+    acp.serve_forever()
+```

package/extensions/services/proxy/agentcp/samples/query_llm_from_agent/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://img0.baidu.com/it/u=727206602,4114969606&fm=253&fmt=auto&app=138&f=JPEG?w=285&h=285",
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": "大模型调用中转器",
+        "description": "转发用户信息到其它大模型Agent",
+        "capabilities": {
+            "core": ["chat"],
+            "extended": []
+        },
+        "llm":{
+            "model":"", #模型名称，或使用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='')
+    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/query_llm_from_agent/main.py

@@ -0,0 +1,23 @@
+import agentcp
+
+if __name__ == "__main__":
+    llm_agent_id = "your_llm_agent_id_from_mu"
+    agent_id = 'your_agent_id_from_profile'
+    acp = agentcp.AgentCP('.', seed_password='')
+    aid = acp.load_aid(agent_id)
+    async def reply_message_handler(reply_msg, sender, session_id):
+        reply_text = aid.get_content_from_message(reply_msg)
+        aid.send_message_content(to_aid_list=[sender], session_id=session_id, llm_content=reply_text)
+
+    @aid.message_handler()
+    async def sync_message_handler(msg):
+        receiver = aid.get_receiver_from_message(msg)
+        if aid.id not in receiver:
+            return
+        session_id = aid.get_session_id_from_message(msg)
+        sender = aid.get_sender_from_message(msg)
+        sender_content = aid.get_content_from_message(msg)
+        aid.quick_send_messsage_content(llm_agent_id, sender_content, lambda reply_msg: reply_message_handler(reply_msg, sender, session_id))
+        return True
+    aid.online()
+    acp.serve_forever()