nexforge-cli 0.1.0__py3-none-any.whl

nexforge/__init__.py

@@ -0,0 +1,7 @@
+"""NexForgeCLI — 面向 DeepSeek V4 的终端智能体。
+
+TUI(全屏交互)+ CLI(one-shot / 管道)双模式。
+"""
+
+__version__ = "0.1.0"
+__all__ = ["__version__"]

nexforge/__main__.py

@@ -0,0 +1,6 @@
+"""支持 `python -m nexforge`。"""
+
+from nexforge.cli import main
+
+if __name__ == "__main__":
+    raise SystemExit(main())

nexforge/app.py

@@ -0,0 +1,38 @@
+"""Textual TUI 入口。
+
+NexForgeApp 提到模块级,便于无头测试(App.run_test)。
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from textual.app import App
+
+if TYPE_CHECKING:
+    from nexforge.config import NexForgeConfig
+
+
+class NexForgeApp(App):
+    CSS_PATH = "theme.tcss"
+    TITLE = "NexForgeCLI"
+    USE_ALT_SCREEN = True
+
+    def __init__(self, config: "NexForgeConfig", *, resume: bool = False) -> None:
+        super().__init__()
+        self._config = config
+        self._resume = resume
+
+    def get_default_screen(self):
+        from nexforge.ui.screens.chat import ChatScreen
+        return ChatScreen(config=self._config, resume_session_id=None)
+
+
+def run_tui(config: "NexForgeConfig", *, resume: bool = False) -> int:
+    app = NexForgeApp(config, resume=resume)
+    try:
+        # 鼠标开启(Button 需要点击);子进程 stdin=DEVNULL 防鼠标事件回显乱码
+        app.run()
+    except KeyboardInterrupt:
+        return 0
+    return 0

nexforge/assets/icons/agent.svg

@@ -0,0 +1,13 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 32 32" fill="none">
+  <defs>
+    <linearGradient id="nf" x1="0" y1="0" x2="32" y2="32" gradientUnits="userSpaceOnUse">
+      <stop stop-color="#6D5EF5"/>
+      <stop offset="1" stop-color="#22D3EE"/>
+    </linearGradient>
+  </defs>
+  <!-- NexForge 铁砧 / 分叉标记 -->
+  <path d="M16 3 4 9v8c0 6.5 5.1 10.7 12 12 6.9-1.3 12-5.5 12-12V9L16 3z"
+        stroke="url(#nf)" stroke-width="2" stroke-linejoin="round"/>
+  <path d="M11 15h10M16 11v10M11 15l-2 3M21 15l2 3" stroke="url(#nf)"
+        stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
+</svg>

nexforge/assets/icons/cost.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#22D3EE" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- cost / coin/dollar -->
+  <circle cx="12" cy="12" r="10"/>
+  <path d="M12 7v10M10 9h4"/>
+</svg>

nexforge/assets/icons/edit.svg

@@ -0,0 +1,4 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#22D3EE" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- edit: same as write(alias) -->
+  <path d="M17 3a2.828 2.828 0 1 1 4 4L7.5 20.5 2 22l1.5-5.5L17 3z"/>
+</svg>

nexforge/assets/icons/err.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#E04040" stroke-width="3" stroke-linecap="round">
+  <!-- error / x -->
+  <line x1="6" y1="6" x2="18" y2="18"/>
+  <line x1="18" y1="6" x2="6" y2="18"/>
+</svg>

nexforge/assets/icons/file.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- file / generic document -->
+  <path d="M14 2H6a2 2 0 0 0-2 2v16a2 2 0 0 0 2 2h12a2 2 0 0 0 2-2V8z"/>
+  <polyline points="14,2 14,8 20,8"/>
+</svg>

nexforge/assets/icons/model.svg

@@ -0,0 +1,7 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- model / diamond with AI dots -->
+  <rect x="4" y="4" width="16" height="16" rx="2" transform="rotate(45 12 12)"/>
+  <circle cx="10" cy="10" r="1.5" fill="#22D3EE" stroke="none"/>
+  <circle cx="14" cy="10" r="1.5" fill="#22D3EE" stroke="none"/>
+  <circle cx="12" cy="14" r="1.5" fill="#22D3EE" stroke="none"/>
+</svg>

nexforge/assets/icons/ok.svg

@@ -0,0 +1,4 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#22D3EE" stroke-width="3" stroke-linecap="round" stroke-linejoin="round">
+  <!-- ok / check -->
+  <polyline points="4,12 10,19 20,5"/>
+</svg>

nexforge/assets/icons/read.svg

@@ -0,0 +1,8 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- read_file / document -->
+  <path d="M14 2H6a2 2 0 0 0-2 2v16a2 2 0 0 0 2 2h12a2 2 0 0 0 2-2V8z"/>
+  <polyline points="14,2 14,8 20,8"/>
+  <line x1="16" y1="13" x2="8" y2="13"/>
+  <line x1="16" y1="17" x2="8" y2="17"/>
+  <polyline points="10,9 9,9 8,9"/>
+</svg>

nexforge/assets/icons/running.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- running / spinner dot + arc -->
+  <circle cx="12" cy="12" r="10"/>
+  <path stroke="#22D3EE" d="M12 2a10 10 0 0 1 10 10"/>
+</svg>

nexforge/assets/icons/search.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- search / magnifying glass -->
+  <circle cx="11" cy="11" r="8"/>
+  <line x1="21" y1="21" x2="16.65" y2="16.65"/>
+</svg>

nexforge/assets/icons/shell.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- shell / terminal / >_ -->
+  <polyline points="4,17 10,11 4,5"/>
+  <line x1="12" y1="19" x2="20" y2="19"/>
+</svg>

nexforge/assets/icons/thinking.svg

@@ -0,0 +1,5 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#6D5EF5" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- thinking / brain -->
+  <path d="M12 2a5 5 0 0 0-5 5c0 1.2.4 2.3 1 3.2L5 21h14l-3-10.8a5 5 0 0 0-4-8.2z"/>
+  <path d="M9 13h6M10 17h4" stroke="#22D3EE"/>
+</svg>

nexforge/assets/icons/web.svg

@@ -0,0 +1,6 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#22D3EE" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- web / globe -->
+  <circle cx="12" cy="12" r="10"/>
+  <line x1="2" y1="12" x2="22" y2="12"/>
+  <path d="M12 2a15.3 15.3 0 0 1 4 10 15.3 15.3 0 0 1-4 10 15.3 15.3 0 0 1-4-10 15.3 15.3 0 0 1 4-10z"/>
+</svg>

nexforge/assets/icons/write.svg

@@ -0,0 +1,4 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="#22D3EE" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <!-- write / pencil -->
+  <path d="M17 3a2.828 2.828 0 1 1 4 4L7.5 20.5 2 22l1.5-5.5L17 3z"/>
+</svg>

nexforge/cli.py

@@ -0,0 +1,159 @@
+"""NexForgeCLI 命令行入口 —— 解析参数并分发到 TUI 或 one-shot 模式。
+
+用法::
+
+    nexforge                     # 进入全屏 TUI
+    nexforge -p "这段代码做什么?"  # one-shot,打印答案后退出
+    echo "review main.py" | nexforge   # 管道输入 → one-shot
+    nexforge --model pro          # 指定档位
+    nexforge --no-thinking        # 关闭思考模式
+
+重量级模块(Textual / provider)在各分支内延迟导入,保证
+`--version` / `--help` 在缺失可选依赖时也能工作。
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+
+from nexforge import __version__
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="nexforge",
+        description="NexForgeCLI — DeepSeek V4 终端智能体(TUI + CLI 双模式)",
+        epilog="不带参数直接运行进入全屏 TUI;-p 或管道输入进入 one-shot 模式。",
+    )
+    parser.add_argument(
+        "prompt",
+        nargs="*",
+        help="one-shot 提问;省略则进入 TUI。",
+    )
+    parser.add_argument(
+        "-p",
+        "--print",
+        dest="print_prompt",
+        metavar="TEXT",
+        help="one-shot 模式:回答后打印并退出(可与位置参数二选一)。",
+    )
+    parser.add_argument(
+        "-m",
+        "--model",
+        choices=["flash", "pro", "auto"],
+        help="模型档位(覆盖配置默认值)。",
+    )
+    parser.add_argument(
+        "--think",
+        dest="thinking",
+        action="store_true",
+        default=None,
+        help="强制开启思考模式。",
+    )
+    parser.add_argument(
+        "--no-think",
+        dest="thinking",
+        action="store_false",
+        help="关闭思考模式(更快、更省)。",
+    )
+    parser.add_argument(
+        "--resume",
+        action="store_true",
+        help="恢复最近一次会话(M4)。",
+    )
+    parser.add_argument(
+        "--auto",
+        action="store_true",
+        help="启动时即设为全自动模式(警告后)。",
+    )
+    parser.add_argument(
+        "--config",
+        action="store_true",
+        help="打印当前生效配置后退出。",
+    )
+    parser.add_argument(
+        "--login",
+        action="store_true",
+        help="安全地保存 API Key 到凭据文件(~/.nexforge/auth.toml)。",
+    )
+    parser.add_argument(
+        "-V",
+        "--version",
+        action="version",
+        version=f"NexForgeCLI {__version__}",
+    )
+    return parser
+
+
+def _collect_oneshot_prompt(args: argparse.Namespace) -> str | None:
+    """按优先级取 one-shot 提示词:-p > 位置参数 > 管道 stdin。"""
+    if args.print_prompt:
+        return args.print_prompt
+    if args.prompt:
+        return " ".join(args.prompt)
+    # 非交互式 stdin(被管道 / 重定向)→ 读取全部作为提示。
+    if not sys.stdin.isatty():
+        piped = sys.stdin.read().strip()
+        if piped:
+            return piped
+    return None
+
+
+def main(argv: list[str] | None = None) -> int:
+    args = build_parser().parse_args(argv)
+
+    # login:交互式保存 key 后退出(不依赖已有配置)。
+    if args.login or (args.prompt and len(args.prompt) == 1 and args.prompt[0] == "login"):
+        return _do_login()
+
+    # 延迟导入:让 --version / --help 不依赖可选运行时。
+    from nexforge.config import load_config
+
+    config = load_config()
+    if args.model:
+        config.models.default_tier = args.model
+    if args.thinking is not None:
+        config.models.thinking = args.thinking
+    if args.auto:
+        config.ui.mode = "auto"
+
+    if args.config:
+        print(config.pretty())
+        return 0
+
+    prompt = _collect_oneshot_prompt(args)
+    if prompt is not None:
+        # one-shot:无需 TUI,跑一次智能体循环打印结果。
+        from nexforge.core.agent import run_oneshot
+
+        return run_oneshot(config, prompt)
+
+    # 默认:全屏 TUI。
+    from nexforge.app import run_tui
+
+    return run_tui(config, resume=args.resume)
+
+
+def _do_login() -> int:
+    """交互式输入 API Key 并保存到凭据文件,不会回显输入内容。"""
+    from getpass import getpass
+
+    from nexforge.config import AUTH_PATH, save_api_key
+
+    print("NexForgeCLI — API Key 配置")
+    print(f"凭据将写入: {AUTH_PATH}")
+    print()
+    key = getpass("粘贴或输入 API Key (以 sk- 开头): ").strip()
+    if not key:
+        print("✖ 未输入 Key,已取消。")
+        return 1
+    if not key.startswith("sk-"):
+        print("⚠ Key 似乎不以 'sk-' 开头,但仍将保存。")
+    save_api_key(key)
+    print(f"✔ 已保存到 {AUTH_PATH}")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

nexforge/config.py

@@ -0,0 +1,192 @@
+"""NexForgeCLI 配置系统。
+
+优先级:内置默认值 → ``~/.nexforge/config.toml`` 覆盖 → 环境变量(API key)。
+
+API Key 解析链:环境变量 > 凭据文件(nexforge login 写入) > config.toml 内 api_key。
+凭据文件 ``~/.nexforge/auth.toml`` 独立于主配置,不会被 git 追踪。
+
+所有 DeepSeek V4 的真实参数(model id / endpoint / 定价)作为内置默认值,
+用户可在 ``config.toml`` 覆盖。数据核实自 api-docs.deepseek.com(2026-04)。
+"""
+
+from __future__ import annotations
+
+import os
+import tomllib
+from pathlib import Path
+from typing import Literal
+
+from pydantic import BaseModel, Field, PrivateAttr
+
+Tier = Literal["flash", "pro", "auto"]
+Effort = Literal["high", "max"]
+
+CONFIG_DIR = Path.home() / ".nexforge"
+CONFIG_PATH = CONFIG_DIR / "config.toml"
+AUTH_PATH = CONFIG_DIR / "auth.toml"
+SESSIONS_DIR = CONFIG_DIR / "sessions"
+
+
+class PricingTier(BaseModel):
+    input_cache_hit: float
+    input_cache_miss: float
+    output: float
+
+
+class ProviderConfig(BaseModel):
+    base_url: str = "https://api.deepseek.com"
+    api_key_env: str = "DEEPSEEK_API_KEY"
+    api_key: str | None = None
+    timeout: float = 120.0
+    max_retries: int = 2
+
+
+class ModelsConfig(BaseModel):
+    flash: str = "deepseek-v4-flash"
+    pro: str = "deepseek-v4-pro"
+    default_tier: Tier = "auto"
+    thinking: bool = True
+    reasoning_effort: Effort = "high"
+    context_window: int = 1_000_000
+    max_output_tokens: int = 8192
+    temperature: float | None = None  # 0-2,仅 thinking=off 生效
+    top_p: float | None = None
+    frequency_penalty: float | None = None
+    presence_penalty: float | None = None
+    stop: str | list[str] | None = None
+    use_max_completion_tokens: bool = True  # DeepSeek 推荐用新参数名
+    pricing: dict[str, PricingTier] = Field(
+        default_factory=lambda: {
+            "flash": PricingTier(input_cache_hit=0.0028, input_cache_miss=0.14, output=0.28),
+            "pro": PricingTier(input_cache_hit=0.003625, input_cache_miss=0.435, output=0.87),
+        }
+    )
+
+
+class RouterConfig(BaseModel):
+    entry: Tier = "flash"
+    escalate_to: Tier = "pro"
+    escalate_on_tool_failures: int = 2
+    escalate_keywords: list[str] = Field(
+        default_factory=lambda: [
+            "重构", "refactor", "架构", "architecture", "算法", "algorithm",
+            "证明", "prove", "调试", "debug", "为什么", "why", "设计", "design",
+        ]
+    )
+    escalate_context_tokens: int = 200_000
+
+
+class UIConfig(BaseModel):
+    theme: Literal["dark", "light"] = "dark"
+    show_thinking: bool = True
+    icons: Literal["auto", "image", "nerd", "glyph", "ascii"] = "auto"
+    mode: Literal["review", "auto"] = "review"
+    plan_mode: bool = False  # /plan 切换
+    json_mode: bool = False  # /json 开关,强制 JSON 输出
+    accent: str = "#6D5EF5"
+    accent2: str = "#22D3EE"
+
+
+class ToolsConfig(BaseModel):
+    enable_git_commit: bool = False
+    max_tool_iters: int = 50  # 工具轮数安全上限(防死循环)
+    turn_timeout: int = 300   # 单轮总超时(秒),防卡死
+    tool_timeout: int = 60    # 单个工具超时(秒)
+
+class PermissionsConfig(BaseModel):
+    profile: str = "supervised"  # safe | supervised | auto
+
+
+class NexForgeConfig(BaseModel):
+    provider: ProviderConfig = Field(default_factory=ProviderConfig)
+    models: ModelsConfig = Field(default_factory=ModelsConfig)
+    router: RouterConfig = Field(default_factory=RouterConfig)
+    ui: UIConfig = Field(default_factory=UIConfig)
+    tools: ToolsConfig = Field(default_factory=ToolsConfig)
+    permissions: PermissionsConfig = Field(default_factory=PermissionsConfig)
+    _raw_toml: dict = PrivateAttr(default_factory=dict)
+
+    def resolve_api_key(self) -> str | None:
+        return os.environ.get(self.provider.api_key_env) or load_saved_api_key() or self.provider.api_key
+
+    def model_id(self, tier: Tier | None = None) -> str:
+        t = tier or self.models.default_tier
+        if t == "auto":
+            t = self.router.entry
+        return self.models.pro if t == "pro" else self.models.flash
+
+    def price(self, tier: Tier) -> PricingTier:
+        key = "pro" if tier == "pro" else "flash"
+        return self.models.pricing[key]
+
+    def pretty(self) -> str:
+        key = self.resolve_api_key()
+        masked = "(未设置,运行 nexforge login 配置)" if not key else f"{key[:6]}…{key[-4:]}"
+        if os.environ.get(self.provider.api_key_env):
+            source = f"env:{self.provider.api_key_env}"
+        elif load_saved_api_key():
+            source = f"凭据文件 {AUTH_PATH}"
+        elif self.provider.api_key:
+            source = "config.toml"
+        else:
+            source = "无"
+        return (
+            "NexForgeCLI 配置\n"
+            f"  endpoint      : {self.provider.base_url}\n"
+            f"  api_key       : {masked}  (来源 {source})\n"
+            f"  default_tier  : {self.models.default_tier}\n"
+            f"  flash / pro   : {self.models.flash} / {self.models.pro}\n"
+            f"  thinking      : {self.models.thinking}  (effort={self.models.reasoning_effort})\n"
+            f"  theme / icons : {self.ui.theme} / {self.ui.icons}\n"
+            f"  mode          : {self.ui.mode}"
+            f"{' (plan)' if self.ui.plan_mode else ''}\n"
+            f"  max_tool_iters: {self.tools.max_tool_iters}\n"
+            f"  config file   : {CONFIG_PATH}  ({'存在' if CONFIG_PATH.exists() else '缺省'})"
+        )
+
+
+def _deep_merge(base: dict, override: dict) -> dict:
+    out = dict(base)
+    for k, v in override.items():
+        if isinstance(v, dict) and isinstance(out.get(k), dict):
+            out[k] = _deep_merge(out[k], v)
+        else:
+            out[k] = v
+    return out
+
+
+def load_config(path: Path | None = None) -> NexForgeConfig:
+    cfg_path = path or CONFIG_PATH
+    defaults = NexForgeConfig().model_dump()
+    raw = {}
+    if cfg_path.exists():
+        with cfg_path.open("rb") as f:
+            raw = tomllib.load(f)
+        merged = _deep_merge(defaults, raw)
+        cfg = NexForgeConfig.model_validate(merged)
+    else:
+        cfg = NexForgeConfig.model_validate(defaults)
+    object.__setattr__(cfg, "_raw_toml", raw)
+    return cfg
+
+
+def save_api_key(key: str) -> None:
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    AUTH_PATH.write_text(f'api_key = "{key}"\n', encoding="utf-8")
+
+
+def load_saved_api_key() -> str | None:
+    if not AUTH_PATH.exists():
+        return None
+    try:
+        with AUTH_PATH.open("r", encoding="utf-8") as f:
+            for line in f:
+                s = line.strip()
+                if s.startswith("api_key"):
+                    val = s.split("=", 1)[-1].strip()
+                    if len(val) >= 2 and val[0] in ('"', "'") and val[0] == val[-1]:
+                        val = val[1:-1]
+                    return val if val else None
+    except Exception:
+        pass
+    return None

nexforge/core/__init__.py

@@ -0,0 +1 @@
+"""核心层:智能体循环、会话、流式处理、auto 路由。"""