abyss-cli 0.1.0__py3-none-any.whl

abyss/tools/registry.py

@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+"""
+工具注册中心
+替代硬编码 TOOLS dict，支持动态注册/注销，为 MCP 工具接入预留扩展点。
+"""
+import json
+import time
+from typing import Any, Dict, List, Optional
+
+from .base import Tool
+from .file_read import FileReadTool
+from .file_write import FileWriteTool
+from .file_edit import FileEditTool
+from .shell_exec import ShellExecTool
+from .web_search import WebSearchTool
+
+
+def _get_tc_field(tc, name: str, default=None):
+    """兼容 dict 和 SDK 对象两种 tool_call 格式"""
+    if isinstance(tc, dict):
+        return tc.get(name, default)
+    return getattr(tc, name, default)
+
+
+class ToolRegistry:
+    """工具注册中心：统一管理内置工具与外部扩展工具（MCP）。"""
+
+    def __init__(self):
+        self._tools: Dict[str, Tool] = {}
+
+    def register(self, tool: Tool) -> None:
+        """注册单个工具，按 tool.name 索引"""
+        self._tools[tool.name] = tool
+
+    def unregister(self, name: str) -> None:
+        """按名注销工具"""
+        self._tools.pop(name, None)
+
+    def get(self, name: str) -> Optional[Tool]:
+        """按名获取工具，未注册返回 None"""
+        return self._tools.get(name)
+
+    def list_names(self) -> List[str]:
+        """返回所有已注册工具名"""
+        return list(self._tools.keys())
+
+    def to_openai_schemas(self) -> List[Dict[str, Any]]:
+        """生成所有工具的 OpenAI function calling 定义"""
+        return [t.to_openai_schema() for t in self._tools.values()]
+
+    def register_builtins(self) -> None:
+        """注册 5 个内置工具"""
+        for tool in (
+            FileReadTool(), FileWriteTool(), FileEditTool(),
+            ShellExecTool(), WebSearchTool()
+        ):
+            self.register(tool)
+
+    def __contains__(self, name: str) -> bool:
+        return name in self._tools
+
+    def execute_tool_call(self, tool_call) -> str:
+        """执行单个 tool_call，返回给 AI 的结果 JSON 字符串。
+
+        参数解析失败或工具未注册时返回 {"success": False, "error": ...}。
+        对 shell_exec 输出做长度截断以保护上下文窗口。
+        """
+        func = _get_tc_field(tool_call, "function", {})
+        name = _get_tc_field(func, "name", "")
+        args_str = _get_tc_field(func, "arguments", "")
+        try:
+            arguments = json.loads(args_str) if args_str else {}
+        except json.JSONDecodeError:
+            return json.dumps({"success": False, "error": f"参数解析失败: {args_str}"})
+
+        tool = self.get(name)
+        if tool is None:
+            return json.dumps({"success": False, "error": f"未知工具: {name}"})
+
+        result = tool.execute(**arguments)
+        return self._format_result_for_llm(name, result)
+
+    @staticmethod
+    def _format_result_for_llm(name: str, result: Dict[str, Any]) -> str:
+        """格式化工具结果给 LLM，对 shell_exec 输出做截断"""
+        if name == "shell_exec":
+            max_len = 4000
+            stdout = (result.get("stdout", "") or "")
+            stderr = (result.get("stderr", "") or "")
+            if len(stdout) > max_len:
+                stdout = stdout[:max_len] + f"\n...[截断, 共 {result.get('stdout','').count(chr(10))+1} 行]"
+            if len(stderr) > max_len:
+                stderr = stderr[:max_len] + f"\n...[截断]"
+            return json.dumps({
+                "success": result.get("success", False),
+                "stdout": stdout,
+                "stderr": stderr,
+                "returncode": result.get("returncode", None),
+            }, ensure_ascii=False)
+        return json.dumps(result, ensure_ascii=False)
+
+
+def create_default_registry() -> ToolRegistry:
+    """创建并返回已注册全部内置工具的 ToolRegistry 实例"""
+    reg = ToolRegistry()
+    reg.register_builtins()
+    return reg

abyss/tools/shell_exec.py

@@ -0,0 +1,181 @@
+# -*- coding: utf-8 -*-
+"""
+终端命令执行工具
+执行终端命令，带安全限制和失败重试
+"""
+import re
+import sys
+import time
+import subprocess
+from .base import Tool
+
+
+class ShellExecTool(Tool):
+    """执行终端命令的工具，带安全限制。失败时自动重试，避免临时网络/IO 抖动。"""
+
+    # 最大重试次数（不含首次执行）。例如 max_retries=2 表示最多执行 3 次
+    MAX_RETRIES = 2
+
+    # 重试退避基数（秒），实际等待 = BASE_BACKOFF * 2^attempt
+    BASE_BACKOFF = 0.5
+
+    DANGEROUS_PATTERNS = [
+        "rm -rf /", "rm -rf /*", "> /dev/sda", "dd if=/dev/zero",
+        "mkfs.", "chmod -R 777 /", ":(){ :|:& };:", "> /etc/",
+        "del /f /q", "fdisk"
+    ]
+
+    # 正则匹配的危险模式：format 是 Windows 格式化磁盘命令（format C: /Q），
+    # 但要避免误伤 PowerShell 的 "Get-Date -Format" 这类合法用法。
+    DANGEROUS_REGEX = [
+        re.compile(r'(^|[\s;&|])format\s+[a-zA-Z]:', re.IGNORECASE),
+    ]
+
+    @property
+    def name(self) -> str:
+        return "shell_exec"
+
+    @property
+    def description(self) -> str:
+        return "执行终端命令，支持设置工作目录和超时。失败时自动重试。"
+
+    @property
+    def parameters(self) -> dict:
+        return {
+            "type": "object",
+            "properties": {
+                "command": {"type": "string", "description": "要执行的命令"},
+                "cwd": {"type": "string", "description": "工作目录"},
+                "timeout": {"type": "integer", "description": "超时秒数，默认60"}
+            },
+            "required": ["command"]
+        }
+
+    def _is_safe(self, command: str) -> bool:
+        """检查命令是否安全"""
+        lowered = command.lower()
+        for pattern in self.DANGEROUS_PATTERNS:
+            if pattern in lowered:
+                return False
+        for pattern in self.DANGEROUS_REGEX:
+            if pattern.search(command):
+                return False
+        return True
+
+    @staticmethod
+    def _enrich_timeout_error(command: str, timeout: int, result: dict) -> dict:
+        """为超时错误附加针对性建议。目标：让 AI 看到错误后能立刻换策略。"""
+        cmd_lower = command.lower()
+        suggestions = []
+        if re.search(r"\b(pip|pip3|python\s+-m\s+pip)\b", cmd_lower) and "install" in cmd_lower:
+            suggestions.extend([
+                "pip install 在国内常因 pypi 源慢而超时。",
+                "建议加 -i https://pypi.tuna.tsinghua.edu.cn/simple（清华源）",
+                "或阿里源 https://mirrors.aliyun.com/pypi/simple/",
+                "并把 timeout 加大到 300+ 秒。",
+            ])
+        elif re.search(r"\bgit\s+clone\b", cmd_lower):
+            suggestions.extend([
+                "git clone 慢可换镜像：git clone https://mirror.ghproxy.com/<原github url>",
+                "或 https://gitclone.com/<原 url>",
+            ])
+        elif re.search(r"\b(curl|wget|invoke-webrequest)\b", cmd_lower):
+            suggestions.extend([
+                "下载慢可换国内镜像或在浏览器中下载。",
+            ])
+        else:
+            suggestions.append(
+                f"已强制终止。建议把 timeout 调大（当前 {timeout}s），或拆成更小的步骤。"
+            )
+        if suggestions:
+            new_err = result.get("error", "") + "\n[建议] " + " ".join(suggestions)
+            result = dict(result)
+            result["error"] = new_err
+        return result
+
+    def _wrap_powershell(self, command: str) -> str:
+        """包装用户命令，强制 PowerShell 用 UTF-8 输出（避免中文乱码）。
+        同时把 PYTHONIOENCODING=utf-8 写到子进程环境，让子 Python 进程也用 UTF-8 输出，
+        避免 print('✓') 这类 Unicode 字符在 Windows GBK 终端上崩溃。
+        """
+        prefix = (
+            "$OutputEncoding = [System.Text.Encoding]::UTF8; "
+            "[Console]::OutputEncoding = [System.Text.Encoding]::UTF8; "
+            "chcp 65001 | Out-Null; "
+        )
+        return prefix + command
+
+    def _build_env(self) -> dict:
+        """构造子进程环境：保留系统环境，并注入 PYTHONIOENCODING=utf-8"""
+        import os
+        env = os.environ.copy()
+        env["PYTHONIOENCODING"] = "utf-8"
+        env["PYTHONUTF8"] = "1"
+        return env
+
+    def _run_once(self, command: str, cwd, timeout) -> dict:
+        """单次执行命令（不重试）"""
+        env = self._build_env()
+        if sys.platform == "win32":
+            result = subprocess.run(
+                ["powershell.exe", "-NoProfile", "-Command", self._wrap_powershell(command)],
+                cwd=cwd,
+                capture_output=True,
+                encoding="utf-8",
+                errors="replace",
+                timeout=timeout,
+                env=env
+            )
+        else:
+            result = subprocess.run(
+                command,
+                shell=True,
+                cwd=cwd,
+                capture_output=True,
+                encoding="utf-8",
+                errors="replace",
+                timeout=timeout,
+                env=env
+            )
+        return {
+            "command": command,
+            "success": result.returncode == 0,
+            "returncode": result.returncode,
+            "stdout": result.stdout,
+            "stderr": result.stderr
+        }
+
+    def execute(self, command: str, cwd: str = None, timeout: int = 60) -> dict:
+        """执行命令，失败时按指数退避自动重试。超时场景不重试（重试也没用）"""
+        if not self._is_safe(command):
+            return {"success": False, "error": "命令被安全策略拦截"}
+
+        last_result = None
+        for attempt in range(self.MAX_RETRIES + 1):
+            try:
+                last_result = self._run_once(command, cwd, timeout)
+            except subprocess.TimeoutExpired:
+                # 超时不重试：第一次就附加建议，避免无意义的重试浪费时间
+                last_result = {
+                    "command": command,
+                    "success": False,
+                    "error": f"命令执行超时（{timeout}秒），已强制终止",
+                }
+                return self._enrich_timeout_error(command, timeout, last_result)
+            except Exception as e:
+                last_result = {"command": command, "success": False, "error": str(e)}
+
+            # 如果是超时（不抛异常时由 _run_once 返回 success=False 且 error 含"超时"），
+            # 不再重试，附加建议后直接返回。
+            err_msg = (last_result or {}).get("error", "") or ""
+            if not last_result.get("success") and "超时" in err_msg:
+                return self._enrich_timeout_error(command, timeout, last_result)
+
+            if last_result.get("success"):
+                return last_result
+
+            # 失败：除最后一次外都退避
+            if attempt < self.MAX_RETRIES:
+                time.sleep(self.BASE_BACKOFF * (2 ** attempt))
+
+        return last_result

abyss/tools/web_search.py

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+"""
+联网搜索工具
+使用 Bing 后端搜索信息（国内可用），避免尝试被墙的其他搜索引擎
+"""
+from .base import Tool
+
+
+class WebSearchTool(Tool):
+    """联网搜索工具（需要安装 ddgs）。"""
+
+    @property
+    def name(self) -> str:
+        return "web_search"
+
+    @property
+    def description(self) -> str:
+        return "联网搜索信息，返回搜索结果列表。检索数量由 AI 自行决定，无需询问用户。"
+
+    @property
+    def parameters(self) -> dict:
+        return {
+            "type": "object",
+            "properties": {
+                "query": {"type": "string", "description": "搜索关键词"},
+                "num": {
+                    "type": "integer",
+                    "description": "返回结果数量，由 AI 根据查询复杂度自行决定（一般 3-10 条）"
+                }
+            },
+            "required": ["query", "num"]
+        }
+
+    def execute(self, query: str, num: int = 5) -> dict:
+        """执行搜索"""
+        try:
+            try:
+                from ddgs import DDGS
+            except ImportError:
+                from duckduckgo_search import DDGS
+
+            # 限制 num 在合理范围，避免 AI 误传过大值
+            num = max(1, min(int(num), 20))
+
+            with DDGS(timeout=10) as ddgs:
+                results = list(ddgs.text(
+                    query,
+                    max_results=num,
+                    backend="bing"
+                ))
+
+            return {
+                "success": True,
+                "query": query,
+                "results": [
+                    {"title": r["title"], "url": r["href"], "snippet": r["body"]}
+                    for r in results
+                ]
+            }
+        except ImportError:
+            return {"success": False, "error": "请安装 ddgs: pip install ddgs"}
+        except Exception as e:
+            return {"success": False, "error": str(e)}

abyss_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: abyss-cli
+Version: 0.1.0
+Summary: 终端AI开发助手
+Author: Abyss Team
+Requires-Python: >=3.9
+Requires-Dist: openai>=1.0.0
+Requires-Dist: click>=8.0.0
+Requires-Dist: prompt-toolkit>=3.0.0
+Requires-Dist: ddgs>=9.0.0
+Dynamic: requires-python

abyss_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,35 @@
+abyss/__init__.py,sha256=0Fud8wvwYRa0b0oXRR9WG3IK_UaXGG4nEYn4yattnCM,82
+abyss/ansi_menu.py,sha256=fB2bBYo5UomBsv-Apm_mQccEWQKCW1ilVtW3hZOCcdo,20475
+abyss/api_client.py,sha256=0ajbtOOQ2NAsIDjdZhWWYEsqxjIDR2qo3uzuvzaOxY8,4664
+abyss/config.py,sha256=j_cMCA-t2myZ-vVCRv66XvaUqEgBhZKEqEJ5-fPp-oo,3967
+abyss/custom_input.py,sha256=UazB6SUkQGQClXmYxq4T732chOVkUKQxrFpKqnZZuTU,12002
+abyss/logger.py,sha256=dlX8UGJ7lkruyHDpi-lrF1ykCMf5TROvYyeaqOOS5-k,6320
+abyss/main.py,sha256=tzV9p-uG9bJLqQ69kF5aZO7d5WuroGMC2a62OnD5NNw,31027
+abyss/session.py,sha256=wQIlIG_VtCd40-eenIQNl8SBGNEWHaPgM4W8fo-hYwc,2772
+abyss/commands/__init__.py,sha256=OOLYmwmOx2qszvZuwmKqP-gBiU0w0xWdJ9p3xRGToRo,268
+abyss/commands/slash.py,sha256=DeClT6lM3B9ccY0mqdRI0WpsPI7tZOFDHWfUqc4svWU,2661
+abyss/extensions/__init__.py,sha256=BbbKwSx-OMsGcgIS5Y2u4ZWAjdg-KPcIq8k2Maz-AQ0,516
+abyss/extensions/cli.py,sha256=Xy-TXjmZ3bsdvfYRMPPVqMDn--gs9plF38oc9krjPV0,5400
+abyss/extensions/installer.py,sha256=xfg3QETsBoVtyijm5bPjnHX1nq9om02A_HsJGfOmESA,16620
+abyss/extensions/registry.py,sha256=B9ap3PbkTAeYbXePD7bziA5IbBJFz0sPoUBY8lSIZAI,3850
+abyss/extensions/url_parser.py,sha256=E5l5-eGLhrcZGokyfazfbzX-uz9KYqasg-XRUPVNoX4,2689
+abyss/hooks/__init__.py,sha256=tT1gucCipUuRg8MdLuQ0VBKjB-0iXQJzLbzu-LCvEnQ,274
+abyss/hooks/runner.py,sha256=i0D36SMg29bKC-TJNMpBRPXThtUcZzBKL6xZvEofBhI,4899
+abyss/mcp/__init__.py,sha256=kiNGPmrzGoQvO0jyi_lffDFY4zlVLko8592R9YsiGG8,316
+abyss/mcp/manager.py,sha256=IzjKnQ-_VBU0NUc3sORXwdYityQIuJ2XTIp8CZbYDkg,6977
+abyss/prompts/__init__.py,sha256=nNd2zNTncRphmM0RFsAa4L8i4-2sryvEQ_n9lh1buFA,862
+abyss/skills/__init__.py,sha256=ab1XjEz295VM0r8Le_7NN00WX3ei5V9Yte7rZHyP5PQ,247
+abyss/skills/loader.py,sha256=NSZ678hCDzqRWNMRgznhrtF-kfiPZPE5B-JFVkh2xkc,5651
+abyss/tools/__init__.py,sha256=GwoWzJAn1VMhjsBaRfNQfVwT10_jPW0kiTtoN4sj1ok,442
+abyss/tools/base.py,sha256=gsZPD0MO9_eMUz9mKPHpTpUzt3gn_VWAQDrSBqgyAsw,1089
+abyss/tools/file_edit.py,sha256=Lobs2uz2EPslG2wuah5lBacl1wAJaAV3i2jsoQg1g4Q,1552
+abyss/tools/file_read.py,sha256=--9v-su3cOwpW5xryB-4KF_5_QE-fkzmiq3TT_FDLyk,1677
+abyss/tools/file_write.py,sha256=5eD1cIT4wGSbdwA4Lsio3d02Fzt1clvu-SlGjxC9l9U,1386
+abyss/tools/registry.py,sha256=4q8TeA7B90LIPO6W63gF06Or-NKwAx8GtgvtB_4Ili4,3868
+abyss/tools/shell_exec.py,sha256=XTWHNEEzOzatPj4qv_o0Su6RAzx-YdPQ-Yq8TCT7Zrs,7149
+abyss/tools/web_search.py,sha256=cYp9p9YAdY48pPM7lgoK3AgorNSmWM3DW4Q-M4512Vs,1978
+abyss_cli-0.1.0.dist-info/METADATA,sha256=S8j3aq_l2ejW1hpzZNLj1hGumnNPswLm-Zv4AdOpIoY,282
+abyss_cli-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+abyss_cli-0.1.0.dist-info/entry_points.txt,sha256=j_8IOp0dIAhyarOiN0v0G0wznEJV7F1Iy3GldgxN4FY,41
+abyss_cli-0.1.0.dist-info/top_level.txt,sha256=66PtKGxE4PwMGB48A5isqqRFciZhenwLMiZugJCC8SE,6
+abyss_cli-0.1.0.dist-info/RECORD,,

abyss_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

abyss_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+abyss = abyss.main:cli

abyss_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+abyss