@baimingtao/lbs-agent 1.0.0

package/tools/tool/todo_provider.py

@@ -0,0 +1,161 @@
+"""Todo 工具 Provider：会话内任务清单。基于 LBS todo_tool 的低耦合能力迁移。"""
+from __future__ import annotations
+
+from typing import Any
+
+from tools.tool.schema import ToolCall, ToolContext, ToolResult, ToolSafety, ToolSource, ToolSpec
+
+VALID_STATUSES = {"pending", "in_progress", "completed", "cancelled"}
+MAX_TODO_CONTENT_CHARS = 4000
+MAX_TODO_ITEMS = 256
+_TRUNCATION_MARKER = "… [truncated]"
+
+
+class TodoStore:
+    """单会话内存 todo 列表。列表顺序即优先级。"""
+
+    def __init__(self):
+        self._items: list[dict[str, str]] = []
+
+    def write(self, todos: list[dict[str, Any]], merge: bool = False) -> list[dict[str, str]]:
+        if not merge:
+            self._items = [self._validate(t) for t in self._dedupe_by_id(todos)]
+        else:
+            existing = {item["id"]: item for item in self._items}
+            for t in self._dedupe_by_id(todos):
+                item_id = str(t.get("id", "")).strip()
+                if not item_id:
+                    continue
+                if item_id in existing:
+                    if "content" in t and t["content"]:
+                        existing[item_id]["content"] = self._cap_content(str(t["content"]).strip())
+                    if "status" in t and t["status"]:
+                        status = str(t["status"]).strip().lower()
+                        if status in VALID_STATUSES:
+                            existing[item_id]["status"] = status
+                else:
+                    validated = self._validate(t)
+                    existing[validated["id"]] = validated
+                    self._items.append(validated)
+            seen: set[str] = set()
+            rebuilt: list[dict[str, str]] = []
+            for item in self._items:
+                current = existing.get(item["id"], item)
+                if current["id"] not in seen:
+                    rebuilt.append(current)
+                    seen.add(current["id"])
+            self._items = rebuilt
+        if len(self._items) > MAX_TODO_ITEMS:
+            self._items = self._items[:MAX_TODO_ITEMS]
+        return self.read()
+
+    def read(self) -> list[dict[str, str]]:
+        return [item.copy() for item in self._items]
+
+    def format_for_injection(self) -> str | None:
+        if not self._items:
+            return None
+        markers = {"completed": "[x]", "in_progress": "[>]", "pending": "[ ]", "cancelled": "[~]"}
+        active = [item for item in self._items if item["status"] in {"pending", "in_progress"}]
+        if not active:
+            return None
+        lines = ["[会话任务清单已保留]"]
+        for item in active:
+            lines.append(f"- {markers.get(item['status'], '[?]')} {item['id']}. {item['content']} ({item['status']})")
+        return "\n".join(lines)
+
+    @staticmethod
+    def _cap_content(content: str) -> str:
+        if len(content) > MAX_TODO_CONTENT_CHARS:
+            keep = MAX_TODO_CONTENT_CHARS - len(_TRUNCATION_MARKER)
+            return content[:keep] + _TRUNCATION_MARKER
+        return content
+
+    @staticmethod
+    def _validate(item: dict[str, Any]) -> dict[str, str]:
+        item_id = str(item.get("id", "")).strip() or "?"
+        content = str(item.get("content", "")).strip() or "(no description)"
+        content = TodoStore._cap_content(content)
+        status = str(item.get("status", "pending")).strip().lower()
+        if status not in VALID_STATUSES:
+            status = "pending"
+        return {"id": item_id, "content": content, "status": status}
+
+    @staticmethod
+    def _dedupe_by_id(todos: list[dict[str, Any]]) -> list[dict[str, Any]]:
+        last_index: dict[str, int] = {}
+        for i, item in enumerate(todos):
+            item_id = str(item.get("id", "")).strip() or "?"
+            last_index[item_id] = i
+        return [todos[i] for i in sorted(last_index.values())]
+
+
+TODO_SCHEMA = {
+    "type": "object",
+    "properties": {
+        "todos": {
+            "type": "array",
+            "description": "要写入的任务项；省略则读取当前任务清单。",
+            "items": {
+                "type": "object",
+                "properties": {
+                    "id": {"type": "string", "description": "唯一任务 ID"},
+                    "content": {"type": "string", "description": "任务描述"},
+                    "status": {"type": "string", "enum": ["pending", "in_progress", "completed", "cancelled"], "description": "当前状态"},
+                },
+                "required": ["id", "content", "status"],
+            },
+        },
+        "merge": {"type": "boolean", "description": "true 按 id 合并更新；false 覆盖整个列表。", "default": False},
+    },
+}
+
+
+class TodoProvider:
+    provider_id = "builtin.todo"
+    source = ToolSource.BUILTIN
+
+    def __init__(self, store: TodoStore | None = None):
+        self.store = store or TodoStore()
+
+    def discover(self) -> list[ToolSpec]:
+        return [
+            ToolSpec(
+                name="todo",
+                description=(
+                    "管理当前会话任务清单。复杂任务或多步骤任务必须用它拆解、更新进度；"
+                    "不传参数读取清单，传 todos 写入/更新。始终返回完整任务列表和状态汇总。"
+                ),
+                input_schema=TODO_SCHEMA,
+                source=self.source,
+                provider_id=self.provider_id,
+                capability_tags=["planning", "state"],
+                safety=ToolSafety.LOCAL_READ,
+                metadata={"origin": "D:/work/chit/hermes/tools/todo_tool.py"},
+            )
+        ]
+
+    def invoke(self, call: ToolCall, context: ToolContext) -> ToolResult:
+        args = call.arguments or {}
+        try:
+            todos = args.get("todos")
+            merge = bool(args.get("merge", False))
+            items = self.store.write(todos, merge) if todos is not None else self.store.read()
+            summary = {
+                "total": len(items),
+                "pending": sum(1 for i in items if i["status"] == "pending"),
+                "in_progress": sum(1 for i in items if i["status"] == "in_progress"),
+                "completed": sum(1 for i in items if i["status"] == "completed"),
+                "cancelled": sum(1 for i in items if i["status"] == "cancelled"),
+            }
+            return ToolResult(
+                call_id=call.id,
+                ok=True,
+                content={"todos": items, "summary": summary},
+                source=self.source,
+                provider_id=self.provider_id,
+                safety=ToolSafety.LOCAL_READ,
+                evidence={"tool_name": "todo", "source": self.source.value, "provider_id": self.provider_id},
+            )
+        except Exception as exc:
+            return ToolResult(call_id=call.id, ok=False, error=f"{type(exc).__name__}: {exc}", source=self.source, provider_id=self.provider_id, safety=ToolSafety.LOCAL_READ)

package/tools/tool/tools.yaml

@@ -0,0 +1,115 @@
+# tools.yaml — 统一工具层配置
+# toolset 是能力包；profile 是 Agent 可用工具与安全边界。
+default_profile: default
+
+toolsets:
+  file_read:
+    description: 文件只读工具
+    tools: [read_file, search_files, list_dir]
+  file_write:
+    description: 文件写入工具
+    tools: [write_file]
+  shell:
+    description: 本地命令执行工具
+    tools: [terminal]
+  skills:
+    description: 本地技能查看与管理工具
+    tools: [skills_list, skill_view, skill_manage]
+  planning:
+    description: 会话任务规划与进度跟踪工具
+    tools: [todo]
+  web_research:
+    description: 网页搜索与百科查询
+    tools: [web_search, wikipedia_query]
+  email:
+    description: 邮件收发
+    tools: [email_send, email_read]
+  data_query:
+    description: 数据库查询
+    tools: [sql_query]
+  browser:
+    description: 浏览器自动化工具
+    tools: [browser_navigate, browser_click, browser_type, browser_snapshot,
+           browser_screenshot, browser_scroll, browser_extract, browser_task,
+           browser_go_back, browser_go_forward, browser_refresh, browser_send_keys,
+           browser_list_tabs, browser_switch_tab, browser_close_tab,
+           browser_upload, browser_wait, browser_get_html, browser_get_url,
+           browser_close]
+  advanced:
+    description: 高级工具（patch/code_execution/clarify）
+    tools: [patch, code_execution, clarify]
+  hermes:
+    description: Hermes 对标工具（memory/session_search）
+    tools: [memory, session_search]
+  media:
+    description: 媒体工具（视觉分析/图片生成/web搜索）
+    tools: [vision_analyze, image_gen, web_search]
+  mcp_github:
+    description: GitHub MCP 工具
+    tools: [mcp_github_*]
+
+profiles:
+  default:
+    description: 默认只读配置；禁止写入、命令执行和外部写操作
+    include_toolsets: [file_read]
+    deny_safety: [local_write, local_exec, external_write, destructive]
+  audit:
+    description: 审计/安全只读配置；只读文件+搜索
+    include_toolsets: [file_read]
+    deny_safety: [local_write, local_exec, external_write, destructive]
+  chat:
+    description: 对话工具配置；文件读取、网页搜索和本地命令
+    include_toolsets: [file_read, web_research, shell]
+    deny_safety: [local_write, external_write, destructive]
+  vision:
+    description: 视觉分析配置；图片分析和文件读取
+    include_toolsets: [file_read, media]
+    include_tools: [vision_analyze]
+    deny_safety: [local_write, local_exec, external_write, destructive]
+  verifier:
+    description: 结果验证配置；视觉分析和文件读取
+    include_toolsets: [file_read, media]
+    include_tools: [vision_analyze]
+    deny_safety: [local_write, local_exec, external_write, destructive]
+  planner:
+    description: 任务规划配置；只允许更新任务进度
+    include_toolsets: [planning]
+    deny_safety: [local_write, local_exec, external_write, destructive]
+  code:
+    description: 代码修改配置；允许读写文件、本地命令、技能和高级工具
+    include_toolsets: [file_read, file_write, shell, skills, planning, advanced]
+  document:
+    description: 文档生成配置；允许读写文件和本地命令
+    include_toolsets: [file_read, file_write, shell]
+  research:
+    description: 研究配置；网页搜索+百科+文件读写
+    include_toolsets: [web_research, file_read, file_write]
+  data_analyst:
+    description: 数据分析配置；SQL+Shell+文件读写+高级工具
+    include_toolsets: [data_query, file_read, file_write, shell, advanced]
+  email:
+    description: 邮件配置；邮件收发
+    include_toolsets: [email]
+  browser:
+    description: 浏览器全权限配置；全部 browser_* 工具
+    include_toolsets: [browser, file_read, file_write]
+    auto_confirm: true
+  github:
+    description: GitHub MCP 与本地 Git 工作流
+    include_toolsets: [file_read, file_write, shell, mcp_github]
+    deny_safety: [destructive]
+
+agent_profiles:
+  audit: audit
+  auditor: audit
+  planner: planner
+  code: code
+  document: document
+  browser_worker: browser
+  verifier: verifier
+  research: research
+  data_analyst: data_analyst
+  email_agent: email
+  github_agent: github
+  chat: chat
+  vision: vision