Initial commit: Air Agent Framework v2：全双工对话智能体（打断、插话、主动开口）

2026-09-19 12:09:51 +08:00
commit fa0cb5824d
26 changed files with 2362 additions and 0 deletions
@@ -0,0 +1,73 @@
+from __future__ import annotations
+import os
+
+from .registry import ToolRegistry
+
+
+def register_file_tools(registry: ToolRegistry):
+
+    @registry.tool(
+        name="read_file",
+        description="读取文件内容",
+        parameters={
+            "type": "object",
+            "properties": {
+                "path": {
+                    "type": "string",
+                    "description": "文件路径",
+                },
+            },
+            "required": ["path"],
+        },
+    )
+    async def read_file(path: str) -> str:
+        if not os.path.exists(path):
+            return f"[文件不存在: {path}]"
+        with open(path, "r", encoding="utf-8", errors="replace") as f:
+            content = f.read()
+        if len(content) > 10000:
+            content = content[:10000] + "\n\n[内容过长，已截断]"
+        return content
+
+    @registry.tool(
+        name="write_file",
+        description="写入文件",
+        parameters={
+            "type": "object",
+            "properties": {
+                "path": {"type": "string", "description": "文件路径"},
+                "content": {"type": "string", "description": "文件内容"},
+            },
+            "required": ["path", "content"],
+        },
+    )
+    async def write_file(path: str, content: str) -> str:
+        os.makedirs(os.path.dirname(path) or ".", exist_ok=True)
+        with open(path, "w", encoding="utf-8") as f:
+            f.write(content)
+        return f"[已写入 {len(content)} 字节到 {path}]"
+
+    @registry.tool(
+        name="list_files",
+        description="列出目录内容",
+        parameters={
+            "type": "object",
+            "properties": {
+                "path": {
+                    "type": "string",
+                    "description": "目录路径",
+                },
+            },
+            "required": ["path"],
+        },
+    )
+    async def list_files(path: str = ".") -> str:
+        if not os.path.exists(path):
+            return f"[目录不存在: {path}]"
+        items = os.listdir(path)
+        lines = []
+        for item in sorted(items):
+            full = os.path.join(path, item)
+            suffix = "/" if os.path.isdir(full) else ""
+            lines.append(f"  {item}{suffix}")
+        return f"{path} ({len(lines)} 项):\n" + "\n".join(lines)