aria-code 0.167.0__py3-none-win_amd64.whl

aria_code/apps/cli/i18n.py

@@ -0,0 +1,310 @@
+"""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
+        # getlocale(), not getdefaultlocale(): the latter is deprecated and
+        # removed in Python 3.15, where this fallback would stop working.
+        lang_code, _ = _locale.getlocale()
+        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 ui_text(owner, zh_text: str, en_text: str) -> str:
+    """The text for the configured UI language; English unless ui_lang is zh.
+
+    ``owner`` is a command object (its ``terminal.config``), a terminal (its
+    ``config``) or a config dict.
+    """
+    config = owner if isinstance(owner, dict) else (
+        getattr(owner, "config", None) or getattr(getattr(owner, "terminal", None), "config", None) or {})
+    return zh_text if str(config.get("ui_lang", "en")).lower().startswith("zh") else en_text
+
+
+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)