我用Python 实现了一个可以添加多Agent的类似“微信”的聊天桌面程序2-添加了本地tools机制-CSDN博客

前言
一、工具集成的目标与意义
二、核心：工具加载与发现机制
- 1. 工具文件夹结构
- 2. 动态加载工具的魔法
三、工具选择与配置界面
四、工具调用的实际案例
- 1. 文件查找工具
- 2. 文件打开工具
五、工具与Agent的结合
六、工具调用的实际效果
七、总结与展望
PS.扩展阅读

前言

Hello！上一篇文章我们聊了如何用Python打造一个支持多Agent的桌面聊天程序，今天我们深入探讨如何为这个程序添加本地tools机制，让我们的聊天程序更强大、更实用！

一、工具集成的目标与意义

在第一版程序中，我们实现了基本的多Agent聊天功能。但仅仅发送和接收文本消息显然不够。我们希望通过集成本地工具，让Agent能执行更复杂的任务，例如文件操作、数据查询等。
在这里插入图片描述

二、核心：工具加载与发现机制

1. 工具文件夹结构

我们创建了一个toolbox\localtools文件夹，所有工具模块都放在这里。每个工具模块是一个.py文件，包含具体的工具函数。

2. 动态加载工具的魔法

local_package_path = '.\\toolbox\\localtools'
modules = import_all_py_files_in_folder(local_package_path)
TOOLBOXTOOLS = {}
for module in modules:
    TOOLBOXTOOLS[module.__name__] = []
    functions = get_functions(module)
    for function in functions:
        TOOLBOXTOOLS[module.__name__].append(function.__name__)

这段代码会扫描localtools文件夹，动态加载所有.py文件，并提取其中的函数名称。最终TOOLBOXTOOLS会是一个字典，结构类似：

{
    'findtools': ['find_files_with_powershell_regex', 'find_paths_with_powershell_regex'],
    'filetools': ['open_file_by_path']
}

这样，我们就能清楚知道每个工具模块提供了哪些功能。

三、工具选择与配置界面

在聊天程序中，我们添加了一个工具选择对话框：

dialog = ToolSelectionDialog(TOOLBOXTOOLS, self)
if dialog.exec() == QDialog.Accepted:
    selected_tools = dialog.selected_tools
    stools = []
    for tool in selected_tools:
        stools.append(tool[1])
    tools_str = str(stools)

这个对话框会以树状结构展示所有可用工具，用户可以选择需要的工具。选择结果会以字符串形式保存在Agent配置中。

在这里插入图片描述

四、工具调用的实际案例

1. 文件查找工具

我们实现了find_files_with_powershell_regex函数，它能根据正则表达式查找文件：

def find_files_with_powershell_regex(root_dir: str, regex_pattern: str) -> dict:
    # ...[函数实现部分]...

通过PowerShell命令，这个工具可以在指定目录及其子目录中查找匹配文件名模式的文件。
在这里插入图片描述

2. 文件打开工具

另一个实用工具是open_file_by_path，它能打开指定路径的文件：

def open_file_by_path(file_path: str) -> dict:
    try:
        subprocess.run(['start', '', file_path], shell=True)
        return {"status": "success", "response": f"文件已经打开"}
    except FileNotFoundError:
        return {"status": "error", "response": f"文件未找到: {file_path}"}

这个工具利用系统默认程序打开文件，方便用户快速访问文件内容。

五、工具与Agent的结合

当我们创建新Agent时，可以选择为其配置工具：

def create_new_agent(self):
    # ...[前面代码]...
    dialog = ToolSelectionDialog(TOOLBOXTOOLS, self)
    if dialog.exec() == QDialog.Accepted:
        selected_tools = dialog.selected_tools
        stools = []
        for tool in selected_tools:
            stools.append(tool[1])
        tools_str = str(stools)