aria-code 0.52.0__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.
- aria_code/__init__.py +42 -0
- aria_code/_version.py +20 -0
- aria_code/adk_apps/__init__.py +1 -0
- aria_code/adk_apps/aria_code_review/__init__.py +1 -0
- aria_code/adk_apps/aria_code_review/agent.py +32 -0
- aria_code/adk_apps/aria_research/__init__.py +1 -0
- aria_code/adk_apps/aria_research/agent.py +31 -0
- aria_code/agents/__init__.py +34 -0
- aria_code/agents/base.py +318 -0
- aria_code/agents/code_review.py +200 -0
- aria_code/agents/deep/__init__.py +37 -0
- aria_code/agents/deep/calibration_loop.py +144 -0
- aria_code/agents/deep/critic.py +125 -0
- aria_code/agents/deep/deepen.py +193 -0
- aria_code/agents/deep/models.py +149 -0
- aria_code/agents/deep/pipeline.py +164 -0
- aria_code/agents/deep/quant_fusion.py +192 -0
- aria_code/agents/deep/themes.py +95 -0
- aria_code/agents/deep/tiers.py +106 -0
- aria_code/agents/engineering/__init__.py +14 -0
- aria_code/agents/engineering/coder.py +222 -0
- aria_code/agents/engineering/tester.py +145 -0
- aria_code/agents/financial/__init__.py +21 -0
- aria_code/agents/financial/cashflow_burn.py +119 -0
- aria_code/agents/financial/catalyst.py +279 -0
- aria_code/agents/financial/corporate_finance.py +175 -0
- aria_code/agents/financial/debate.py +224 -0
- aria_code/agents/financial/earnings.py +303 -0
- aria_code/agents/financial/fundamental.py +191 -0
- aria_code/agents/financial/macro.py +99 -0
- aria_code/agents/financial/news.py +207 -0
- aria_code/agents/financial/risk.py +165 -0
- aria_code/agents/financial/sector.py +279 -0
- aria_code/agents/financial/strategist.py +121 -0
- aria_code/agents/financial/stripe_revenue.py +127 -0
- aria_code/agents/financial/synthesis.py +274 -0
- aria_code/agents/financial/technical.py +258 -0
- aria_code/agents/orchestrator.py +44 -0
- aria_code/agents/portfolio_agent.py +363 -0
- aria_code/agents/realty/__init__.py +62 -0
- aria_code/agents/realty/asset_diagnosis.py +152 -0
- aria_code/agents/realty/business_match.py +166 -0
- aria_code/agents/realty/cashflow_verify.py +209 -0
- aria_code/agents/realty/contract_rules.py +210 -0
- aria_code/agents/realty/energy_anomaly.py +189 -0
- aria_code/agents/realty/exit_settlement.py +208 -0
- aria_code/agents/realty/fulfillment_risk.py +206 -0
- aria_code/agents/realty/ops_optimize.py +160 -0
- aria_code/agents/realty/revenue_share.py +215 -0
- aria_code/agents/registry.py +158 -0
- aria_code/agents/signal_scheme.py +111 -0
- aria_code/agents/sports/__init__.py +0 -0
- aria_code/agents/sports/football_agent.py +197 -0
- aria_code/agents/supervisor.py +113 -0
- aria_code/agents/team.py +492 -0
- aria_code/agents/warehouse/__init__.py +25 -0
- aria_code/agents/warehouse/contracts.py +26 -0
- aria_code/agents/warehouse/dashboard.py +155 -0
- aria_code/agents/warehouse/fulfillment_leadtime.py +95 -0
- aria_code/agents/warehouse/inbound_exceptions.py +25 -0
- aria_code/agents/warehouse/inventory_health.py +24 -0
- aria_code/agents/warehouse/local_run.py +71 -0
- aria_code/agents/warehouse/local_snapshot.py +132 -0
- aria_code/agents/warehouse/logistics_cost.py +84 -0
- aria_code/agents/warehouse/logistics_sync.py +21 -0
- aria_code/agents/warehouse/workflow.py +31 -0
- aria_code/aliyun_data_client.py +4 -0
- aria_code/apps/README.md +12 -0
- aria_code/apps/__init__.py +2 -0
- aria_code/apps/channels/README.md +15 -0
- aria_code/apps/channels/__init__.py +17 -0
- aria_code/apps/channels/intake.py +92 -0
- aria_code/apps/channels/registry.py +95 -0
- aria_code/apps/channels/tradingview.py +94 -0
- aria_code/apps/cli/README.md +13 -0
- aria_code/apps/cli/__init__.py +2 -0
- aria_code/apps/cli/_optional.py +120 -0
- aria_code/apps/cli/bootstrap.py +145 -0
- aria_code/apps/cli/broker_render.py +206 -0
- aria_code/apps/cli/codegen_paths.py +29 -0
- aria_code/apps/cli/commands/__init__.py +16 -0
- aria_code/apps/cli/commands/_business_render.py +78 -0
- aria_code/apps/cli/commands/_ui.py +71 -0
- aria_code/apps/cli/commands/analysis_cmds.py +369 -0
- aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
- aria_code/apps/cli/commands/auth_cmds.py +239 -0
- aria_code/apps/cli/commands/backtest_cmds.py +2543 -0
- aria_code/apps/cli/commands/broker_cmds.py +1234 -0
- aria_code/apps/cli/commands/business_workflow_cmds.py +319 -0
- aria_code/apps/cli/commands/canvas_cmds.py +77 -0
- aria_code/apps/cli/commands/catalog.py +87 -0
- aria_code/apps/cli/commands/core_cmds.py +2520 -0
- aria_code/apps/cli/commands/data_cmds.py +457 -0
- aria_code/apps/cli/commands/diagnostic_cmds.py +230 -0
- aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
- aria_code/apps/cli/commands/file_cmds.py +312 -0
- aria_code/apps/cli/commands/finance_service_cmds.py +60 -0
- aria_code/apps/cli/commands/fx_commodity_cmds.py +123 -0
- aria_code/apps/cli/commands/market.py +434 -0
- aria_code/apps/cli/commands/market_cmds.py +1381 -0
- aria_code/apps/cli/commands/market_context.py +425 -0
- aria_code/apps/cli/commands/market_render.py +7 -0
- aria_code/apps/cli/commands/model_cmds.py +2153 -0
- aria_code/apps/cli/commands/ops_cmds.py +721 -0
- aria_code/apps/cli/commands/orchestrator_cmds.py +179 -0
- aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
- aria_code/apps/cli/commands/portfolio_cmds.py +1278 -0
- aria_code/apps/cli/commands/report.py +437 -0
- aria_code/apps/cli/commands/scaffold_templates.py +617 -0
- aria_code/apps/cli/commands/session_cmds.py +236 -0
- aria_code/apps/cli/commands/session_ux_cmds.py +359 -0
- aria_code/apps/cli/commands/team.py +669 -0
- aria_code/apps/cli/commands/team_render.py +8 -0
- aria_code/apps/cli/commands/ui_cmds.py +401 -0
- aria_code/apps/cli/commands/warehouse_cmds.py +79 -0
- aria_code/apps/cli/commands/workflow_cmds.py +480 -0
- aria_code/apps/cli/commands/workspace_cmds.py +1494 -0
- aria_code/apps/cli/config_paths.py +73 -0
- aria_code/apps/cli/config_store.py +78 -0
- aria_code/apps/cli/constants.py +88 -0
- aria_code/apps/cli/context.py +34 -0
- aria_code/apps/cli/deterministic.py +191 -0
- aria_code/apps/cli/direct.py +48 -0
- aria_code/apps/cli/finance_service_catalog.py +111 -0
- aria_code/apps/cli/football_reports.py +231 -0
- aria_code/apps/cli/github_app_auth.py +135 -0
- aria_code/apps/cli/google_login.py +196 -0
- aria_code/apps/cli/handlers/__init__.py +11 -0
- aria_code/apps/cli/handlers/broker_handlers.py +122 -0
- aria_code/apps/cli/handlers/chart_handlers.py +1309 -0
- aria_code/apps/cli/handlers/market_handlers.py +2513 -0
- aria_code/apps/cli/handlers/realty_handlers.py +114 -0
- aria_code/apps/cli/handlers/sports_handlers.py +22 -0
- aria_code/apps/cli/handlers/strategy_advice.py +82 -0
- aria_code/apps/cli/helpers.py +302 -0
- aria_code/apps/cli/hooks.py +180 -0
- aria_code/apps/cli/i18n.py +297 -0
- aria_code/apps/cli/intent.py +136 -0
- aria_code/apps/cli/intent_router.py +276 -0
- aria_code/apps/cli/intent_signals.py +344 -0
- aria_code/apps/cli/lifecycle_hooks.py +48 -0
- aria_code/apps/cli/local_tool_schemas.py +536 -0
- aria_code/apps/cli/main.py +29 -0
- aria_code/apps/cli/market_metadata.py +135 -0
- aria_code/apps/cli/market_universe.py +487 -0
- aria_code/apps/cli/message_processing.py +310 -0
- aria_code/apps/cli/model_catalog.py +620 -0
- aria_code/apps/cli/pdf_report.py +977 -0
- aria_code/apps/cli/plan_mode.py +139 -0
- aria_code/apps/cli/plotly_html.py +15 -0
- aria_code/apps/cli/prediction_feedback.py +202 -0
- aria_code/apps/cli/preflight.py +502 -0
- aria_code/apps/cli/project_aria.py +60 -0
- aria_code/apps/cli/prompt_assembly.py +59 -0
- aria_code/apps/cli/prompts/__init__.py +0 -0
- aria_code/apps/cli/prompts/coding.py +53 -0
- aria_code/apps/cli/prompts/select.py +142 -0
- aria_code/apps/cli/prompts/system_prompts.py +552 -0
- aria_code/apps/cli/prompts/ui.py +436 -0
- aria_code/apps/cli/provider_endpoints.py +237 -0
- aria_code/apps/cli/provider_keys.py +73 -0
- aria_code/apps/cli/providers/__init__.py +1 -0
- aria_code/apps/cli/providers/base.py +521 -0
- aria_code/apps/cli/providers/chat_routing.py +128 -0
- aria_code/apps/cli/providers/collaboration.py +98 -0
- aria_code/apps/cli/providers/llm/__init__.py +1 -0
- aria_code/apps/cli/providers/llm/offline_responses.py +68 -0
- aria_code/apps/cli/providers/llm/ollama_stream.py +1594 -0
- aria_code/apps/cli/providers/llm/response_cache.py +60 -0
- aria_code/apps/cli/providers/llm/sse_stream.py +257 -0
- aria_code/apps/cli/providers/runtime_bridge.py +353 -0
- aria_code/apps/cli/providers/vertexai_stream.py +316 -0
- aria_code/apps/cli/runtime_consumer.py +657 -0
- aria_code/apps/cli/sandbox.py +32 -0
- aria_code/apps/cli/session_export.py +120 -0
- aria_code/apps/cli/session_jsonl.py +208 -0
- aria_code/apps/cli/session_store.py +112 -0
- aria_code/apps/cli/skills_catalog.py +273 -0
- aria_code/apps/cli/todo_tracker.py +190 -0
- aria_code/apps/cli/tool_executor.py +823 -0
- aria_code/apps/cli/tool_registry.py +34 -0
- aria_code/apps/cli/tool_scope.py +106 -0
- aria_code/apps/cli/tools/__init__.py +40 -0
- aria_code/apps/cli/tools/context.py +46 -0
- aria_code/apps/cli/tools/file_tools.py +180 -0
- aria_code/apps/cli/tools/market_tools.py +620 -0
- aria_code/apps/cli/tools/notebook_tools.py +111 -0
- aria_code/apps/cli/tools/system_tools.py +726 -0
- aria_code/apps/cli/tools/write_tools.py +845 -0
- aria_code/apps/cli/tradingview_bridge.py +529 -0
- aria_code/apps/cli/turn_planning.py +137 -0
- aria_code/apps/cli/update_check.py +153 -0
- aria_code/apps/cli/utils/__init__.py +0 -0
- aria_code/apps/cli/utils/market_detect.py +1597 -0
- aria_code/apps/daemon/README.md +14 -0
- aria_code/apps/vscode/README.md +115 -0
- aria_code/apps/vscode/package.json +70 -0
- aria_code/aria_cli.py +6743 -0
- aria_code/aria_daemon.py +1398 -0
- aria_code/aria_feishu_bot.py +1423 -0
- aria_code/aria_mcp_server.py +7 -0
- aria_code/aria_relay_client.py +4 -0
- aria_code/aria_relay_server.py +490 -0
- aria_code/aria_telegram_bot.py +202 -0
- aria_code/ariarc.py +385 -0
- aria_code/artifacts.py +581 -0
- aria_code/backtest_engine.py +4 -0
- aria_code/backtest_report.py +4 -0
- aria_code/brokers/__init__.py +78 -0
- aria_code/brokers/automation.py +193 -0
- aria_code/brokers/base.py +231 -0
- aria_code/brokers/capabilities.py +264 -0
- aria_code/brokers/cn/__init__.py +10 -0
- aria_code/brokers/cn/easytrader_broker.py +193 -0
- aria_code/brokers/cn/futu_broker.py +194 -0
- aria_code/brokers/cn/longbridge_broker.py +190 -0
- aria_code/brokers/cn/tiger_broker.py +196 -0
- aria_code/brokers/cn/xtquant_broker.py +225 -0
- aria_code/brokers/config.py +414 -0
- aria_code/brokers/intl/__init__.py +5 -0
- aria_code/brokers/intl/alpaca_broker.py +183 -0
- aria_code/brokers/intl/ibkr_broker.py +215 -0
- aria_code/brokers/intl/trading212_broker.py +183 -0
- aria_code/brokers/intl/webull_broker.py +156 -0
- aria_code/brokers/paper_broker.py +266 -0
- aria_code/brokers/planning.py +296 -0
- aria_code/brokers/registry.py +213 -0
- aria_code/brokers/trading.py +267 -0
- aria_code/canva_client.py +4 -0
- aria_code/change_store.py +127 -0
- aria_code/clients/__init__.py +0 -0
- aria_code/clients/aliyun_data_client.py +660 -0
- aria_code/clients/aria_relay_client.py +182 -0
- aria_code/clients/canva_client.py +284 -0
- aria_code/clients/figma_client.py +100 -0
- aria_code/clients/football_data_client.py +1671 -0
- aria_code/clients/kling_video_client.py +182 -0
- aria_code/clients/market_data_client.py +1976 -0
- aria_code/clients/openai_image_client.py +242 -0
- aria_code/clients/runway_video_client.py +157 -0
- aria_code/clients/warehouse_erp_client.py +105 -0
- aria_code/cloud_memory.py +181 -0
- aria_code/command_safety.py +19 -0
- aria_code/computer_use_tools.py +4 -0
- aria_code/dashboard_generator.py +579 -0
- aria_code/data_analysis_tools.py +4 -0
- aria_code/data_cleaner.py +483 -0
- aria_code/data_service.py +513 -0
- aria_code/datasources/__init__.py +23 -0
- aria_code/datasources/base.py +166 -0
- aria_code/datasources/router.py +253 -0
- aria_code/datasources/sources/__init__.py +15 -0
- aria_code/datasources/sources/akshare_source.py +269 -0
- aria_code/datasources/sources/alpha_vantage_source.py +203 -0
- aria_code/datasources/sources/edgar_source.py +219 -0
- aria_code/datasources/sources/finnhub_source.py +198 -0
- aria_code/datasources/sources/fred_source.py +223 -0
- aria_code/datasources/sources/tushare_source.py +143 -0
- aria_code/datasources/sources/web_scraper_source.py +278 -0
- aria_code/datasources/sources/world_bank_source.py +206 -0
- aria_code/datasources/sources/yfinance_source.py +152 -0
- aria_code/demo_player.py +207 -0
- aria_code/doctor.py +813 -0
- aria_code/domain/__init__.py +0 -0
- aria_code/domain/backtest_engine.py +354 -0
- aria_code/domain/backtest_report.py +472 -0
- aria_code/domain/finance_formulas.py +389 -0
- aria_code/domain/portfolio_ledger.py +307 -0
- aria_code/domain/strategy_vault.py +421 -0
- aria_code/evals/__init__.py +37 -0
- aria_code/evals/harness.py +598 -0
- aria_code/evals/runner.py +238 -0
- aria_code/external_agent_runner.py +115 -0
- aria_code/figma_client.py +4 -0
- aria_code/file_analysis_tools.py +4 -0
- aria_code/finance_formulas.py +4 -0
- aria_code/football_data_client.py +4 -0
- aria_code/image_gen_tools.py +4 -0
- aria_code/intent_classifier.py +386 -0
- aria_code/kling_video_client.py +4 -0
- aria_code/licensing.py +151 -0
- aria_code/local_finance_tools.py +4 -0
- aria_code/local_image_provider.py +4 -0
- aria_code/local_llm_provider.py +4 -0
- aria_code/macro_tools.py +4 -0
- aria_code/markdown_pdf.py +289 -0
- aria_code/market_data_client.py +4 -0
- aria_code/mcp_client.py +639 -0
- aria_code/memory_manager.py +246 -0
- aria_code/model_capability.py +486 -0
- aria_code/notification_tools.py +4 -0
- aria_code/openai_image_client.py +4 -0
- aria_code/packages/__init__.py +23 -0
- aria_code/packages/adk_bridge/__init__.py +11 -0
- aria_code/packages/adk_bridge/code_review_tools.py +49 -0
- aria_code/packages/adk_bridge/market_tools.py +125 -0
- aria_code/packages/aria_agents/__init__.py +5 -0
- aria_code/packages/aria_agents/manifest.py +78 -0
- aria_code/packages/aria_core/__init__.py +34 -0
- aria_code/packages/aria_core/architecture.py +269 -0
- aria_code/packages/aria_core/export.py +147 -0
- aria_code/packages/aria_core/manifest.py +65 -0
- aria_code/packages/aria_core/paths.py +57 -0
- aria_code/packages/aria_core/secure_file.py +68 -0
- aria_code/packages/aria_infra/__init__.py +15 -0
- aria_code/packages/aria_infra/arthera.py +52 -0
- aria_code/packages/aria_infra/doctor.py +246 -0
- aria_code/packages/aria_infra/product.py +40 -0
- aria_code/packages/aria_mcp/__init__.py +25 -0
- aria_code/packages/aria_mcp/bridge.py +201 -0
- aria_code/packages/aria_mcp/circuit.py +82 -0
- aria_code/packages/aria_mcp/config.py +111 -0
- aria_code/packages/aria_mcp/installer.py +95 -0
- aria_code/packages/aria_mcp/server.py +1550 -0
- aria_code/packages/aria_mcp/tools.py +81 -0
- aria_code/packages/aria_sdk/__init__.py +19 -0
- aria_code/packages/aria_sdk/client.py +396 -0
- aria_code/packages/aria_sdk/providers.py +70 -0
- aria_code/packages/aria_sdk/streaming.py +79 -0
- aria_code/packages/aria_sdk/types.py +86 -0
- aria_code/packages/aria_services/__init__.py +92 -0
- aria_code/packages/aria_services/cache.py +60 -0
- aria_code/packages/aria_services/context.py +356 -0
- aria_code/packages/aria_services/context_checkpoints.py +157 -0
- aria_code/packages/aria_services/data.py +11 -0
- aria_code/packages/aria_services/provider_health.py +268 -0
- aria_code/packages/aria_services/references.py +247 -0
- aria_code/packages/aria_services/registry.py +249 -0
- aria_code/packages/aria_services/research_protocol.py +57 -0
- aria_code/packages/aria_services/research_quality.py +186 -0
- aria_code/packages/aria_services/settings.py +157 -0
- aria_code/packages/aria_services/usage.py +151 -0
- aria_code/packages/aria_skills/__init__.py +46 -0
- aria_code/packages/aria_skills/catalog.py +149 -0
- aria_code/packages/aria_skills/loader.py +455 -0
- aria_code/packages/aria_skills/registry.py +84 -0
- aria_code/packages/aria_tools/__init__.py +5 -0
- aria_code/packages/aria_tools/financial/__init__.py +8 -0
- aria_code/packages/aria_tools/financial/_paths.py +130 -0
- aria_code/packages/aria_tools/financial/compliance.py +31 -0
- aria_code/packages/aria_tools/financial/factors.py +33 -0
- aria_code/packages/aria_tools/financial/risk_tools.py +33 -0
- aria_code/packages/aria_tools/financial/strategy.py +31 -0
- aria_code/packages/aria_tools/registry.py +128 -0
- aria_code/packages/quant_engine/__init__.py +34 -0
- aria_code/packages/quant_engine/_namespace.py +62 -0
- aria_code/packages/quant_engine/backtest/__init__.py +14 -0
- aria_code/packages/quant_engine/backtest/core.py +53 -0
- aria_code/packages/quant_engine/backtest/engine.py +96 -0
- aria_code/packages/quant_engine/portfolio/__init__.py +13 -0
- aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
- aria_code/packages/quant_engine/sports/__init__.py +80 -0
- aria_code/packages/quant_engine/sports/calibrator.py +354 -0
- aria_code/packages/quant_engine/sports/dixon_coles.py +234 -0
- aria_code/packages/quant_engine/sports/elo.py +300 -0
- aria_code/packages/quant_engine/sports/form.py +188 -0
- aria_code/packages/quant_engine/sports/h2h.py +195 -0
- aria_code/packages/quant_engine/sports/ml_model.py +377 -0
- aria_code/packages/quant_engine/sports/predictor.py +312 -0
- aria_code/packages/quant_engine/sports/tracker.py +665 -0
- aria_code/packages/quant_engine/stochastic/__init__.py +35 -0
- aria_code/packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
- aria_code/packages/quant_engine/stochastic/ito_calculus.py +477 -0
- aria_code/packages/quant_engine/stochastic/kelly_criterion.py +181 -0
- aria_code/packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
- aria_code/packages/quant_engine/stochastic/options_pricing.py +576 -0
- aria_code/packages/quant_engine/stochastic/stochastic_processes.py +90 -0
- aria_code/packs/__init__.py +85 -0
- aria_code/packs/base.py +168 -0
- aria_code/packs/finance/__init__.py +250 -0
- aria_code/packs/logistics/__init__.py +169 -0
- aria_code/packs/payments/__init__.py +139 -0
- aria_code/packs/realty/__init__.py +190 -0
- aria_code/packs/registry.py +202 -0
- aria_code/packs/rules.py +90 -0
- aria_code/plan_utils.py +194 -0
- aria_code/plugin_loader.py +328 -0
- aria_code/portfolio_ledger.py +4 -0
- aria_code/preview_server.py +449 -0
- aria_code/privacy/__init__.py +5 -0
- aria_code/privacy/feedback.py +123 -0
- aria_code/project_review.py +278 -0
- aria_code/project_review_api.py +235 -0
- aria_code/project_review_server.py +18 -0
- aria_code/project_tools.py +4 -0
- aria_code/providers/__init__.py +30 -0
- aria_code/providers/llm/__init__.py +19 -0
- aria_code/providers/llm/anthropic.py +184 -0
- aria_code/providers/llm/autoconfig.py +213 -0
- aria_code/providers/llm/base.py +139 -0
- aria_code/providers/llm/custom.py +158 -0
- aria_code/providers/llm/ollama.py +129 -0
- aria_code/providers/llm/openai_compat.py +313 -0
- aria_code/providers/llm/registry.py +444 -0
- aria_code/providers/local_image_provider.py +202 -0
- aria_code/providers/local_llm_provider.py +589 -0
- aria_code/realty_data_tools.py +4 -0
- aria_code/report_exporters.py +150 -0
- aria_code/report_generator.py +1764 -0
- aria_code/runtime/__init__.py +169 -0
- aria_code/runtime/acceptance.py +481 -0
- aria_code/runtime/agent_loop.py +1520 -0
- aria_code/runtime/approval.py +57 -0
- aria_code/runtime/budget.py +232 -0
- aria_code/runtime/checkpoints.py +432 -0
- aria_code/runtime/crash_report.py +174 -0
- aria_code/runtime/events.py +113 -0
- aria_code/runtime/gateway.py +150 -0
- aria_code/runtime/lsp.py +484 -0
- aria_code/runtime/orchestrator.py +55 -0
- aria_code/runtime/repo_map.py +700 -0
- aria_code/runtime/run_state.py +127 -0
- aria_code/runtime/run_store.py +450 -0
- aria_code/runtime/self_healing.py +422 -0
- aria_code/runtime/subagent.py +547 -0
- aria_code/runtime/task_ledger.py +86 -0
- aria_code/runtime/task_orchestrator.py +90 -0
- aria_code/runtime/tool_executor.py +217 -0
- aria_code/runtime/tool_policy.py +107 -0
- aria_code/runtime/worktrees.py +215 -0
- aria_code/runway_video_client.py +4 -0
- aria_code/safety/__init__.py +23 -0
- aria_code/safety/permissions.py +306 -0
- aria_code/safety/service.py +102 -0
- aria_code/setup_wizard.py +673 -0
- aria_code/spreadsheet_tools.py +4 -0
- aria_code/strategy_vault.py +4 -0
- aria_code/tools/__init__.py +0 -0
- aria_code/tools/artifact_tools.py +252 -0
- aria_code/tools/broker_tools.py +92 -0
- aria_code/tools/build_quant_engine.py +97 -0
- aria_code/tools/code_audit_tools.py +192 -0
- aria_code/tools/computer_use_tools.py +504 -0
- aria_code/tools/data_analysis_tools.py +809 -0
- aria_code/tools/enterprise_finance_tools.py +156 -0
- aria_code/tools/extended_tools.py +240 -0
- aria_code/tools/file_analysis_tools.py +885 -0
- aria_code/tools/image_gen_tools.py +114 -0
- aria_code/tools/local_finance_tools.py +3417 -0
- aria_code/tools/logistics_tools.py +135 -0
- aria_code/tools/macro_tools.py +369 -0
- aria_code/tools/notification_tools.py +249 -0
- aria_code/tools/project_tools.py +525 -0
- aria_code/tools/realty_data_tools.py +659 -0
- aria_code/tools/spreadsheet_tools.py +388 -0
- aria_code/tools/stripe_tools.py +115 -0
- aria_code/ui/__init__.py +105 -0
- aria_code/ui/banner.py +508 -0
- aria_code/ui/completer.py +501 -0
- aria_code/ui/console.py +281 -0
- aria_code/ui/image_render.py +243 -0
- aria_code/ui/input_box.py +488 -0
- aria_code/ui/picker.py +201 -0
- aria_code/ui/render/__init__.py +11 -0
- aria_code/ui/render/finance.py +1705 -0
- aria_code/ui/render/market.py +225 -0
- aria_code/ui/render/output.py +775 -0
- aria_code/ui/render/responsive.py +75 -0
- aria_code/ui/render/team.py +354 -0
- aria_code/ui/robot.py +235 -0
- aria_code/ui/startup_dashboard.py +109 -0
- aria_code/utils/vscode_uri.py +37 -0
- aria_code/video_analysis.py +147 -0
- aria_code/video_editor.py +271 -0
- aria_code/workspace/__init__.py +6 -0
- aria_code/workspace/files.py +193 -0
- aria_code/workspace/verify.py +113 -0
- aria_code/workspace_changes.py +199 -0
- aria_code/workspace_context.py +233 -0
- aria_code/workspace_service.py +196 -0
- aria_code/workspace_tools.py +150 -0
- aria_code-0.52.0.dist-info/METADATA +237 -0
- aria_code-0.52.0.dist-info/RECORD +478 -0
- aria_code-0.52.0.dist-info/WHEEL +5 -0
- aria_code-0.52.0.dist-info/entry_points.txt +3 -0
- aria_code-0.52.0.dist-info/licenses/LICENSE +201 -0
- aria_code-0.52.0.dist-info/licenses/NOTICE +49 -0
- aria_code-0.52.0.dist-info/top_level.txt +1 -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)
|