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.
- agents/__init__.py +32 -0
- agents/base.py +190 -0
- agents/deep/__init__.py +37 -0
- agents/deep/calibration_loop.py +144 -0
- agents/deep/critic.py +125 -0
- agents/deep/deepen.py +193 -0
- agents/deep/models.py +149 -0
- agents/deep/pipeline.py +164 -0
- agents/deep/quant_fusion.py +192 -0
- agents/deep/themes.py +95 -0
- agents/deep/tiers.py +106 -0
- agents/financial/__init__.py +10 -0
- agents/financial/catalyst.py +279 -0
- agents/financial/debate.py +145 -0
- agents/financial/earnings.py +303 -0
- agents/financial/fundamental.py +159 -0
- agents/financial/macro.py +99 -0
- agents/financial/news.py +207 -0
- agents/financial/risk.py +132 -0
- agents/financial/sector.py +279 -0
- agents/financial/synthesis.py +274 -0
- agents/financial/technical.py +258 -0
- agents/portfolio_agent.py +333 -0
- agents/realty/__init__.py +62 -0
- agents/realty/asset_diagnosis.py +150 -0
- agents/realty/business_match.py +165 -0
- agents/realty/cashflow_verify.py +208 -0
- agents/realty/contract_rules.py +209 -0
- agents/realty/energy_anomaly.py +188 -0
- agents/realty/exit_settlement.py +207 -0
- agents/realty/fulfillment_risk.py +205 -0
- agents/realty/ops_optimize.py +159 -0
- agents/realty/revenue_share.py +214 -0
- agents/registry.py +144 -0
- agents/sports/__init__.py +0 -0
- agents/sports/football_agent.py +169 -0
- agents/team.py +289 -0
- aliyun_data_client.py +660 -0
- apps/README.md +12 -0
- apps/__init__.py +2 -0
- apps/channels/README.md +15 -0
- apps/cli/README.md +13 -0
- apps/cli/__init__.py +2 -0
- apps/cli/bootstrap.py +99 -0
- apps/cli/codegen_paths.py +29 -0
- apps/cli/commands/__init__.py +16 -0
- apps/cli/commands/analysis_cmds.py +288 -0
- apps/cli/commands/backtest_cmds.py +1887 -0
- apps/cli/commands/broker_cmds.py +1154 -0
- apps/cli/commands/business_workflow_cmds.py +289 -0
- apps/cli/commands/catalog.py +84 -0
- apps/cli/commands/data_cmds.py +405 -0
- apps/cli/commands/diagnostic_cmds.py +179 -0
- apps/cli/commands/diagnostic_ops_cmds.py +696 -0
- apps/cli/commands/finance_render.py +12 -0
- apps/cli/commands/market.py +399 -0
- apps/cli/commands/market_cmds.py +1276 -0
- apps/cli/commands/market_context.py +425 -0
- apps/cli/commands/market_render.py +7 -0
- apps/cli/commands/model_cmds.py +1579 -0
- apps/cli/commands/ops_cmds.py +668 -0
- apps/cli/commands/portfolio_cmds.py +962 -0
- apps/cli/commands/report.py +377 -0
- apps/cli/commands/scaffold_templates.py +617 -0
- apps/cli/commands/session_cmds.py +179 -0
- apps/cli/commands/session_ux_cmds.py +280 -0
- apps/cli/commands/team.py +588 -0
- apps/cli/commands/team_render.py +8 -0
- apps/cli/commands/ui_cmds.py +358 -0
- apps/cli/commands/workflow_cmds.py +279 -0
- apps/cli/commands/workspace_cmds.py +1414 -0
- apps/cli/config_paths.py +70 -0
- apps/cli/config_store.py +61 -0
- apps/cli/deterministic.py +122 -0
- apps/cli/direct.py +48 -0
- apps/cli/github_app_auth.py +135 -0
- apps/cli/handlers/__init__.py +11 -0
- apps/cli/handlers/broker_handlers.py +122 -0
- apps/cli/handlers/chart_handlers.py +1309 -0
- apps/cli/handlers/market_handlers.py +2509 -0
- apps/cli/handlers/realty_handlers.py +114 -0
- apps/cli/handlers/strategy_advice.py +82 -0
- apps/cli/hooks.py +180 -0
- apps/cli/i18n.py +284 -0
- apps/cli/intent.py +136 -0
- apps/cli/intent_router.py +217 -0
- apps/cli/lifecycle_hooks.py +48 -0
- apps/cli/main.py +29 -0
- apps/cli/market_metadata.py +135 -0
- apps/cli/market_universe.py +265 -0
- apps/cli/message_processing.py +257 -0
- apps/cli/plan_mode.py +139 -0
- apps/cli/plotly_html.py +15 -0
- apps/cli/prediction_feedback.py +202 -0
- apps/cli/preflight.py +497 -0
- apps/cli/project_aria.py +60 -0
- apps/cli/prompts/__init__.py +0 -0
- apps/cli/prompts/coding.py +658 -0
- apps/cli/prompts/system_prompts.py +531 -0
- apps/cli/prompts/ui.py +434 -0
- apps/cli/providers/__init__.py +1 -0
- apps/cli/providers/base.py +271 -0
- apps/cli/providers/chat_routing.py +80 -0
- apps/cli/providers/llm/__init__.py +1 -0
- apps/cli/providers/llm/ollama_stream.py +1170 -0
- apps/cli/providers/llm/sse_stream.py +216 -0
- apps/cli/providers/runtime_bridge.py +185 -0
- apps/cli/runtime_consumer.py +489 -0
- apps/cli/session_export.py +87 -0
- apps/cli/session_jsonl.py +207 -0
- apps/cli/session_store.py +112 -0
- apps/cli/todo_tracker.py +190 -0
- apps/cli/tools/__init__.py +40 -0
- apps/cli/tools/context.py +46 -0
- apps/cli/tools/file_tools.py +112 -0
- apps/cli/tools/market_tools.py +549 -0
- apps/cli/tools/notebook_tools.py +111 -0
- apps/cli/tools/system_tools.py +669 -0
- apps/cli/tools/write_tools.py +715 -0
- apps/cli/tradingview_bridge.py +434 -0
- apps/cli/update_check.py +152 -0
- apps/cli/utils/__init__.py +0 -0
- apps/cli/utils/market_detect.py +1578 -0
- apps/daemon/README.md +14 -0
- apps/vscode/README.md +115 -0
- apps/vscode/package.json +70 -0
- aria_cli.py +11636 -0
- aria_code-4.1.3.dist-info/METADATA +952 -0
- aria_code-4.1.3.dist-info/RECORD +284 -0
- aria_code-4.1.3.dist-info/WHEEL +5 -0
- aria_code-4.1.3.dist-info/entry_points.txt +2 -0
- aria_code-4.1.3.dist-info/licenses/LICENSE +121 -0
- aria_code-4.1.3.dist-info/top_level.txt +50 -0
- aria_daemon.py +1295 -0
- aria_feishu_bot.py +1359 -0
- aria_relay_client.py +182 -0
- aria_relay_server.py +405 -0
- aria_telegram_bot.py +202 -0
- ariarc.py +328 -0
- artifacts.py +491 -0
- backtest_report.py +472 -0
- brokers/__init__.py +72 -0
- brokers/base.py +207 -0
- brokers/capabilities.py +264 -0
- brokers/cn/__init__.py +10 -0
- brokers/cn/easytrader_broker.py +193 -0
- brokers/cn/futu_broker.py +194 -0
- brokers/cn/longbridge_broker.py +190 -0
- brokers/cn/tiger_broker.py +196 -0
- brokers/cn/xtquant_broker.py +175 -0
- brokers/config.py +364 -0
- brokers/intl/__init__.py +5 -0
- brokers/intl/alpaca_broker.py +183 -0
- brokers/intl/ibkr_broker.py +215 -0
- brokers/intl/webull_broker.py +156 -0
- brokers/paper_broker.py +259 -0
- brokers/planning.py +296 -0
- brokers/registry.py +181 -0
- brokers/trading.py +237 -0
- change_store.py +127 -0
- command_safety.py +19 -0
- computer_use_tools.py +504 -0
- dashboard_generator.py +578 -0
- data_analysis_tools.py +808 -0
- data_cleaner.py +483 -0
- data_service.py +481 -0
- datasources/__init__.py +23 -0
- datasources/base.py +166 -0
- datasources/router.py +221 -0
- datasources/sources/__init__.py +15 -0
- datasources/sources/akshare_source.py +269 -0
- datasources/sources/alpha_vantage_source.py +202 -0
- datasources/sources/edgar_source.py +218 -0
- datasources/sources/finnhub_source.py +197 -0
- datasources/sources/fred_source.py +219 -0
- datasources/sources/tushare_source.py +141 -0
- datasources/sources/web_scraper_source.py +278 -0
- datasources/sources/world_bank_source.py +205 -0
- datasources/sources/yfinance_source.py +152 -0
- demo_player.py +204 -0
- doctor.py +508 -0
- file_analysis_tools.py +734 -0
- finance_formulas.py +389 -0
- football_data_client.py +1670 -0
- intent_classifier.py +358 -0
- local_finance_tools.py +3221 -0
- local_llm_provider.py +552 -0
- macro_tools.py +368 -0
- market_data_client.py +1899 -0
- mcp_client.py +506 -0
- memory_manager.py +245 -0
- model_capability.py +416 -0
- notification_tools.py +248 -0
- packages/__init__.py +23 -0
- packages/aria_agents/__init__.py +5 -0
- packages/aria_agents/manifest.py +69 -0
- packages/aria_core/__init__.py +34 -0
- packages/aria_core/architecture.py +192 -0
- packages/aria_core/export.py +124 -0
- packages/aria_core/manifest.py +65 -0
- packages/aria_infra/__init__.py +15 -0
- packages/aria_infra/arthera.py +52 -0
- packages/aria_infra/doctor.py +246 -0
- packages/aria_infra/product.py +37 -0
- packages/aria_mcp/__init__.py +25 -0
- packages/aria_mcp/bridge.py +38 -0
- packages/aria_mcp/config.py +97 -0
- packages/aria_mcp/tools.py +61 -0
- packages/aria_sdk/__init__.py +19 -0
- packages/aria_sdk/client.py +396 -0
- packages/aria_sdk/providers.py +70 -0
- packages/aria_sdk/streaming.py +73 -0
- packages/aria_sdk/types.py +86 -0
- packages/aria_services/__init__.py +55 -0
- packages/aria_services/context.py +258 -0
- packages/aria_services/data.py +11 -0
- packages/aria_services/provider_health.py +189 -0
- packages/aria_services/registry.py +213 -0
- packages/aria_services/usage.py +138 -0
- packages/aria_skills/__init__.py +5 -0
- packages/aria_skills/registry.py +59 -0
- packages/aria_tools/__init__.py +5 -0
- packages/aria_tools/registry.py +128 -0
- packages/quant_engine/__init__.py +6 -0
- packages/quant_engine/sports/__init__.py +72 -0
- packages/quant_engine/sports/calibrator.py +353 -0
- packages/quant_engine/sports/dixon_coles.py +234 -0
- packages/quant_engine/sports/elo.py +299 -0
- packages/quant_engine/sports/form.py +188 -0
- packages/quant_engine/sports/h2h.py +195 -0
- packages/quant_engine/sports/ml_model.py +354 -0
- packages/quant_engine/sports/predictor.py +311 -0
- packages/quant_engine/sports/tracker.py +664 -0
- packages/quant_engine/stochastic/__init__.py +27 -0
- packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
- packages/quant_engine/stochastic/ito_calculus.py +477 -0
- packages/quant_engine/stochastic/kelly_criterion.py +181 -0
- packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
- packages/quant_engine/stochastic/options_pricing.py +573 -0
- packages/quant_engine/stochastic/stochastic_processes.py +90 -0
- plan_utils.py +194 -0
- plugin_loader.py +328 -0
- portfolio_ledger.py +262 -0
- privacy/__init__.py +5 -0
- privacy/feedback.py +123 -0
- project_tools.py +525 -0
- providers/__init__.py +30 -0
- providers/llm/__init__.py +19 -0
- providers/llm/anthropic.py +184 -0
- providers/llm/base.py +139 -0
- providers/llm/ollama.py +128 -0
- providers/llm/openai_compat.py +282 -0
- providers/llm/registry.py +358 -0
- realty_data_tools.py +659 -0
- report_generator.py +1314 -0
- runtime/__init__.py +103 -0
- runtime/agent_loop.py +1183 -0
- runtime/approval.py +51 -0
- runtime/events.py +102 -0
- runtime/gateway.py +128 -0
- runtime/lsp.py +346 -0
- runtime/subagent.py +258 -0
- runtime/tool_executor.py +104 -0
- runtime/tool_policy.py +106 -0
- safety/__init__.py +21 -0
- safety/permissions.py +275 -0
- setup_wizard.py +653 -0
- strategy_vault.py +420 -0
- ui/__init__.py +100 -0
- ui/banner.py +310 -0
- ui/completer.py +391 -0
- ui/console.py +271 -0
- ui/image_render.py +243 -0
- ui/input_box.py +376 -0
- ui/picker.py +195 -0
- ui/render/__init__.py +11 -0
- ui/render/finance.py +1480 -0
- ui/render/market.py +225 -0
- ui/render/output.py +681 -0
- ui/render/team.py +346 -0
- ui/robot.py +235 -0
- workspace/__init__.py +6 -0
- workspace/files.py +170 -0
- 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
|
+
)
|