aria-code 4.1.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (284) hide show
  1. agents/__init__.py +32 -0
  2. agents/base.py +190 -0
  3. agents/deep/__init__.py +37 -0
  4. agents/deep/calibration_loop.py +144 -0
  5. agents/deep/critic.py +125 -0
  6. agents/deep/deepen.py +193 -0
  7. agents/deep/models.py +149 -0
  8. agents/deep/pipeline.py +164 -0
  9. agents/deep/quant_fusion.py +192 -0
  10. agents/deep/themes.py +95 -0
  11. agents/deep/tiers.py +106 -0
  12. agents/financial/__init__.py +10 -0
  13. agents/financial/catalyst.py +279 -0
  14. agents/financial/debate.py +145 -0
  15. agents/financial/earnings.py +303 -0
  16. agents/financial/fundamental.py +159 -0
  17. agents/financial/macro.py +99 -0
  18. agents/financial/news.py +207 -0
  19. agents/financial/risk.py +132 -0
  20. agents/financial/sector.py +279 -0
  21. agents/financial/synthesis.py +274 -0
  22. agents/financial/technical.py +258 -0
  23. agents/portfolio_agent.py +333 -0
  24. agents/realty/__init__.py +62 -0
  25. agents/realty/asset_diagnosis.py +150 -0
  26. agents/realty/business_match.py +165 -0
  27. agents/realty/cashflow_verify.py +208 -0
  28. agents/realty/contract_rules.py +209 -0
  29. agents/realty/energy_anomaly.py +188 -0
  30. agents/realty/exit_settlement.py +207 -0
  31. agents/realty/fulfillment_risk.py +205 -0
  32. agents/realty/ops_optimize.py +159 -0
  33. agents/realty/revenue_share.py +214 -0
  34. agents/registry.py +144 -0
  35. agents/sports/__init__.py +0 -0
  36. agents/sports/football_agent.py +169 -0
  37. agents/team.py +289 -0
  38. aliyun_data_client.py +660 -0
  39. apps/README.md +12 -0
  40. apps/__init__.py +2 -0
  41. apps/channels/README.md +15 -0
  42. apps/cli/README.md +13 -0
  43. apps/cli/__init__.py +2 -0
  44. apps/cli/bootstrap.py +99 -0
  45. apps/cli/codegen_paths.py +29 -0
  46. apps/cli/commands/__init__.py +16 -0
  47. apps/cli/commands/analysis_cmds.py +288 -0
  48. apps/cli/commands/backtest_cmds.py +1887 -0
  49. apps/cli/commands/broker_cmds.py +1154 -0
  50. apps/cli/commands/business_workflow_cmds.py +289 -0
  51. apps/cli/commands/catalog.py +84 -0
  52. apps/cli/commands/data_cmds.py +405 -0
  53. apps/cli/commands/diagnostic_cmds.py +179 -0
  54. apps/cli/commands/diagnostic_ops_cmds.py +696 -0
  55. apps/cli/commands/finance_render.py +12 -0
  56. apps/cli/commands/market.py +399 -0
  57. apps/cli/commands/market_cmds.py +1276 -0
  58. apps/cli/commands/market_context.py +425 -0
  59. apps/cli/commands/market_render.py +7 -0
  60. apps/cli/commands/model_cmds.py +1579 -0
  61. apps/cli/commands/ops_cmds.py +668 -0
  62. apps/cli/commands/portfolio_cmds.py +962 -0
  63. apps/cli/commands/report.py +377 -0
  64. apps/cli/commands/scaffold_templates.py +617 -0
  65. apps/cli/commands/session_cmds.py +179 -0
  66. apps/cli/commands/session_ux_cmds.py +280 -0
  67. apps/cli/commands/team.py +588 -0
  68. apps/cli/commands/team_render.py +8 -0
  69. apps/cli/commands/ui_cmds.py +358 -0
  70. apps/cli/commands/workflow_cmds.py +279 -0
  71. apps/cli/commands/workspace_cmds.py +1414 -0
  72. apps/cli/config_paths.py +70 -0
  73. apps/cli/config_store.py +61 -0
  74. apps/cli/deterministic.py +122 -0
  75. apps/cli/direct.py +48 -0
  76. apps/cli/github_app_auth.py +135 -0
  77. apps/cli/handlers/__init__.py +11 -0
  78. apps/cli/handlers/broker_handlers.py +122 -0
  79. apps/cli/handlers/chart_handlers.py +1309 -0
  80. apps/cli/handlers/market_handlers.py +2509 -0
  81. apps/cli/handlers/realty_handlers.py +114 -0
  82. apps/cli/handlers/strategy_advice.py +82 -0
  83. apps/cli/hooks.py +180 -0
  84. apps/cli/i18n.py +284 -0
  85. apps/cli/intent.py +136 -0
  86. apps/cli/intent_router.py +217 -0
  87. apps/cli/lifecycle_hooks.py +48 -0
  88. apps/cli/main.py +29 -0
  89. apps/cli/market_metadata.py +135 -0
  90. apps/cli/market_universe.py +265 -0
  91. apps/cli/message_processing.py +257 -0
  92. apps/cli/plan_mode.py +139 -0
  93. apps/cli/plotly_html.py +15 -0
  94. apps/cli/prediction_feedback.py +202 -0
  95. apps/cli/preflight.py +497 -0
  96. apps/cli/project_aria.py +60 -0
  97. apps/cli/prompts/__init__.py +0 -0
  98. apps/cli/prompts/coding.py +658 -0
  99. apps/cli/prompts/system_prompts.py +531 -0
  100. apps/cli/prompts/ui.py +434 -0
  101. apps/cli/providers/__init__.py +1 -0
  102. apps/cli/providers/base.py +271 -0
  103. apps/cli/providers/chat_routing.py +80 -0
  104. apps/cli/providers/llm/__init__.py +1 -0
  105. apps/cli/providers/llm/ollama_stream.py +1170 -0
  106. apps/cli/providers/llm/sse_stream.py +216 -0
  107. apps/cli/providers/runtime_bridge.py +185 -0
  108. apps/cli/runtime_consumer.py +489 -0
  109. apps/cli/session_export.py +87 -0
  110. apps/cli/session_jsonl.py +207 -0
  111. apps/cli/session_store.py +112 -0
  112. apps/cli/todo_tracker.py +190 -0
  113. apps/cli/tools/__init__.py +40 -0
  114. apps/cli/tools/context.py +46 -0
  115. apps/cli/tools/file_tools.py +112 -0
  116. apps/cli/tools/market_tools.py +549 -0
  117. apps/cli/tools/notebook_tools.py +111 -0
  118. apps/cli/tools/system_tools.py +669 -0
  119. apps/cli/tools/write_tools.py +715 -0
  120. apps/cli/tradingview_bridge.py +434 -0
  121. apps/cli/update_check.py +152 -0
  122. apps/cli/utils/__init__.py +0 -0
  123. apps/cli/utils/market_detect.py +1578 -0
  124. apps/daemon/README.md +14 -0
  125. apps/vscode/README.md +115 -0
  126. apps/vscode/package.json +70 -0
  127. aria_cli.py +11636 -0
  128. aria_code-4.1.3.dist-info/METADATA +952 -0
  129. aria_code-4.1.3.dist-info/RECORD +284 -0
  130. aria_code-4.1.3.dist-info/WHEEL +5 -0
  131. aria_code-4.1.3.dist-info/entry_points.txt +2 -0
  132. aria_code-4.1.3.dist-info/licenses/LICENSE +121 -0
  133. aria_code-4.1.3.dist-info/top_level.txt +50 -0
  134. aria_daemon.py +1295 -0
  135. aria_feishu_bot.py +1359 -0
  136. aria_relay_client.py +182 -0
  137. aria_relay_server.py +405 -0
  138. aria_telegram_bot.py +202 -0
  139. ariarc.py +328 -0
  140. artifacts.py +491 -0
  141. backtest_report.py +472 -0
  142. brokers/__init__.py +72 -0
  143. brokers/base.py +207 -0
  144. brokers/capabilities.py +264 -0
  145. brokers/cn/__init__.py +10 -0
  146. brokers/cn/easytrader_broker.py +193 -0
  147. brokers/cn/futu_broker.py +194 -0
  148. brokers/cn/longbridge_broker.py +190 -0
  149. brokers/cn/tiger_broker.py +196 -0
  150. brokers/cn/xtquant_broker.py +175 -0
  151. brokers/config.py +364 -0
  152. brokers/intl/__init__.py +5 -0
  153. brokers/intl/alpaca_broker.py +183 -0
  154. brokers/intl/ibkr_broker.py +215 -0
  155. brokers/intl/webull_broker.py +156 -0
  156. brokers/paper_broker.py +259 -0
  157. brokers/planning.py +296 -0
  158. brokers/registry.py +181 -0
  159. brokers/trading.py +237 -0
  160. change_store.py +127 -0
  161. command_safety.py +19 -0
  162. computer_use_tools.py +504 -0
  163. dashboard_generator.py +578 -0
  164. data_analysis_tools.py +808 -0
  165. data_cleaner.py +483 -0
  166. data_service.py +481 -0
  167. datasources/__init__.py +23 -0
  168. datasources/base.py +166 -0
  169. datasources/router.py +221 -0
  170. datasources/sources/__init__.py +15 -0
  171. datasources/sources/akshare_source.py +269 -0
  172. datasources/sources/alpha_vantage_source.py +202 -0
  173. datasources/sources/edgar_source.py +218 -0
  174. datasources/sources/finnhub_source.py +197 -0
  175. datasources/sources/fred_source.py +219 -0
  176. datasources/sources/tushare_source.py +141 -0
  177. datasources/sources/web_scraper_source.py +278 -0
  178. datasources/sources/world_bank_source.py +205 -0
  179. datasources/sources/yfinance_source.py +152 -0
  180. demo_player.py +204 -0
  181. doctor.py +508 -0
  182. file_analysis_tools.py +734 -0
  183. finance_formulas.py +389 -0
  184. football_data_client.py +1670 -0
  185. intent_classifier.py +358 -0
  186. local_finance_tools.py +3221 -0
  187. local_llm_provider.py +552 -0
  188. macro_tools.py +368 -0
  189. market_data_client.py +1899 -0
  190. mcp_client.py +506 -0
  191. memory_manager.py +245 -0
  192. model_capability.py +416 -0
  193. notification_tools.py +248 -0
  194. packages/__init__.py +23 -0
  195. packages/aria_agents/__init__.py +5 -0
  196. packages/aria_agents/manifest.py +69 -0
  197. packages/aria_core/__init__.py +34 -0
  198. packages/aria_core/architecture.py +192 -0
  199. packages/aria_core/export.py +124 -0
  200. packages/aria_core/manifest.py +65 -0
  201. packages/aria_infra/__init__.py +15 -0
  202. packages/aria_infra/arthera.py +52 -0
  203. packages/aria_infra/doctor.py +246 -0
  204. packages/aria_infra/product.py +37 -0
  205. packages/aria_mcp/__init__.py +25 -0
  206. packages/aria_mcp/bridge.py +38 -0
  207. packages/aria_mcp/config.py +97 -0
  208. packages/aria_mcp/tools.py +61 -0
  209. packages/aria_sdk/__init__.py +19 -0
  210. packages/aria_sdk/client.py +396 -0
  211. packages/aria_sdk/providers.py +70 -0
  212. packages/aria_sdk/streaming.py +73 -0
  213. packages/aria_sdk/types.py +86 -0
  214. packages/aria_services/__init__.py +55 -0
  215. packages/aria_services/context.py +258 -0
  216. packages/aria_services/data.py +11 -0
  217. packages/aria_services/provider_health.py +189 -0
  218. packages/aria_services/registry.py +213 -0
  219. packages/aria_services/usage.py +138 -0
  220. packages/aria_skills/__init__.py +5 -0
  221. packages/aria_skills/registry.py +59 -0
  222. packages/aria_tools/__init__.py +5 -0
  223. packages/aria_tools/registry.py +128 -0
  224. packages/quant_engine/__init__.py +6 -0
  225. packages/quant_engine/sports/__init__.py +72 -0
  226. packages/quant_engine/sports/calibrator.py +353 -0
  227. packages/quant_engine/sports/dixon_coles.py +234 -0
  228. packages/quant_engine/sports/elo.py +299 -0
  229. packages/quant_engine/sports/form.py +188 -0
  230. packages/quant_engine/sports/h2h.py +195 -0
  231. packages/quant_engine/sports/ml_model.py +354 -0
  232. packages/quant_engine/sports/predictor.py +311 -0
  233. packages/quant_engine/sports/tracker.py +664 -0
  234. packages/quant_engine/stochastic/__init__.py +27 -0
  235. packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
  236. packages/quant_engine/stochastic/ito_calculus.py +477 -0
  237. packages/quant_engine/stochastic/kelly_criterion.py +181 -0
  238. packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
  239. packages/quant_engine/stochastic/options_pricing.py +573 -0
  240. packages/quant_engine/stochastic/stochastic_processes.py +90 -0
  241. plan_utils.py +194 -0
  242. plugin_loader.py +328 -0
  243. portfolio_ledger.py +262 -0
  244. privacy/__init__.py +5 -0
  245. privacy/feedback.py +123 -0
  246. project_tools.py +525 -0
  247. providers/__init__.py +30 -0
  248. providers/llm/__init__.py +19 -0
  249. providers/llm/anthropic.py +184 -0
  250. providers/llm/base.py +139 -0
  251. providers/llm/ollama.py +128 -0
  252. providers/llm/openai_compat.py +282 -0
  253. providers/llm/registry.py +358 -0
  254. realty_data_tools.py +659 -0
  255. report_generator.py +1314 -0
  256. runtime/__init__.py +103 -0
  257. runtime/agent_loop.py +1183 -0
  258. runtime/approval.py +51 -0
  259. runtime/events.py +102 -0
  260. runtime/gateway.py +128 -0
  261. runtime/lsp.py +346 -0
  262. runtime/subagent.py +258 -0
  263. runtime/tool_executor.py +104 -0
  264. runtime/tool_policy.py +106 -0
  265. safety/__init__.py +21 -0
  266. safety/permissions.py +275 -0
  267. setup_wizard.py +653 -0
  268. strategy_vault.py +420 -0
  269. ui/__init__.py +100 -0
  270. ui/banner.py +310 -0
  271. ui/completer.py +391 -0
  272. ui/console.py +271 -0
  273. ui/image_render.py +243 -0
  274. ui/input_box.py +376 -0
  275. ui/picker.py +195 -0
  276. ui/render/__init__.py +11 -0
  277. ui/render/finance.py +1480 -0
  278. ui/render/market.py +225 -0
  279. ui/render/output.py +681 -0
  280. ui/render/team.py +346 -0
  281. ui/robot.py +235 -0
  282. workspace/__init__.py +6 -0
  283. workspace/files.py +170 -0
  284. workspace/verify.py +113 -0
@@ -0,0 +1,192 @@
1
+ """P2 — fuse quantitative ground truth and calibrate confidence.
2
+
3
+ The qualitative team votes a signal with a self-reported confidence. That number
4
+ is uncalibrated — an agent saying "80%" doesn't mean it's right 80% of the time.
5
+ This layer:
6
+
7
+ 1. gathers quant signals (AI signal, risk metrics, backtest) as ground truth,
8
+ 2. nudges confidence by whether quant *agrees* with the qualitative verdict,
9
+ 3. scales by a *reliability* factor learned from realised outcomes (CalibrationStore),
10
+
11
+ so confidence drifts toward the historical hit-rate as outcomes accumulate.
12
+ """
13
+
14
+ from __future__ import annotations
15
+
16
+ import json
17
+ import os
18
+ import time
19
+ from pathlib import Path
20
+ from typing import Callable, Dict, Optional, Tuple
21
+
22
+ from .models import QuantEvidence, Provenance
23
+
24
+ _BULL = ("STRONG_BUY", "BUY")
25
+ _BEAR = ("STRONG_SELL", "SELL")
26
+
27
+
28
+ # ── gather quant evidence ─────────────────────────────────────────────────────
29
+ def _default_provider(symbol: str) -> Dict[str, Dict]:
30
+ """Best-effort pull of quant signals from local_finance_tools (each optional)."""
31
+ out: Dict[str, Dict] = {}
32
+ try:
33
+ import local_finance_tools as lft
34
+ except Exception:
35
+ return out
36
+ for key, fn in (("ai", "_get_ai_signal"), ("risk", "_get_risk_metrics"),
37
+ ("backtest", "_backtest_strategy"), ("factors", "_calculate_factors")):
38
+ f = getattr(lft, fn, None)
39
+ if not f:
40
+ continue
41
+ try:
42
+ res = f({"symbol": symbol})
43
+ if isinstance(res, dict) and res.get("success"):
44
+ out[key] = res
45
+ except Exception:
46
+ pass
47
+ return out
48
+
49
+
50
+ def _num(d: Dict, *keys) -> Optional[float]:
51
+ for k in keys:
52
+ v = d.get(k)
53
+ if isinstance(v, (int, float)):
54
+ return float(v)
55
+ return None
56
+
57
+
58
+ def gather_quant_evidence(
59
+ symbol: str,
60
+ provider: Optional[Callable[[str], Dict[str, Dict]]] = None,
61
+ ) -> Tuple[QuantEvidence, list]:
62
+ """Return (QuantEvidence, [Provenance]). Never raises."""
63
+ raw = {}
64
+ try:
65
+ raw = (provider or _default_provider)(symbol)
66
+ except Exception:
67
+ raw = {}
68
+
69
+ prov: list = []
70
+ ev = QuantEvidence()
71
+ if not raw:
72
+ ev.note = "no quant backend available"
73
+ return ev, prov
74
+
75
+ ai = raw.get("ai") or {}
76
+ if ai:
77
+ action = str(ai.get("action") or ai.get("signal") or "").upper()
78
+ ev.ai_signal = action or None
79
+ conf = _num(ai, "confidence") or 0.0
80
+ if action in _BULL:
81
+ ev.ai_score = conf
82
+ elif action in _BEAR:
83
+ ev.ai_score = -conf
84
+ else:
85
+ ev.ai_score = 0.0
86
+ prov.append(Provenance("ai_signal", str(ai.get("provider", "quant")), note=action))
87
+
88
+ risk = raw.get("risk") or {}
89
+ if risk:
90
+ ev.sharpe = _num(risk, "sharpe", "sharpe_ratio")
91
+ ev.max_drawdown = _num(risk, "max_drawdown", "max_dd")
92
+ prov.append(Provenance("risk_metrics", "local_finance", note="VaR/Sharpe"))
93
+
94
+ bt = raw.get("backtest") or {}
95
+ if bt:
96
+ ev.backtest_return = _num(bt, "total_return", "return", "cagr")
97
+ if ev.sharpe is None:
98
+ ev.sharpe = _num(bt, "sharpe", "sharpe_ratio")
99
+ if ev.max_drawdown is None:
100
+ ev.max_drawdown = _num(bt, "max_drawdown", "max_dd")
101
+ prov.append(Provenance("backtest", "local_finance", note=str(bt.get("strategy", ""))))
102
+
103
+ fac = raw.get("factors") or {}
104
+ if fac:
105
+ ev.ic = _num(fac, "ic", "information_coefficient")
106
+ ev.factors = {k: v for k, v in fac.items()
107
+ if k not in ("success", "symbol") and isinstance(v, (int, float))}
108
+ prov.append(Provenance("factors", "local_finance"))
109
+
110
+ ev.available = bool(ai or risk or bt or fac)
111
+ return ev, prov
112
+
113
+
114
+ # ── confidence calibration ────────────────────────────────────────────────────
115
+ def _bucket(conf: float) -> str:
116
+ if conf < 0.4:
117
+ return "lo"
118
+ if conf < 0.7:
119
+ return "mid"
120
+ return "hi"
121
+
122
+
123
+ def _side(signal: str) -> str:
124
+ if signal in _BULL:
125
+ return "bull"
126
+ if signal in _BEAR:
127
+ return "bear"
128
+ return "neutral"
129
+
130
+
131
+ def agreement(agent_signal: str, quant_verdict: str) -> str:
132
+ a = _side(agent_signal)
133
+ q = {"BULLISH": "bull", "BEARISH": "bear", "NEUTRAL": "neutral"}.get(quant_verdict, "neutral")
134
+ if a == "neutral" or q == "neutral":
135
+ return "neutral"
136
+ return "agree" if a == q else "disagree"
137
+
138
+
139
+ class CalibrationStore:
140
+ """Tracks realised hit-rate per (signal-side, confidence-bucket) on disk.
141
+
142
+ ``reliability(conf, signal)`` returns observed_hit_rate / nominal_confidence,
143
+ clamped to a sane band, so a chronically over-confident bucket gets damped and
144
+ an under-confident one gets a small boost. With no history it returns 1.0.
145
+ """
146
+
147
+ _NOMINAL = {"lo": 0.30, "mid": 0.55, "hi": 0.80}
148
+
149
+ def __init__(self, path: Optional[Path] = None):
150
+ self.path = Path(path or os.path.expanduser("~/.arthera/deep_calibration.json"))
151
+ self._data: Dict[str, Dict[str, int]] = {}
152
+ try:
153
+ if self.path.exists():
154
+ self._data = json.loads(self.path.read_text())
155
+ except Exception:
156
+ self._data = {}
157
+
158
+ def _key(self, side: str, bucket: str) -> str:
159
+ return f"{side}:{bucket}"
160
+
161
+ def reliability(self, conf: float, signal: str = "") -> float:
162
+ rec = self._data.get(self._key(_side(signal), _bucket(conf)))
163
+ if not rec or rec.get("n", 0) < 8:
164
+ return 1.0
165
+ hit = rec["hit"] / rec["n"]
166
+ nominal = self._NOMINAL[_bucket(conf)] or 1.0
167
+ return max(0.6, min(1.25, hit / nominal))
168
+
169
+ def record_outcome(self, signal: str, conf: float, correct: bool) -> None:
170
+ k = self._key(_side(signal), _bucket(conf))
171
+ rec = self._data.setdefault(k, {"n": 0, "hit": 0})
172
+ rec["n"] += 1
173
+ rec["hit"] += 1 if correct else 0
174
+ try:
175
+ self.path.parent.mkdir(parents=True, exist_ok=True)
176
+ self.path.write_text(json.dumps(self._data, indent=2))
177
+ except Exception:
178
+ pass
179
+
180
+
181
+ def calibrate_confidence(
182
+ raw_conf: float,
183
+ agent_signal: str,
184
+ quant: QuantEvidence,
185
+ store: Optional[CalibrationStore] = None,
186
+ ) -> Tuple[float, str]:
187
+ """Return (calibrated_confidence, agreement_label)."""
188
+ agree = agreement(agent_signal, quant.verdict()) if quant and quant.available else "neutral"
189
+ factor = {"agree": 1.15, "neutral": 1.0, "disagree": 0.70}[agree]
190
+ rel = store.reliability(raw_conf, agent_signal) if store else 1.0
191
+ cal = max(0.0, min(1.0, raw_conf * factor * rel))
192
+ return cal, agree
agents/deep/themes.py ADDED
@@ -0,0 +1,95 @@
1
+ """P1a — group agent results by theme and build a per-theme sub-synthesis.
2
+
3
+ Flat synthesis ("here are 8 opinions, here's the average") loses structure. Real
4
+ research clusters evidence: what does *valuation* say, what does *momentum* say,
5
+ what does *risk* say — then reconciles across clusters. This module does the
6
+ clustering and a deterministic per-cluster roll-up (no LLM needed).
7
+ """
8
+
9
+ from __future__ import annotations
10
+
11
+ from typing import Dict, List, Tuple
12
+
13
+ from ..base import AgentResult
14
+ from .models import ThemeGroup
15
+
16
+ # Which theme each agent belongs to. Unknown agents fall into "other".
17
+ _AGENT_THEME = {
18
+ "fundamental": "valuation",
19
+ "earnings": "valuation",
20
+ "technical": "momentum",
21
+ "risk": "risk",
22
+ "news": "catalysts",
23
+ "catalyst": "catalysts",
24
+ "macro": "macro",
25
+ "sector": "macro",
26
+ "northbound": "macro",
27
+ "debate": "reconciliation",
28
+ }
29
+
30
+ # Display order + human labels.
31
+ _THEME_ORDER = ["valuation", "momentum", "macro", "catalysts", "risk",
32
+ "reconciliation", "other"]
33
+ _THEME_LABEL = {
34
+ "valuation": "估值/基本面",
35
+ "momentum": "动量/技术",
36
+ "macro": "宏观/板块",
37
+ "catalysts": "催化/消息",
38
+ "risk": "风险",
39
+ "reconciliation": "分歧调解",
40
+ "other": "其他",
41
+ }
42
+
43
+ _SCORE = {"STRONG_BUY": 2, "BUY": 1, "HOLD": 0, "SELL": -1, "STRONG_SELL": -2}
44
+
45
+
46
+ def theme_of(agent_name: str) -> str:
47
+ return _AGENT_THEME.get(agent_name, "other")
48
+
49
+
50
+ def _vote(results: List[AgentResult]) -> Tuple[str, float]:
51
+ """Confidence-weighted majority within a single theme."""
52
+ valid = [r for r in results if r.success and r.signal in _SCORE]
53
+ if not valid:
54
+ return "HOLD", 0.0
55
+ avg_score = sum(_SCORE[r.signal] * r.confidence for r in valid) / len(valid)
56
+ avg_conf = sum(r.confidence for r in valid) / len(valid)
57
+ if avg_score >= 1.5:
58
+ return "STRONG_BUY", avg_conf
59
+ if avg_score >= 0.5:
60
+ return "BUY", avg_conf
61
+ if avg_score <= -1.5:
62
+ return "STRONG_SELL", avg_conf
63
+ if avg_score <= -0.5:
64
+ return "SELL", avg_conf
65
+ return "HOLD", avg_conf
66
+
67
+
68
+ def group_by_theme(results: List[AgentResult]) -> List[ThemeGroup]:
69
+ """Cluster agent results into themes with a per-theme signal + summary."""
70
+ buckets: Dict[str, List[AgentResult]] = {}
71
+ for r in results:
72
+ buckets.setdefault(theme_of(r.agent), []).append(r)
73
+
74
+ groups: List[ThemeGroup] = []
75
+ for theme in _THEME_ORDER:
76
+ members = buckets.get(theme)
77
+ if not members:
78
+ continue
79
+ signal, conf = _vote(members)
80
+ points: List[str] = []
81
+ for r in members:
82
+ if r.success:
83
+ points.extend((r.key_points or [])[:2])
84
+ ok = sum(1 for r in members if r.success)
85
+ summary = (f"{_THEME_LABEL[theme]}: {signal}({ok}/{len(members)} agent 有效,"
86
+ f"置信度 {conf:.0%})")
87
+ groups.append(ThemeGroup(
88
+ theme=_THEME_LABEL[theme],
89
+ agents=[r.agent for r in members],
90
+ signal=signal,
91
+ confidence=conf,
92
+ summary=summary,
93
+ key_points=points[:5],
94
+ ))
95
+ return groups
agents/deep/tiers.py ADDED
@@ -0,0 +1,106 @@
1
+ """P3 — render the deep result at three depths, with data provenance.
2
+
3
+ brief — one glance: signal, calibrated confidence, headline
4
+ standard — + per-theme roll-up + synthesis
5
+ deep — + quant evidence + critique + provenance (data lineage) + agent points
6
+
7
+ Pure text/markdown so it renders in the terminal; the HTML report layer can reuse
8
+ the same DeepAnalysisResult.
9
+ """
10
+
11
+ from __future__ import annotations
12
+
13
+ from .models import DeepAnalysisResult
14
+
15
+ _SIGNAL_ICON = {
16
+ "STRONG_BUY": "🟢🟢", "BUY": "🟢", "HOLD": "⚪",
17
+ "SELL": "🔴", "STRONG_SELL": "🔴🔴",
18
+ }
19
+
20
+
21
+ def _headline(r: DeepAnalysisResult) -> str:
22
+ icon = _SIGNAL_ICON.get(r.final_signal, "⚪")
23
+ conf = f"{r.calibrated_confidence:.0%}"
24
+ note = ""
25
+ if r.raw_confidence and abs(r.calibrated_confidence - r.raw_confidence) >= 0.05:
26
+ note = f"(原始 {r.raw_confidence:.0%} → 校准 {conf})"
27
+ else:
28
+ note = f"(置信度 {conf})"
29
+ return f"{icon} **{r.symbol} · {r.final_signal}** {note}"
30
+
31
+
32
+ def render_brief(r: DeepAnalysisResult) -> str:
33
+ lines = [_headline(r)]
34
+ if r.quant and r.quant.available:
35
+ lines.append(f" 量化: {r.quant.verdict()}"
36
+ + (f" · IC {r.quant.ic:.2f}" if r.quant.ic is not None else "")
37
+ + (f" · Sharpe {r.quant.sharpe:.2f}" if r.quant.sharpe is not None else ""))
38
+ if r.critique and r.critique.high:
39
+ lines.append(f" ⚠️ {r.critique.high[0].message}")
40
+ return "\n".join(lines)
41
+
42
+
43
+ def render_standard(r: DeepAnalysisResult) -> str:
44
+ parts = [render_brief(r), ""]
45
+ if r.themes:
46
+ parts.append("### 分主题")
47
+ for t in r.themes:
48
+ parts.append(f"- {t.summary}")
49
+ parts.append("")
50
+ if r.synthesis:
51
+ parts.append("### 综合")
52
+ parts.append(r.synthesis)
53
+ return "\n".join(parts).rstrip()
54
+
55
+
56
+ def render_deep(r: DeepAnalysisResult) -> str:
57
+ parts = [render_standard(r), ""]
58
+
59
+ if r.quant and r.quant.available:
60
+ q = r.quant
61
+ parts.append("### 量化地面真值")
62
+ row = []
63
+ if q.ai_signal: row.append(f"AI信号 {q.ai_signal}")
64
+ if q.ai_score is not None:row.append(f"分值 {q.ai_score:+.2f}")
65
+ if q.ic is not None: row.append(f"IC {q.ic:.3f}")
66
+ if q.sharpe is not None: row.append(f"Sharpe {q.sharpe:.2f}")
67
+ if q.max_drawdown is not None: row.append(f"MaxDD {q.max_drawdown:.1%}")
68
+ if q.backtest_return is not None: row.append(f"回测收益 {q.backtest_return:+.1%}")
69
+ parts.append("- " + " · ".join(row) if row else "- (无)")
70
+ parts.append("")
71
+
72
+ if r.critique is not None:
73
+ parts.append("### 自检 (Critic)")
74
+ mark = "✅ 通过" if r.critique.passed else "❌ 存在高危问题"
75
+ parts.append(f"结论: {mark}" + ("(无问题)" if not r.critique.issues else ""))
76
+ for i in r.critique.issues:
77
+ sev = {"high": "🔴", "medium": "🟡", "low": "⚪"}.get(i.severity, "·")
78
+ parts.append(f"- {sev} [{i.kind}] {i.message}")
79
+ parts.append("")
80
+
81
+ if r.provenance:
82
+ parts.append("### 数据血缘")
83
+ parts.append("| 字段 | 来源 | 时效 | 备注 |")
84
+ parts.append("|------|------|------|------|")
85
+ for p in r.provenance:
86
+ parts.append(f"| {p.field} | {p.source} | {p.freshness} | {p.note} |")
87
+ parts.append("")
88
+
89
+ if r.agent_results:
90
+ parts.append("### 各 Agent 要点")
91
+ for a in r.agent_results:
92
+ if a.get("error"):
93
+ parts.append(f"- **{a['agent']}** ⚠️ {a['error']}")
94
+ continue
95
+ pts = a.get("key_points") or []
96
+ head = f"- **{a['agent']}** ({a.get('signal','?')}, {a.get('confidence',0):.0%})"
97
+ parts.append(head)
98
+ for pt in pts[:3]:
99
+ parts.append(f" • {pt}")
100
+
101
+ return "\n".join(parts).rstrip()
102
+
103
+
104
+ def render_tier(r: DeepAnalysisResult, tier: str = "standard") -> str:
105
+ return {"brief": render_brief, "standard": render_standard,
106
+ "deep": render_deep}.get(tier, render_standard)(r)
@@ -0,0 +1,10 @@
1
+ from .technical import TechnicalAgent
2
+ from .macro import MacroAgent
3
+ from .fundamental import FundamentalAgent
4
+ from .risk import RiskAgent
5
+ from .synthesis import SynthesisAgent
6
+
7
+ __all__ = [
8
+ "TechnicalAgent", "MacroAgent",
9
+ "FundamentalAgent", "RiskAgent", "SynthesisAgent",
10
+ ]
@@ -0,0 +1,279 @@
1
+ """
2
+ agents/financial/catalyst.py — 催化剂检测 Agent
3
+ ================================================
4
+ 识别近期/即将到来的价格催化剂:财报日、股息除权、
5
+ 分析师评级变化、大宗交易/大股东增减持。
6
+ """
7
+
8
+ from __future__ import annotations
9
+
10
+ import logging
11
+ from datetime import datetime, timedelta, timezone
12
+ from typing import Any, Dict, List, Optional
13
+
14
+ from ..base import BaseAgent, AgentResult
15
+
16
+ logger = logging.getLogger(__name__)
17
+
18
+
19
+ class CatalystAgent(BaseAgent):
20
+
21
+ name = "catalyst"
22
+ description = "催化剂检测 — 财报日、股息除权、分析师评级变化"
23
+
24
+ _SYSTEM = (
25
+ "You are an event-driven equity analyst. Analyze the upcoming and recent "
26
+ "catalysts for a stock: earnings dates, ex-dividend dates, analyst rating "
27
+ "changes, and insider activity. Assess whether these catalysts are "
28
+ "likely to be POSITIVE (price-driving), NEUTRAL, or NEGATIVE. "
29
+ "Focus on timing: catalysts within 14 days are high-impact. "
30
+ "Conclude with: POSITIVE / NEUTRAL / NEGATIVE"
31
+ )
32
+
33
+ async def fetch_data(self, symbol: str) -> Dict[str, Any]:
34
+ data = await super().fetch_data(symbol)
35
+ catalysts: Dict[str, Any] = {}
36
+
37
+ try:
38
+ import yfinance as yf
39
+ ticker = yf.Ticker(symbol)
40
+
41
+ # 1. Earnings calendar
42
+ try:
43
+ cal = ticker.calendar
44
+ if cal is not None:
45
+ if hasattr(cal, "to_dict"):
46
+ cal = cal.to_dict()
47
+ if isinstance(cal, dict):
48
+ for k in ("Earnings Date", "earnings_date", "earningsDate"):
49
+ v = cal.get(k)
50
+ if v:
51
+ catalysts["earnings_date"] = str(v[0] if isinstance(v, list) else v)
52
+ break
53
+ except Exception as e:
54
+ logger.debug("[catalyst] calendar fetch %s: %s", symbol, e)
55
+
56
+ # 2. Recent analyst recommendations
57
+ try:
58
+ recs = ticker.recommendations
59
+ if recs is not None and not recs.empty:
60
+ recs = recs.sort_index(ascending=False)
61
+ recent_recs = recs.head(5)
62
+ rec_list = []
63
+ for ts, row in recent_recs.iterrows():
64
+ rec_list.append({
65
+ "date": str(ts)[:10],
66
+ "firm": str(row.get("Firm", row.get("firm", ""))),
67
+ "grade": str(row.get("To Grade", row.get("toGrade", row.get("action", "")))),
68
+ "prev": str(row.get("From Grade", row.get("fromGrade", ""))),
69
+ "action": str(row.get("Action", row.get("action", ""))),
70
+ })
71
+ catalysts["recommendations"] = rec_list
72
+ except Exception as e:
73
+ logger.debug("[catalyst] recs fetch %s: %s", symbol, e)
74
+
75
+ # 3. Upcoming dividend
76
+ try:
77
+ info = ticker.info or {}
78
+ ex_div = info.get("exDividendDate")
79
+ div_rate = info.get("dividendRate") or info.get("trailingAnnualDividendRate")
80
+ if ex_div:
81
+ from datetime import date
82
+ ex_date = datetime.fromtimestamp(ex_div, tz=timezone.utc).date()
83
+ days_to_exdiv = (ex_date - date.today()).days
84
+ catalysts["ex_dividend"] = {
85
+ "date": str(ex_date),
86
+ "days_away": days_to_exdiv,
87
+ "annual_rate": div_rate,
88
+ }
89
+ except Exception as e:
90
+ logger.debug("[catalyst] dividend fetch %s: %s", symbol, e)
91
+
92
+ # 4. Short interest (proxy for contrarian catalyst)
93
+ try:
94
+ info = ticker.info or {}
95
+ short_ratio = info.get("shortRatio")
96
+ short_pct = info.get("shortPercentOfFloat")
97
+ if short_ratio or short_pct:
98
+ catalysts["short_interest"] = {
99
+ "ratio": short_ratio,
100
+ "pct_float": short_pct,
101
+ }
102
+ except Exception as e:
103
+ logger.debug("[catalyst] short interest %s: %s", symbol, e)
104
+
105
+ except Exception as e:
106
+ logger.debug("[catalyst] yfinance init %s: %s", symbol, e)
107
+
108
+ data["catalysts"] = catalysts
109
+ return data
110
+
111
+ async def analyze(self, symbol: str, data: Dict[str, Any]) -> AgentResult:
112
+ catalysts = data.get("catalysts", {})
113
+ quote = data.get("quote", {})
114
+ price = quote.get("price", 0)
115
+
116
+ if not catalysts:
117
+ return AgentResult(
118
+ agent=self.name, symbol=symbol,
119
+ analysis=f"{symbol}: 未获取到催化剂数据。",
120
+ confidence=0.3, signal="HOLD",
121
+ key_points=["无近期催化剂数据"],
122
+ )
123
+
124
+ catalyst_block = _format_catalysts(catalysts)
125
+ urgency = _assess_urgency(catalysts)
126
+
127
+ prompt = (
128
+ f"Stock: {symbol} Price: {price}\n\n"
129
+ f"Catalysts:\n{catalyst_block}\n\n"
130
+ "Evaluate:\n"
131
+ "1. Most impactful upcoming catalyst and timing\n"
132
+ "2. Analyst sentiment trend (upgrades vs downgrades)\n"
133
+ "3. Event-driven trade setup (if any)\n"
134
+ "4. Risk of negative surprise\n"
135
+ "5. Conclude: POSITIVE / NEUTRAL / NEGATIVE"
136
+ )
137
+
138
+ analysis = await self._call_llm(self._SYSTEM, prompt, max_tokens=450)
139
+ if not analysis:
140
+ analysis = _template_analysis(symbol, catalysts, urgency)
141
+
142
+ signal, confidence = _derive_signal(analysis, catalysts, urgency)
143
+ key_points = _build_key_points(catalysts, urgency)
144
+
145
+ return AgentResult(
146
+ agent=self.name, symbol=symbol,
147
+ analysis=analysis,
148
+ confidence=confidence,
149
+ signal=signal,
150
+ key_points=key_points,
151
+ data_used=catalysts,
152
+ )
153
+
154
+
155
+ # ── Helpers ───────────────────────────────────────────────────────────────────
156
+
157
+ def _format_catalysts(c: Dict) -> str:
158
+ lines = []
159
+
160
+ if "earnings_date" in c:
161
+ ed = c["earnings_date"]
162
+ try:
163
+ ed_dt = datetime.fromisoformat(str(ed).split()[0])
164
+ days = (ed_dt.date() - datetime.today().date()).days
165
+ timing = f"({days}天后)" if days >= 0 else f"({-days}天前)"
166
+ except Exception:
167
+ timing = ""
168
+ lines.append(f"财报日: {ed}{timing}")
169
+
170
+ if "ex_dividend" in c:
171
+ d = c["ex_dividend"]
172
+ days_away = d.get("days_away", 999)
173
+ rate = d.get("annual_rate", "")
174
+ rate_str = f" 年化股息 {rate:.2f}" if rate else ""
175
+ lines.append(f"除息日: {d.get('date','')}({days_away}天后){rate_str}")
176
+
177
+ if "recommendations" in c:
178
+ for r in c["recommendations"][:3]:
179
+ action = r.get("action", "").upper()
180
+ firm = r.get("firm", "")
181
+ grade = r.get("grade", "")
182
+ prev = r.get("prev", "")
183
+ change = f"{prev} → {grade}" if prev and prev != grade else grade
184
+ lines.append(f"分析师评级: [{r.get('date','')}] {firm} {action} {change}")
185
+
186
+ if "short_interest" in c:
187
+ si = c["short_interest"]
188
+ ratio = si.get("ratio")
189
+ pct = si.get("pct_float")
190
+ parts = []
191
+ if ratio: parts.append(f"空头比率 {ratio:.1f}x")
192
+ if pct: parts.append(f"流通股空仓 {pct*100:.1f}%")
193
+ if parts: lines.append(f"空头数据: {', '.join(parts)}")
194
+
195
+ return "\n".join(lines) or "无催化剂数据"
196
+
197
+
198
+ def _assess_urgency(c: Dict) -> str:
199
+ if "earnings_date" in c:
200
+ try:
201
+ ed_dt = datetime.fromisoformat(str(c["earnings_date"]).split()[0])
202
+ days = (ed_dt.date() - datetime.today().date()).days
203
+ if 0 <= days <= 7:
204
+ return "high"
205
+ if 0 <= days <= 14:
206
+ return "medium"
207
+ except Exception:
208
+ pass
209
+ exdiv = c.get("ex_dividend", {})
210
+ if 0 <= exdiv.get("days_away", 999) <= 5:
211
+ return "high"
212
+ return "low"
213
+
214
+
215
+ def _derive_signal(analysis: str, c: Dict, urgency: str) -> tuple[str, float]:
216
+ text = analysis.upper()
217
+ recs = c.get("recommendations", [])
218
+ upgrades = sum(1 for r in recs if "UPGRAD" in r.get("action", "").upper())
219
+ downgrades = sum(1 for r in recs if "DOWNGRAD" in r.get("action", "").upper())
220
+
221
+ if "POSITIVE" in text:
222
+ conf = 0.65 if urgency == "high" else 0.55
223
+ return "BUY", conf
224
+ if "NEGATIVE" in text:
225
+ conf = 0.65 if urgency == "high" else 0.55
226
+ return "SELL", conf
227
+
228
+ if upgrades > downgrades:
229
+ return "BUY", 0.5
230
+ if downgrades > upgrades:
231
+ return "SELL", 0.5
232
+ return "HOLD", 0.4
233
+
234
+
235
+ def _build_key_points(c: Dict, urgency: str) -> List[str]:
236
+ points = []
237
+ if "earnings_date" in c:
238
+ try:
239
+ ed_dt = datetime.fromisoformat(str(c["earnings_date"]).split()[0])
240
+ days = (ed_dt.date() - datetime.today().date()).days
241
+ points.append(f"财报日在 {days} 天后" if days >= 0 else f"财报 {-days} 天前已公布")
242
+ except Exception:
243
+ points.append(f"财报日: {c['earnings_date']}")
244
+ if "ex_dividend" in c:
245
+ d = c["ex_dividend"]
246
+ points.append(f"除息日 {d.get('date','')}({d.get('days_away','')}天)")
247
+ recs = c.get("recommendations", [])
248
+ if recs:
249
+ latest = recs[0]
250
+ points.append(f"最新评级: {latest.get('firm','')} {latest.get('grade','')}")
251
+ if urgency == "high":
252
+ points.append("⚡ 高优先级催化剂(7天内)")
253
+ return points[:5]
254
+
255
+
256
+ def _template_analysis(symbol: str, c: Dict, urgency: str) -> str:
257
+ parts = []
258
+ if "earnings_date" in c:
259
+ try:
260
+ ed_dt = datetime.fromisoformat(str(c["earnings_date"]).split()[0])
261
+ days = (ed_dt.date() - datetime.today().date()).days
262
+ parts.append(f"财报日在 {days} 天{'后' if days >= 0 else '前'}")
263
+ except Exception:
264
+ parts.append(f"财报日: {c['earnings_date']}")
265
+ recs = c.get("recommendations", [])
266
+ upgrades = sum(1 for r in recs if "UPGRAD" in r.get("action", "").upper())
267
+ downgrades = sum(1 for r in recs if "DOWNGRAD" in r.get("action", "").upper())
268
+ if upgrades or downgrades:
269
+ parts.append(f"近期评级: {upgrades}次升级 / {downgrades}次降级")
270
+ if "ex_dividend" in c:
271
+ d = c["ex_dividend"]
272
+ parts.append(f"除息日: {d.get('date','')}({d.get('days_away','')}天)")
273
+
274
+ sentiment = "POSITIVE" if upgrades > downgrades else ("NEGATIVE" if downgrades > upgrades else "NEUTRAL")
275
+ return (
276
+ f"{symbol} 催化剂摘要:\n"
277
+ + "\n".join(f" • {p}" for p in parts)
278
+ + f"\n结论:{sentiment}"
279
+ )