aria-code 0.167.0__py3-none-win_amd64.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 (601) hide show
  1. aria_bots/__init__.py +7 -0
  2. aria_bots/channels/README.md +97 -0
  3. aria_bots/channels/__init__.py +1 -0
  4. aria_bots/channels/approvals.py +263 -0
  5. aria_bots/channels/conversation.py +280 -0
  6. aria_bots/channels/digest.py +186 -0
  7. aria_bots/channels/registry.py +95 -0
  8. aria_bots/engine.py +144 -0
  9. aria_bots/feishu_bot.py +1385 -0
  10. aria_bots/logistics/__init__.py +1 -0
  11. aria_bots/logistics/carriers.py +253 -0
  12. aria_bots/logistics/freight.py +154 -0
  13. aria_bots/logistics/inventory.py +242 -0
  14. aria_bots/logistics/mcp_server.py +156 -0
  15. aria_bots/logistics/records.py +119 -0
  16. aria_bots/logistics/tenancy.py +152 -0
  17. aria_bots/market.py +91 -0
  18. aria_bots/relay_client.py +326 -0
  19. aria_bots/telegram_bot.py +299 -0
  20. aria_code/__init__.py +111 -0
  21. aria_code/_bin/aria-native.exe +0 -0
  22. aria_code/_bin/aria-runtime.exe +0 -0
  23. aria_code/_version.py +20 -0
  24. aria_code/account_memory.py +296 -0
  25. aria_code/account_session.py +166 -0
  26. aria_code/adk_apps/__init__.py +1 -0
  27. aria_code/adk_apps/aria_code_review/__init__.py +1 -0
  28. aria_code/adk_apps/aria_code_review/agent.py +32 -0
  29. aria_code/adk_apps/aria_research/__init__.py +1 -0
  30. aria_code/adk_apps/aria_research/agent.py +31 -0
  31. aria_code/agents/__init__.py +49 -0
  32. aria_code/agents/base.py +329 -0
  33. aria_code/agents/code_review.py +200 -0
  34. aria_code/agents/deep/__init__.py +37 -0
  35. aria_code/agents/deep/calibration_loop.py +144 -0
  36. aria_code/agents/deep/critic.py +125 -0
  37. aria_code/agents/deep/deepen.py +193 -0
  38. aria_code/agents/deep/models.py +149 -0
  39. aria_code/agents/deep/pipeline.py +164 -0
  40. aria_code/agents/deep/quant_fusion.py +192 -0
  41. aria_code/agents/deep/themes.py +95 -0
  42. aria_code/agents/deep/tiers.py +106 -0
  43. aria_code/agents/engineering/__init__.py +14 -0
  44. aria_code/agents/engineering/coder.py +222 -0
  45. aria_code/agents/engineering/tester.py +145 -0
  46. aria_code/agents/financial/__init__.py +21 -0
  47. aria_code/agents/financial/cashflow_burn.py +119 -0
  48. aria_code/agents/financial/catalyst.py +279 -0
  49. aria_code/agents/financial/corporate_finance.py +175 -0
  50. aria_code/agents/financial/debate.py +224 -0
  51. aria_code/agents/financial/earnings.py +315 -0
  52. aria_code/agents/financial/fundamental.py +178 -0
  53. aria_code/agents/financial/macro.py +111 -0
  54. aria_code/agents/financial/news.py +231 -0
  55. aria_code/agents/financial/risk.py +146 -0
  56. aria_code/agents/financial/sector.py +279 -0
  57. aria_code/agents/financial/strategist.py +121 -0
  58. aria_code/agents/financial/stripe_revenue.py +127 -0
  59. aria_code/agents/financial/synthesis.py +274 -0
  60. aria_code/agents/financial/technical.py +277 -0
  61. aria_code/agents/orchestrator.py +44 -0
  62. aria_code/agents/portfolio_agent.py +392 -0
  63. aria_code/agents/realty/__init__.py +62 -0
  64. aria_code/agents/realty/asset_diagnosis.py +152 -0
  65. aria_code/agents/realty/business_match.py +166 -0
  66. aria_code/agents/realty/cashflow_verify.py +209 -0
  67. aria_code/agents/realty/contract_rules.py +210 -0
  68. aria_code/agents/realty/energy_anomaly.py +189 -0
  69. aria_code/agents/realty/exit_settlement.py +208 -0
  70. aria_code/agents/realty/fulfillment_risk.py +206 -0
  71. aria_code/agents/realty/ops_optimize.py +160 -0
  72. aria_code/agents/realty/revenue_share.py +215 -0
  73. aria_code/agents/registry.py +168 -0
  74. aria_code/agents/signal_scheme.py +111 -0
  75. aria_code/agents/sports/__init__.py +0 -0
  76. aria_code/agents/sports/football_agent.py +197 -0
  77. aria_code/agents/supervisor.py +113 -0
  78. aria_code/agents/team.py +500 -0
  79. aria_code/agents/video/__init__.py +16 -0
  80. aria_code/agents/video/director.py +376 -0
  81. aria_code/agents/video/qa_agent.py +143 -0
  82. aria_code/agents/video/script_doctor.py +112 -0
  83. aria_code/agents/warehouse/__init__.py +25 -0
  84. aria_code/agents/warehouse/contracts.py +26 -0
  85. aria_code/agents/warehouse/dashboard.py +155 -0
  86. aria_code/agents/warehouse/fulfillment_leadtime.py +95 -0
  87. aria_code/agents/warehouse/inbound_exceptions.py +25 -0
  88. aria_code/agents/warehouse/inventory_health.py +24 -0
  89. aria_code/agents/warehouse/local_run.py +71 -0
  90. aria_code/agents/warehouse/local_snapshot.py +132 -0
  91. aria_code/agents/warehouse/logistics_cost.py +84 -0
  92. aria_code/agents/warehouse/logistics_sync.py +21 -0
  93. aria_code/agents/warehouse/navigation.schema.json +110 -0
  94. aria_code/agents/warehouse/task_agent.py +172 -0
  95. aria_code/agents/warehouse/workflow.py +31 -0
  96. aria_code/aliyun_data_client.py +4 -0
  97. aria_code/apps/README.md +12 -0
  98. aria_code/apps/__init__.py +2 -0
  99. aria_code/apps/channels/README.md +9 -0
  100. aria_code/apps/channels/__init__.py +17 -0
  101. aria_code/apps/channels/approvals.py +11 -0
  102. aria_code/apps/channels/conversation.py +11 -0
  103. aria_code/apps/channels/digest.py +11 -0
  104. aria_code/apps/channels/intake.py +92 -0
  105. aria_code/apps/channels/registry.py +11 -0
  106. aria_code/apps/channels/tradingview.py +94 -0
  107. aria_code/apps/cli/README.md +13 -0
  108. aria_code/apps/cli/__init__.py +2 -0
  109. aria_code/apps/cli/_optional.py +120 -0
  110. aria_code/apps/cli/app_server.py +656 -0
  111. aria_code/apps/cli/auto_update.py +125 -0
  112. aria_code/apps/cli/bootstrap.py +206 -0
  113. aria_code/apps/cli/broker_render.py +206 -0
  114. aria_code/apps/cli/chat_turn.py +1327 -0
  115. aria_code/apps/cli/cloud_models.py +231 -0
  116. aria_code/apps/cli/codegen_paths.py +29 -0
  117. aria_code/apps/cli/commands/__init__.py +34 -0
  118. aria_code/apps/cli/commands/_business_render.py +78 -0
  119. aria_code/apps/cli/commands/_ui.py +71 -0
  120. aria_code/apps/cli/commands/analysis_cmds.py +378 -0
  121. aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
  122. aria_code/apps/cli/commands/auth_cmds.py +215 -0
  123. aria_code/apps/cli/commands/backtest_cmds.py +2518 -0
  124. aria_code/apps/cli/commands/broker_cmds.py +1236 -0
  125. aria_code/apps/cli/commands/business_workflow_cmds.py +379 -0
  126. aria_code/apps/cli/commands/canvas_cmds.py +77 -0
  127. aria_code/apps/cli/commands/catalog.py +189 -0
  128. aria_code/apps/cli/commands/core_cmds.py +2748 -0
  129. aria_code/apps/cli/commands/data_cmds.py +521 -0
  130. aria_code/apps/cli/commands/diagnostic_cmds.py +285 -0
  131. aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
  132. aria_code/apps/cli/commands/engineering_cmds.py +399 -0
  133. aria_code/apps/cli/commands/file_cmds.py +312 -0
  134. aria_code/apps/cli/commands/finance_service_cmds.py +60 -0
  135. aria_code/apps/cli/commands/fx_commodity_cmds.py +123 -0
  136. aria_code/apps/cli/commands/market.py +544 -0
  137. aria_code/apps/cli/commands/market_cmds.py +1419 -0
  138. aria_code/apps/cli/commands/market_context.py +432 -0
  139. aria_code/apps/cli/commands/market_render.py +7 -0
  140. aria_code/apps/cli/commands/model_cmds.py +2374 -0
  141. aria_code/apps/cli/commands/ops_cmds.py +721 -0
  142. aria_code/apps/cli/commands/orchestrator_cmds.py +181 -0
  143. aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
  144. aria_code/apps/cli/commands/portfolio_cmds.py +1316 -0
  145. aria_code/apps/cli/commands/report.py +476 -0
  146. aria_code/apps/cli/commands/scaffold_templates.py +617 -0
  147. aria_code/apps/cli/commands/session_cmds.py +236 -0
  148. aria_code/apps/cli/commands/session_ux_cmds.py +359 -0
  149. aria_code/apps/cli/commands/team.py +723 -0
  150. aria_code/apps/cli/commands/team_render.py +8 -0
  151. aria_code/apps/cli/commands/ui_cmds.py +401 -0
  152. aria_code/apps/cli/commands/warehouse_cmds.py +297 -0
  153. aria_code/apps/cli/commands/workflow_cmds.py +629 -0
  154. aria_code/apps/cli/commands/workspace_cmds.py +1543 -0
  155. aria_code/apps/cli/config_paths.py +73 -0
  156. aria_code/apps/cli/config_store.py +158 -0
  157. aria_code/apps/cli/constants.py +88 -0
  158. aria_code/apps/cli/context.py +34 -0
  159. aria_code/apps/cli/conversation_policy.py +53 -0
  160. aria_code/apps/cli/deterministic.py +200 -0
  161. aria_code/apps/cli/direct.py +48 -0
  162. aria_code/apps/cli/earnings_view.py +172 -0
  163. aria_code/apps/cli/endpoint_migration.py +64 -0
  164. aria_code/apps/cli/exec_events.py +255 -0
  165. aria_code/apps/cli/file_references.py +155 -0
  166. aria_code/apps/cli/finance_service_catalog.py +111 -0
  167. aria_code/apps/cli/football_reports.py +231 -0
  168. aria_code/apps/cli/github_app_auth.py +135 -0
  169. aria_code/apps/cli/google_login.py +239 -0
  170. aria_code/apps/cli/handlers/__init__.py +11 -0
  171. aria_code/apps/cli/handlers/broker_handlers.py +122 -0
  172. aria_code/apps/cli/handlers/chart_handlers.py +1314 -0
  173. aria_code/apps/cli/handlers/market_handlers.py +2570 -0
  174. aria_code/apps/cli/handlers/realty_handlers.py +114 -0
  175. aria_code/apps/cli/handlers/sports_handlers.py +22 -0
  176. aria_code/apps/cli/handlers/strategy_advice.py +82 -0
  177. aria_code/apps/cli/headless.py +352 -0
  178. aria_code/apps/cli/helpers.py +326 -0
  179. aria_code/apps/cli/hooks.py +180 -0
  180. aria_code/apps/cli/i18n.py +310 -0
  181. aria_code/apps/cli/intent.py +136 -0
  182. aria_code/apps/cli/intent_router.py +305 -0
  183. aria_code/apps/cli/intent_shadow.py +72 -0
  184. aria_code/apps/cli/intent_signals.py +347 -0
  185. aria_code/apps/cli/lifecycle_hooks.py +48 -0
  186. aria_code/apps/cli/local_tool_schemas.py +550 -0
  187. aria_code/apps/cli/main.py +39 -0
  188. aria_code/apps/cli/market_hours.py +100 -0
  189. aria_code/apps/cli/market_metadata.py +135 -0
  190. aria_code/apps/cli/market_universe.py +487 -0
  191. aria_code/apps/cli/message_processing.py +310 -0
  192. aria_code/apps/cli/mixin_binding.py +32 -0
  193. aria_code/apps/cli/model_catalog.py +652 -0
  194. aria_code/apps/cli/model_probe.py +169 -0
  195. aria_code/apps/cli/model_retirement.py +65 -0
  196. aria_code/apps/cli/native_branding.py +12 -0
  197. aria_code/apps/cli/native_bridge.py +118 -0
  198. aria_code/apps/cli/native_entry.py +58 -0
  199. aria_code/apps/cli/pdf_report.py +977 -0
  200. aria_code/apps/cli/plan_mode.py +139 -0
  201. aria_code/apps/cli/plotly_html.py +15 -0
  202. aria_code/apps/cli/prediction_feedback.py +202 -0
  203. aria_code/apps/cli/preflight.py +502 -0
  204. aria_code/apps/cli/project_aria.py +60 -0
  205. aria_code/apps/cli/prompt_assembly.py +78 -0
  206. aria_code/apps/cli/prompts/__init__.py +0 -0
  207. aria_code/apps/cli/prompts/coding.py +71 -0
  208. aria_code/apps/cli/prompts/select.py +193 -0
  209. aria_code/apps/cli/prompts/system_prompts.py +562 -0
  210. aria_code/apps/cli/prompts/ui.py +436 -0
  211. aria_code/apps/cli/provider_endpoints.py +237 -0
  212. aria_code/apps/cli/provider_keys.py +73 -0
  213. aria_code/apps/cli/providers/__init__.py +1 -0
  214. aria_code/apps/cli/providers/base.py +775 -0
  215. aria_code/apps/cli/providers/chat_routing.py +139 -0
  216. aria_code/apps/cli/providers/collaboration.py +98 -0
  217. aria_code/apps/cli/providers/llm/__init__.py +1 -0
  218. aria_code/apps/cli/providers/llm/offline_responses.py +68 -0
  219. aria_code/apps/cli/providers/llm/ollama_stream.py +1574 -0
  220. aria_code/apps/cli/providers/llm/response_cache.py +60 -0
  221. aria_code/apps/cli/providers/llm/sse_stream.py +298 -0
  222. aria_code/apps/cli/providers/runtime_bridge.py +586 -0
  223. aria_code/apps/cli/providers/vertexai_stream.py +592 -0
  224. aria_code/apps/cli/release_download.py +159 -0
  225. aria_code/apps/cli/review_runner.py +111 -0
  226. aria_code/apps/cli/runtime_consumer.py +769 -0
  227. aria_code/apps/cli/sandbox.py +32 -0
  228. aria_code/apps/cli/session_export.py +120 -0
  229. aria_code/apps/cli/session_jsonl.py +208 -0
  230. aria_code/apps/cli/session_refresh.py +123 -0
  231. aria_code/apps/cli/session_store.py +125 -0
  232. aria_code/apps/cli/skills_catalog.py +273 -0
  233. aria_code/apps/cli/task_isolation.py +211 -0
  234. aria_code/apps/cli/terminal_chrome.py +412 -0
  235. aria_code/apps/cli/todo_tracker.py +190 -0
  236. aria_code/apps/cli/tool_executor.py +932 -0
  237. aria_code/apps/cli/tool_registry.py +34 -0
  238. aria_code/apps/cli/tool_scope.py +113 -0
  239. aria_code/apps/cli/tools/__init__.py +40 -0
  240. aria_code/apps/cli/tools/context.py +46 -0
  241. aria_code/apps/cli/tools/file_tools.py +181 -0
  242. aria_code/apps/cli/tools/market_tools.py +626 -0
  243. aria_code/apps/cli/tools/notebook_tools.py +111 -0
  244. aria_code/apps/cli/tools/system_tools.py +656 -0
  245. aria_code/apps/cli/tools/write_tools.py +961 -0
  246. aria_code/apps/cli/tradingview_bridge.py +529 -0
  247. aria_code/apps/cli/transactions.py +269 -0
  248. aria_code/apps/cli/turn_planning.py +137 -0
  249. aria_code/apps/cli/update_check.py +257 -0
  250. aria_code/apps/cli/updater.py +373 -0
  251. aria_code/apps/cli/utils/__init__.py +0 -0
  252. aria_code/apps/cli/utils/market_detect.py +1663 -0
  253. aria_code/apps/cli/web_login.py +136 -0
  254. aria_code/apps/cli/workflow_runner.py +127 -0
  255. aria_code/apps/cli/workspace_route.py +121 -0
  256. aria_code/apps/daemon/README.md +14 -0
  257. aria_code/apps/vscode/README.md +115 -0
  258. aria_code/apps/vscode/package.json +70 -0
  259. aria_code/aria_cli.py +5244 -0
  260. aria_code/aria_daemon.py +1372 -0
  261. aria_code/aria_feishu_bot.py +16 -0
  262. aria_code/aria_mcp_server.py +61 -0
  263. aria_code/aria_relay_client.py +4 -0
  264. aria_code/aria_telegram_bot.py +11 -0
  265. aria_code/ariarc.py +385 -0
  266. aria_code/artifacts.py +581 -0
  267. aria_code/backtest_engine.py +4 -0
  268. aria_code/backtest_report.py +4 -0
  269. aria_code/brokers/__init__.py +78 -0
  270. aria_code/brokers/automation.py +193 -0
  271. aria_code/brokers/base.py +231 -0
  272. aria_code/brokers/capabilities.py +264 -0
  273. aria_code/brokers/cn/__init__.py +10 -0
  274. aria_code/brokers/cn/easytrader_broker.py +193 -0
  275. aria_code/brokers/cn/futu_broker.py +194 -0
  276. aria_code/brokers/cn/longbridge_broker.py +190 -0
  277. aria_code/brokers/cn/tiger_broker.py +196 -0
  278. aria_code/brokers/cn/xtquant_broker.py +225 -0
  279. aria_code/brokers/config.py +414 -0
  280. aria_code/brokers/intl/__init__.py +5 -0
  281. aria_code/brokers/intl/alpaca_broker.py +183 -0
  282. aria_code/brokers/intl/ibkr_broker.py +215 -0
  283. aria_code/brokers/intl/trading212_broker.py +183 -0
  284. aria_code/brokers/intl/webull_broker.py +156 -0
  285. aria_code/brokers/paper_broker.py +266 -0
  286. aria_code/brokers/planning.py +296 -0
  287. aria_code/brokers/registry.py +213 -0
  288. aria_code/brokers/trading.py +267 -0
  289. aria_code/canva_client.py +4 -0
  290. aria_code/change_store.py +127 -0
  291. aria_code/clients/__init__.py +0 -0
  292. aria_code/clients/aliyun_data_client.py +660 -0
  293. aria_code/clients/aria_relay_client.py +16 -0
  294. aria_code/clients/canva_client.py +284 -0
  295. aria_code/clients/figma_client.py +100 -0
  296. aria_code/clients/football_data_client.py +1671 -0
  297. aria_code/clients/kling_video_client.py +182 -0
  298. aria_code/clients/market_data_client.py +1985 -0
  299. aria_code/clients/openai_image_client.py +242 -0
  300. aria_code/clients/runway_video_client.py +157 -0
  301. aria_code/clients/warehouse_erp_client.py +105 -0
  302. aria_code/cloud_memory.py +190 -0
  303. aria_code/command_safety.py +19 -0
  304. aria_code/computer_use_server.py +276 -0
  305. aria_code/computer_use_tools.py +4 -0
  306. aria_code/dashboard_generator.py +579 -0
  307. aria_code/data_analysis_tools.py +4 -0
  308. aria_code/data_cleaner.py +483 -0
  309. aria_code/data_service.py +513 -0
  310. aria_code/datasources/__init__.py +23 -0
  311. aria_code/datasources/base.py +166 -0
  312. aria_code/datasources/router.py +253 -0
  313. aria_code/datasources/sec_statements.py +217 -0
  314. aria_code/datasources/sources/__init__.py +15 -0
  315. aria_code/datasources/sources/akshare_source.py +269 -0
  316. aria_code/datasources/sources/alpha_vantage_source.py +203 -0
  317. aria_code/datasources/sources/edgar_source.py +219 -0
  318. aria_code/datasources/sources/finnhub_source.py +198 -0
  319. aria_code/datasources/sources/fred_source.py +223 -0
  320. aria_code/datasources/sources/tushare_source.py +143 -0
  321. aria_code/datasources/sources/web_scraper_source.py +278 -0
  322. aria_code/datasources/sources/world_bank_source.py +206 -0
  323. aria_code/datasources/sources/yfinance_source.py +152 -0
  324. aria_code/doctor.py +860 -0
  325. aria_code/domain/__init__.py +0 -0
  326. aria_code/domain/_finance_core/__init__.py +5 -0
  327. aria_code/domain/_finance_core/backtest_engine.py +325 -0
  328. aria_code/domain/_finance_core/performance_stats.py +101 -0
  329. aria_code/domain/backtest_engine.py +75 -0
  330. aria_code/domain/backtest_report.py +488 -0
  331. aria_code/domain/finance_formulas.py +389 -0
  332. aria_code/domain/performance_stats.py +6 -0
  333. aria_code/domain/portfolio_ledger.py +307 -0
  334. aria_code/domain/strategy_vault.py +421 -0
  335. aria_code/evals/__init__.py +37 -0
  336. aria_code/evals/behavior.py +113 -0
  337. aria_code/evals/gap.py +93 -0
  338. aria_code/evals/harness.py +949 -0
  339. aria_code/evals/review_probe.py +132 -0
  340. aria_code/evals/runner.py +425 -0
  341. aria_code/evals/scorecard.py +322 -0
  342. aria_code/evals/summary.py +125 -0
  343. aria_code/external_agent_runner.py +115 -0
  344. aria_code/figma_client.py +4 -0
  345. aria_code/file_analysis_tools.py +4 -0
  346. aria_code/finance_formulas.py +4 -0
  347. aria_code/football_data_client.py +4 -0
  348. aria_code/image_gen_tools.py +4 -0
  349. aria_code/intent_classifier.py +459 -0
  350. aria_code/kling_video_client.py +4 -0
  351. aria_code/licensing.py +151 -0
  352. aria_code/local_finance_tools.py +4 -0
  353. aria_code/local_image_provider.py +4 -0
  354. aria_code/local_llm_provider.py +4 -0
  355. aria_code/macro_tools.py +4 -0
  356. aria_code/markdown_pdf.py +289 -0
  357. aria_code/market_data_client.py +4 -0
  358. aria_code/mcp_client.py +744 -0
  359. aria_code/memory_manager.py +273 -0
  360. aria_code/model_capability.py +512 -0
  361. aria_code/notification_tools.py +4 -0
  362. aria_code/openai_image_client.py +4 -0
  363. aria_code/packages/__init__.py +26 -0
  364. aria_code/packages/adk_bridge/__init__.py +11 -0
  365. aria_code/packages/adk_bridge/code_review_tools.py +49 -0
  366. aria_code/packages/adk_bridge/market_tools.py +125 -0
  367. aria_code/packages/aria_agents/__init__.py +5 -0
  368. aria_code/packages/aria_agents/manifest.py +78 -0
  369. aria_code/packages/aria_core/__init__.py +34 -0
  370. aria_code/packages/aria_core/architecture.py +269 -0
  371. aria_code/packages/aria_core/export.py +147 -0
  372. aria_code/packages/aria_core/manifest.py +65 -0
  373. aria_code/packages/aria_core/paths.py +57 -0
  374. aria_code/packages/aria_core/secure_file.py +68 -0
  375. aria_code/packages/aria_infra/__init__.py +15 -0
  376. aria_code/packages/aria_infra/arthera.py +58 -0
  377. aria_code/packages/aria_infra/doctor.py +246 -0
  378. aria_code/packages/aria_infra/product.py +40 -0
  379. aria_code/packages/aria_mcp/__init__.py +25 -0
  380. aria_code/packages/aria_mcp/bridge.py +232 -0
  381. aria_code/packages/aria_mcp/circuit.py +82 -0
  382. aria_code/packages/aria_mcp/config.py +162 -0
  383. aria_code/packages/aria_mcp/installer.py +95 -0
  384. aria_code/packages/aria_mcp/server.py +1705 -0
  385. aria_code/packages/aria_mcp/tools.py +81 -0
  386. aria_code/packages/aria_sdk/__init__.py +19 -0
  387. aria_code/packages/aria_sdk/client.py +393 -0
  388. aria_code/packages/aria_sdk/providers.py +70 -0
  389. aria_code/packages/aria_sdk/streaming.py +75 -0
  390. aria_code/packages/aria_sdk/types.py +86 -0
  391. aria_code/packages/aria_services/__init__.py +92 -0
  392. aria_code/packages/aria_services/cache.py +60 -0
  393. aria_code/packages/aria_services/context.py +356 -0
  394. aria_code/packages/aria_services/context_checkpoints.py +157 -0
  395. aria_code/packages/aria_services/data.py +11 -0
  396. aria_code/packages/aria_services/provider_health.py +304 -0
  397. aria_code/packages/aria_services/references.py +247 -0
  398. aria_code/packages/aria_services/registry.py +249 -0
  399. aria_code/packages/aria_services/research_protocol.py +137 -0
  400. aria_code/packages/aria_services/research_quality.py +186 -0
  401. aria_code/packages/aria_services/settings.py +160 -0
  402. aria_code/packages/aria_services/usage.py +151 -0
  403. aria_code/packages/aria_skills/__init__.py +46 -0
  404. aria_code/packages/aria_skills/catalog.py +149 -0
  405. aria_code/packages/aria_skills/loader.py +455 -0
  406. aria_code/packages/aria_skills/registry.py +84 -0
  407. aria_code/packages/aria_tools/__init__.py +5 -0
  408. aria_code/packages/aria_tools/financial/__init__.py +5 -0
  409. aria_code/packages/aria_tools/financial/_paths.py +92 -0
  410. aria_code/packages/aria_tools/registry.py +129 -0
  411. aria_code/packages/quant_engine/__init__.py +33 -0
  412. aria_code/packages/quant_engine/_namespace.py +24 -0
  413. aria_code/packages/quant_engine/backtest/__init__.py +10 -0
  414. aria_code/packages/quant_engine/backtest/core.py +53 -0
  415. aria_code/packages/quant_engine/backtest/engine.py +102 -0
  416. aria_code/packages/quant_engine/portfolio/__init__.py +9 -0
  417. aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
  418. aria_code/packages/quant_engine/sports/__init__.py +76 -0
  419. aria_code/packages/quant_engine/sports/calibrator.py +354 -0
  420. aria_code/packages/quant_engine/sports/dixon_coles.py +234 -0
  421. aria_code/packages/quant_engine/sports/elo.py +300 -0
  422. aria_code/packages/quant_engine/sports/form.py +188 -0
  423. aria_code/packages/quant_engine/sports/h2h.py +195 -0
  424. aria_code/packages/quant_engine/sports/predictor.py +312 -0
  425. aria_code/packages/quant_engine/sports/tracker.py +665 -0
  426. aria_code/packages/quant_engine/stochastic/__init__.py +31 -0
  427. aria_code/packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
  428. aria_code/packages/quant_engine/stochastic/ito_calculus.py +477 -0
  429. aria_code/packages/quant_engine/stochastic/kelly_criterion.py +181 -0
  430. aria_code/packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
  431. aria_code/packages/quant_engine/stochastic/options_pricing.py +576 -0
  432. aria_code/packages/quant_engine/stochastic/stochastic_processes.py +90 -0
  433. aria_code/packs/__init__.py +85 -0
  434. aria_code/packs/base.py +168 -0
  435. aria_code/packs/finance/__init__.py +250 -0
  436. aria_code/packs/healthcare/__init__.py +40 -0
  437. aria_code/packs/logistics/__init__.py +171 -0
  438. aria_code/packs/payments/__init__.py +139 -0
  439. aria_code/packs/realty/__init__.py +190 -0
  440. aria_code/packs/registry.py +202 -0
  441. aria_code/packs/rules.py +90 -0
  442. aria_code/plan_utils.py +194 -0
  443. aria_code/plugin_loader.py +328 -0
  444. aria_code/portfolio_ledger.py +4 -0
  445. aria_code/preview_server.py +449 -0
  446. aria_code/privacy/__init__.py +5 -0
  447. aria_code/privacy/feedback.py +123 -0
  448. aria_code/project_review.py +278 -0
  449. aria_code/project_review_api.py +235 -0
  450. aria_code/project_review_server.py +18 -0
  451. aria_code/project_tools.py +4 -0
  452. aria_code/providers/__init__.py +30 -0
  453. aria_code/providers/google_video_provider.py +256 -0
  454. aria_code/providers/llm/__init__.py +19 -0
  455. aria_code/providers/llm/anthropic.py +184 -0
  456. aria_code/providers/llm/arthera_gateway.py +90 -0
  457. aria_code/providers/llm/autoconfig.py +213 -0
  458. aria_code/providers/llm/base.py +139 -0
  459. aria_code/providers/llm/custom.py +158 -0
  460. aria_code/providers/llm/ollama.py +137 -0
  461. aria_code/providers/llm/openai_compat.py +350 -0
  462. aria_code/providers/llm/registry.py +457 -0
  463. aria_code/providers/local_image_provider.py +202 -0
  464. aria_code/providers/local_llm_provider.py +549 -0
  465. aria_code/providers/ollama_models.py +79 -0
  466. aria_code/providers/openai_params.py +32 -0
  467. aria_code/providers/tool_messages.py +100 -0
  468. aria_code/realty_data_tools.py +4 -0
  469. aria_code/report_exporters.py +150 -0
  470. aria_code/report_generator.py +1764 -0
  471. aria_code/review_service.py +497 -0
  472. aria_code/runtime/__init__.py +176 -0
  473. aria_code/runtime/acceptance.py +632 -0
  474. aria_code/runtime/agent_loop.py +1851 -0
  475. aria_code/runtime/agent_profiles.py +75 -0
  476. aria_code/runtime/approval.py +71 -0
  477. aria_code/runtime/baseline.py +219 -0
  478. aria_code/runtime/budget.py +232 -0
  479. aria_code/runtime/checkpoints.py +481 -0
  480. aria_code/runtime/consult.py +65 -0
  481. aria_code/runtime/contract.py +282 -0
  482. aria_code/runtime/delivery.py +432 -0
  483. aria_code/runtime/events.py +113 -0
  484. aria_code/runtime/gateway.py +167 -0
  485. aria_code/runtime/lsp.py +484 -0
  486. aria_code/runtime/native_index.py +153 -0
  487. aria_code/runtime/orchestrator.py +55 -0
  488. aria_code/runtime/processes.py +343 -0
  489. aria_code/runtime/project_graph.py +614 -0
  490. aria_code/runtime/repo_map.py +712 -0
  491. aria_code/runtime/research_bundle.py +105 -0
  492. aria_code/runtime/review.py +269 -0
  493. aria_code/runtime/run_state.py +127 -0
  494. aria_code/runtime/run_store.py +522 -0
  495. aria_code/runtime/scoped_runs.py +113 -0
  496. aria_code/runtime/self_healing.py +422 -0
  497. aria_code/runtime/semantic_diff.py +171 -0
  498. aria_code/runtime/subagent.py +721 -0
  499. aria_code/runtime/symbol_edit.py +117 -0
  500. aria_code/runtime/task_brief.py +46 -0
  501. aria_code/runtime/task_context.py +145 -0
  502. aria_code/runtime/task_ledger.py +108 -0
  503. aria_code/runtime/task_orchestrator.py +90 -0
  504. aria_code/runtime/task_outcome.py +33 -0
  505. aria_code/runtime/task_verification.py +266 -0
  506. aria_code/runtime/task_worktree.py +448 -0
  507. aria_code/runtime/tool_executor.py +329 -0
  508. aria_code/runtime/tool_policy.py +107 -0
  509. aria_code/runtime/transactions.py +216 -0
  510. aria_code/runtime/windows_job.py +115 -0
  511. aria_code/runtime/workflows.py +222 -0
  512. aria_code/runtime/worktrees.py +215 -0
  513. aria_code/runway_video_client.py +4 -0
  514. aria_code/safety/__init__.py +27 -0
  515. aria_code/safety/client_isolation.py +178 -0
  516. aria_code/safety/permissions.py +339 -0
  517. aria_code/safety/risk.py +699 -0
  518. aria_code/safety/sandbox.py +148 -0
  519. aria_code/safety/service.py +102 -0
  520. aria_code/service_endpoints.py +14 -0
  521. aria_code/setup_wizard.py +684 -0
  522. aria_code/spreadsheet_tools.py +4 -0
  523. aria_code/strategy_vault.py +4 -0
  524. aria_code/tools/__init__.py +0 -0
  525. aria_code/tools/artifact_tools.py +252 -0
  526. aria_code/tools/broker_tools.py +92 -0
  527. aria_code/tools/build_quant_engine.py +97 -0
  528. aria_code/tools/code_audit_tools.py +192 -0
  529. aria_code/tools/computer_use_tools.py +509 -0
  530. aria_code/tools/data_analysis_tools.py +831 -0
  531. aria_code/tools/enterprise_finance_tools.py +202 -0
  532. aria_code/tools/extended_tools.py +240 -0
  533. aria_code/tools/file_analysis_tools.py +880 -0
  534. aria_code/tools/image_gen_tools.py +114 -0
  535. aria_code/tools/local_finance_tools.py +3488 -0
  536. aria_code/tools/logistics_carriers.py +11 -0
  537. aria_code/tools/logistics_inventory.py +11 -0
  538. aria_code/tools/logistics_io.py +11 -0
  539. aria_code/tools/logistics_tenancy.py +11 -0
  540. aria_code/tools/logistics_tools.py +11 -0
  541. aria_code/tools/macro_tools.py +412 -0
  542. aria_code/tools/notification_tools.py +249 -0
  543. aria_code/tools/project_tools.py +525 -0
  544. aria_code/tools/realty_data_tools.py +662 -0
  545. aria_code/tools/spreadsheet_tools.py +399 -0
  546. aria_code/tools/stripe_tools.py +115 -0
  547. aria_code/tools/video_tools.py +500 -0
  548. aria_code/training/__init__.py +1 -0
  549. aria_code/training/compare.py +117 -0
  550. aria_code/training/prelude_commands.py +111 -0
  551. aria_code/training/prelude_eval.py +390 -0
  552. aria_code/training/prelude_real.py +183 -0
  553. aria_code/training/prelude_train.py +248 -0
  554. aria_code/training/sft_dataset.py +230 -0
  555. aria_code/training/trajectory_dataset.py +263 -0
  556. aria_code/ui/__init__.py +105 -0
  557. aria_code/ui/banner.py +404 -0
  558. aria_code/ui/completer.py +571 -0
  559. aria_code/ui/console.py +285 -0
  560. aria_code/ui/input_box.py +505 -0
  561. aria_code/ui/mentions.py +206 -0
  562. aria_code/ui/picker.py +248 -0
  563. aria_code/ui/render/__init__.py +11 -0
  564. aria_code/ui/render/actions.py +297 -0
  565. aria_code/ui/render/finance.py +1732 -0
  566. aria_code/ui/render/market.py +253 -0
  567. aria_code/ui/render/output.py +954 -0
  568. aria_code/ui/render/responsive.py +75 -0
  569. aria_code/ui/render/team.py +367 -0
  570. aria_code/ui/robot.py +239 -0
  571. aria_code/ui/startup_dashboard.py +110 -0
  572. aria_code/video_analysis.py +147 -0
  573. aria_code/video_editor.py +518 -0
  574. aria_code/video_workflow/__init__.py +8 -0
  575. aria_code/video_workflow/__main__.py +61 -0
  576. aria_code/video_workflow/engines/__init__.py +18 -0
  577. aria_code/video_workflow/engines/base.py +69 -0
  578. aria_code/video_workflow/engines/custom_model.py +204 -0
  579. aria_code/video_workflow/engines/google_veo.py +94 -0
  580. aria_code/video_workflow/engines/local_engine.py +71 -0
  581. aria_code/video_workflow/engines/registry.py +98 -0
  582. aria_code/video_workflow/google_models.py +278 -0
  583. aria_code/video_workflow/music/__init__.py +5 -0
  584. aria_code/video_workflow/music/foley_service.py +132 -0
  585. aria_code/video_workflow/music/service.py +212 -0
  586. aria_code/video_workflow/pipeline.py +620 -0
  587. aria_code/video_workflow/plan.py +202 -0
  588. aria_code/video_workflow/session.py +297 -0
  589. aria_code/workspace/__init__.py +6 -0
  590. aria_code/workspace/files.py +219 -0
  591. aria_code/workspace/verify.py +167 -0
  592. aria_code/workspace_changes.py +199 -0
  593. aria_code/workspace_service.py +196 -0
  594. aria_code/workspace_tools.py +150 -0
  595. aria_code-0.167.0.dist-info/METADATA +330 -0
  596. aria_code-0.167.0.dist-info/RECORD +601 -0
  597. aria_code-0.167.0.dist-info/WHEEL +5 -0
  598. aria_code-0.167.0.dist-info/entry_points.txt +5 -0
  599. aria_code-0.167.0.dist-info/licenses/LICENSE +201 -0
  600. aria_code-0.167.0.dist-info/licenses/NOTICE +49 -0
  601. aria_code-0.167.0.dist-info/top_level.txt +2 -0
@@ -0,0 +1,665 @@
1
+ """
2
+ sports/tracker.py — 预测追踪、准确率统计、自动 Elo 同步
3
+ =========================================================
4
+ 功能:
5
+ 1. 记录每次预测(赛前调用 record_prediction)
6
+ — 自动计算并存储比分概率矩阵 top_scorelines + predicted_score
7
+ 2. 赛后记录实际结果,自动计算:
8
+ — 1X2 Brier Score / Log-Loss
9
+ — 比分精确命中 score_exact / 比分排名 score_rank
10
+ — 比分 RPS(Ranked Probability Score over scoreline space)
11
+ — 进球 MAE(预期进球 vs 实际进球的平均绝对误差)
12
+ 3. 从 football-data.org API 自动同步已结束 WC 比赛 Elo
13
+ 4. 动态计算赛事实际场均进球(替换硬编码 1.32)
14
+ 5. backfill_score_metrics() — 为历史无比分指标的记录补算
15
+
16
+ 持久化路径:
17
+ ~/.arthera/football_predictions.json — 预测记录
18
+ ~/.arthera/elo_synced_matches.json — 已同步比赛 ID(防重复)
19
+ """
20
+
21
+ from __future__ import annotations
22
+
23
+ import json
24
+ import math
25
+ import time
26
+ from pathlib import Path
27
+ from typing import Dict, List, Optional, Tuple
28
+ from aria_code.packages.aria_core.paths import aria_home
29
+
30
+ _PRED_PATH = aria_home() / "football_predictions.json"
31
+ _SYNCED_PATH = aria_home() / "elo_synced_matches.json"
32
+ _AVG_PATH = aria_home() / "wc_league_avg.json"
33
+
34
+ _SCORE_MAX_G = 9 # 评估矩阵上限:0..8 进球
35
+
36
+
37
+ # ── 比分概率工具 ──────────────────────────────────────────────────────────────
38
+
39
+ def _poisson_pmf(k: int, lam: float) -> float:
40
+ if lam <= 0:
41
+ return 1.0 if k == 0 else 0.0
42
+ return (lam ** k) * math.exp(-lam) / math.factorial(k)
43
+
44
+
45
+ def _scoreline_matrix(lh: float, la: float, max_g: int = _SCORE_MAX_G) -> Dict[Tuple[int, int], float]:
46
+ """全比分概率矩阵 {(home_goals, away_goals): prob},0..max_g-1 进球。"""
47
+ matrix: Dict[Tuple[int, int], float] = {}
48
+ for h in range(max_g):
49
+ ph = _poisson_pmf(h, lh)
50
+ for a in range(max_g):
51
+ matrix[(h, a)] = ph * _poisson_pmf(a, la)
52
+ return matrix
53
+
54
+
55
+ def _top_scorelines(lh: float, la: float, n: int = 10) -> List[Dict]:
56
+ """返回概率最高的 n 个比分,含排名。"""
57
+ matrix = _scoreline_matrix(lh, la)
58
+ ranked = sorted(matrix.items(), key=lambda x: -x[1])[:n]
59
+ return [
60
+ {"score": f"{h}-{a}", "h": h, "a": a, "prob": round(p * 100, 2), "rank": i + 1}
61
+ for i, ((h, a), p) in enumerate(ranked)
62
+ ]
63
+
64
+
65
+ def _score_rps(matrix: Dict[Tuple[int, int], float], actual_h: int, actual_a: int) -> float:
66
+ """
67
+ 比分空间上的 Ranked Probability Score。
68
+
69
+ 在进球总数维度上累积 CDF 误差:
70
+ RPS = mean_over_g( (F_pred(g) - F_actual(g))^2 )
71
+ 其中 F(g) = P(total_goals ≤ g),范围 0..2*(max_g-1)。
72
+ 值域 [0, 1],越低越好;完美预测 = 0。
73
+ """
74
+ max_total = 2 * (_SCORE_MAX_G - 1)
75
+ actual_total = actual_h + actual_a
76
+
77
+ # CDF for predicted total goals
78
+ total_probs: Dict[int, float] = {}
79
+ for (h, a), p in matrix.items():
80
+ t = h + a
81
+ total_probs[t] = total_probs.get(t, 0.0) + p
82
+
83
+ cum_pred = 0.0
84
+ cum_actual = 0.0
85
+ rps = 0.0
86
+ for g in range(max_total + 1):
87
+ cum_pred += total_probs.get(g, 0.0)
88
+ cum_actual += 1.0 if g == actual_total else 0.0
89
+ rps += (cum_pred - cum_actual) ** 2
90
+
91
+ return round(rps / (max_total + 1), 5)
92
+
93
+
94
+ # ── 持久化工具 ────────────────────────────────────────────────────────────────
95
+
96
+ def _load_json(path: Path, default):
97
+ try:
98
+ if path.exists():
99
+ return json.loads(path.read_text(encoding="utf-8"))
100
+ except Exception:
101
+ pass
102
+ return default
103
+
104
+
105
+ def _save_json(path: Path, data) -> None:
106
+ try:
107
+ path.parent.mkdir(parents=True, exist_ok=True)
108
+ path.write_text(json.dumps(data, ensure_ascii=False, indent=2), encoding="utf-8")
109
+ except Exception:
110
+ pass
111
+
112
+
113
+ # ── 1. 预测记录 ───────────────────────────────────────────────────────────────
114
+
115
+ def record_prediction(
116
+ home_team: str,
117
+ away_team: str,
118
+ home_win: float,
119
+ draw: float,
120
+ away_win: float,
121
+ match_date: str = "",
122
+ competition: str = "WC",
123
+ extra: Optional[Dict] = None,
124
+ ) -> str:
125
+ """
126
+ 赛前记录一次预测,返回 prediction_id。
127
+
128
+ 用法:
129
+ pid = record_prediction("germany", "curacao", 0.714, 0.165, 0.121,
130
+ match_date="2026-06-14", competition="WC")
131
+ """
132
+ records = _load_json(_PRED_PATH, [])
133
+ pid = f"{home_team}_vs_{away_team}_{match_date}".replace(" ", "_")
134
+ entry = {
135
+ "id": pid,
136
+ "home_team": home_team,
137
+ "away_team": away_team,
138
+ "home_win": round(home_win, 4),
139
+ "draw": round(draw, 4),
140
+ "away_win": round(away_win, 4),
141
+ "match_date": match_date,
142
+ "competition": competition,
143
+ "timestamp": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
144
+ "result": None,
145
+ "brier_score": None,
146
+ "log_loss": None,
147
+ **(extra or {}),
148
+ }
149
+ # Auto-compute scoreline distribution from lambdas if available
150
+ lh = (extra or {}).get("lambda_home")
151
+ la = (extra or {}).get("lambda_away")
152
+ if lh and la and lh > 0 and la > 0:
153
+ top = _top_scorelines(float(lh), float(la), n=10)
154
+ entry["top_scorelines"] = top
155
+ entry["predicted_score"] = top[0]["score"] if top else None
156
+
157
+ records = [r for r in records if r.get("id") != pid]
158
+ records.append(entry)
159
+ _save_json(_PRED_PATH, records)
160
+ return pid
161
+
162
+
163
+ def record_result(
164
+ prediction_id: str,
165
+ actual_outcome: str, # "home" | "draw" | "away"
166
+ home_goals: Optional[int] = None,
167
+ away_goals: Optional[int] = None,
168
+ ) -> Optional[Dict]:
169
+ """
170
+ 赛后填入实际结果,自动计算 Brier Score 和 Log-Loss。
171
+
172
+ Brier Score = (p_home-1_home)² + (p_draw-1_draw)² + (p_away-1_away)²
173
+ Log-Loss = -log(p_actual)
174
+ """
175
+ records = _load_json(_PRED_PATH, [])
176
+ for r in records:
177
+ if r.get("id") == prediction_id:
178
+ oc = actual_outcome.lower()
179
+ i_home = 1.0 if oc == "home" else 0.0
180
+ i_draw = 1.0 if oc == "draw" else 0.0
181
+ i_away = 1.0 if oc == "away" else 0.0
182
+
183
+ brier = (
184
+ (r["home_win"] - i_home) ** 2
185
+ + (r["draw"] - i_draw) ** 2
186
+ + (r["away_win"] - i_away) ** 2
187
+ )
188
+ p_act = r["home_win"] * i_home + r["draw"] * i_draw + r["away_win"] * i_away
189
+ logloss = -math.log(max(p_act, 1e-7))
190
+
191
+ r["result"] = oc
192
+ r["brier_score"] = round(brier, 4)
193
+ r["log_loss"] = round(logloss, 4)
194
+ if home_goals is not None:
195
+ r["actual_home_goals"] = home_goals
196
+ if away_goals is not None:
197
+ r["actual_away_goals"] = away_goals
198
+
199
+ # ── 比分级别评估 ──────────────────────────────────────────────────
200
+ if home_goals is not None and away_goals is not None:
201
+ ah, aa = int(home_goals), int(away_goals)
202
+ actual_score_str = f"{ah}-{aa}"
203
+
204
+ # 1. 精确比分命中
205
+ r["score_exact"] = (r.get("predicted_score") == actual_score_str)
206
+
207
+ # 2. 比分排名(在存储的 top_scorelines 里的位置)
208
+ top = r.get("top_scorelines", [])
209
+ ranks = [s["rank"] for s in top if s["score"] == actual_score_str]
210
+ r["score_rank"] = ranks[0] if ranks else None
211
+
212
+ # 3. 用 lambda 重算完整矩阵 → RPS + 精确概率
213
+ lh = r.get("lambda_home")
214
+ la = r.get("lambda_away")
215
+ if lh and la and lh > 0 and la > 0:
216
+ matrix = _scoreline_matrix(float(lh), float(la))
217
+ r["score_prob"] = round(matrix.get((ah, aa), 0.0) * 100, 3)
218
+ r["score_rps"] = _score_rps(matrix, ah, aa)
219
+ r["goals_mae"] = round((abs(float(lh) - ah) + abs(float(la) - aa)) / 2, 3)
220
+ r["goals_mae_h"] = round(abs(float(lh) - ah), 3)
221
+ r["goals_mae_a"] = round(abs(float(la) - aa), 3)
222
+ # rank from full matrix (not just stored top 10)
223
+ ranked_all = sorted(matrix.items(), key=lambda x: -x[1])
224
+ for rank_idx, ((rh, ra), _) in enumerate(ranked_all):
225
+ if rh == ah and ra == aa:
226
+ r["score_rank_full"] = rank_idx + 1
227
+ break
228
+ else:
229
+ r["score_rank_full"] = len(ranked_all)
230
+
231
+ _save_json(_PRED_PATH, records)
232
+ return r
233
+ return None
234
+
235
+
236
+ def get_accuracy_stats() -> Dict:
237
+ """返回所有已结算预测的准确率统计(含比分级别指标)。"""
238
+ records = _load_json(_PRED_PATH, [])
239
+ settled = [r for r in records if r.get("result") and r.get("brier_score") is not None]
240
+ if not settled:
241
+ return {"total": 0, "message": "暂无已结算预测"}
242
+
243
+ correct = sum(
244
+ 1 for r in settled
245
+ if (r["result"] == "home" and r["home_win"] >= max(r["draw"], r["away_win"]))
246
+ or (r["result"] == "draw" and r["draw"] >= max(r["home_win"], r["away_win"]))
247
+ or (r["result"] == "away" and r["away_win"] >= max(r["home_win"], r["draw"]))
248
+ )
249
+ avg_brier = sum(r["brier_score"] for r in settled) / len(settled)
250
+ avg_logloss = sum(r["log_loss"] for r in settled) / len(settled)
251
+
252
+ stats: Dict = {
253
+ "total": len(settled),
254
+ "correct": correct,
255
+ "accuracy": round(correct / len(settled), 3),
256
+ "avg_brier_score": round(avg_brier, 4),
257
+ "avg_log_loss": round(avg_logloss, 4),
258
+ }
259
+
260
+ # ── 比分级别统计 ──────────────────────────────────────────────────────────
261
+ has_score = [r for r in settled if r.get("actual_home_goals") is not None and r.get("score_rps") is not None]
262
+ if has_score:
263
+ exact_hits = [r for r in has_score if r.get("score_exact")]
264
+ rank_vals = [r["score_rank_full"] for r in has_score if r.get("score_rank_full")]
265
+ rps_vals = [r["score_rps"] for r in has_score if r.get("score_rps") is not None]
266
+ mae_vals = [r["goals_mae"] for r in has_score if r.get("goals_mae") is not None]
267
+ mae_h_vals = [r["goals_mae_h"] for r in has_score if r.get("goals_mae_h") is not None]
268
+ mae_a_vals = [r["goals_mae_a"] for r in has_score if r.get("goals_mae_a") is not None]
269
+ prob_vals = [r["score_prob"] for r in has_score if r.get("score_prob") is not None]
270
+
271
+ stats["score"] = {
272
+ "total_with_score": len(has_score),
273
+ "exact_hits": len(exact_hits),
274
+ "exact_rate": round(len(exact_hits) / len(has_score), 3),
275
+ "avg_score_rank": round(sum(rank_vals) / len(rank_vals), 1) if rank_vals else None,
276
+ "avg_score_rps": round(sum(rps_vals) / len(rps_vals), 5) if rps_vals else None,
277
+ "avg_goals_mae": round(sum(mae_vals) / len(mae_vals), 3) if mae_vals else None,
278
+ "avg_goals_mae_home": round(sum(mae_h_vals) / len(mae_h_vals), 3) if mae_h_vals else None,
279
+ "avg_goals_mae_away": round(sum(mae_a_vals) / len(mae_a_vals), 3) if mae_a_vals else None,
280
+ "avg_score_prob_pct": round(sum(prob_vals) / len(prob_vals), 3) if prob_vals else None,
281
+ }
282
+
283
+ stats["records"] = settled
284
+ return stats
285
+
286
+
287
+ # ── 历史记录比分指标补算 ──────────────────────────────────────────────────────
288
+
289
+ def backfill_score_metrics() -> Dict:
290
+ """
291
+ 为历史已结算但缺少比分指标的记录补算:
292
+ top_scorelines, predicted_score, score_exact, score_rank,
293
+ score_rps, goals_mae, score_rank_full, score_prob
294
+
295
+ 幂等:已有完整指标的记录跳过。
296
+ 返回 {"updated": n, "skipped": n}。
297
+ """
298
+ records = _load_json(_PRED_PATH, [])
299
+ updated = 0
300
+
301
+ for r in records:
302
+ lh = r.get("lambda_home")
303
+ la = r.get("lambda_away")
304
+ if not (lh and la and lh > 0 and la > 0):
305
+ continue
306
+
307
+ # 补 top_scorelines / predicted_score
308
+ if not r.get("top_scorelines"):
309
+ top = _top_scorelines(float(lh), float(la), n=10)
310
+ r["top_scorelines"] = top
311
+ r["predicted_score"] = top[0]["score"] if top else None
312
+ updated += 1
313
+
314
+ # 补比分评估(仅对已有实际进球的记录)
315
+ ah = r.get("actual_home_goals")
316
+ aa = r.get("actual_away_goals")
317
+ if ah is None or aa is None:
318
+ continue
319
+ if r.get("score_rps") is not None:
320
+ continue # 已有,跳过
321
+
322
+ ah, aa = int(ah), int(aa)
323
+ actual_score_str = f"{ah}-{aa}"
324
+ top = r.get("top_scorelines", [])
325
+ r["score_exact"] = (r.get("predicted_score") == actual_score_str)
326
+ ranks = [s["rank"] for s in top if s["score"] == actual_score_str]
327
+ r["score_rank"] = ranks[0] if ranks else None
328
+
329
+ matrix = _scoreline_matrix(float(lh), float(la))
330
+ r["score_prob"] = round(matrix.get((ah, aa), 0.0) * 100, 3)
331
+ r["score_rps"] = _score_rps(matrix, ah, aa)
332
+ r["goals_mae"] = round((abs(float(lh) - ah) + abs(float(la) - aa)) / 2, 3)
333
+ r["goals_mae_h"] = round(abs(float(lh) - ah), 3)
334
+ r["goals_mae_a"] = round(abs(float(la) - aa), 3)
335
+
336
+ ranked_all = sorted(matrix.items(), key=lambda x: -x[1])
337
+ for rank_idx, ((rh, ra), _) in enumerate(ranked_all):
338
+ if rh == ah and ra == aa:
339
+ r["score_rank_full"] = rank_idx + 1
340
+ break
341
+ else:
342
+ r["score_rank_full"] = len(ranked_all)
343
+
344
+ updated += 1
345
+
346
+ _save_json(_PRED_PATH, records)
347
+ return {"updated": updated, "skipped": len(records) - updated}
348
+
349
+
350
+ # ── 队名规范化(结算匹配用)──────────────────────────────────────────────────
351
+ # football-data.org 的官方队名与预测记录里的简称/中文常常不一致,
352
+ # 例如 API "Cape Verde Islands" vs 预测 "cape verde",导致结算 ID 对不上,
353
+ # 模型永远学不到这场比赛。统一规范化后按「身份」匹配,而非字符串 ID。
354
+
355
+ _TEAM_ALIASES = {
356
+ "cape verde islands": "cape verde",
357
+ "korea republic": "south korea",
358
+ "korea dpr": "north korea",
359
+ "ir iran": "iran",
360
+ "iran islamic republic": "iran",
361
+ "cote d'ivoire": "ivory coast",
362
+ "côte d'ivoire": "ivory coast",
363
+ "usa": "united states",
364
+ "united states of america": "united states",
365
+ "curaçao": "curacao",
366
+ "türkiye": "turkey",
367
+ "turkiye": "turkey",
368
+ "czech republic": "czechia",
369
+ "china pr": "china",
370
+ }
371
+
372
+
373
+ def _canon(name: str) -> str:
374
+ """Canonical team key: lowercase, alias-resolved, spaces→underscore."""
375
+ s = (name or "").strip().lower().replace("_", " ")
376
+ s = " ".join(s.split())
377
+ s = _TEAM_ALIASES.get(s, s)
378
+ return s.replace(" ", "_")
379
+
380
+
381
+ def _find_pred_record(records: List[Dict], home: str, away: str, date: str) -> Optional[Dict]:
382
+ """Match a prediction record by canonical (home, away, date) identity.
383
+
384
+ Tolerant of name variants (API vs short name vs alias) so name suffixes
385
+ like 'Cape Verde Islands' settle against a 'cape verde' prediction.
386
+ """
387
+ ch, ca = _canon(home), _canon(away)
388
+ for r in records:
389
+ if r.get("match_date") != date:
390
+ continue
391
+ if _canon(r.get("home_team", "")) == ch and _canon(r.get("away_team", "")) == ca:
392
+ return r
393
+ return None
394
+
395
+
396
+ # ── 2. 自动 Elo 同步 ──────────────────────────────────────────────────────────
397
+
398
+ def sync_elo_from_wc(api_get_fn, competition_code: str = "WC") -> Dict:
399
+ """
400
+ 从 football-data.org API 同步已结束的 WC 比赛到 Elo 系统。
401
+ 防重复处理:已同步的 match_id 记录在 elo_synced_matches.json。
402
+
403
+ api_get_fn: football_data_client._get
404
+ 返回: {"synced": int, "skipped": int, "details": [...]}
405
+ """
406
+ from .elo import get_elo
407
+
408
+ synced_ids: List[int] = _load_json(_SYNCED_PATH, [])
409
+ synced_set = set(synced_ids)
410
+
411
+ data = api_get_fn(f"/competitions/{competition_code}/matches", {"status": "FINISHED"})
412
+ if not data:
413
+ return {"synced": 0, "skipped": 0, "error": "API 无响应"}
414
+
415
+ matches = data.get("matches", [])
416
+ elo = get_elo()
417
+ pred_records = _load_json(_PRED_PATH, [])
418
+ results = []
419
+ new_synced = 0
420
+ newly_settled = 0
421
+
422
+ for m in matches:
423
+ mid = m.get("id")
424
+ ht_name = m.get("homeTeam", {}).get("name", "").lower()
425
+ at_name = m.get("awayTeam", {}).get("name", "").lower()
426
+ ft = m.get("score", {}).get("fullTime", {})
427
+ hg = ft.get("home")
428
+ ag = ft.get("away")
429
+ stage = m.get("stage", "GROUP_STAGE").lower()
430
+
431
+ if hg is None or ag is None:
432
+ continue
433
+
434
+ winner = "home" if hg > ag else ("draw" if hg == ag else "away")
435
+ date_str = m.get("utcDate", "")[:10]
436
+
437
+ # ── Elo 更新:只对新比赛执行(防重复计分)──────────────────────────
438
+ if mid not in synced_set:
439
+ match_type = _stage_to_match_type(stage)
440
+ da, db = elo.update(
441
+ ht_name, at_name,
442
+ int(hg), int(ag),
443
+ match_type=match_type,
444
+ neutral_venue=True,
445
+ )
446
+ synced_ids.append(mid)
447
+ synced_set.add(mid)
448
+ new_synced += 1
449
+ results.append({
450
+ "match": f"{ht_name} {hg}-{ag} {at_name}",
451
+ "type": match_type,
452
+ "elo_chg": f"{ht_name} {da:+.1f} / {at_name} {db:+.1f}",
453
+ })
454
+
455
+ # ── 结算预测:对所有已结束比赛执行(幂等),按规范化身份匹配 ────────
456
+ # 与 Elo 同步解耦,确保名称变体(cape verde / cape verde islands)也能
457
+ # 结算,否则模型永远学不到这些「大冷门」。
458
+ rec = _find_pred_record(pred_records, ht_name, at_name, date_str)
459
+ if rec and rec.get("result") is None:
460
+ settled = record_result(rec["id"], winner, home_goals=int(hg), away_goals=int(ag))
461
+ if settled:
462
+ newly_settled += 1
463
+ try:
464
+ from .calibrator import update_team_goal_bias
465
+ if settled.get("lambda_home"):
466
+ update_team_goal_bias(ht_name, settled["lambda_home"], int(hg))
467
+ if settled.get("lambda_away"):
468
+ update_team_goal_bias(at_name, settled["lambda_away"], int(ag))
469
+ except Exception:
470
+ pass
471
+
472
+ elo.save()
473
+ _save_json(_SYNCED_PATH, synced_ids)
474
+
475
+ return {
476
+ "synced": new_synced,
477
+ "newly_settled": newly_settled,
478
+ "skipped": len(matches) - new_synced,
479
+ "details": results,
480
+ }
481
+
482
+
483
+ def auto_calibrate(api_get_fn=None) -> Dict:
484
+ """
485
+ 自动校准主函数 — 在 sync_elo_from_wc 后调用。
486
+
487
+ 触发条件(累积满足即执行):
488
+ ≥ 5 场已结算 → λ 偏差修正
489
+ ≥ 10 场已结算 → 攻防斜率网格搜索
490
+ 有 api_get_fn → ρ 重新校准
491
+
492
+ 返回校准报告 dict。
493
+ """
494
+ from .calibrator import (
495
+ optimize_lambda_bias,
496
+ optimize_slopes_from_outcomes,
497
+ optimize_confidence_temp,
498
+ save_calibration,
499
+ get_calibrated_params,
500
+ )
501
+
502
+ records = _load_json(_PRED_PATH, [])
503
+ settled = [r for r in records if r.get("result") and r.get("brier_score") is not None]
504
+ n = len(settled)
505
+ report: Dict = {"settled_count": n, "actions": []}
506
+
507
+ if n < 5:
508
+ report["status"] = "waiting"
509
+ report["message"] = f"数据不足,当前 {n} 场(需 ≥5 场触发校准)"
510
+ return report
511
+
512
+ params = get_calibrated_params()
513
+
514
+ # ── 队伍进球偏差更新(遍历所有已结算含实际进球的记录)────────────────────
515
+ try:
516
+ from .calibrator import update_team_goal_bias
517
+ for r in settled:
518
+ if r.get("actual_home_goals") is not None and r.get("lambda_home"):
519
+ update_team_goal_bias(r["home_team"], r["lambda_home"], r["actual_home_goals"])
520
+ if r.get("actual_away_goals") is not None and r.get("lambda_away"):
521
+ update_team_goal_bias(r["away_team"], r["lambda_away"], r["actual_away_goals"])
522
+ except Exception:
523
+ pass
524
+
525
+ # ── λ 偏差修正(双路径:比值法 + MAE 网格搜索,取 MAE 更优者)────────────
526
+ bias = optimize_lambda_bias(settled)
527
+ try:
528
+ from .calibrator import optimize_lambda_bias_from_scores as _score_bias_fn
529
+ score_bias = _score_bias_fn(settled)
530
+ except Exception:
531
+ score_bias = {"status": "error"}
532
+
533
+ if score_bias.get("status") == "optimized":
534
+ # 使用 MAE 网格搜索结果(更稳健,对极端比分不敏感)
535
+ params["lambda_home_bias"] = score_bias["home_bias"]
536
+ params["lambda_away_bias"] = score_bias["away_bias"]
537
+ params["n_matches"] = n
538
+ report["actions"].append(
539
+ f"λ 偏差(MAE网格): 主队×{score_bias['home_bias']:.3f} "
540
+ f"客队×{score_bias['away_bias']:.3f} "
541
+ f"goals_MAE={score_bias['goals_mae']:.3f}(n={score_bias['n']})"
542
+ )
543
+ elif bias["n_home"] >= 5 or bias["n_away"] >= 5:
544
+ params["lambda_home_bias"] = bias["home_bias"]
545
+ params["lambda_away_bias"] = bias["away_bias"]
546
+ params["n_matches"] = n
547
+ report["actions"].append(
548
+ f"λ 偏差(比值法): 主队×{bias['home_bias']:.3f}(n={bias['n_home']}) "
549
+ f"客队×{bias['away_bias']:.3f}(n={bias['n_away']})"
550
+ )
551
+
552
+ # ── 斜率网格搜索(≥10 场)────────────────────────────────────────────────
553
+ if n >= 10:
554
+ slopes = optimize_slopes_from_outcomes(settled)
555
+ if slopes.get("status") == "optimized":
556
+ prev_brier = get_calibrated_params().get("avg_brier")
557
+ new_brier = slopes["avg_brier"]
558
+ if prev_brier is None or new_brier < prev_brier - 0.002:
559
+ params.update({k: slopes[k] for k in ("a1", "a2", "d1", "d2", "avg_brier")})
560
+ report["actions"].append(
561
+ f"斜率优化: a1={slopes['a1']} a2={slopes['a2']} "
562
+ f"Brier {prev_brier}→{new_brier}"
563
+ )
564
+
565
+ # ── 概率温度校准(对抗过度自信,≥8 场)──────────────────────────────────
566
+ temp = optimize_confidence_temp(settled)
567
+ if temp.get("status") == "optimized" and temp["temp"] < 1.0:
568
+ params["conf_temp"] = temp["temp"]
569
+ report["actions"].append(
570
+ f"概率温度: ×{temp['temp']} "
571
+ f"(Brier {temp['brier_before']}→{temp['brier_after']}, n={temp['n']})"
572
+ )
573
+ else:
574
+ params.setdefault("conf_temp", 1.0)
575
+
576
+ # ── ρ 重新校准 ────────────────────────────────────────────────────────────
577
+ if api_get_fn:
578
+ rho = fetch_wc_rho(api_get_fn)
579
+ report["rho"] = rho
580
+ report["actions"].append(f"ρ 校准: {rho:.3f}")
581
+
582
+ save_calibration(params)
583
+ report["status"] = "ok"
584
+ report["message"] = f"校准完成 ({n} 场数据)"
585
+ report["params"] = {k: params[k] for k in ("a1", "a2", "lambda_home_bias", "lambda_away_bias")}
586
+ return report
587
+
588
+
589
+ def _stage_to_match_type(stage: str) -> str:
590
+ if "final" in stage and "semi" not in stage and "quarter" not in stage:
591
+ return "wc_final"
592
+ if "semi" in stage:
593
+ return "wc_semifinal"
594
+ if "quarter" in stage:
595
+ return "wc_quarterfinal"
596
+ if "round_of_16" in stage or "r16" in stage:
597
+ return "wc_r16"
598
+ return "wc_group"
599
+
600
+
601
+ # ── 3. 动态场均进球 ───────────────────────────────────────────────────────────
602
+
603
+ _CACHE_TTL = 1800 # 30 min
604
+
605
+
606
+ def fetch_wc_rho(api_get_fn, competition_code: str = "WC") -> float:
607
+ """
608
+ 从已结束 WC 比赛估计最优 ρ,保存到 ~/.arthera/wc_rho.json。
609
+ 需要 ≥20 场数据才校准,否则保持 -0.10。
610
+ """
611
+ from .dixon_coles import estimate_rho_from_results
612
+
613
+ _RHO_PATH = aria_home() / "wc_rho.json"
614
+ cached = _load_json(_RHO_PATH, {})
615
+ now = time.time()
616
+ if cached.get("ts", 0) + _CACHE_TTL > now:
617
+ return cached.get("rho", -0.10)
618
+
619
+ data = api_get_fn(f"/competitions/{competition_code}/matches", {"status": "FINISHED"})
620
+ if not data:
621
+ return cached.get("rho", -0.10)
622
+
623
+ results = []
624
+ for m in data.get("matches", []):
625
+ ft = m.get("score", {}).get("fullTime", {})
626
+ hg, ag = ft.get("home"), ft.get("away")
627
+ if hg is not None and ag is not None:
628
+ results.append((int(hg), int(ag)))
629
+
630
+ if len(results) < 20:
631
+ return -0.10
632
+
633
+ rho = estimate_rho_from_results(results)
634
+ _save_json(_RHO_PATH, {"rho": rho, "ts": now, "matches": len(results)})
635
+ return rho
636
+
637
+
638
+ def fetch_wc_league_avg(api_get_fn, competition_code: str = "WC") -> float:
639
+ """
640
+ 从 API 计算本届赛事实际场均进球,缓存 30 分钟。
641
+ 数据不足 5 场时返回默认值 1.35。
642
+ """
643
+ cached = _load_json(_AVG_PATH, {})
644
+ now = time.time()
645
+ if cached.get("ts", 0) + _CACHE_TTL > now:
646
+ return cached.get("avg", 1.35)
647
+
648
+ data = api_get_fn(f"/competitions/{competition_code}/matches", {"status": "FINISHED"})
649
+ if not data:
650
+ return cached.get("avg", 1.35)
651
+
652
+ matches = data.get("matches", [])
653
+ totals = []
654
+ for m in matches:
655
+ ft = m.get("score", {}).get("fullTime", {})
656
+ hg, ag = ft.get("home"), ft.get("away")
657
+ if hg is not None and ag is not None:
658
+ totals.append(hg + ag)
659
+
660
+ if len(totals) < 5:
661
+ return 1.35
662
+
663
+ avg = sum(totals) / (len(totals) * 2)
664
+ _save_json(_AVG_PATH, {"avg": round(avg, 3), "ts": now, "matches": len(totals)})
665
+ return round(avg, 3)