64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
"""工具类别配置。
|
|
|
|
提供前端和终端公用的工具分组定义，方便按类别控制启用状态。
|
|
"""
|
|
|
|
from typing import Dict, List
|
|
|
|
|
|
class ToolCategory:
|
|
"""工具类别的结构化定义。"""
|
|
|
|
def __init__(self, label: str, tools: List[str]):
|
|
self.label = label
|
|
self.tools = tools
|
|
|
|
|
|
TOOL_CATEGORIES: Dict[str, ToolCategory] = {
|
|
"network": ToolCategory(
|
|
label="网络检索",
|
|
tools=["web_search", "extract_webpage", "save_webpage"],
|
|
),
|
|
"file_edit": ToolCategory(
|
|
label="文件编辑",
|
|
tools=[
|
|
"create_file",
|
|
"append_to_file",
|
|
"modify_file",
|
|
"delete_file",
|
|
"rename_file",
|
|
"create_folder",
|
|
],
|
|
),
|
|
"read_focus": ToolCategory(
|
|
label="阅读聚焦",
|
|
tools=["read_file", "focus_file", "unfocus_file", "ocr_image"],
|
|
),
|
|
"terminal_realtime": ToolCategory(
|
|
label="实时终端",
|
|
tools=[
|
|
"terminal_session",
|
|
"terminal_input",
|
|
"terminal_snapshot",
|
|
"terminal_reset",
|
|
"sleep",
|
|
],
|
|
),
|
|
"terminal_command": ToolCategory(
|
|
label="终端指令",
|
|
tools=["run_command", "run_python"],
|
|
),
|
|
"memory": ToolCategory(
|
|
label="记忆",
|
|
tools=["update_memory"],
|
|
),
|
|
"todo": ToolCategory(
|
|
label="待办事项",
|
|
tools=["todo_create", "todo_update_task", "todo_finish", "todo_finish_confirm"],
|
|
),
|
|
"sub_agent": ToolCategory(
|
|
label="子智能体",
|
|
tools=["create_sub_agent", "wait_sub_agent", "close_sub_agent"],
|
|
),
|
|
}
|