{"id":1282,"date":"2025-12-04T11:05:25","date_gmt":"2025-12-04T03:05:25","guid":{"rendered":"http:\/\/www.preluna.xyz\/?p=1282"},"modified":"2025-12-04T11:13:15","modified_gmt":"2025-12-04T03:13:15","slug":"agentservertools","status":"publish","type":"post","link":"http:\/\/www.preluna.xyz\/index.php\/2025\/12\/04\/agentservertools\/preluna\/technology\/career-skills\/open-source-analysis\/","title":{"rendered":"agentserver\\tools"},"content":{"rendered":"\n

file_edit_toolkit.py<\/h2>\n\n\n\n
#!\/usr\/bin\/env python3
# -*- coding: utf-8 -*-
\"\"\"
<\/em>\u6587\u4ef6\u7f16\u8f91\u5de5\u5177\u5305<\/em> - <\/em>\u9002\u914d<\/em>agentserver<\/em>\u67b6\u6784
<\/em>\u63d0\u4f9b\u6587\u4ef6\u8bfb\u5199\u3001\u7f16\u8f91\u529f\u80fd\uff0c\u652f\u6301<\/em>SEARCH\/REPLACE<\/em>\u683c\u5f0f\u7684<\/em>diff<\/em>\u64cd\u4f5c<\/em>
<\/em>\"\"\"
<\/em>
<\/em>import re          # \u6b63\u5219\u8868\u8fbe\u5f0f\uff0c\u7528\u6765\u641c\u7d22\u548c\u66ff\u6362\u6587\u5b57
import shutil      # \u6587\u4ef6\u64cd\u4f5c\u5de5\u5177\uff0c\u7528\u6765\u590d\u5236\u6587\u4ef6
import logging     # \u8bb0\u5f55\u65e5\u5fd7\uff08\u5c31\u50cf\u5199\u65e5\u8bb0\uff0c\u8bb0\u5f55\u53d1\u751f\u4e86\u4ec0\u4e48\uff09
from datetime import datetime  # \u83b7\u53d6\u65f6\u95f4
from pathlib import Path       # \u5904\u7406\u6587\u4ef6\u8def\u5f84
#\u524d5\u884c\u662f\u5bfc\u5165Python\u81ea\u5e26\u7684\u5de5\u5177

from .base_toolkit import AsyncBaseToolkit, register_tool, ToolkitConfig
#\u6700\u540e\u4e00\u884c\u662f\u5bfc\u5165\u81ea\u5df1\u5199\u7684\u5176\u4ed6\u4ee3\u7801\uff08\u4ece\u5f53\u524d\u6587\u4ef6\u5939\u7684base_toolkit.py\u5bfc\u5165\uff09

logger = logging.getLogger(__name__)


#\u521b\u5efa\u4e00\u4e2a\u53ebFileEditToolkit\u7684\u7c7b
#\u5b83\u7ee7\u627f\u81eaAsyncBaseToolkit\uff08\u610f\u601d\u662f\u5b83\u62e5\u6709\u7236\u7c7b\u7684\u6240\u6709\u529f\u80fd\uff09
class FileEditToolkit(AsyncBaseToolkit):
    \"\"\"<\/em>\u6587\u4ef6\u7f16\u8f91\u5de5\u5177\u5305<\/em>\"\"\"
<\/em>
<\/em>    <\/em>def __init__(self, config: ToolkitConfig = None) -> None: #\u5b83\u7ee7\u627f\u81eaAsyncBaseToolkit\uff08\u610f\u601d\u662f\u5b83\u62e5\u6709\u7236\u7c7b\u7684\u6240\u6709\u529f\u80fd\uff09
        super().__init__(config)
        workspace_root = self.config.config.get(\"workspace_root\", \"\/tmp\/\")
        self.setup_workspace(workspace_root)

        self.default_encoding = self.config.config.get(\"default_encoding\", \"utf-8\")
        self.backup_enabled = self.config.config.get(\"backup_enabled\", False)
        logger.info(
            f\"FileEditToolkit\u521d\u59cb\u5316\u5b8c\u6210 - \u5de5\u4f5c\u76ee\u5f55: {self.work_dir}, \u7f16\u7801: {self.default_encoding}\"
        )
    '''
    \u505a\u7684\u4e8b\u60c5\uff1a
     \u5148\u8c03\u7528\u7236\u7c7b\u7684\u8bbe\u7f6e\u65b9\u6cd5
     \u8bbe\u7f6e\u5de5\u4f5c\u76ee\u5f55\uff08\u6587\u4ef6\u5b58\u653e\u7684\u5730\u65b9\uff09
     \u8bbe\u7f6e\u6587\u4ef6\u7f16\u7801\uff08\u9ed8\u8ba4\u7528UTF-8\uff0c\u53ef\u4ee5\u663e\u793a\u4e2d\u6587\uff09
     \u8bbe\u7f6e\u662f\u5426\u5907\u4efd\u6587\u4ef6
     \u8bb0\u5f55\u4e00\u6761\u65e5\u5fd7\u8bf4\u521d\u59cb\u5316\u5b8c\u6210\u4e86
    '''

    def setup_workspace(self, workspace_root: str): #\u521b\u5efa\u4e00\u4e2a\u6587\u4ef6\u5939\u4f5c\u4e3a\u5de5\u4f5c\u7a7a\u95f4
        \"\"\"<\/em>\u8bbe\u7f6e\u5de5\u4f5c\u7a7a\u95f4<\/em>\"\"\"
<\/em>        <\/em>self.work_dir = Path(workspace_root).resolve()  #\u628a\u4f20\u5165\u7684\u8def\u5f84\u53d8\u6210Path\u5bf9\u8c61
        self.work_dir.mkdir(parents=True, exist_ok=True) #\u521b\u5efa\u8fd9\u4e2a\u6587\u4ef6\u5939\uff08\u5982\u679c\u4e0d\u5b58\u5728\u7684\u8bdd\uff09

    def _sanitize_filename(self, filename: str) -> str:  #\u628a\u6587\u4ef6\u540d\u4e2d\u7684\u7279\u6b8a\u5b57\u7b26\uff08\u9664\u4e86\u5b57\u6bcd\u3001\u6570\u5b57\u3001\u4e0b\u5212\u7ebf\u3001\u6a2a\u6760\u3001\u70b9\uff09\u90fd\u53d8\u6210\u4e0b\u5212\u7ebf
        \"\"\"<\/em>\u6e05\u7406\u6587\u4ef6\u540d\uff0c\u79fb\u9664\u4e0d\u5b89\u5168\u5b57\u7b26<\/em>\"\"\"
<\/em>        <\/em>safe = re.sub(r\"[^\\w\\-.]\", \"_\", filename) #\u4f8b\u5b50\uff1atest&file.txt \u2192 test_file.txt
        return safe

    def _resolve_filepath(self, file_path: str) -> Path: 
        \"\"\"<\/em>\u89e3\u6790\u6587\u4ef6\u8def\u5f84\uff0c\u5904\u7406\u76f8\u5bf9\u8def\u5f84\u548c\u5b89\u5168\u68c0\u67e5<\/em>\"\"\"
<\/em>        <\/em>path_obj = Path(file_path)
        if not path_obj.is_absolute():           #\u5982\u679c\u662f\u76f8\u5bf9\u8def\u5f84\uff0c\u5c31\u52a0\u4e0a\u5de5\u4f5c\u76ee\u5f55\u53d8\u6210\u7edd\u5bf9\u8def\u5f84
            path_obj = self.work_dir \/ path_obj  

        sanitized_filename = self._sanitize_filename(path_obj.name)     #\u6e05\u7406\u6587\u4ef6\u540d
        path_obj = path_obj.parent \/ sanitized_filename
        resolved_path = path_obj.resolve()
        self._create_backup(resolved_path)  #\u5982\u679c\u9700\u8981\uff0c\u521b\u5efa\u5907\u4efd
        return resolved_path  #\u8fd4\u56de\u5904\u7406\u540e\u7684\u8def\u5f84

    def _create_backup(self, file_path: Path) -> None:
        \"\"\"<\/em>\u521b\u5efa\u6587\u4ef6\u5907\u4efd<\/em>\"\"\"  <\/em>#\u5728\u4fee\u6539\u6587\u4ef6\u524d\u5148\u5907\u4efd
        if not self.backup_enabled or not file_path.exists():   #\u53ea\u6709\u5f53backup_enabled\u4e3aTrue\u4e14\u6587\u4ef6\u5b58\u5728\u65f6\u624d\u5907\u4efd
            return

        timestamp = datetime.now().strftime(\"%Y%m%d_%H%M%S\")
        backup_path = file_path.parent \/ f\"{file_path.name}.{timestamp}.bak\"  #\u4efd\u6587\u4ef6\u547d\u540d\uff1a\u539f\u6587\u4ef6\u540d + \u65f6\u95f4 + .bak
        shutil.copy2(file_path, backup_path)
        logger.info(f\"\u521b\u5efa\u5907\u4efd\u6587\u4ef6: {backup_path}\")

    @register_tool
    async def edit_file(self, path: str, diff: str) -> str:
        \"\"\"<\/em>\u7f16\u8f91\u6587\u4ef6\uff0c\u4f7f\u7528<\/em>SEARCH\/REPLACE<\/em>\u683c\u5f0f\u7684<\/em>diff
<\/em>
<\/em>        <\/em>\u53c2\u6570<\/em>:
<\/em>            path (str): <\/em>\u8981\u7f16\u8f91\u7684\u6587\u4ef6\u8def\u5f84<\/em>
<\/em>            diff (str): SEARCH\/REPLACE<\/em>\u683c\u5f0f\u7684<\/em>diff<\/em>\u5185\u5bb9\uff0c\u683c\u5f0f\u5982\u4e0b<\/em>:
<\/em>                ```
<\/em>                <<<<<<< SEARCH
<\/em>                [<\/em>\u8981\u67e5\u627e\u7684\u786e\u5207\u5185\u5bb9<\/em>]
<\/em>                =======
<\/em>                [<\/em>\u8981\u66ff\u6362\u7684\u65b0\u5185\u5bb9<\/em>]
<\/em>                >>>>>>> REPLACE
<\/em>                ```
<\/em>        \"\"\"
<\/em>        <\/em>try:
            resolved_path = self._resolve_filepath(path)

            with open(resolved_path, encoding=self.default_encoding) as f:
                content = f.read()
            modified_content = content
            pattern = r\"<<<<<<< SEARCH\\n(.*?)\\n=======\\n(.*?)\\n>>>>>>> REPLACE\"
            matches = re.findall(pattern, diff, re.DOTALL)
            if not matches:
                return \"\u9519\u8bef\uff01\u5728\u63d0\u4f9b\u7684diff\u4e2d\u672a\u627e\u5230\u6709\u6548\u7684SEARCH\/REPLACE\u5757\"

            # \u5e94\u7528\u6bcf\u4e2a\u641c\u7d22\/\u66ff\u6362\u5bf9
            for search_text, replace_text in matches:
                if search_text in modified_content:
                    modified_content = modified_content.replace(search_text, replace_text)
                else:
                    logger.warning(f\"\u5728\u6587\u4ef6\u4e2d\u672a\u627e\u5230\u641c\u7d22\u6587\u672c: {search_text[:50]}...\")

            with open(resolved_path, \"w\", encoding=self.default_encoding) as f:
                f.write(modified_content)
            return f\"\u6210\u529f\u7f16\u8f91\u6587\u4ef6: {resolved_path}\"
        except Exception as e:
            logger.error(f\"\u7f16\u8f91\u6587\u4ef6\u5931\u8d25: {e}\")
            return f\"\u7f16\u8f91\u6587\u4ef6\u5931\u8d25: {str(e)}\"

    @register_tool
    async def write_file(self, path: str, file_text: str) -> str:
        \"\"\"<\/em>\u5199\u5165\u6587\u672c\u5185\u5bb9\u5230\u6587\u4ef6
<\/em>
<\/em>        \u53c2\u6570<\/em>:
<\/em>            path (str): <\/em>\u8981\u5199\u5165\u7684\u6587\u4ef6\u8def\u5f84<\/em>
<\/em>            file_text (str): <\/em>\u8981\u5199\u5165\u7684\u5b8c\u6574\u6587\u672c\u5185\u5bb9<\/em>
<\/em>        \"\"\"
<\/em>        <\/em>try:
            path_obj = self._resolve_filepath(path)
            path_obj.parent.mkdir(parents=True, exist_ok=True)
            path_obj.write_text(file_text, encoding=self.default_encoding)
            return f\"\u6210\u529f\u5199\u5165\u6587\u4ef6: {path_obj}\"
        except Exception as e:
            logger.error(f\"\u5199\u5165\u6587\u4ef6\u5931\u8d25: {e}\")
            return f\"\u5199\u5165\u6587\u4ef6\u5931\u8d25: {str(e)}\"

    @register_tool
    async def read_file(self, path: str) -> str:
        \"\"\"<\/em>\u8bfb\u53d6\u5e76\u8fd4\u56de\u6587\u4ef6\u5185\u5bb9
<\/em>
<\/em>        \u53c2\u6570<\/em>:
<\/em>            path (str): <\/em>\u8981\u8bfb\u53d6\u7684\u6587\u4ef6\u8def\u5f84<\/em>
<\/em>        \"\"\"
<\/em>        <\/em>try:
            path_obj = self._resolve_filepath(path)
            if not path_obj.exists():
                return f\"\u6587\u4ef6\u4e0d\u5b58\u5728: {path_obj}\"
            return path_obj.read_text(encoding=self.default_encoding)
        except Exception as e:
            logger.error(f\"\u8bfb\u53d6\u6587\u4ef6\u5931\u8d25: {e}\")
            return f\"\u8bfb\u53d6\u6587\u4ef6\u5931\u8d25: {str(e)}\"

    @register_tool
    async def list_files(self, directory: str = \".\") -> str:
        \"\"\"<\/em>\u5217\u51fa\u76ee\u5f55\u4e2d\u7684\u6587\u4ef6
<\/em>
<\/em>        \u53c2\u6570<\/em>:
<\/em>            directory (str): <\/em>\u8981\u5217\u51fa\u7684\u76ee\u5f55\u8def\u5f84\uff0c\u9ed8\u8ba4\u4e3a\u5f53\u524d\u76ee\u5f55<\/em>
<\/em>        \"\"\"
<\/em>        <\/em>try:
            if directory == \".\":
                target_dir = self.work_dir
            else:
                target_dir = self.work_dir \/ directory

            if not target_dir.exists():
                return f\"\u76ee\u5f55\u4e0d\u5b58\u5728: {target_dir}\"

            files = []
            for item in target_dir.iterdir():
                if item.is_file():
                    files.append(f\"\u6587\u4ef6: {item.name}\")
                elif item.is_dir():
                    files.append(f\"\u76ee\u5f55: {item.name}\/\")

            if not files:
                return f\"\u76ee\u5f55\u4e3a\u7a7a: {target_dir}\"

            return f\"\u76ee\u5f55 {target_dir} \u5185\u5bb9:\\n\" + \"\\n\".join(files)
        except Exception as e:
            logger.error(f\"\u5217\u51fa\u6587\u4ef6\u5931\u8d25: {e}\")
            return f\"\u5217\u51fa\u6587\u4ef6\u5931\u8d25: {str(e)}\"

    @register_tool
    async def create_directory(self, path: str) -> str:
        \"\"\"<\/em>\u521b\u5efa\u76ee\u5f55
<\/em>
<\/em>        \u53c2\u6570<\/em>:
<\/em>            path (str): <\/em>\u8981\u521b\u5efa\u7684\u76ee\u5f55\u8def\u5f84<\/em>
<\/em>        \"\"\"
<\/em>        <\/em>try:
            path_obj = self.work_dir \/ path
            path_obj.mkdir(parents=True, exist_ok=True)
            return f\"\u6210\u529f\u521b\u5efa\u76ee\u5f55: {path_obj}\"
        except Exception as e:
            logger.error(f\"\u521b\u5efa\u76ee\u5f55\u5931\u8d25: {e}\")
            return f\"\u521b\u5efa\u76ee\u5f55\u5931\u8d25: {str(e)}\"

    @register_tool
    async def delete_file(self, path: str) -> str:
        \"\"\"<\/em>\u5220\u9664\u6587\u4ef6
<\/em>
<\/em>        \u53c2\u6570<\/em>:
<\/em>            path (str): <\/em>\u8981\u5220\u9664\u7684\u6587\u4ef6\u8def\u5f84<\/em>
<\/em>        \"\"\"
<\/em>        <\/em>try:
            path_obj = self._resolve_filepath(path)
            if not path_obj.exists():
                return f\"\u6587\u4ef6\u4e0d\u5b58\u5728: {path_obj}\"
            path_obj.unlink()
            return f\"\u6210\u529f\u5220\u9664\u6587\u4ef6: {path_obj}\"
        except Exception as e:
            logger.error(f\"\u5220\u9664\u6587\u4ef6\u5931\u8d25: {e}\")
            return f\"\u5220\u9664\u6587\u4ef6\u5931\u8d25: {str(e)}\"<\/code><\/pre>\n\n\n\n
base_toolkit.py<\/h2>\n\n\n\n
#!\/usr\/bin\/env python3\n# -*- coding: utf-8 -*-\n\"\"\"\n\u57fa\u7840\u5de5\u5177\u5305\u7c7b - \u9002\u914dagentserver\u67b6\u6784\n\"\"\"\n\nimport logging\nfrom collections.abc import Callable  #Callable\uff1a\u8868\u793a\"\u53ef\u4ee5\u88ab\u8c03\u7528\u7684\u4e1c\u897f\"\uff0c\u6bd4\u5982\u51fd\u6570\nfrom typing import Any, Dict, Optional\n'''\ntyping\u6a21\u5757\uff1a\u7528\u6765\u505a\u7c7b\u578b\u63d0\u793a\u7684\uff0c\u544a\u8bc9Python\u6bcf\u4e2a\u53d8\u91cf\u5e94\u8be5\u662f\u4ec0\u4e48\u7c7b\u578b\n   Any\uff1a\u53ef\u4ee5\u662f\u4efb\u4f55\u7c7b\u578b\n   Dict\uff1a\u5b57\u5178\u7c7b\u578b\n   Optional\uff1a\u53ef\u9009\u7c7b\u578b\uff08\u53ef\u80fd\u6709\u503c\uff0c\u53ef\u80fd\u6ca1\u6709\uff09\n'''\n\nlogger = logging.getLogger(__name__)\n\n\n#\u8fd9\u662f\u4e00\u4e2a\u7528\u6765\u5b58\u50a8\u914d\u7f6e\u4fe1\u606f\u7684\u7c7b\u3002\nclass ToolkitConfig:\n    \"\"\"\u5de5\u5177\u5305\u914d\u7f6e\u7c7b\"\"\"\n    \n    def __init__(self, config: Dict[str, Any] = None, name: str = None):\n        self.config = config or {}\n        self.name = name or self.__class__.__name__\n        self.mode = \"builtin\"\n        self.activated_tools = None\n        '''\n        \u8fd9\u4e2a\u7c7b\u67094\u4e2a\u5c5e\u6027\uff08\u53ef\u4ee5\u7406\u89e3\u4e3a4\u4e2a\u76d2\u5b50\uff09\uff1a\n          1.self.config - \u914d\u7f6e\u5b57\u5178\n             \u5982\u679c\u4f20\u5165\u4e86\u914d\u7f6e\uff0c\u5c31\u7528\u4f20\u5165\u7684\n             \u5982\u679c\u6ca1\u6709\u4f20\u5165\uff0c\u5c31\u7528\u7a7a\u5b57\u5178{}\n          2.self.name - \u540d\u5b57\n             \u5982\u679c\u4f20\u5165\u4e86\u540d\u5b57\uff0c\u5c31\u7528\u4f20\u5165\u7684\n             \u5982\u679c\u6ca1\u6709\uff0c\u5c31\u7528\u7c7b\u7684\u540d\u5b57\uff08\"ToolkitConfig\"\uff09\n          3.self.mode - \u6a21\u5f0f\n             \u56fa\u5b9a\u4e3a\"builtin\"\uff08\u5185\u7f6e\u6a21\u5f0f\uff09\n          4.self.activated_tools - \u6fc0\u6d3b\u7684\u5de5\u5177\n             \u521d\u59cb\u4e3aNone\uff08\u7a7a\uff09\n             \u7528\u6765\u51b3\u5b9a\u54ea\u4e9b\u5de5\u5177\u53ef\u4ee5\u7528\n        '''\n    \n\nclass AsyncBaseToolkit:\n    \"\"\"\u5de5\u5177\u5305\u57fa\u7c7b - \u9002\u914dagentserver\u67b6\u6784\"\"\"\n\n    def __init__(self, config: ToolkitConfig | dict | None = None):\n        # \u770b\u770b\u4f20\u5165\u7684config\u662f\u4e0d\u662fToolkitConfig\u7c7b\u578b,\u5982\u679c\u4e0d\u662f\uff1a\u5c31\u521b\u5efa\u4e00\u4e2a\u65b0\u7684ToolkitConfig\u5bf9\u8c61,\u5982\u679c\u662f\uff1a\u5c31\u76f4\u63a5\u4f7f\u7528\n        if not isinstance(config, ToolkitConfig): \n            config = config or {}\n            config = ToolkitConfig(config=config, name=self.__class__.__name__)\n\n        self.config: ToolkitConfig = config               #self.config\uff1a\u5b58\u50a8\u914d\u7f6e\u5bf9\u8c61\n        self._tools_map: dict[str, Callable] = None       #self._tools_map\uff1a\u5b58\u50a8\u5de5\u5177\u6620\u5c04\uff08\u521d\u59cb\u4e3aNone\uff09\n\n    @property\n    # \u8fd9\u4e2a\u5c5e\u6027\u4f1a\u81ea\u52a8\u6536\u96c6\u6240\u6709\u88ab@register_tool\u88c5\u9970\u8fc7\u7684\u65b9\u6cd5\n    def tools_map(self) -> dict[str, Callable]:  #\u8fd4\u56de\u7c7b\u578b\uff1adict[str, Callable]\uff08\u5b57\u5178\uff0c\u952e\u662f\u5b57\u7b26\u4e32\uff0c\u503c\u662f\u53ef\u8c03\u7528\u5bf9\u8c61\uff09\n        #\u6ce8\u610f\uff1a\u56e0\u4e3a\u6709 @property\uff0c\u6240\u4ee5\u5b9e\u9645\u4e0a\u662f\u4e2a\"\u5c5e\u6027\"\n        \"\"\"\u61d2\u52a0\u8f7d\u5de5\u5177\u6620\u5c04 - \u6536\u96c6\u901a\u8fc7@register_tool\u6ce8\u518c\u7684\u5de5\u5177\"\"\"\n        if self._tools_map is None #\u68c0\u67e5 self._tools_map \u662f\u5426\u4e3a None.\u5982\u679c\u5df2\u7ecf\u662f\u5b57\u5178\uff0c\u8bf4\u660e\u5df2\u7ecf\u6536\u96c6\u8fc7\u4e86\uff0c\u4e0d\u7528\u518d\u6536\u96c6\n            self._tools_map = {} #\u5982\u679c\u8fd8\u6ca1\u6536\u96c6\u8fc7\uff1a\u521b\u5efa\u4e00\u4e2a\u7a7a\u5b57\u5178,self._tools_map \u662f\u79c1\u6709\u5c5e\u6027\uff0c\u5b58\u50a8\u5de5\u5177\u6620\u5c04\n            # \u904d\u5386\u7c7b\u65b9\u6cd5\uff0c\u627e\u5230\u5e26\u6709@register_tool\u88c5\u9970\u5668\u7684\u65b9\u6cd5\n            for attr_name in dir(self):  #dir(self)\uff1a\u83b7\u53d6\u5bf9\u8c61\u6240\u6709\u5c5e\u6027\u548c\u65b9\u6cd5\u7684\u540d\u79f0\u5217\u8868,\u5faa\u73af\uff1a\u904d\u5386\u6bcf\u4e2a\u5c5e\u6027\u548c\u65b9\u6cd5\u7684\u540d\u79f0\n                #\u4f8b\u5b50\uff1a\u5982\u679c\u5bf9\u8c61\u6709 edit_file, read_file \u65b9\u6cd5\uff0cdir(self) \u4f1a\u8fd4\u56de\u5305\u542b\u8fd9\u4e9b\u540d\u5b57\u7684\u5217\u8868\n                attr = getattr(self, attr_name)  #getattr(self, attr_name)\uff1a\u6839\u636e\u540d\u79f0\u83b7\u53d6\u5b9e\u9645\u7684\u5c5e\u6027\u6216\u65b9\u6cd5.\u5982\u679c attr_name \u662f \"edit_file\"\uff0c\u90a3\u4e48 attr \u5c31\u662f edit_file \u65b9\u6cd5\u672c\u8eab\n                #\u8fd9\u662f\u5173\u952e\u5224\u65ad\u6761\u4ef6\uff0c\u5206\u4e24\u90e8\u5206\uff1a\n                if callable(attr) and getattr(attr, \"_is_tool\", False):\n                    '''\n                    \u7b2c\u4e00\u90e8\u5206\uff1acallable(attr)\n                       \u68c0\u67e5 attr \u662f\u5426\"\u53ef\u8c03\u7528\"\n                       \u4ec0\u4e48\u662f\"\u53ef\u8c03\u7528\"\uff1f\u5c31\u662f\u53ef\u4ee5\u52a0\u62ec\u53f7\u6267\u884c\u7684\u4e1c\u897f\n                             \u51fd\u6570 \u221a \u53ef\u8c03\u7528\n                             \u65b9\u6cd5 \u221a \u53ef\u8c03\u7528\n                             \u5b57\u7b26\u4e32 \u00d7 \u4e0d\u53ef\u8c03\u7528\n                            \u6574\u6570 \u00d7 \u4e0d\u53ef\u8c03\u7528\n\n                    \u7b2c\u4e8c\u90e8\u5206\uff1agetattr(attr, \"_is_tool\", False)\n                       getattr(attr, \"_is_tool\", False)\uff1a\u5c1d\u8bd5\u83b7\u53d6 attr \u7684 _is_tool \u5c5e\u6027\n                       \u4e09\u4e2a\u53c2\u6570\uff1a\n                          attr\uff1a\u8981\u68c0\u67e5\u7684\u5bf9\u8c61\n                          \"_is_tool\"\uff1a\u8981\u83b7\u53d6\u7684\u5c5e\u6027\u540d\n                          False\uff1a\u5982\u679c\u5c5e\u6027\u4e0d\u5b58\u5728\uff0c\u8fd4\u56de False\n                      \u4f5c\u7528\uff1a\u68c0\u67e5\u8fd9\u4e2a\u51fd\u6570\u662f\u5426\u88ab @register_tool \u88c5\u9970\u8fc7\n\n                   \u7ec4\u5408\u8d77\u6765\uff1a\n                     \u5982\u679c\u662f\u53ef\u8c03\u7528\u7684\uff08\u662f\u51fd\u6570\/\u65b9\u6cd5\uff09\n                     \u5e76\u4e14\u6709 _is_tool \u5c5e\u6027\uff08\u503c\u4e3a True\uff09\n                     \u90a3\u4e48\u8bf4\u660e\u8fd9\u662f\u4e00\u4e2a\u5de5\u5177\u65b9\u6cd5\n                    '''\n                    self._tools_map[attr._tool_name] = attr #\u5982\u679c\u901a\u8fc7\u4e86\u4e0a\u9762\u7684\u68c0\u67e5\uff1a\u628a\u5de5\u5177\u540d\u4f5c\u4e3a\u952e\uff1aattr._tool_name,\u628a\u5de5\u5177\u51fd\u6570\u4f5c\u4e3a\u503c\uff1aattr,\u5b58\u5165\u5b57\u5178 self._tools_map\n        return self._tools_map\n       '''\n        \u5982\u4f55\u5de5\u4f5c\uff1f\n          1.\u7b2c\u4e00\u6b21\u8bbf\u95eetools_map\u65f6\uff0cself._tools_map\u662fNone\n          2\u7136\u540e\u5b83\u5f00\u59cb\u641c\u7d22\uff1a\n             \u904d\u5386\u8fd9\u4e2a\u7c7b\u7684\u6240\u6709\u5c5e\u6027\u548c\u65b9\u6cd5\n             \u68c0\u67e5\u6bcf\u4e2a\u5c5e\u6027\u662f\u5426\"\u53ef\u8c03\u7528\"\uff08\u662f\u51fd\u6570\uff09\u5e76\u4e14\u6709_is_tool\u6807\u8bb0\n             \u5982\u679c\u6709\uff0c\u5c31\u628a\u5b83\u6dfb\u52a0\u5230\u5de5\u5177\u6620\u5c04\u4e2d\n        '''\n    \n    \n    #\u5b9a\u4e49\u4e00\u4e2a\u65b9\u6cd5\uff0c\u8fd4\u56de\u5b57\u5178\uff08\u5de5\u5177\u540d -> \u5de5\u5177\u51fd\u6570\uff09\n    def get_tools_map_func(self) -> dict[str, Callable]:\n        \"\"\"\u83b7\u53d6\u5de5\u5177\u6620\u5c04\uff0c\u6839\u636e\u914d\u7f6e\u8fc7\u6ee4\u6fc0\u6d3b\u7684\u5de5\u5177\"\"\"\n        if self.config.activated_tools: #\u68c0\u67e5\u914d\u7f6e\u4e2d\u662f\u5426\u6709 activated_tools,\u5982\u679c\u6709\uff0c\u8bf4\u660e\u9700\u8981\u8fc7\u6ee4\uff08\u53ea\u8fd4\u56de\u6307\u5b9a\u7684\u5de5\u5177\uff09,\u5982\u679c\u6ca1\u6709\uff0c\u8fd4\u56de\u6240\u6709\u5de5\u5177\n            #\u8fd9\u662f\u6700\u96be\u7406\u89e3\u7684\u90e8\u5206\uff0c\u6211\u4eec\u62c6\u5f00\u770b\uff1aall()\uff1a\u68c0\u67e5\u6240\u6709\u6761\u4ef6\u662f\u5426\u90fd\u4e3a True\n            #\u91cc\u9762\u7684\u8868\u8fbe\u5f0f\uff1atool_name in self.tools_map for tool_name in self.config.activated_tools,\u610f\u601d\uff1a\u5bf9\u4e8e\u914d\u7f6e\u4e2d\u7684\u6bcf\u4e2a\u5de5\u5177\u540d\uff0c\u68c0\u67e5\u5b83\u662f\u5426\u5728 tools_map \u4e2d\u5b58\u5728\n            #\u8fd4\u56de\u503c\uff1a\u5982\u679c\u6240\u6709\u5de5\u5177\u540d\u90fd\u5b58\u5728\uff0c\u8fd4\u56deTrue\uff1b\u5426\u5219\u8fd4\u56deFalse\n            '''assert \u8bed\u53e5\uff1a\n            assert \u6761\u4ef6, \"\u9519\u8bef\u4fe1\u606f\"\n            \u4f5c\u7528\uff1a\u5982\u679c\u6761\u4ef6\u4e3a False\uff0c\u5c31\u62a5\u9519\uff0c\u663e\u793a\u9519\u8bef\u4fe1\u606f\\\u8fd9\u91cc\uff1a\u5982\u679c\u914d\u7f6e\u7684\u5de5\u5177\u540d\u6709\u4e0d\u5b58\u5728\u4e8e\u5b9e\u9645\u5de5\u5177\u4e2d\u7684\uff0c\u5c31\u62a5\u9519\n            '''\n            assert all(tool_name in self.tools_map for tool_name in self.config.activated_tools), (\n                f\"\u9519\u8bef\u914d\u7f6e\u6fc0\u6d3b\u5de5\u5177: {self.config.activated_tools}! \u53ef\u7528\u5de5\u5177: {self.tools_map.keys()}\" #\u683c\u5f0f\u5316\u5b57\u7b26\u4e32\uff0c\u663e\u793a\uff1a\u914d\u7f6e\u4e86\u54ea\u4e9b\u5de5\u5177,\u5b9e\u9645\u6709\u54ea\u4e9b\u5de5\u5177\u53ef\u7528\n            )\n            tools_map = {tool_name: self.tools_map[tool_name] for tool_name in self.config.activated_tools}#\u5feb\u901f\u521b\u5efa\u5b57\u5178\u7684\u65b9\u6cd5\n            '''\n            \u62c6\u89e3\uff1a\n            # \u4f2a\u4ee3\u7801\u7406\u89e3\uff1a\n            tools_map = {}\n            for tool_name in self.config.activated_tools:\n                tools_map[tool_name] = self.tools_map[tool_name]\n            '''\n            #\u7ed3\u679c\uff1a\u521b\u5efa\u4e00\u4e2a\u65b0\u5b57\u5178\uff0c\u53ea\u5305\u542b\u914d\u7f6e\u4e2d\u6307\u5b9a\u7684\u5de5\u5177\n        else: #\u5426\u5219\uff08\u5982\u679c\u6ca1\u6709\u914d\u7f6e activated_tools\uff09,\u8fd4\u56de\u6240\u6709\u5de5\u5177\n            tools_map = self.tools_map\n        return tools_map\n    '''\n    \u529f\u80fd\u89e3\u91ca\uff1a\n      \u83b7\u53d6\u5de5\u5177\u6620\u5c04\n      \u5982\u679c\u914d\u7f6e\u4e86activated_tools\uff08\u6fc0\u6d3b\u7684\u5de5\u5177\u5217\u8868\uff09\uff0c\u5c31\u53ea\u8fd4\u56de\u8fd9\u4e9b\u5de5\u5177\n      \u5982\u679c\u6ca1\u6709\u914d\u7f6e\uff0c\u5c31\u8fd4\u56de\u6240\u6709\u5de5\u5177\n    '''\n\n    def get_tools_list(self) -> list[Dict[str, Any]]: #\u5b9a\u4e49\u4e00\u4e2a\u65b9\u6cd5\uff0c\u8fd4\u56de\u5217\u8868,\u5217\u8868\u4e2d\u7684\u6bcf\u4e2a\u5143\u7d20\u662f\u5b57\u5178\n        \"\"\"\u83b7\u53d6\u5de5\u5177\u5217\u8868 - \u9002\u914dagentserver\u683c\u5f0f\"\"\"\n        tools_map = self.get_tools_map_func() #\u8c03\u7528\u521a\u624d\u7684\u65b9\u6cd5\uff0c\u83b7\u53d6\u5de5\u5177\u6620\u5c04\uff08\u53ef\u80fd\u5df2\u8fc7\u6ee4\uff09\n        tools = []  #\u521b\u5efa\u4e00\u4e2a\u7a7a\u5217\u8868\uff0c\u51c6\u5907\u5b58\u50a8\u5de5\u5177\u4fe1\u606f\n        for tool_name, tool_func in tools_map.items(): #\u904d\u5386\u5de5\u5177\u5b57\u5178\u7684\u6bcf\u4e00\u9879\n            #\u6bcf\u6b21\u5faa\u73af\u5f97\u5230\uff1atool_name\uff1a\u5de5\u5177\u540d\uff08\u5b57\u7b26\u4e32\uff09,tool_func\uff1a\u5de5\u5177\u51fd\u6570\uff08\u53ef\u8c03\u7528\u5bf9\u8c61\uff09\n            tools.append({   #\u8fd9\u662f\u5411\u5217\u8868\u6dfb\u52a0\u4e00\u4e2a\u5b57\u5178\uff1a\u5b57\u5178\u5305\u542b\u4e09\u4e2a\u952e\uff1a\n                \"name\": tool_name,  \n                \"function\": tool_func,\n                \"description\": tool_func.__doc__ or f\"\u5de5\u5177: {tool_name}\" #tool_func.__doc__ \u83b7\u53d6\u51fd\u6570\u7684\u6587\u6863\u5b57\u7b26\u4e32\n                #or \u8fd0\u7b97\u7b26\uff1a\u5982\u679c\u5de6\u8fb9\u4e3a\u771f\uff08\u6709\u6587\u6863\uff09\uff0c\u5c31\u7528\u5de6\u8fb9\u7684,\u5982\u679c\u5de6\u8fb9\u4e3a\u5047\uff08\u7a7a\u5b57\u7b26\u4e32\u6216None\uff09\uff0c\u5c31\u7528\u53f3\u8fb9\u7684\n            })\n        return tools\n    '''\n    \u529f\u80fd\u89e3\u91ca\uff1a\n    \u628a\u5de5\u5177\u8f6c\u6362\u6210\u5217\u8868\u683c\u5f0f\uff0c\u6bcf\u4e2a\u5de5\u5177\u5305\u542b\uff1a\n      1.\u540d\u5b57\uff08name\uff09\n      2.\u51fd\u6570\uff08function\uff09\n      3.\u63cf\u8ff0\uff08description\uff09\n         \u7528\u51fd\u6570\u7684\u6587\u6863\u5b57\u7b26\u4e32\uff08__doc__\uff09\n         \u5982\u679c\u6ca1\u6709\u6587\u6863\uff0c\u5c31\u663e\u793a\"\u5de5\u5177: \u5de5\u5177\u540d\"\n    \u4e3a\u4ec0\u4e48\u9700\u8981\u8fd9\u4e2a\uff1f\n      \u6709\u4e9b\u7cfb\u7edf\u9700\u8981\u5de5\u5177\u4ee5\u7279\u5b9a\u683c\u5f0f\u63d0\u4f9b\n      \u8fd9\u4e2a\u683c\u5f0f\u8ba9\u5176\u4ed6\u7cfb\u7edf\u77e5\u9053\u6bcf\u4e2a\u5de5\u5177\u662f\u5e72\u4ec0\u4e48\u7684\n    '''\n\n    async def call_tool(self, name: str, arguments: dict) -> str:\n        \"\"\"\u901a\u8fc7\u540d\u79f0\u8c03\u7528\u5de5\u5177\"\"\"\n        tools_map = self.get_tools_map_func() #\u83b7\u53d6\u5de5\u5177\u6620\u5c04\n        if name not in tools_map:\n            raise ValueError(f\"\u5de5\u5177 {name} \u672a\u627e\u5230\") #\u5982\u679c\u4e0d\u5b58\u5728\uff0c\u629b\u51fa\u9519\u8bef\n        tool = tools_map[name]\n        #\u8fd9\u662f\u5173\u952e\u8c03\u7528\u90e8\u5206\uff1a\n        return await tool(**arguments) #**arguments\uff1a** \u662f\"\u89e3\u5305\"\u8fd0\u7b97\u7b26,\u628a\u5b57\u5178\u89e3\u5305\u6210\u5173\u952e\u5b57\u53c2\u6570\n    '''\n    \u5de5\u4f5c\u6b65\u9aa4\uff1a\n     \u83b7\u53d6\u5de5\u5177\u6620\u5c04\n     \u68c0\u67e5\u8981\u8c03\u7528\u7684\u5de5\u5177\u662f\u5426\u5b58\u5728\n     \u5982\u679c\u4e0d\u5b58\u5728\uff0c\u62a5\u9519\n     \u5982\u679c\u5b58\u5728\uff0c\u8c03\u7528\u8fd9\u4e2a\u5de5\u5177\n     \u628a\u53c2\u6570\u4f20\u7ed9\u5de5\u5177\n     \u8fd4\u56de\u7ed3\u679c\n    '''\n\n#\u8fd9\u662f\u4e00\u4e2a\u88c5\u9970\u5668\uff0c\u7528\u6765\u6807\u8bb0\u54ea\u4e9b\u65b9\u6cd5\u662f\u5de5\u5177\ndef register_tool(name: str = None):\n'''\n \u4f7f\u7528\u65b9\u6cd5\uff1a\n   1.\u6700\u7b80\u5355\u7684\u65b9\u5f0f\uff1a@register_tool\n       \u4f7f\u7528\u51fd\u6570\u672c\u8eab\u7684\u540d\u5b57\u4f5c\u4e3a\u5de5\u5177\u540d\n   2.\u4f7f\u7528\u62ec\u53f7\uff1a@register_tool()\n       \u540c\u4e0a\uff0c\u4f7f\u7528\u51fd\u6570\u540d\n   3.\u81ea\u5b9a\u4e49\u540d\u5b57\uff1a@register_tool(\"custom_name\")\n       \u4f7f\u7528\u81ea\u5b9a\u4e49\u540d\u5b57\u4f5c\u4e3a\u5de5\u5177\u540d\n'''\n    \"\"\"\u88c5\u9970\u5668\uff1a\u5c06\u65b9\u6cd5\u6ce8\u518c\u4e3a\u5de5\u5177\n    \n    \u7528\u6cd5:\n        @register_tool  # \u4f7f\u7528\u65b9\u6cd5\u540d\n        @register_tool()  # \u4f7f\u7528\u65b9\u6cd5\u540d  \n        @register_tool(\"custom_name\")  # \u4f7f\u7528\u81ea\u5b9a\u4e49\u540d\u79f0\n        \n    \u53c2\u6570:\n        name (str, optional): \u5de5\u5177\u540d\u79f0\n    \"\"\"\n\n    def decorator(func: Callable):\n        if isinstance(name, str): \n            tool_name = name\n        else:\n            tool_name = func.__name__\n        func._is_tool = True     # \u6807\u8bb0\u8fd9\u4e2a\u51fd\u6570\u662f\u5de5\u5177\n        func._tool_name = tool_name     # \u8bbe\u7f6e\u5de5\u5177\u540d\n        return func\n    '''\n    \u5173\u952e\u70b9\uff1a\n     1.\u7ed9\u51fd\u6570\u6dfb\u52a0_is_tool = True\u6807\u8bb0\n     2.\u7ed9\u51fd\u6570\u6dfb\u52a0_tool_name\u5c5e\u6027\n    3.\u8fd9\u6837tools_map\u5c5e\u6027\u5c31\u80fd\u8bc6\u522b\u8fd9\u4e2a\u51fd\u6570\u662f\u5de5\u5177\u4e86\n   \u6bd4\u55bb\u7406\u89e3\uff1a\n   \u8fd9\u4e2a\u88c5\u9970\u5668\u5c31\u50cf\u7ed9\u5de5\u5177\u8d34\u6807\u7b7e\uff1a\n     \u8d34\u4e0a_is_tool\u6807\u7b7e\uff1a\u8868\u793a\u8fd9\u662f\u4e2a\u5de5\u5177\n     \u8d34\u4e0a_tool_name\u6807\u7b7e\uff1a\u5199\u6e05\u695a\u5de5\u5177\u53eb\u4ec0\u4e48\u540d\u5b57\n     \u8d34\u4e86\u6807\u7b7e\u7684\u5de5\u5177\u624d\u80fd\u88ab\u5de5\u5177\u7bb1\u8bc6\u522b\u548c\u6536\u96c6\n    '''\n\n    if callable(name): #\u5904\u7406\u88c5\u9970\u5668\u7684\u4e0d\u540c\u4f7f\u7528\u65b9\u5f0f\u3002\n        return decorator(name)\n    return decorator\n    '''\n    \u88c5\u9970\u5668\u7684\u4e24\u79cd\u8c03\u7528\u65b9\u5f0f\uff1a\n      \u65e0\u62ec\u53f7\uff1a\u88c5\u9970\u5668\u76f4\u63a5\u63a5\u6536\u51fd\u6570\n      \u6709\u62ec\u53f7\uff1a\u88c5\u9970\u5668\u5148\u63a5\u6536\u53c2\u6570\uff0c\u8fd4\u56de\u53e6\u4e00\u4e2a\u51fd\u6570\uff0c\u518d\u63a5\u6536\u51fd\u6570\n      '''\n<\/code><\/pre>\n","protected":false},"excerpt":{"rendered":"
file_edit_toolkit.py base_toolkit.py<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[27,55],"tags":[],"class_list":["post-1282","post","type-post","status-publish","format-standard","hentry","category-open-source-analysis","category-technology"],"_links":{"self":[{"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/posts\/1282","targetHints":{"allow":["GET"]}}],"collection":[{"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/comments?post=1282"}],"version-history":[{"count":2,"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/posts\/1282\/revisions"}],"predecessor-version":[{"id":1284,"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/posts\/1282\/revisions\/1284"}],"wp:attachment":[{"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/media?parent=1282"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/categories?post=1282"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/www.preluna.xyz\/index.php\/wp-json\/wp\/v2\/tags?post=1282"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}