aria-code 0.52.0__py3-none-any.whl

aria_code/apps/cli/hooks.py

@@ -0,0 +1,180 @@
+"""Aria Code lifecycle hooks — JSON-configurable, shell-executable.
+
+Config file: ~/.arthera/hooks.json
+Project-local override: .aria/hooks.json  (takes precedence)
+
+Schema
+------
+{
+  "PreToolUse":   [{"tool": "run_command", "command": "echo $ARIA_TOOL"}],
+  "PostToolUse":  [{"command": "notify-send 'done'"}],
+  "ResponseDone": [{"command": "afplay /System/Sounds/Glass.aiff"}],
+  "SessionStart": [{"command": "echo starting $ARIA_SESSION"}],
+  "SessionEnd":   [{"command": "echo ended"}]
+}
+
+Each hook entry:
+  command  — shell command to execute (required)
+  tool     — if set, only fires when tool name matches (PreToolUse / PostToolUse)
+  timeout  — seconds before the hook is killed (default 10)
+  blocking — if true AND exit code != 0, execution is blocked (PreToolUse only)
+
+Env vars injected into every hook:
+  ARIA_EVENT        — hook event name
+  ARIA_TOOL         — tool name (Pre/PostToolUse only)
+  ARIA_TOOL_PARAMS  — JSON-encoded params (Pre/PostToolUse only)
+  ARIA_RESULT       — JSON-encoded result (PostToolUse only)
+  ARIA_RESPONSE     — first 500 chars of response (ResponseDone)
+  ARIA_SESSION      — session ID
+"""
+from __future__ import annotations
+
+import json
+import logging
+import os
+import subprocess
+from pathlib import Path
+from typing import Optional
+
+from aria_code.apps.cli.config_paths import resolve_config_dir
+
+logger = logging.getLogger(__name__)
+
+_CONFIG_DIR = resolve_config_dir()
+_GLOBAL_HOOKS_FILE = _CONFIG_DIR / "hooks.json"
+_LOCAL_HOOKS_FILE = Path.cwd() / ".aria" / "hooks.json"
+
+_VALID_EVENTS = frozenset([
+    "PreToolUse", "PostToolUse",
+    "ResponseDone",
+    "SessionStart", "SessionEnd",
+])
+
+
+def load_hooks() -> dict:
+    """Load and merge global + project-local hooks config."""
+    merged: dict[str, list] = {e: [] for e in _VALID_EVENTS}
+
+    for path in (_GLOBAL_HOOKS_FILE, _LOCAL_HOOKS_FILE):
+        if path.exists():
+            try:
+                data = json.loads(path.read_text(encoding="utf-8"))
+                for event, entries in data.items():
+                    if event in _VALID_EVENTS and isinstance(entries, list):
+                        merged[event].extend(entries)
+            except Exception as exc:
+                logger.debug("hooks.json parse error (%s): %s", path, exc)
+
+    return merged
+
+
+def fire(
+    event: str,
+    *,
+    tool: Optional[str] = None,
+    params: Optional[dict] = None,
+    result: Optional[dict] = None,
+    response: str = "",
+    session_id: str = "",
+    hooks: Optional[dict] = None,
+) -> bool:
+    """Fire all matching hooks for an event.
+
+    Returns False if any *blocking* PreToolUse hook exits non-zero
+    (meaning the tool call should be suppressed). Returns True otherwise.
+    """
+    if hooks is None:
+        hooks = load_hooks()
+
+    entries = hooks.get(event, [])
+    if not entries:
+        return True
+
+    base_env = dict(os.environ)
+    base_env["ARIA_EVENT"] = event
+    base_env["ARIA_SESSION"] = session_id or ""
+    if tool:
+        base_env["ARIA_TOOL"] = tool
+    if params is not None:
+        base_env["ARIA_TOOL_PARAMS"] = json.dumps(params, ensure_ascii=False)[:2000]
+    if result is not None:
+        base_env["ARIA_RESULT"] = json.dumps(result, ensure_ascii=False)[:2000]
+    if response:
+        base_env["ARIA_RESPONSE"] = response[:500]
+
+    blocked = False
+    for entry in entries:
+        if not isinstance(entry, dict):
+            continue
+        cmd = entry.get("command", "").strip()
+        if not cmd:
+            continue
+
+        # Tool filter: skip if hook is scoped to a different tool
+        if entry.get("tool") and tool and entry["tool"] != tool:
+            continue
+
+        timeout = int(entry.get("timeout", 10))
+        is_blocking = bool(entry.get("blocking", False))
+
+        try:
+            proc = subprocess.run(
+                cmd,
+                shell=True,
+                env=base_env,
+                timeout=timeout,
+                capture_output=True,
+                text=True,
+            )
+            if is_blocking and proc.returncode != 0 and event == "PreToolUse":
+                logger.info(
+                    "Blocking hook vetoed tool '%s': exit %d — %s",
+                    tool, proc.returncode, proc.stderr[:120],
+                )
+                blocked = True
+        except subprocess.TimeoutExpired:
+            logger.debug("Hook timed out after %ds: %s", timeout, cmd[:60])
+        except Exception as exc:
+            logger.debug("Hook error: %s", exc)
+
+    return not blocked
+
+
+def list_hooks() -> list[dict]:
+    """Return a flat list of configured hooks for display."""
+    hooks = load_hooks()
+    rows = []
+    for event, entries in hooks.items():
+        for entry in entries:
+            if isinstance(entry, dict) and entry.get("command"):
+                rows.append({
+                    "event": event,
+                    "tool": entry.get("tool", "*"),
+                    "command": entry["command"][:80],
+                    "blocking": entry.get("blocking", False),
+                    "timeout": entry.get("timeout", 10),
+                })
+    return rows
+
+
+def hooks_file_path(scope: str = "global") -> Path:
+    """Return the hooks.json path for a given scope."""
+    if scope == "local":
+        return Path.cwd() / ".aria" / "hooks.json"
+    return _GLOBAL_HOOKS_FILE
+
+
+def create_example_hooks(path: Path) -> None:
+    """Write an annotated example hooks.json if it doesn't exist."""
+    if path.exists():
+        return
+    path.parent.mkdir(parents=True, exist_ok=True)
+    example = {
+        "_comment": "Aria Code hooks — https://aria.code/docs/hooks",
+        "PreToolUse": [],
+        "PostToolUse": [],
+        "ResponseDone": [],
+        "SessionStart": [],
+        "SessionEnd": [],
+    }
+    path.write_text(json.dumps(example, indent=2, ensure_ascii=False), encoding="utf-8")

aria_code/apps/cli/i18n.py

@@ -0,0 +1,297 @@
+"""i18n.py — System language detection and UI string translations for Aria Code.
+
+Priority order for UI language:
+  1. config["ui_lang"]  (user explicitly set via /config set ui_lang=zh)
+  2. OS locale at first-run  (written into config automatically)
+  3. Fallback: "en"
+
+Supported languages: "zh" (Simplified Chinese), "en" (English)
+Future: "ja", "ko" — extend STRINGS dict below.
+"""
+from __future__ import annotations
+
+import os
+from typing import Optional
+
+
+# ---------------------------------------------------------------------------
+# String catalogue
+# ---------------------------------------------------------------------------
+
+STRINGS: dict[str, dict[str, str]] = {
+    # Banner labels
+    "local_first_agent": {
+        "zh": "本地优先智能体",
+        "en": "local-first agent",
+    },
+    "model": {
+        "zh": "模型",
+        "en": "model",
+    },
+    "workspace": {
+        "zh": "工作区",
+        "en": "workspace",
+    },
+    "mode": {
+        "zh": "模式",
+        "en": "mode",
+    },
+    "status": {
+        "zh": "状态",
+        "en": "status",
+    },
+    "tools": {
+        "zh": "工具",
+        "en": "tools",
+    },
+    "skills": {
+        "zh": "技能",
+        "en": "skills",
+    },
+    "quant": {
+        "zh": "量化",
+        "en": "quant",
+    },
+    # Status labels
+    "sharing_on": {
+        "zh": "数据共享",
+        "en": "sharing on",
+    },
+    "local_only": {
+        "zh": "纯本地",
+        "en": "local-only",
+    },
+    "local_retention": {
+        "zh": "本地留存",
+        "en": "local retention",
+    },
+    "cloud": {
+        "zh": "云端",
+        "en": "cloud",
+    },
+    "local": {
+        "zh": "本地",
+        "en": "local",
+    },
+    "lite": {
+        "zh": "轻量",
+        "en": "lite",
+    },
+    "ollama_online": {
+        "zh": "Ollama 在线",
+        "en": "Ollama online",
+    },
+    "ollama_offline": {
+        "zh": "Ollama 离线",
+        "en": "Ollama offline",
+    },
+    "cloud_connected": {
+        "zh": "云端已连接",
+        "en": "cloud ✓",
+    },
+    "model_singular": {
+        "zh": "个模型",
+        "en": "model",
+    },
+    "model_plural": {
+        "zh": "个模型",
+        "en": "models",
+    },
+    # Hint line
+    "try": {
+        "zh": "试试",
+        "en": "try",
+    },
+    # Model picker
+    "select_model": {
+        "zh": "选择模型",
+        "en": "Select Model",
+    },
+    "installed": {
+        "zh": "已安装",
+        "en": "installed",
+    },
+    "not_installed": {
+        "zh": "未安装",
+        "en": "not installed",
+    },
+    "cancelled": {
+        "zh": "已取消",
+        "en": "Cancelled",
+    },
+    "no_change": {
+        "zh": "未更改",
+        "en": "No change",
+    },
+    "community_models": {
+        "zh": "社区模型 (Ollama)",
+        "en": "Community (Ollama)",
+    },
+    "ollama_unreachable": {
+        "zh": "Ollama 无法连接",
+        "en": "Ollama unreachable",
+    },
+    "switch_model_hint": {
+        "zh": "使用 /model <id> 切换模型",
+        "en": "Use /model <id> to switch",
+    },
+    "current": {
+        "zh": "当前",
+        "en": "current",
+    },
+    # Permission / mode labels
+    "network_on": {
+        "zh": "网络开",
+        "en": "network on",
+    },
+    "network_off": {
+        "zh": "网络关",
+        "en": "network off",
+    },
+    "privacy": {
+        "zh": "隐私",
+        "en": "privacy",
+    },
+    # Setup wizard
+    "detecting_lang": {
+        "zh": "检测到系统语言：中文",
+        "en": "Detected system language: English",
+    },
+    "first_run_welcome": {
+        "zh": "欢迎使用 Aria Code！首次运行，正在自动配置...",
+        "en": "Welcome to Aria Code! First run — auto-configuring...",
+    },
+    "auto_model_selected": {
+        "zh": "已自动选择本地模型：",
+        "en": "Auto-selected local model: ",
+    },
+    "no_ollama_models": {
+        "zh": "未发现本地模型，使用默认配置",
+        "en": "No local models found, using default config",
+    },
+    # Thinking mode
+    "thinking_mode": {
+        "zh": "思考模式",
+        "en": "Thinking Mode",
+    },
+    # Misc
+    "tip": {
+        "zh": "提示",
+        "en": "tip",
+    },
+    "auto_matched": {
+        "zh": "自动匹配",
+        "en": "auto-matched",
+    },
+}
+
+
+# ---------------------------------------------------------------------------
+# Detection
+# ---------------------------------------------------------------------------
+
+def detect_system_lang() -> str:
+    """Detect OS language. Returns 'zh' for Chinese systems, 'en' for everything else.
+
+    Checks (in order): $LANGUAGE, $LANG, $LC_ALL, $LC_MESSAGES, Python locale module.
+    Treats zh_CN, zh_TW, zh_HK, zh_SG as 'zh'.
+    """
+    for var in ("LANGUAGE", "LANG", "LC_ALL", "LC_MESSAGES"):
+        raw = os.environ.get(var, "").strip()
+        if not raw or raw in ("C", "POSIX", "C.UTF-8"):
+            continue
+        # LANGUAGE may be colon-separated list: zh_CN:en
+        code = raw.split(":")[0].split(".")[0].split("_")[0].lower()
+        if code == "zh":
+            return "zh"
+        if code in ("en", "fr", "de", "es", "pt", "ja", "ko", "ru", "ar"):
+            return code if code in ("ja", "ko") else "en"
+
+    # Fallback: Python locale module (works on macOS/Windows when env vars absent)
+    try:
+        import locale as _locale
+        lang_code, _ = _locale.getdefaultlocale()
+        if lang_code:
+            prefix = lang_code.split("_")[0].lower()
+            if prefix == "zh":
+                return "zh"
+            if prefix in ("ja", "ko"):
+                return prefix
+    except Exception:
+        pass
+
+    return "en"
+
+
+def get_ui_lang(config: Optional[dict] = None) -> str:
+    """Return effective UI language.
+
+    1. config["ui_lang"] if explicitly set
+    2. detect_system_lang()
+    """
+    if config:
+        saved = config.get("ui_lang", "")
+        if saved in STRINGS.get("model", {}):  # valid lang code
+            return saved
+    return detect_system_lang()
+
+
+def t(key: str, lang: Optional[str] = None, config: Optional[dict] = None) -> str:
+    """Translate a UI key to the given language (or auto-detect).
+
+    Falls back to 'en' if key/lang not found.
+    """
+    if lang is None:
+        lang = get_ui_lang(config)
+    entry = STRINGS.get(key, {})
+    return entry.get(lang) or entry.get("en") or key
+
+
+# ---------------------------------------------------------------------------
+# Ollama model helpers. No longer called during config load — see
+# SettingsService.load(), which no longer overrides the configured model.
+# ---------------------------------------------------------------------------
+
+_MODEL_PRIORITY = [
+    "qwen2.5:7b",
+    "qwen2.5-coder:7b",
+    "deepseek-r1:7b",
+    "qwen2.5:3b",
+    "qwen2.5-coder:3b",
+    "llama3.2:3b",
+    "mistral:7b",
+    "qwen2.5-coder:1.5b",
+    "qwen2.5:1.5b",
+    "llama3.2:1b",
+]
+
+
+def installed_ollama_models(ollama_url: str = "http://localhost:11434") -> set[str]:
+    """The model names Ollama reports, or an empty set if it cannot be reached."""
+    try:
+        import urllib.request as _ur
+        import json as _json
+        with _ur.urlopen(f"{ollama_url}/api/tags", timeout=3) as resp:
+            data = _json.loads(resp.read())
+        return {m["name"] for m in data.get("models", [])}
+    except Exception:
+        return set()
+
+
+def preferred_installed_model(installed: set[str]) -> Optional[str]:
+    """The best model from *installed* that this list vouches for, else None.
+
+    Pure, so it is testable without a running Ollama.
+
+    Note what this does NOT do: there is no alphabetical last resort. The
+    function it replaced (``auto_select_model``) ended with
+    ``sorted(installed)[0]``, and it was called during config load to overwrite
+    the caller's default model. A machine with one stray 0.5B model got that
+    model as its agent. Declining to answer is the correct answer when nothing
+    on the curated list is present; ``model_catalog.pick_best_installed_model``
+    is the richer version that also consults the capability registry.
+    """
+    for pref in _MODEL_PRIORITY:
+        if pref in installed:
+            return pref
+    return None

aria_code/apps/cli/intent.py

@@ -0,0 +1,136 @@
+"""Intent classification helpers extracted from aria_cli.py.
+
+All functions are pure — no I/O, no globals, fully unit-testable.
+"""
+from __future__ import annotations
+
+# ── Keyword lists ─────────────────────────────────────────────────────────────
+
+CODING_KEYWORDS = (
+    "write", "generate", "create", "script", "code", "plot", "backtest",
+    "策略", "代码", "回测", "编写", "生成", "k线", "k-line", "kline",
+    "analyze and save", "analysis script", "python", "dashboard",
+    "写一个", "生成代码", "写代码", "编写代码",
+)
+
+VISUAL_ARTIFACT_KEYWORDS = (
+    "图表", "走势图", "k线图", "k线", "k-line", "kline", "candlestick",
+    "chart", "plot", "dashboard", "看板", "晨报", "日报", "周报", "月报",
+    "report", "热力图", "heatmap",
+)
+
+ANALYSIS_KEYWORDS = (
+    "analyze", "analysis", "分析", "研究", "评估", "研判",
+    "技术面", "基本面", "走势", "趋势", "行情",
+    "技术分析", "技术指标", "支撑", "阻力", "支撑位", "阻力位",
+    "rsi", "macd", "bollinger", "布林", "均线", "kdj", "kdj指标",
+    "stock analysis", "technical analysis", "fundamental",
+    "valuation", "estimate", "outlook", "投资建议", "买入", "卖出",
+    "看多", "看空", "多头", "空头", "金叉", "死叉",
+)
+
+ANALYSIS_NON_STOCK_TOPICS = (
+    "房价", "楼市", "房产", "房地产", "租金", "二手房", "新房", "商铺", "折旧",
+    "宏观", "宏观经济", "宏观政策", "宏观角度", "经济政策", "货币政策",
+    "财政政策", "gdp", "通胀", "通货膨胀", "cpi", "ppi", "利率政策",
+    "黄金", "原油", "大宗商品", "汇率", "外汇", "美元指数",
+)
+
+GENERAL_KNOWLEDGE_KEYWORDS = (
+    "什么是", "what is", "what are", "how does", "explain", "define",
+    "解释", "定义", "概念", "原理", "介绍", "步骤", "流程", "怎么",
+    "如何理解", "是什么", "为什么", "区别", "difference between",
+    "tell me about", "describe", "how to", "注册", "成立", "公司",
+    "基本概念", "简介", "举例", "example", "例子",
+    "足球", "篮球", "网球", "棒球", "橄榄球", "排球", "乒乓球", "羽毛球",
+    "世界杯", "欧冠", "英超", "德甲", "西甲", "意甲", "法甲", "bundesliga",
+    "比赛", "赛事", "比分", "进球", "射门", "门将", "球队", "球员", "教练",
+    "联赛", "积分榜", "赛程", "晋级", "淘汰赛", "决赛", "半决赛",
+    "football", "soccer", "match", "goal", "league", "champion",
+    "nba", "nfl", "mlb", "f1", "赛车", "奥运", "olympic",
+)
+
+FINANCE_CONCEPT_TERMS = (
+    "dcf", "pe", "pb", "ps", "ev", "ebitda", "ebit", "wacc", "capm",
+    "beta", "alpha", "sharpe", "sortino", "var", "cvar", "drawdown",
+    "black-scholes", "bs模型", "期权", "期货", "衍生品", "套利",
+    "量化", "quant", "回测", "factor", "因子", "ic值", "ir值",
+    "市盈率", "市净率", "净利润", "营业收入", "自由现金流", "贴现",
+    "折现", "估值", "valuation", "ipo", "etf", "reits", "债券",
+    "利率", "收益率", "久期", "凸性", "信用利差", "风险溢价",
+    "动量", "均值回归", "布林带", "macd", "rsi", "kdj", "技术指标",
+    "北向资金", "融资融券", "股指期货", "沪深300", "中证500",
+)
+
+SPORTS_KEYWORDS = (
+    "足球", "世界杯", "欧冠", "英超", "德甲", "西甲", "意甲", "法甲",
+    "篮球", "nba", "网球", "f1", "赛车", "奥运", "olympic",
+    "比赛", "赛事", "比分", "进球", "球队", "球员", "联赛",
+    "football", "soccer", "world cup", "champions league",
+    "match", "score", "league", "premier league", "bundesliga",
+)
+
+
+# ── Classifier functions ──────────────────────────────────────────────────────
+
+def is_coding_request(message: str) -> bool:
+    """Return True if message looks like a coding/file-generation task."""
+    low = message.lower()
+    if any(k in low for k in VISUAL_ARTIFACT_KEYWORDS):
+        return True
+    if any(k in low for k in CODING_KEYWORDS):
+        return True
+    if low.startswith("/code") or low.startswith("/gen-"):
+        return True
+    return False
+
+
+def is_sports_query(message: str) -> bool:
+    """Return True if the message is about sports/football."""
+    low = message.lower()
+    return any(k in low for k in SPORTS_KEYWORDS)
+
+
+def is_analysis_request(message: str) -> bool:
+    """Return True if message is a stock/crypto technical analysis request (not coding).
+
+    Excludes real-estate, pure macro, and sports questions: those match keywords
+    like '分析'/'走势' but should NOT use the stock technical-analysis template
+    (which requires injected market data to be useful).
+    """
+    if is_coding_request(message):
+        return False
+    low = message.lower()
+    if any(k in low for k in ANALYSIS_NON_STOCK_TOPICS):
+        return False
+    if is_sports_query(message):
+        return False
+    return any(k in low for k in ANALYSIS_KEYWORDS)
+
+
+def is_general_knowledge(message: str) -> bool:
+    """Return True for pure knowledge/explanation questions that don't need tools.
+
+    Finance/quant terms are excluded so they keep the full FINANCE_CHAT_PROMPT
+    even when phrased as "X是什么" explanatory questions.
+
+    Pure macro/conceptual analysis questions are treated as general knowledge:
+    routing them to the finance prompt with tools causes the model to fetch live
+    prices and output the stock-analysis template instead of thoughtful commentary.
+    """
+    if is_coding_request(message) or is_analysis_request(message):
+        return False
+    low = message.lower().strip()
+    if any(term in low for term in FINANCE_CONCEPT_TERMS):
+        return False
+    _macro_conceptual = (
+        "宏观", "宏观经济", "宏观政策", "宏观角度", "宏观分析",
+        "货币政策", "财政政策", "值得投资吗", "应该投资吗", "是否值得",
+        "投资逻辑", "长期展望", "未来前景",
+    )
+    if (any(k in low for k in _macro_conceptual)
+            and not any(c.isdigit() for c in low)):
+        return True
+    if len(low) < 30 and not any(c.isdigit() for c in low):
+        return True
+    return any(k in low for k in GENERAL_KNOWLEDGE_KEYWORDS)