aria-code 4.1.3__py3-none-any.whl

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)

apps/cli/intent_router.py

@@ -0,0 +1,217 @@
+"""Canonical lightweight intent routing for CLI services.
+
+This module keeps routing side-effect free.  It does not call tools, install
+packages, or fetch data; it only classifies a user message into stable service
+intents that other layers can reuse.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Iterable
+
+
+def _contains_any(text: str, keywords: Iterable[str]) -> bool:
+    return any(keyword in text for keyword in keywords)
+
+
+def _first_token(text: str) -> str:
+    stripped = text.strip()
+    if not stripped:
+        return ""
+    return stripped.split(maxsplit=1)[0].lower()
+
+
+def _add_unique(items: list[str], name: str) -> None:
+    if name not in items:
+        items.append(name)
+
+
+@dataclass(frozen=True)
+class IntentRoute:
+    message: str
+    primary: str
+    intents: tuple[str, ...]
+    services: tuple[str, ...]
+    explicit_code: bool = False
+    visual_artifact: bool = False
+    market_related: bool = False
+
+    @property
+    def wants_market_prefetch(self) -> bool:
+        return self.market_related and not self.visual_artifact and self.primary != "general"
+
+    @property
+    def allows_code_autorun(self) -> bool:
+        return self.primary in {"code", "strategy", "backtest"} and self.explicit_code
+
+
+COMMAND_INTENTS = {
+    "/chart": "chart",
+    "/dashboard": "dashboard",
+    "/report": "report",
+    "/team": "market_research",
+    "/analyze": "market_analysis",
+    "/ta": "market_analysis",
+    "/quote": "market_snapshot",
+    "/backtest": "backtest",
+    "/auto-strategy": "strategy",
+    "/strategy": "strategy",
+    "/ui": "ui_artifact",
+    "/vision": "vision",
+    "/upload-image": "vision",
+    "/screenshot": "screenshot",
+    "/browser": "browser",
+    "/file": "file_analysis",
+    "/github": "github",
+    "/mcp": "mcp",
+    "/cloud": "cloud",
+}
+
+
+def detect_intents(message: str) -> tuple[str, ...]:
+    low = message.lower().strip()
+    cmd = _first_token(low)
+    intents: list[str] = []
+
+    mapped = COMMAND_INTENTS.get(cmd)
+    if mapped:
+        _add_unique(intents, mapped)
+
+    if _contains_any(low, ("k线图", "k线", "k-line", "kline", "candlestick", "走势图", "图表", "chart", "plot")):
+        _add_unique(intents, "chart")
+    if _contains_any(low, ("看板", "晨报", "日报", "dashboard", "market board", "持仓看板")):
+        _add_unique(intents, "dashboard")
+    if _contains_any(low, ("研究报告", "财报", "报告", "report", "研报")):
+        _add_unique(intents, "report")
+    if _contains_any(low, ("回测", "backtest", "收益曲线", "最大回撤")):
+        _add_unique(intents, "backtest")
+    if _contains_any(low, ("策略代码", "写策略", "量化策略", "交易策略", "strategy code", "trading bot")):
+        _add_unique(intents, "strategy")
+    if _contains_any(low, ("技术分析", "技术指标", "rsi", "macd", "均线", "支撑", "阻力")):
+        _add_unique(intents, "market_analysis")
+    if _contains_any(low, ("股票", "行情", "持仓", "portfolio", "quote", "market data")):
+        _add_unique(intents, "market_snapshot")
+    if _contains_any(low, ("图片", "图像", "截图", "上传图片", "分析图片", "识别图片", "image", "screenshot")):
+        _add_unique(intents, "vision")
+    if _contains_any(low, ("浏览器", "网页截图", "打开网页", "browser", "playwright")):
+        _add_unique(intents, "browser")
+    if _contains_any(low, ("pdf", "docx", "word", "excel", "xlsx", "csv", "文件分析", "上传文件")):
+        _add_unique(intents, "file_analysis")
+    if _contains_any(low, ("github", "pull request", "pr ", "issue", "ci")):
+        _add_unique(intents, "github")
+    if _contains_any(low, ("mcp", "server", "tools", "skills")):
+        _add_unique(intents, "mcp")
+    if _contains_any(low, ("ollama", "本地模型", "local model")):
+        _add_unique(intents, "local_model")
+    if _contains_any(low, ("阿里云", "aliyun", "cloud service", "云端服务")):
+        _add_unique(intents, "cloud")
+    if _contains_any(low, ("搜索", "联网", "上网查", "web search", "google", "查一下", "搜一下", "最新消息")):
+        _add_unique(intents, "web_search")
+    if _contains_any(low, ("加密货币", "比特币", "以太坊", "bitcoin", "btc", "eth", "crypto", "币安", "binance", "okx", "资金费率", "funding rate")):
+        _add_unique(intents, "crypto")
+    if _contains_any(low, ("足球", "球赛", "比分预测", "世界杯", "欧洲杯", "英超", "西甲", "football", "soccer", "world cup", "premier league")):
+        _add_unique(intents, "sports")
+
+    return tuple(intents)
+
+
+def _service_names(intents: tuple[str, ...]) -> tuple[str, ...]:
+    services: list[str] = []
+
+    def service(name: str) -> None:
+        _add_unique(services, name)
+
+    if any(i in intents for i in ("market_snapshot", "market_analysis", "chart", "dashboard", "report", "backtest", "strategy", "market_research")):
+        service("market_data")
+    if "chart" in intents:
+        service("chart_renderer")
+    if "dashboard" in intents:
+        service("dashboard_generator")
+    if "report" in intents:
+        service("report_generator")
+    if "backtest" in intents or "strategy" in intents:
+        service("backtest_engine")
+    if "vision" in intents:
+        service("vision_input")
+    if "screenshot" in intents:
+        service("screenshot")
+    if "browser" in intents:
+        service("browser")
+    if "file_analysis" in intents:
+        service("file_parser")
+    if "github" in intents:
+        service("github_cli")
+    if "mcp" in intents:
+        service("mcp")
+    if "local_model" in intents:
+        service("local_llm")
+    if "cloud" in intents:
+        service("cloud_runtime")
+    if "web_search" in intents:
+        service("web_search")
+    if "crypto" in intents:
+        service("crypto_data")
+    if "sports" in intents:
+        service("sports_data")
+    return tuple(services)
+
+
+def build_intent_route(message: str) -> IntentRoute:
+    low = message.lower().strip()
+    intents = detect_intents(message)
+    explicit_code = _contains_any(low, (
+        "代码", "脚本", "python", "程序", "实现", "开发", "修改文件",
+        "写代码", "编写代码", "策略代码", "保存为.py", ".py",
+        "script", "code", "program", "implement", "edit file", "write file",
+    ))
+    try:
+        from intent_classifier import (
+            INTENT_ANALYSIS,
+            INTENT_CODING,
+            INTENT_FINANCE,
+            INTENT_GENERAL,
+            INTENT_REALTIME,
+            classify_intent_sync,
+            is_visual_market_artifact_request,
+        )
+        classifier_intent = classify_intent_sync(message)
+        visual_artifact = bool(is_visual_market_artifact_request(message))
+    except Exception:
+        INTENT_ANALYSIS = "analysis"
+        INTENT_CODING = "coding"
+        INTENT_FINANCE = "finance"
+        INTENT_GENERAL = "general"
+        INTENT_REALTIME = "realtime"
+        classifier_intent = INTENT_FINANCE
+        visual_artifact = any(i in intents for i in ("chart", "dashboard", "report", "ui_artifact"))
+
+    if intents:
+        primary = intents[0]
+    elif classifier_intent == INTENT_REALTIME:
+        primary = "market_snapshot"
+    elif classifier_intent == INTENT_ANALYSIS:
+        primary = "market_analysis"
+    elif classifier_intent == INTENT_CODING:
+        primary = "code"
+    elif classifier_intent == INTENT_GENERAL:
+        primary = "general"
+    elif classifier_intent == INTENT_FINANCE:
+        primary = "finance"
+    else:
+        primary = "finance"
+
+    market_related = any(i in intents for i in (
+        "market_snapshot", "market_analysis", "chart", "dashboard", "report",
+        "backtest", "strategy", "market_research",
+    )) or classifier_intent in {INTENT_ANALYSIS, INTENT_REALTIME, INTENT_FINANCE}
+
+    return IntentRoute(
+        message=message,
+        primary=primary,
+        intents=intents,
+        services=_service_names(intents),
+        explicit_code=explicit_code,
+        visual_artifact=visual_artifact,
+        market_related=market_related,
+    )

apps/cli/lifecycle_hooks.py

@@ -0,0 +1,48 @@
+"""Simple lifecycle shell hooks for the CLI adapter."""
+
+from __future__ import annotations
+
+import os
+import subprocess
+from pathlib import Path
+from typing import Mapping
+
+
+def run_event_hook(
+    event: str,
+    *,
+    config_dir: Path,
+    cwd: Path | None = None,
+    env_extra: Mapping[str, str] | None = None,
+    timeout: int = 10,
+) -> None:
+    """Run executable hook scripts for a lifecycle event.
+
+    Script locations are intentionally simple for compatibility:
+    `<config_dir>/hooks/<event>.sh` and `<cwd>/.aria/hooks/<event>.sh`.
+    Hook failures are non-fatal.
+    """
+    root = cwd or Path.cwd()
+    dirs = [
+        config_dir / "hooks",
+        root / ".aria" / "hooks",
+    ]
+    env = dict(os.environ)
+    env["ARIA_EVENT"] = event
+    if env_extra:
+        env.update({str(k): str(v) for k, v in env_extra.items()})
+    for hook_dir in dirs:
+        script = hook_dir / f"{event}.sh"
+        if not script.exists() or script.stat().st_size <= 0:
+            continue
+        try:
+            subprocess.run(
+                [str(script)],
+                env=env,
+                timeout=timeout,
+                capture_output=True,
+                text=True,
+                check=False,
+            )
+        except Exception:
+            continue

apps/cli/main.py

@@ -0,0 +1,29 @@
+"""Installed console entrypoint for Aria Code.
+
+The terminal implementation lives in ``aria_cli`` as an async ``main()``. A
+console script (declared in pyproject ``[project.scripts]`` as
+``aria-code = "apps.cli.main:main"``) must be a *synchronous* callable, so this
+shim wraps the coroutine in ``asyncio.run`` — mirroring the
+``if __name__ == "__main__"`` block in ``aria_cli.py``. Calling the async
+``aria_cli.main`` directly from a console script would only create a coroutine
+that is never awaited.
+"""
+
+from __future__ import annotations
+
+import asyncio
+import sys
+
+
+def main() -> None:
+    """Synchronous entry point for the ``aria-code`` command."""
+    from aria_cli import main as _async_main
+
+    try:
+        asyncio.run(_async_main())
+    except KeyboardInterrupt:
+        print()
+        sys.exit(0)
+
+
+__all__ = ["main"]

apps/cli/market_metadata.py

@@ -0,0 +1,135 @@
+"""Local market symbol metadata used to normalize provider output.
+
+Providers occasionally omit display names or return a generic/default currency.
+This module keeps deterministic exchange metadata close to the CLI so prompt
+injection and tool responses do not ask the model to guess symbol identity.
+"""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Mapping
+
+
+@dataclass(frozen=True)
+class MarketAssetMeta:
+    symbol: str
+    short_label: str
+    display_name: str
+    currency: str
+    exchange: str = ""
+
+
+_SYMBOL_META: dict[str, MarketAssetMeta] = {
+    "MC.PA": MarketAssetMeta(
+        symbol="MC.PA",
+        short_label="LVMH/路易威登",
+        display_name="LVMH Moet Hennessy Louis Vuitton SE",
+        currency="EUR",
+        exchange="Euronext Paris",
+    ),
+    "RMS.PA": MarketAssetMeta(
+        symbol="RMS.PA",
+        short_label="Hermes/爱马仕",
+        display_name="Hermes International SCA",
+        currency="EUR",
+        exchange="Euronext Paris",
+    ),
+    "KER.PA": MarketAssetMeta(
+        symbol="KER.PA",
+        short_label="Kering/开云集团",
+        display_name="Kering SA",
+        currency="EUR",
+        exchange="Euronext Paris",
+    ),
+}
+
+_SUFFIX_CURRENCY: tuple[tuple[str, str], ...] = (
+    (".SS", "CNY"),
+    (".SH", "CNY"),
+    (".SZ", "CNY"),
+    (".HK", "HKD"),
+    (".PA", "EUR"),
+    (".DE", "EUR"),
+    (".MI", "EUR"),
+    (".AS", "EUR"),
+    (".BR", "EUR"),
+    (".MC", "EUR"),
+    (".LS", "EUR"),
+    (".SW", "CHF"),
+    (".L", "GBp"),
+    (".TO", "CAD"),
+    (".AX", "AUD"),
+    (".T", "JPY"),
+)
+
+
+def normalize_market_symbol(symbol: str) -> str:
+    return str(symbol or "").strip().upper()
+
+
+def market_asset_meta(symbol: str) -> MarketAssetMeta | None:
+    return _SYMBOL_META.get(normalize_market_symbol(symbol))
+
+
+def default_currency_for_symbol(symbol: str) -> str:
+    normalized = normalize_market_symbol(symbol)
+    meta = market_asset_meta(normalized)
+    if meta:
+        return meta.currency
+    for suffix, currency in _SUFFIX_CURRENCY:
+        if normalized.endswith(suffix):
+            return currency
+    return ""
+
+
+def _provider_currency_looks_wrong(symbol: str, provider_currency: Any, inferred_currency: str) -> bool:
+    if not inferred_currency:
+        return False
+    currency = str(provider_currency or "").strip().upper()
+    inferred = inferred_currency.upper()
+    if not currency:
+        return True
+    if currency == inferred:
+        return False
+    normalized = normalize_market_symbol(symbol)
+    # Most wrong cases here come from global fallbacks defaulting non-US symbols
+    # to USD. Do not override real US tickers that have no exchange suffix.
+    return currency == "USD" and "." in normalized
+
+
+def enrich_market_quote(symbol: str, quote: Mapping[str, Any] | None) -> dict[str, Any]:
+    """Return a quote copy with deterministic symbol name/currency metadata."""
+    data = dict(quote or {})
+    normalized = normalize_market_symbol(symbol or data.get("symbol") or "")
+    if normalized and not data.get("symbol"):
+        data["symbol"] = normalized
+
+    meta = market_asset_meta(normalized)
+    inferred_currency = default_currency_for_symbol(normalized)
+    if _provider_currency_looks_wrong(normalized, data.get("currency"), inferred_currency):
+        data["currency"] = inferred_currency
+        data["currency_source"] = "symbol_metadata"
+    elif not data.get("currency") and inferred_currency:
+        data["currency"] = inferred_currency
+        data["currency_source"] = "symbol_metadata"
+
+    if meta:
+        current_name = str(data.get("name") or "").strip()
+        if not current_name or current_name.upper() == normalized:
+            data["name"] = meta.display_name
+        data.setdefault("display_name", meta.display_name)
+        data.setdefault("short_label", meta.short_label)
+        data.setdefault("exchange", meta.exchange)
+    return data
+
+
+def market_display_label(symbol: str, quote: Mapping[str, Any] | None = None) -> str:
+    data = dict(quote or {})
+    normalized = normalize_market_symbol(symbol or data.get("symbol") or "")
+    meta = market_asset_meta(normalized)
+    if meta:
+        return f"{meta.short_label}({normalized})"
+    name = str(data.get("name") or data.get("display_name") or "").strip()
+    if name and name.upper() != normalized:
+        return f"{name}({normalized})"
+    return normalized or name or "market_data"