@agentunion/kite 1.3.2 → 1.5.0

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

@@ -0,0 +1,123 @@
+import json
+from datetime import datetime, timezone
+from pathlib import Path  # 新增导入
+
+import agentcp
+
+
+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": "串并行调用",
+        "description": "实现智能体搜索 调用大模型Agent，实现工具选择，多个工具并行调用获取结果。",
+        "capabilities": {
+            "core": ["智能体搜索", "工具选择", "工具调用"],
+            "extended": []
+        },
+        "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": "查询明天上海的天气，并规划一条去北京的路线"
+            },
+            "semantics": [""],
+            "compatibleAids": ["*"]
+        },
+        "output": {
+            "types": ["content"],
+            "formats": ["markdown"],
+            "examples": {
+                "type": "content",
+                "format": "markdown",
+                "content": ""
+            },
+            "semantics": [""],
+            "compatibleAids": [""]
+        },
+        "surportFind": False, # 是否支持智能体发现
+        "avaUrl": "https://bkimg.cdn.bcebos.com/pic/5d6034a85edf8db1dd6336220023dd54564e7422",
+        "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("../../../data","",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_use/llm/create_profile.py

@@ -0,0 +1,129 @@
+import json
+from datetime import datetime, timezone
+from pathlib import Path  # 新增导入
+
+import agentcp
+
+
+def create_financial_analyzer_json(publisherInfo):
+    """创建智能体能力、权限描述"""
+    name = "模拟大模型调用Agent"
+    profile_json_data = {
+        "publisherInfo": publisherInfo,
+        "avaUrl": "https://pic.rmb.bdstatic.com/bjh/news/6b87d6906492735a3b03aa7d0aa9cb02.jpeg",
+        "version": "1.0.0",
+        "lastUpdated": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+        "name": name,
+        "description": name,
+        "capabilities": {
+            "core": [name],
+            "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("../../../../data","",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_use/llm/env.json

@@ -0,0 +1,5 @@
+{
+  "API_KEY":"大模型的api_key",
+  "BASE_URL":"大模型Api接口URL",
+  "MODEL":"模型名称"
+}

package/extensions/services/proxy/agentcp/samples/agent_use/llm/main.py

@@ -0,0 +1,146 @@
+import json
+import time
+
+import agentcp
+from openai import OpenAI
+
+
+class QwenClient:
+    def __init__(self):
+        self.openai_api_key = None
+        self.base_url = None
+        self.model = None
+        self.client = None
+        self.acp = agentcp.AgentCP("../../../../data", seed_password="", 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("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！")
+    _my_aid = "llmdemo007.agentunion.cn"
+    try:
+        client.agentid = client.acp.load_aid(_my_aid)
+
+        @client.agentid.message_handler()
+        async def sync_message_handler(msg):
+            await client.async_message_handler(msg)  # 添加await关键字
+            return True
+
+        print("开始在线...")
+        env_path = "./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/agent_use/main.py

@@ -0,0 +1,123 @@
+import json
+import time
+import traceback
+
+import agentcp
+
+"""
+智能体搜索 使用大模型agent实现工具选择 多工具并行调用
+"""
+class Agent:
+    def __init__(self):
+        self.agentid = None
+        self.acp = agentcp.AgentCP("../../../data",seed_password="")
+        self.llm_agent_id = "llmdemo007.agentunion.cn"
+        self.search_agent_id = "search007.agentunion.cn"
+    async def async_message_handler(self, msg):
+        try:
+            receiver = self.agentid.get_receiver_from_message(msg)
+            if self.agentid.id not in receiver:
+                print("不是发给我的消息，不处理")
+                return
+            sender = self.agentid.get_sender_from_message(msg)
+            session_id = self.agentid.get_session_id_from_message(msg)
+            to_aid_list = [sender]
+
+            # 获取输入
+            llm_content = self.agentid.get_content_from_message(msg)
+            print(f"llm_content={llm_content}\n")
+
+            # 调用工具选择agent
+            self.mult_tool_choose(llm_content,session_id,to_aid_list)
+        except Exception as e:
+            print(f"处理消息时发生错误: {e}\n完整堆栈跟踪:\n{traceback.format_exc()}")
+
+    async def reply_message_handler(self,reply_msg,session_id,to_aid_list):
+        content = []
+        print(f"模型返回结果{reply_msg}")
+        message_json = json.loads(reply_msg.get("message"))
+        if isinstance(message_json, list) and len(message_json) > 0:
+            content = message_json
+            message_json = message_json[0]
+        # 根据大模型返回结果 决定是否调用其他agent
+        if message_json.get("type") == "tool_call":
+            print(f"使用工具 {content}")
+            await self.mult_tool_call(content,session_id, to_aid_list)
+        else:
+            self.agentid.send_message_content(session_id, to_aid_list, self.agentid.get_content_from_message(reply_msg))
+        return
+
+    def mult_tool_choose(self,llm_content,session_id,to_aid_list):
+        def search_agent_handler(search_msg):
+            result = self.agentid.get_content_from_message(search_msg)
+            print(f"search result={result}")
+            agents = json.loads(result)
+            tools = []
+            for ainfo in agents:
+                description = f"我是{ainfo['agent_id']}，我能提供[{ainfo['description']}]服务，我的aid是{ainfo['agent_id']}"
+                tools.append({
+                    "type": "function",
+                    "function": {
+                        "name": "agent_" + ainfo['agent_id'],
+                        "description": description,
+                        "parameters": {
+                            "type": "object",
+                            "properties": {
+                                "aid": {
+                                    "type": "string",
+                                    "description": "",
+                                },
+                                "text": {
+                                    "type": "string",
+                                    "description": ""
+                                }
+                            },
+                            "required": ["aid", "text"]
+                        }
+                    }
+                })
+
+            msg_block = {
+                "type": "content",
+                "status": "success",
+                "timestamp": int(time.time() * 1000),
+                "content": llm_content,
+                "tools": tools,
+                "prompt": ""
+            }
+            self.agentid.quick_send_messsage(self.llm_agent_id, msg_block,
+                                             lambda reply_msg: self.reply_message_handler(reply_msg, session_id,
+                                                                                          to_aid_list))
+
+        self.agentid.quick_send_messsage_content(self.search_agent_id, llm_content,search_agent_handler)
+
+    async def mult_tool_call(self,content,session_id, to_aid_list) :
+        for tool_call in content:
+            tool = json.loads(tool_call.get("content"))
+            tool_args = tool.get("tool_args")
+            print(f"aiddddddd={tool_args['aid']} text={tool_args['text']}")
+            async def async_func_call_result(message):
+                tool_result = self.agentid.get_content_from_message(message)
+                print(f"工具返回的结果={tool_result}")
+                self.agentid.send_message_content(session_id,to_aid_list, tool_result)
+                # self.agentid.quick_send_messsage_content(self.llm_agent_id, tool_result,
+                #     lambda reply_msg: self.reply_message_handler(reply_msg,session_id,to_aid_list))
+                return
+            self.agentid.quick_send_messsage_content(tool_args["aid"], tool_args["text"], async_func_call_result)
+
+if __name__ == "__main__":
+    _my_aid = "mc58009.agentunion.cn"
+
+    agent = Agent()
+    agent.agentid =agent.acp.load_aid(_my_aid)
+    async def sync_message_handler(msg):
+        print(f"收到消息数据: {msg}")
+        await agent.async_message_handler(msg)  # 添加await关键字
+        return True
+    try:
+        agent.agentid.online()
+        # agent.agentid.sync_public_files()
+        agent.agentid.add_message_handler(sync_message_handler)
+        agent.acp.serve_forever()
+    except Exception as e:
+        print(f"AgentID未正确初始化: {e}")