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.
- aria_bots/__init__.py +7 -0
- aria_bots/channels/README.md +97 -0
- aria_bots/channels/__init__.py +1 -0
- aria_bots/channels/approvals.py +263 -0
- aria_bots/channels/conversation.py +280 -0
- aria_bots/channels/digest.py +186 -0
- aria_bots/channels/registry.py +95 -0
- aria_bots/engine.py +144 -0
- aria_bots/feishu_bot.py +1385 -0
- aria_bots/logistics/__init__.py +1 -0
- aria_bots/logistics/carriers.py +253 -0
- aria_bots/logistics/freight.py +154 -0
- aria_bots/logistics/inventory.py +242 -0
- aria_bots/logistics/mcp_server.py +156 -0
- aria_bots/logistics/records.py +119 -0
- aria_bots/logistics/tenancy.py +152 -0
- aria_bots/market.py +91 -0
- aria_bots/relay_client.py +326 -0
- aria_bots/telegram_bot.py +299 -0
- aria_code/__init__.py +111 -0
- aria_code/_bin/aria-native.exe +0 -0
- aria_code/_bin/aria-runtime.exe +0 -0
- aria_code/_version.py +20 -0
- aria_code/account_memory.py +296 -0
- aria_code/account_session.py +166 -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 +49 -0
- aria_code/agents/base.py +329 -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 +315 -0
- aria_code/agents/financial/fundamental.py +178 -0
- aria_code/agents/financial/macro.py +111 -0
- aria_code/agents/financial/news.py +231 -0
- aria_code/agents/financial/risk.py +146 -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 +277 -0
- aria_code/agents/orchestrator.py +44 -0
- aria_code/agents/portfolio_agent.py +392 -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 +168 -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 +500 -0
- aria_code/agents/video/__init__.py +16 -0
- aria_code/agents/video/director.py +376 -0
- aria_code/agents/video/qa_agent.py +143 -0
- aria_code/agents/video/script_doctor.py +112 -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/navigation.schema.json +110 -0
- aria_code/agents/warehouse/task_agent.py +172 -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 +9 -0
- aria_code/apps/channels/__init__.py +17 -0
- aria_code/apps/channels/approvals.py +11 -0
- aria_code/apps/channels/conversation.py +11 -0
- aria_code/apps/channels/digest.py +11 -0
- aria_code/apps/channels/intake.py +92 -0
- aria_code/apps/channels/registry.py +11 -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/app_server.py +656 -0
- aria_code/apps/cli/auto_update.py +125 -0
- aria_code/apps/cli/bootstrap.py +206 -0
- aria_code/apps/cli/broker_render.py +206 -0
- aria_code/apps/cli/chat_turn.py +1327 -0
- aria_code/apps/cli/cloud_models.py +231 -0
- aria_code/apps/cli/codegen_paths.py +29 -0
- aria_code/apps/cli/commands/__init__.py +34 -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 +378 -0
- aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
- aria_code/apps/cli/commands/auth_cmds.py +215 -0
- aria_code/apps/cli/commands/backtest_cmds.py +2518 -0
- aria_code/apps/cli/commands/broker_cmds.py +1236 -0
- aria_code/apps/cli/commands/business_workflow_cmds.py +379 -0
- aria_code/apps/cli/commands/canvas_cmds.py +77 -0
- aria_code/apps/cli/commands/catalog.py +189 -0
- aria_code/apps/cli/commands/core_cmds.py +2748 -0
- aria_code/apps/cli/commands/data_cmds.py +521 -0
- aria_code/apps/cli/commands/diagnostic_cmds.py +285 -0
- aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
- aria_code/apps/cli/commands/engineering_cmds.py +399 -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 +544 -0
- aria_code/apps/cli/commands/market_cmds.py +1419 -0
- aria_code/apps/cli/commands/market_context.py +432 -0
- aria_code/apps/cli/commands/market_render.py +7 -0
- aria_code/apps/cli/commands/model_cmds.py +2374 -0
- aria_code/apps/cli/commands/ops_cmds.py +721 -0
- aria_code/apps/cli/commands/orchestrator_cmds.py +181 -0
- aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
- aria_code/apps/cli/commands/portfolio_cmds.py +1316 -0
- aria_code/apps/cli/commands/report.py +476 -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 +723 -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 +297 -0
- aria_code/apps/cli/commands/workflow_cmds.py +629 -0
- aria_code/apps/cli/commands/workspace_cmds.py +1543 -0
- aria_code/apps/cli/config_paths.py +73 -0
- aria_code/apps/cli/config_store.py +158 -0
- aria_code/apps/cli/constants.py +88 -0
- aria_code/apps/cli/context.py +34 -0
- aria_code/apps/cli/conversation_policy.py +53 -0
- aria_code/apps/cli/deterministic.py +200 -0
- aria_code/apps/cli/direct.py +48 -0
- aria_code/apps/cli/earnings_view.py +172 -0
- aria_code/apps/cli/endpoint_migration.py +64 -0
- aria_code/apps/cli/exec_events.py +255 -0
- aria_code/apps/cli/file_references.py +155 -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 +239 -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 +1314 -0
- aria_code/apps/cli/handlers/market_handlers.py +2570 -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/headless.py +352 -0
- aria_code/apps/cli/helpers.py +326 -0
- aria_code/apps/cli/hooks.py +180 -0
- aria_code/apps/cli/i18n.py +310 -0
- aria_code/apps/cli/intent.py +136 -0
- aria_code/apps/cli/intent_router.py +305 -0
- aria_code/apps/cli/intent_shadow.py +72 -0
- aria_code/apps/cli/intent_signals.py +347 -0
- aria_code/apps/cli/lifecycle_hooks.py +48 -0
- aria_code/apps/cli/local_tool_schemas.py +550 -0
- aria_code/apps/cli/main.py +39 -0
- aria_code/apps/cli/market_hours.py +100 -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/mixin_binding.py +32 -0
- aria_code/apps/cli/model_catalog.py +652 -0
- aria_code/apps/cli/model_probe.py +169 -0
- aria_code/apps/cli/model_retirement.py +65 -0
- aria_code/apps/cli/native_branding.py +12 -0
- aria_code/apps/cli/native_bridge.py +118 -0
- aria_code/apps/cli/native_entry.py +58 -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 +78 -0
- aria_code/apps/cli/prompts/__init__.py +0 -0
- aria_code/apps/cli/prompts/coding.py +71 -0
- aria_code/apps/cli/prompts/select.py +193 -0
- aria_code/apps/cli/prompts/system_prompts.py +562 -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 +775 -0
- aria_code/apps/cli/providers/chat_routing.py +139 -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 +1574 -0
- aria_code/apps/cli/providers/llm/response_cache.py +60 -0
- aria_code/apps/cli/providers/llm/sse_stream.py +298 -0
- aria_code/apps/cli/providers/runtime_bridge.py +586 -0
- aria_code/apps/cli/providers/vertexai_stream.py +592 -0
- aria_code/apps/cli/release_download.py +159 -0
- aria_code/apps/cli/review_runner.py +111 -0
- aria_code/apps/cli/runtime_consumer.py +769 -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_refresh.py +123 -0
- aria_code/apps/cli/session_store.py +125 -0
- aria_code/apps/cli/skills_catalog.py +273 -0
- aria_code/apps/cli/task_isolation.py +211 -0
- aria_code/apps/cli/terminal_chrome.py +412 -0
- aria_code/apps/cli/todo_tracker.py +190 -0
- aria_code/apps/cli/tool_executor.py +932 -0
- aria_code/apps/cli/tool_registry.py +34 -0
- aria_code/apps/cli/tool_scope.py +113 -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 +181 -0
- aria_code/apps/cli/tools/market_tools.py +626 -0
- aria_code/apps/cli/tools/notebook_tools.py +111 -0
- aria_code/apps/cli/tools/system_tools.py +656 -0
- aria_code/apps/cli/tools/write_tools.py +961 -0
- aria_code/apps/cli/tradingview_bridge.py +529 -0
- aria_code/apps/cli/transactions.py +269 -0
- aria_code/apps/cli/turn_planning.py +137 -0
- aria_code/apps/cli/update_check.py +257 -0
- aria_code/apps/cli/updater.py +373 -0
- aria_code/apps/cli/utils/__init__.py +0 -0
- aria_code/apps/cli/utils/market_detect.py +1663 -0
- aria_code/apps/cli/web_login.py +136 -0
- aria_code/apps/cli/workflow_runner.py +127 -0
- aria_code/apps/cli/workspace_route.py +121 -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 +5244 -0
- aria_code/aria_daemon.py +1372 -0
- aria_code/aria_feishu_bot.py +16 -0
- aria_code/aria_mcp_server.py +61 -0
- aria_code/aria_relay_client.py +4 -0
- aria_code/aria_telegram_bot.py +11 -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 +16 -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 +1985 -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 +190 -0
- aria_code/command_safety.py +19 -0
- aria_code/computer_use_server.py +276 -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/sec_statements.py +217 -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/doctor.py +860 -0
- aria_code/domain/__init__.py +0 -0
- aria_code/domain/_finance_core/__init__.py +5 -0
- aria_code/domain/_finance_core/backtest_engine.py +325 -0
- aria_code/domain/_finance_core/performance_stats.py +101 -0
- aria_code/domain/backtest_engine.py +75 -0
- aria_code/domain/backtest_report.py +488 -0
- aria_code/domain/finance_formulas.py +389 -0
- aria_code/domain/performance_stats.py +6 -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/behavior.py +113 -0
- aria_code/evals/gap.py +93 -0
- aria_code/evals/harness.py +949 -0
- aria_code/evals/review_probe.py +132 -0
- aria_code/evals/runner.py +425 -0
- aria_code/evals/scorecard.py +322 -0
- aria_code/evals/summary.py +125 -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 +459 -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 +744 -0
- aria_code/memory_manager.py +273 -0
- aria_code/model_capability.py +512 -0
- aria_code/notification_tools.py +4 -0
- aria_code/openai_image_client.py +4 -0
- aria_code/packages/__init__.py +26 -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 +58 -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 +232 -0
- aria_code/packages/aria_mcp/circuit.py +82 -0
- aria_code/packages/aria_mcp/config.py +162 -0
- aria_code/packages/aria_mcp/installer.py +95 -0
- aria_code/packages/aria_mcp/server.py +1705 -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 +393 -0
- aria_code/packages/aria_sdk/providers.py +70 -0
- aria_code/packages/aria_sdk/streaming.py +75 -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 +304 -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 +137 -0
- aria_code/packages/aria_services/research_quality.py +186 -0
- aria_code/packages/aria_services/settings.py +160 -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 +5 -0
- aria_code/packages/aria_tools/financial/_paths.py +92 -0
- aria_code/packages/aria_tools/registry.py +129 -0
- aria_code/packages/quant_engine/__init__.py +33 -0
- aria_code/packages/quant_engine/_namespace.py +24 -0
- aria_code/packages/quant_engine/backtest/__init__.py +10 -0
- aria_code/packages/quant_engine/backtest/core.py +53 -0
- aria_code/packages/quant_engine/backtest/engine.py +102 -0
- aria_code/packages/quant_engine/portfolio/__init__.py +9 -0
- aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
- aria_code/packages/quant_engine/sports/__init__.py +76 -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/predictor.py +312 -0
- aria_code/packages/quant_engine/sports/tracker.py +665 -0
- aria_code/packages/quant_engine/stochastic/__init__.py +31 -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/healthcare/__init__.py +40 -0
- aria_code/packs/logistics/__init__.py +171 -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/google_video_provider.py +256 -0
- aria_code/providers/llm/__init__.py +19 -0
- aria_code/providers/llm/anthropic.py +184 -0
- aria_code/providers/llm/arthera_gateway.py +90 -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 +137 -0
- aria_code/providers/llm/openai_compat.py +350 -0
- aria_code/providers/llm/registry.py +457 -0
- aria_code/providers/local_image_provider.py +202 -0
- aria_code/providers/local_llm_provider.py +549 -0
- aria_code/providers/ollama_models.py +79 -0
- aria_code/providers/openai_params.py +32 -0
- aria_code/providers/tool_messages.py +100 -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/review_service.py +497 -0
- aria_code/runtime/__init__.py +176 -0
- aria_code/runtime/acceptance.py +632 -0
- aria_code/runtime/agent_loop.py +1851 -0
- aria_code/runtime/agent_profiles.py +75 -0
- aria_code/runtime/approval.py +71 -0
- aria_code/runtime/baseline.py +219 -0
- aria_code/runtime/budget.py +232 -0
- aria_code/runtime/checkpoints.py +481 -0
- aria_code/runtime/consult.py +65 -0
- aria_code/runtime/contract.py +282 -0
- aria_code/runtime/delivery.py +432 -0
- aria_code/runtime/events.py +113 -0
- aria_code/runtime/gateway.py +167 -0
- aria_code/runtime/lsp.py +484 -0
- aria_code/runtime/native_index.py +153 -0
- aria_code/runtime/orchestrator.py +55 -0
- aria_code/runtime/processes.py +343 -0
- aria_code/runtime/project_graph.py +614 -0
- aria_code/runtime/repo_map.py +712 -0
- aria_code/runtime/research_bundle.py +105 -0
- aria_code/runtime/review.py +269 -0
- aria_code/runtime/run_state.py +127 -0
- aria_code/runtime/run_store.py +522 -0
- aria_code/runtime/scoped_runs.py +113 -0
- aria_code/runtime/self_healing.py +422 -0
- aria_code/runtime/semantic_diff.py +171 -0
- aria_code/runtime/subagent.py +721 -0
- aria_code/runtime/symbol_edit.py +117 -0
- aria_code/runtime/task_brief.py +46 -0
- aria_code/runtime/task_context.py +145 -0
- aria_code/runtime/task_ledger.py +108 -0
- aria_code/runtime/task_orchestrator.py +90 -0
- aria_code/runtime/task_outcome.py +33 -0
- aria_code/runtime/task_verification.py +266 -0
- aria_code/runtime/task_worktree.py +448 -0
- aria_code/runtime/tool_executor.py +329 -0
- aria_code/runtime/tool_policy.py +107 -0
- aria_code/runtime/transactions.py +216 -0
- aria_code/runtime/windows_job.py +115 -0
- aria_code/runtime/workflows.py +222 -0
- aria_code/runtime/worktrees.py +215 -0
- aria_code/runway_video_client.py +4 -0
- aria_code/safety/__init__.py +27 -0
- aria_code/safety/client_isolation.py +178 -0
- aria_code/safety/permissions.py +339 -0
- aria_code/safety/risk.py +699 -0
- aria_code/safety/sandbox.py +148 -0
- aria_code/safety/service.py +102 -0
- aria_code/service_endpoints.py +14 -0
- aria_code/setup_wizard.py +684 -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 +509 -0
- aria_code/tools/data_analysis_tools.py +831 -0
- aria_code/tools/enterprise_finance_tools.py +202 -0
- aria_code/tools/extended_tools.py +240 -0
- aria_code/tools/file_analysis_tools.py +880 -0
- aria_code/tools/image_gen_tools.py +114 -0
- aria_code/tools/local_finance_tools.py +3488 -0
- aria_code/tools/logistics_carriers.py +11 -0
- aria_code/tools/logistics_inventory.py +11 -0
- aria_code/tools/logistics_io.py +11 -0
- aria_code/tools/logistics_tenancy.py +11 -0
- aria_code/tools/logistics_tools.py +11 -0
- aria_code/tools/macro_tools.py +412 -0
- aria_code/tools/notification_tools.py +249 -0
- aria_code/tools/project_tools.py +525 -0
- aria_code/tools/realty_data_tools.py +662 -0
- aria_code/tools/spreadsheet_tools.py +399 -0
- aria_code/tools/stripe_tools.py +115 -0
- aria_code/tools/video_tools.py +500 -0
- aria_code/training/__init__.py +1 -0
- aria_code/training/compare.py +117 -0
- aria_code/training/prelude_commands.py +111 -0
- aria_code/training/prelude_eval.py +390 -0
- aria_code/training/prelude_real.py +183 -0
- aria_code/training/prelude_train.py +248 -0
- aria_code/training/sft_dataset.py +230 -0
- aria_code/training/trajectory_dataset.py +263 -0
- aria_code/ui/__init__.py +105 -0
- aria_code/ui/banner.py +404 -0
- aria_code/ui/completer.py +571 -0
- aria_code/ui/console.py +285 -0
- aria_code/ui/input_box.py +505 -0
- aria_code/ui/mentions.py +206 -0
- aria_code/ui/picker.py +248 -0
- aria_code/ui/render/__init__.py +11 -0
- aria_code/ui/render/actions.py +297 -0
- aria_code/ui/render/finance.py +1732 -0
- aria_code/ui/render/market.py +253 -0
- aria_code/ui/render/output.py +954 -0
- aria_code/ui/render/responsive.py +75 -0
- aria_code/ui/render/team.py +367 -0
- aria_code/ui/robot.py +239 -0
- aria_code/ui/startup_dashboard.py +110 -0
- aria_code/video_analysis.py +147 -0
- aria_code/video_editor.py +518 -0
- aria_code/video_workflow/__init__.py +8 -0
- aria_code/video_workflow/__main__.py +61 -0
- aria_code/video_workflow/engines/__init__.py +18 -0
- aria_code/video_workflow/engines/base.py +69 -0
- aria_code/video_workflow/engines/custom_model.py +204 -0
- aria_code/video_workflow/engines/google_veo.py +94 -0
- aria_code/video_workflow/engines/local_engine.py +71 -0
- aria_code/video_workflow/engines/registry.py +98 -0
- aria_code/video_workflow/google_models.py +278 -0
- aria_code/video_workflow/music/__init__.py +5 -0
- aria_code/video_workflow/music/foley_service.py +132 -0
- aria_code/video_workflow/music/service.py +212 -0
- aria_code/video_workflow/pipeline.py +620 -0
- aria_code/video_workflow/plan.py +202 -0
- aria_code/video_workflow/session.py +297 -0
- aria_code/workspace/__init__.py +6 -0
- aria_code/workspace/files.py +219 -0
- aria_code/workspace/verify.py +167 -0
- aria_code/workspace_changes.py +199 -0
- aria_code/workspace_service.py +196 -0
- aria_code/workspace_tools.py +150 -0
- aria_code-0.167.0.dist-info/METADATA +330 -0
- aria_code-0.167.0.dist-info/RECORD +601 -0
- aria_code-0.167.0.dist-info/WHEEL +5 -0
- aria_code-0.167.0.dist-info/entry_points.txt +5 -0
- aria_code-0.167.0.dist-info/licenses/LICENSE +201 -0
- aria_code-0.167.0.dist-info/licenses/NOTICE +49 -0
- aria_code-0.167.0.dist-info/top_level.txt +2 -0
|
@@ -0,0 +1,166 @@
|
|
|
1
|
+
"""
|
|
2
|
+
datasources/base.py — 数据源统一接口
|
|
3
|
+
=====================================
|
|
4
|
+
所有数据源实现 BaseDataSource,输出统一 schema,
|
|
5
|
+
上层代码不关心具体是 akshare / yfinance / tushare。
|
|
6
|
+
"""
|
|
7
|
+
|
|
8
|
+
from __future__ import annotations
|
|
9
|
+
|
|
10
|
+
from abc import ABC, abstractmethod
|
|
11
|
+
from dataclasses import dataclass, field
|
|
12
|
+
from typing import Any, Dict, List, Optional
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
# ── 统一输出 schema ────────────────────────────────────────────────────────────
|
|
16
|
+
|
|
17
|
+
@dataclass
|
|
18
|
+
class QuoteResult:
|
|
19
|
+
symbol: str
|
|
20
|
+
name: str = ""
|
|
21
|
+
price: float = 0.0
|
|
22
|
+
change: float = 0.0
|
|
23
|
+
change_pct: float = 0.0
|
|
24
|
+
volume: float = 0.0
|
|
25
|
+
market_cap: float = 0.0
|
|
26
|
+
pe_ttm: float = 0.0
|
|
27
|
+
pb: float = 0.0
|
|
28
|
+
high_52w: float = 0.0
|
|
29
|
+
low_52w: float = 0.0
|
|
30
|
+
currency: str = "CNY"
|
|
31
|
+
market: str = "" # "a_share" | "us" | "hk" | "crypto"
|
|
32
|
+
source: str = "" # 实际使用的数据源名称
|
|
33
|
+
timestamp: str = ""
|
|
34
|
+
extra: Dict[str, Any] = field(default_factory=dict)
|
|
35
|
+
|
|
36
|
+
def to_dict(self) -> Dict:
|
|
37
|
+
return {
|
|
38
|
+
"symbol": self.symbol, "name": self.name,
|
|
39
|
+
"price": self.price, "change": self.change,
|
|
40
|
+
"change_pct": self.change_pct, "volume": self.volume,
|
|
41
|
+
"market_cap": self.market_cap, "pe_ttm": self.pe_ttm,
|
|
42
|
+
"pb": self.pb, "high_52w": self.high_52w, "low_52w": self.low_52w,
|
|
43
|
+
"currency": self.currency, "market": self.market,
|
|
44
|
+
"source": self.source, "timestamp": self.timestamp,
|
|
45
|
+
**self.extra,
|
|
46
|
+
}
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
@dataclass
|
|
50
|
+
class HistoryResult:
|
|
51
|
+
symbol: str
|
|
52
|
+
data: Any = None # pandas DataFrame
|
|
53
|
+
source: str = ""
|
|
54
|
+
interval: str = "1d"
|
|
55
|
+
|
|
56
|
+
|
|
57
|
+
@dataclass
|
|
58
|
+
class FundamentalsResult:
|
|
59
|
+
symbol: str
|
|
60
|
+
pe_ttm: Optional[float] = None
|
|
61
|
+
pb: Optional[float] = None
|
|
62
|
+
roe: Optional[float] = None
|
|
63
|
+
revenue_growth: Optional[float] = None
|
|
64
|
+
net_profit_growth: Optional[float] = None
|
|
65
|
+
dividend_yield: Optional[float] = None
|
|
66
|
+
total_mv: Optional[float] = None
|
|
67
|
+
source: str = ""
|
|
68
|
+
|
|
69
|
+
|
|
70
|
+
# ── 基类 ──────────────────────────────────────────────────────────────────────
|
|
71
|
+
|
|
72
|
+
class BaseDataSource(ABC):
|
|
73
|
+
"""
|
|
74
|
+
所有数据源的抽象基类。
|
|
75
|
+
|
|
76
|
+
子类实现 `supports()` 判断是否支持该 symbol,
|
|
77
|
+
再实现具体的 `quote()` / `history()` / `fundamentals()`。
|
|
78
|
+
"""
|
|
79
|
+
|
|
80
|
+
name: str = "base" # 数据源唯一标识
|
|
81
|
+
markets: List[str] = [] # 支持的市场: "a_share", "us", "hk", "crypto"
|
|
82
|
+
requires_key: bool = False # 是否需要 API key
|
|
83
|
+
|
|
84
|
+
def __init__(self, config: Dict[str, Any] = None):
|
|
85
|
+
self.config = config or {}
|
|
86
|
+
self._available: Optional[bool] = None
|
|
87
|
+
|
|
88
|
+
def supports(self, symbol: str) -> bool:
|
|
89
|
+
"""判断该数据源是否能处理这个 symbol(子类可覆盖)"""
|
|
90
|
+
market = _detect_market(symbol)
|
|
91
|
+
return market in self.markets
|
|
92
|
+
|
|
93
|
+
def is_configured(self) -> bool:
|
|
94
|
+
"""数据源是否已配置好(有 key 等)"""
|
|
95
|
+
return True
|
|
96
|
+
|
|
97
|
+
@abstractmethod
|
|
98
|
+
def quote(self, symbol: str) -> Optional[QuoteResult]:
|
|
99
|
+
"""获取实时行情(同步)"""
|
|
100
|
+
...
|
|
101
|
+
|
|
102
|
+
def history(
|
|
103
|
+
self,
|
|
104
|
+
symbol: str,
|
|
105
|
+
days: int = 90,
|
|
106
|
+
interval: str = "1d",
|
|
107
|
+
) -> Optional[HistoryResult]:
|
|
108
|
+
"""获取历史 OHLCV(同步,子类按需实现)"""
|
|
109
|
+
return None
|
|
110
|
+
|
|
111
|
+
def fundamentals(self, symbol: str) -> Optional[FundamentalsResult]:
|
|
112
|
+
"""获取基本面数据(子类按需实现)"""
|
|
113
|
+
return None
|
|
114
|
+
|
|
115
|
+
|
|
116
|
+
# ── 工具函数 ──────────────────────────────────────────────────────────────────
|
|
117
|
+
|
|
118
|
+
def _detect_market(symbol: str) -> str:
|
|
119
|
+
"""简单推断 symbol 所属市场"""
|
|
120
|
+
s = symbol.upper().replace(" ", "")
|
|
121
|
+
|
|
122
|
+
# 加密货币
|
|
123
|
+
if "/" in s or s in ("BTC", "ETH", "SOL", "DOGE", "BNB", "ADA", "XRP"):
|
|
124
|
+
return "crypto"
|
|
125
|
+
if s.endswith(("-USD", "-USDT", "-BTC")) or s.endswith("USDT"):
|
|
126
|
+
return "crypto"
|
|
127
|
+
|
|
128
|
+
# 大宗商品
|
|
129
|
+
_COMMODITIES = {"WTI", "BRENT", "GOLD", "SILVER", "COPPER",
|
|
130
|
+
"ALUMINUM", "WHEAT", "CORN", "SOYBEAN", "NATGAS", "GAS"}
|
|
131
|
+
if s in _COMMODITIES:
|
|
132
|
+
return "commodity"
|
|
133
|
+
|
|
134
|
+
# 外汇(格式: USDCNY, USD/CNY, EURUSD 等)
|
|
135
|
+
_FX_CURRENCIES = {"USD", "EUR", "GBP", "JPY", "CNY", "HKD",
|
|
136
|
+
"AUD", "CAD", "CHF", "KRW", "SGD", "INR"}
|
|
137
|
+
if len(s) == 6 and s[:3] in _FX_CURRENCIES and s[3:] in _FX_CURRENCIES:
|
|
138
|
+
return "forex"
|
|
139
|
+
if "/" in s:
|
|
140
|
+
parts = s.split("/")
|
|
141
|
+
if len(parts) == 2 and parts[0] in _FX_CURRENCIES and parts[1] in _FX_CURRENCIES:
|
|
142
|
+
return "forex"
|
|
143
|
+
|
|
144
|
+
# A股:数字代码 or 带前缀
|
|
145
|
+
if s.startswith(("SH", "SZ", "BJ")):
|
|
146
|
+
return "a_share"
|
|
147
|
+
try:
|
|
148
|
+
int(s[:6])
|
|
149
|
+
if len(s) == 6 or (len(s) > 6 and not s[6:].isalpha()):
|
|
150
|
+
return "a_share"
|
|
151
|
+
except ValueError:
|
|
152
|
+
pass
|
|
153
|
+
|
|
154
|
+
# 港股
|
|
155
|
+
if s.endswith(".HK") or (s.isdigit() and len(s) == 5):
|
|
156
|
+
return "hk"
|
|
157
|
+
|
|
158
|
+
# 宏观指标
|
|
159
|
+
_MACRO = {"GDP", "CPI", "CPIYOY", "PCE", "UNRATE", "NFP", "FEDFUNDS",
|
|
160
|
+
"US10Y", "US2Y", "US3M", "VIX", "M2", "MORTGAGE", "HOUSING",
|
|
161
|
+
"SP500", "NASDAQ", "WILSHIRE", "USDCNY", "USDINR", "USDEUR"}
|
|
162
|
+
if s in _MACRO:
|
|
163
|
+
return "macro"
|
|
164
|
+
|
|
165
|
+
# 默认美股
|
|
166
|
+
return "us"
|
|
@@ -0,0 +1,253 @@
|
|
|
1
|
+
"""
|
|
2
|
+
datasources/router.py — 数据源路由器
|
|
3
|
+
=====================================
|
|
4
|
+
读取 ~/.aria/datasources.yaml 配置,按优先级依次尝试每个数据源,
|
|
5
|
+
首个成功的结果直接返回;所有失败则返回 None。
|
|
6
|
+
|
|
7
|
+
配置示例 (~/.aria/datasources.yaml):
|
|
8
|
+
a_shares:
|
|
9
|
+
- akshare
|
|
10
|
+
- tushare
|
|
11
|
+
us_stocks:
|
|
12
|
+
- yfinance
|
|
13
|
+
- finnhub
|
|
14
|
+
crypto:
|
|
15
|
+
- ccxt/binance
|
|
16
|
+
- yfinance
|
|
17
|
+
"""
|
|
18
|
+
|
|
19
|
+
from __future__ import annotations
|
|
20
|
+
|
|
21
|
+
import logging
|
|
22
|
+
import threading
|
|
23
|
+
from pathlib import Path
|
|
24
|
+
from typing import Any, Dict, List, Optional, Type
|
|
25
|
+
|
|
26
|
+
from .base import (
|
|
27
|
+
BaseDataSource, QuoteResult, HistoryResult, FundamentalsResult, _detect_market
|
|
28
|
+
)
|
|
29
|
+
from .sources import (
|
|
30
|
+
AkshareSource, YFinanceSource, TushareSource,
|
|
31
|
+
FREDSource, EDGARSource, AlphaVantageSource, WorldBankSource,
|
|
32
|
+
FinnhubSource, WebScraperSource,
|
|
33
|
+
)
|
|
34
|
+
|
|
35
|
+
logger = logging.getLogger(__name__)
|
|
36
|
+
|
|
37
|
+
# ── 数据源目录 ────────────────────────────────────────────────────────────────
|
|
38
|
+
_SOURCE_REGISTRY: Dict[str, Type[BaseDataSource]] = {
|
|
39
|
+
"akshare": AkshareSource,
|
|
40
|
+
"yfinance": YFinanceSource,
|
|
41
|
+
"finnhub": FinnhubSource,
|
|
42
|
+
"tushare": TushareSource,
|
|
43
|
+
"fred": FREDSource,
|
|
44
|
+
"edgar": EDGARSource,
|
|
45
|
+
"alpha_vantage": AlphaVantageSource,
|
|
46
|
+
"world_bank": WorldBankSource,
|
|
47
|
+
"web_scraper": WebScraperSource,
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
|
|
51
|
+
def register_datasource(name: str, cls: Type[BaseDataSource]) -> None:
|
|
52
|
+
"""注册自定义数据源(供插件/用户扩展)"""
|
|
53
|
+
_SOURCE_REGISTRY[name.lower()] = cls
|
|
54
|
+
logger.info(f"✓ 注册自定义数据源: {name}")
|
|
55
|
+
|
|
56
|
+
|
|
57
|
+
# ── 默认优先级链 ──────────────────────────────────────────────────────────────
|
|
58
|
+
_DEFAULT_CHAINS: Dict[str, List[str]] = {
|
|
59
|
+
"a_share": ["tushare", "akshare"],
|
|
60
|
+
"us": ["yfinance", "finnhub", "alpha_vantage", "edgar"],
|
|
61
|
+
"hk": ["yfinance", "finnhub", "akshare"],
|
|
62
|
+
"crypto": ["yfinance"],
|
|
63
|
+
"macro": ["fred", "world_bank"],
|
|
64
|
+
"forex": ["alpha_vantage", "yfinance"],
|
|
65
|
+
"commodity": ["alpha_vantage"],
|
|
66
|
+
}
|
|
67
|
+
|
|
68
|
+
# ── 配置文件路径 ──────────────────────────────────────────────────────────────
|
|
69
|
+
_CONFIG_PATHS = [
|
|
70
|
+
Path.home() / ".aria" / "datasources.yaml",
|
|
71
|
+
Path.home() / ".aria" / "datasources.json",
|
|
72
|
+
Path(".aria.json"),
|
|
73
|
+
]
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
def _load_user_chains() -> Dict[str, List[str]]:
|
|
77
|
+
for p in _CONFIG_PATHS:
|
|
78
|
+
if not p.exists():
|
|
79
|
+
continue
|
|
80
|
+
try:
|
|
81
|
+
if p.suffix in (".yaml", ".yml"):
|
|
82
|
+
import yaml
|
|
83
|
+
with open(p, encoding="utf-8") as f:
|
|
84
|
+
data = yaml.safe_load(f) or {}
|
|
85
|
+
else:
|
|
86
|
+
import json
|
|
87
|
+
with open(p, encoding="utf-8") as f:
|
|
88
|
+
data = json.load(f)
|
|
89
|
+
ds = data.get("datasources", data)
|
|
90
|
+
if isinstance(ds, dict):
|
|
91
|
+
chains = {}
|
|
92
|
+
key_map = {
|
|
93
|
+
"a_shares": "a_share", "a_share": "a_share",
|
|
94
|
+
"us_stocks": "us", "us": "us",
|
|
95
|
+
"hk_stocks": "hk", "hk": "hk",
|
|
96
|
+
"crypto": "crypto",
|
|
97
|
+
}
|
|
98
|
+
for k, v in ds.items():
|
|
99
|
+
market = key_map.get(k.lower(), k.lower())
|
|
100
|
+
if isinstance(v, list):
|
|
101
|
+
chains[market] = v
|
|
102
|
+
return chains
|
|
103
|
+
except Exception as e:
|
|
104
|
+
logger.debug(f"加载 datasources 配置 {p} 失败: {e}")
|
|
105
|
+
return {}
|
|
106
|
+
|
|
107
|
+
|
|
108
|
+
class DataRouter:
|
|
109
|
+
"""
|
|
110
|
+
统一数据路由器。线程安全(内部缓存用锁保护)。
|
|
111
|
+
|
|
112
|
+
用法:
|
|
113
|
+
router = DataRouter()
|
|
114
|
+
q = router.quote("600519")
|
|
115
|
+
h = router.history("AAPL", days=90)
|
|
116
|
+
f = router.fundamentals("000858")
|
|
117
|
+
"""
|
|
118
|
+
|
|
119
|
+
def __init__(self):
|
|
120
|
+
self._user_chains = _load_user_chains()
|
|
121
|
+
# name → instance, or None for "cannot serve" (unknown, unconfigured,
|
|
122
|
+
# or failed to construct). Caching the negative case is what stops a
|
|
123
|
+
# rebuild on every call; see invalidate_sources() for the escape hatch.
|
|
124
|
+
self._source_cache: Dict[str, Optional[BaseDataSource]] = {}
|
|
125
|
+
self._lock = threading.Lock()
|
|
126
|
+
|
|
127
|
+
def _get_chain(self, market: str) -> List[str]:
|
|
128
|
+
user = self._user_chains.get(market) or self._user_chains.get(
|
|
129
|
+
{"a_share": "a_shares"}.get(market, market)
|
|
130
|
+
)
|
|
131
|
+
if user:
|
|
132
|
+
return user
|
|
133
|
+
return _DEFAULT_CHAINS.get(market, ["yfinance"])
|
|
134
|
+
|
|
135
|
+
def _get_source(self, name: str) -> Optional[BaseDataSource]:
|
|
136
|
+
"""The live instance for *name*, or None if it cannot serve.
|
|
137
|
+
|
|
138
|
+
Negative results are cached too. Only successes used to be, so an
|
|
139
|
+
unconfigured source was constructed again on every single call — for a
|
|
140
|
+
chain with two unkeyed providers ahead of a working one, that is two
|
|
141
|
+
wasted constructions per quote, forever. Some sources read files or
|
|
142
|
+
probe the environment in __init__, which makes it more than a
|
|
143
|
+
micro-cost.
|
|
144
|
+
"""
|
|
145
|
+
with self._lock:
|
|
146
|
+
if name in self._source_cache:
|
|
147
|
+
return self._source_cache[name]
|
|
148
|
+
|
|
149
|
+
cls = _SOURCE_REGISTRY.get(name.lower())
|
|
150
|
+
if not cls:
|
|
151
|
+
logger.debug(f"未知数据源: {name}")
|
|
152
|
+
self._source_cache[name] = None
|
|
153
|
+
return None
|
|
154
|
+
try:
|
|
155
|
+
src = cls()
|
|
156
|
+
except Exception as exc:
|
|
157
|
+
# A source whose constructor raises must not take down the
|
|
158
|
+
# chain; the next provider is very likely fine.
|
|
159
|
+
logger.debug(f"数据源 {name} 初始化失败: {exc}")
|
|
160
|
+
self._source_cache[name] = None
|
|
161
|
+
return None
|
|
162
|
+
if not src.is_configured():
|
|
163
|
+
logger.debug(f"数据源 {name} 未配置(缺少 API key)")
|
|
164
|
+
self._source_cache[name] = None
|
|
165
|
+
return None
|
|
166
|
+
self._source_cache[name] = src
|
|
167
|
+
return src
|
|
168
|
+
|
|
169
|
+
def invalidate_sources(self) -> None:
|
|
170
|
+
"""Forget what is configured, after credentials change at runtime.
|
|
171
|
+
|
|
172
|
+
Without this the negative cache above would outlive an /apikey set for
|
|
173
|
+
the whole session.
|
|
174
|
+
"""
|
|
175
|
+
with self._lock:
|
|
176
|
+
self._source_cache.clear()
|
|
177
|
+
|
|
178
|
+
def quote(self, symbol: str) -> Optional[QuoteResult]:
|
|
179
|
+
market = _detect_market(symbol)
|
|
180
|
+
for src_name in self._get_chain(market):
|
|
181
|
+
if src_name == "edgar":
|
|
182
|
+
continue
|
|
183
|
+
src = self._get_source(src_name)
|
|
184
|
+
if not src or not src.supports(symbol):
|
|
185
|
+
continue
|
|
186
|
+
try:
|
|
187
|
+
result = src.quote(symbol)
|
|
188
|
+
try:
|
|
189
|
+
valid_price = result is not None and float(getattr(result, "price", 0) or 0) > 0
|
|
190
|
+
except Exception:
|
|
191
|
+
valid_price = False
|
|
192
|
+
if result and valid_price:
|
|
193
|
+
logger.debug(f"quote({symbol}) ← {src_name}")
|
|
194
|
+
return result
|
|
195
|
+
except Exception as e:
|
|
196
|
+
logger.debug(f"[{src_name}] quote {symbol} 异常: {e}")
|
|
197
|
+
logger.debug(f"所有数据源均无法获取 {symbol} 行情")
|
|
198
|
+
return None
|
|
199
|
+
|
|
200
|
+
def history(
|
|
201
|
+
self, symbol: str, days: int = 90, interval: str = "1d"
|
|
202
|
+
) -> Optional[HistoryResult]:
|
|
203
|
+
market = _detect_market(symbol)
|
|
204
|
+
for src_name in self._get_chain(market):
|
|
205
|
+
src = self._get_source(src_name)
|
|
206
|
+
if not src or not src.supports(symbol):
|
|
207
|
+
continue
|
|
208
|
+
try:
|
|
209
|
+
result = src.history(symbol, days=days, interval=interval)
|
|
210
|
+
if result is not None:
|
|
211
|
+
logger.debug(f"history({symbol}) ← {src_name}")
|
|
212
|
+
return result
|
|
213
|
+
except Exception as e:
|
|
214
|
+
logger.debug(f"[{src_name}] history {symbol} 异常: {e}")
|
|
215
|
+
return None
|
|
216
|
+
|
|
217
|
+
def fundamentals(self, symbol: str) -> Optional[FundamentalsResult]:
|
|
218
|
+
market = _detect_market(symbol)
|
|
219
|
+
for src_name in self._get_chain(market):
|
|
220
|
+
src = self._get_source(src_name)
|
|
221
|
+
if not src or not src.supports(symbol):
|
|
222
|
+
continue
|
|
223
|
+
try:
|
|
224
|
+
result = src.fundamentals(symbol)
|
|
225
|
+
if result:
|
|
226
|
+
logger.debug(f"fundamentals({symbol}) ← {src_name}")
|
|
227
|
+
return result
|
|
228
|
+
except Exception as e:
|
|
229
|
+
logger.debug(f"[{src_name}] fundamentals {symbol} 异常: {e}")
|
|
230
|
+
return None
|
|
231
|
+
|
|
232
|
+
def list_sources(self) -> List[Dict[str, Any]]:
|
|
233
|
+
"""列出所有数据源及其状态(用于 /config 展示)"""
|
|
234
|
+
result = []
|
|
235
|
+
for name, cls in _SOURCE_REGISTRY.items():
|
|
236
|
+
src = cls()
|
|
237
|
+
result.append({
|
|
238
|
+
"name": name,
|
|
239
|
+
"markets": cls.markets,
|
|
240
|
+
"needs_key": cls.requires_key,
|
|
241
|
+
"configured": src.is_configured(),
|
|
242
|
+
})
|
|
243
|
+
return result
|
|
244
|
+
|
|
245
|
+
|
|
246
|
+
# ── 单例 ──────────────────────────────────────────────────────────────────────
|
|
247
|
+
_router: Optional[DataRouter] = None
|
|
248
|
+
|
|
249
|
+
def get_router() -> DataRouter:
|
|
250
|
+
global _router
|
|
251
|
+
if _router is None:
|
|
252
|
+
_router = DataRouter()
|
|
253
|
+
return _router
|
|
@@ -0,0 +1,217 @@
|
|
|
1
|
+
"""Period-aligned annual SEC facts; no estimates or cross-filing joins."""
|
|
2
|
+
from __future__ import annotations
|
|
3
|
+
|
|
4
|
+
from collections import OrderedDict
|
|
5
|
+
from copy import deepcopy
|
|
6
|
+
from datetime import date
|
|
7
|
+
import json
|
|
8
|
+
import math
|
|
9
|
+
import os
|
|
10
|
+
import re
|
|
11
|
+
import ssl
|
|
12
|
+
import threading
|
|
13
|
+
import time
|
|
14
|
+
import urllib.error
|
|
15
|
+
import urllib.request
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
class SECError(ValueError):
|
|
19
|
+
def __init__(self, code: str):
|
|
20
|
+
self.code = code
|
|
21
|
+
super().__init__(code)
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
INCOME_TAGS = {
|
|
25
|
+
'revenue': ('RevenueFromContractWithCustomerExcludingAssessedTax', 'Revenues', 'SalesRevenueNet'),
|
|
26
|
+
'net_income': ('NetIncomeLoss',),
|
|
27
|
+
'gross_profit': ('GrossProfit',),
|
|
28
|
+
'operating_income': ('OperatingIncomeLoss',),
|
|
29
|
+
'cost_of_goods_sold': ('CostOfRevenue', 'CostOfGoodsAndServicesSold'),
|
|
30
|
+
}
|
|
31
|
+
BALANCE_TAGS = {
|
|
32
|
+
'total_assets': ('Assets',),
|
|
33
|
+
'total_equity': ('StockholdersEquity',),
|
|
34
|
+
'current_assets': ('AssetsCurrent',),
|
|
35
|
+
'current_liabilities': ('LiabilitiesCurrent',),
|
|
36
|
+
'inventory': ('InventoryNet',),
|
|
37
|
+
'retained_earnings': ('RetainedEarningsAccumulatedDeficit',),
|
|
38
|
+
}
|
|
39
|
+
_REQUEST_LOCK = threading.Lock()
|
|
40
|
+
_LAST_REQUEST = 0.0
|
|
41
|
+
_CACHE_LOCK = threading.Lock()
|
|
42
|
+
_CACHE = OrderedDict()
|
|
43
|
+
_MAX_BYTES = 20 * 1024 * 1024
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
def _value(entry):
|
|
47
|
+
value = entry.get('val')
|
|
48
|
+
if isinstance(value, bool) or not isinstance(value, (int, float)):
|
|
49
|
+
return None
|
|
50
|
+
try:
|
|
51
|
+
if not math.isfinite(value):
|
|
52
|
+
return None
|
|
53
|
+
except OverflowError:
|
|
54
|
+
return None
|
|
55
|
+
return value
|
|
56
|
+
|
|
57
|
+
|
|
58
|
+
def _rows(facts, tags):
|
|
59
|
+
for tag in tags:
|
|
60
|
+
concept = facts.get(tag, {})
|
|
61
|
+
units = concept.get('units', {}) if isinstance(concept, dict) else {}
|
|
62
|
+
rows = units.get('USD', []) if isinstance(units, dict) else []
|
|
63
|
+
for row in rows if isinstance(rows, list) else []:
|
|
64
|
+
if isinstance(row, dict) and _value(row) is not None:
|
|
65
|
+
yield tag, row
|
|
66
|
+
|
|
67
|
+
|
|
68
|
+
def parse_annual_statements(payload: dict, *, today: date | None = None) -> dict:
|
|
69
|
+
"""Require core income and balance values from one annual USD filing.
|
|
70
|
+
|
|
71
|
+
A 10-K also repeats older years and sometimes contains quarterly facts.
|
|
72
|
+
Match accession/start/end explicitly; never choose latest per metric.
|
|
73
|
+
"""
|
|
74
|
+
today = today or date.today()
|
|
75
|
+
if not isinstance(payload, dict) or not isinstance(payload.get('facts'), dict):
|
|
76
|
+
raise SECError('sec_invalid_response')
|
|
77
|
+
facts = payload['facts'].get('us-gaap', {})
|
|
78
|
+
if not isinstance(facts, dict):
|
|
79
|
+
raise SECError('sec_invalid_response')
|
|
80
|
+
anchors = []
|
|
81
|
+
for tag, row in _rows(facts, INCOME_TAGS['revenue']):
|
|
82
|
+
try:
|
|
83
|
+
start, end, filed = (date.fromisoformat(row[k]) for k in ('start', 'end', 'filed'))
|
|
84
|
+
except (KeyError, TypeError, ValueError):
|
|
85
|
+
continue
|
|
86
|
+
if (row.get('form') in {'10-K', '10-K/A'} and row.get('accn')
|
|
87
|
+
and 335 <= (end - start).days <= 395 and start <= end <= filed <= today):
|
|
88
|
+
anchors.append((tag, row))
|
|
89
|
+
anchors.sort(key=lambda item: (item[1]['end'], item[1]['filed']), reverse=True)
|
|
90
|
+
for revenue_tag, anchor in anchors:
|
|
91
|
+
accession = anchor['accn']
|
|
92
|
+
def extract(mapping, duration):
|
|
93
|
+
values, provenance = {}, {}
|
|
94
|
+
for field, tags in mapping.items():
|
|
95
|
+
for tag in tags:
|
|
96
|
+
rows = [row for _, row in _rows(facts, (tag,))
|
|
97
|
+
if row.get('accn') == accession and row.get('end') == anchor['end']
|
|
98
|
+
and row.get('form') in {'10-K', '10-K/A'} and row.get('filed') == anchor['filed']
|
|
99
|
+
and (row.get('start') == anchor['start'] if duration else not row.get('start'))]
|
|
100
|
+
if not rows:
|
|
101
|
+
continue
|
|
102
|
+
if len({_value(row) for row in rows}) != 1:
|
|
103
|
+
raise SECError('sec_conflicting_facts')
|
|
104
|
+
row = rows[0]
|
|
105
|
+
values[field] = _value(row)
|
|
106
|
+
provenance[field] = {'tag': tag, 'unit': 'USD', 'accession': accession,
|
|
107
|
+
'start': row.get('start'), 'end': row['end'], 'filed': row['filed']}
|
|
108
|
+
break
|
|
109
|
+
return values, provenance
|
|
110
|
+
income, income_provenance = extract(INCOME_TAGS, True)
|
|
111
|
+
balance, balance_provenance = extract(BALANCE_TAGS, False)
|
|
112
|
+
if not {'revenue', 'net_income'} <= income.keys() or not {'total_assets', 'total_equity'} <= balance.keys():
|
|
113
|
+
continue
|
|
114
|
+
return {
|
|
115
|
+
'income_statement': income, 'balance_sheet': balance, 'cashflow': {},
|
|
116
|
+
'period': anchor['end'], 'period_start': anchor['start'], 'period_type': 'annual',
|
|
117
|
+
'currency': 'USD', 'accession': accession, 'filed': anchor['filed'],
|
|
118
|
+
'age_days': (today - date.fromisoformat(anchor['end'])).days,
|
|
119
|
+
'stale': (today - date.fromisoformat(anchor['end'])).days > 550,
|
|
120
|
+
'entity_name': payload.get('entityName'), 'cik': str(payload.get('cik', '')).zfill(10),
|
|
121
|
+
'provenance': {**income_provenance, **balance_provenance},
|
|
122
|
+
}
|
|
123
|
+
raise SECError('sec_no_aligned_annual_statements')
|
|
124
|
+
|
|
125
|
+
|
|
126
|
+
def _get_json(url: str, deadline: float):
|
|
127
|
+
agent = os.environ.get('EDGAR_USER_AGENT', '').strip()
|
|
128
|
+
if not re.fullmatch(r'[^\r\n]{3,240}', agent) or not re.search(r'[^\s@]+@[^\s@]+\.[^\s@]+', agent) or 'example.com' in agent:
|
|
129
|
+
raise SECError('sec_identity_required')
|
|
130
|
+
global _LAST_REQUEST
|
|
131
|
+
with _REQUEST_LOCK:
|
|
132
|
+
delay = max(0.0, 0.5 - (time.monotonic() - _LAST_REQUEST))
|
|
133
|
+
if time.monotonic() + delay >= deadline:
|
|
134
|
+
raise SECError('sec_timeout')
|
|
135
|
+
if delay:
|
|
136
|
+
time.sleep(delay)
|
|
137
|
+
_LAST_REQUEST = time.monotonic()
|
|
138
|
+
remaining = deadline - time.monotonic()
|
|
139
|
+
if remaining <= 0:
|
|
140
|
+
raise SECError('sec_timeout')
|
|
141
|
+
request = urllib.request.Request(url, headers={'User-Agent': agent, 'Accept': 'application/json', 'Accept-Encoding': 'identity'})
|
|
142
|
+
try:
|
|
143
|
+
context = ssl.create_default_context()
|
|
144
|
+
try:
|
|
145
|
+
import certifi
|
|
146
|
+
context.load_verify_locations(certifi.where())
|
|
147
|
+
except ImportError:
|
|
148
|
+
pass
|
|
149
|
+
with urllib.request.urlopen(request, timeout=min(remaining, 6), context=context) as response:
|
|
150
|
+
chunks, size = [], 0
|
|
151
|
+
while size <= _MAX_BYTES:
|
|
152
|
+
if time.monotonic() >= deadline:
|
|
153
|
+
raise SECError('sec_timeout')
|
|
154
|
+
chunk = response.read1(min(65536, _MAX_BYTES + 1 - size))
|
|
155
|
+
if not chunk:
|
|
156
|
+
break
|
|
157
|
+
chunks.append(chunk)
|
|
158
|
+
size += len(chunk)
|
|
159
|
+
raw = b''.join(chunks)
|
|
160
|
+
if time.monotonic() >= deadline:
|
|
161
|
+
raise SECError('sec_timeout')
|
|
162
|
+
if len(raw) > _MAX_BYTES:
|
|
163
|
+
raise SECError('sec_response_too_large')
|
|
164
|
+
result = json.loads(raw)
|
|
165
|
+
if not isinstance(result, dict):
|
|
166
|
+
raise SECError('sec_invalid_response')
|
|
167
|
+
return result
|
|
168
|
+
except urllib.error.HTTPError as exc:
|
|
169
|
+
raise SECError({403: 'sec_access_denied', 429: 'sec_rate_limited'}.get(exc.code, 'sec_http_error')) from None
|
|
170
|
+
except TimeoutError:
|
|
171
|
+
raise SECError('sec_timeout') from None
|
|
172
|
+
except urllib.error.URLError as exc:
|
|
173
|
+
raise SECError('sec_timeout' if isinstance(exc.reason, TimeoutError) else 'sec_source_unavailable') from None
|
|
174
|
+
except OSError:
|
|
175
|
+
raise SECError('sec_source_unavailable') from None
|
|
176
|
+
except (json.JSONDecodeError, UnicodeDecodeError):
|
|
177
|
+
raise SECError('sec_invalid_response') from None
|
|
178
|
+
|
|
179
|
+
|
|
180
|
+
def _cached_json(url, deadline, ttl):
|
|
181
|
+
with _CACHE_LOCK:
|
|
182
|
+
item = _CACHE.get(url)
|
|
183
|
+
if item and time.monotonic() - item[0] < ttl:
|
|
184
|
+
return deepcopy(item[1]), item[2], True
|
|
185
|
+
payload = _get_json(url, deadline)
|
|
186
|
+
retrieved = time.strftime('%Y-%m-%dT%H:%M:%SZ', time.gmtime())
|
|
187
|
+
with _CACHE_LOCK:
|
|
188
|
+
_CACHE[url] = (time.monotonic(), deepcopy(payload), retrieved)
|
|
189
|
+
_CACHE.move_to_end(url)
|
|
190
|
+
while len(_CACHE) > 32:
|
|
191
|
+
_CACHE.popitem(last=False)
|
|
192
|
+
return payload, retrieved, False
|
|
193
|
+
|
|
194
|
+
|
|
195
|
+
def fetch_annual_statements(symbol: str, *, timeout: float = 10) -> dict:
|
|
196
|
+
if not re.fullmatch(r'[A-Z][A-Z0-9.-]{0,9}', symbol):
|
|
197
|
+
raise SECError('sec_invalid_ticker')
|
|
198
|
+
deadline = time.monotonic() + timeout
|
|
199
|
+
mapping, _, _ = _cached_json('https://www.sec.gov/files/company_tickers.json', deadline, 86400)
|
|
200
|
+
matches = {str(item.get('cik_str')).zfill(10) for item in mapping.values()
|
|
201
|
+
if isinstance(item, dict) and item.get('ticker') == symbol}
|
|
202
|
+
if len(matches) != 1:
|
|
203
|
+
raise SECError('sec_ticker_not_found')
|
|
204
|
+
cik = matches.pop()
|
|
205
|
+
if not re.fullmatch(r'\d{10}', cik):
|
|
206
|
+
raise SECError('sec_invalid_response')
|
|
207
|
+
url = f'https://data.sec.gov/api/xbrl/companyfacts/CIK{cik}.json'
|
|
208
|
+
payload, retrieved, cached = _cached_json(url, deadline, 300)
|
|
209
|
+
try:
|
|
210
|
+
returned_cik = str(int(payload.get('cik'))).zfill(10)
|
|
211
|
+
except (ValueError, TypeError):
|
|
212
|
+
raise SECError('sec_invalid_response') from None
|
|
213
|
+
if returned_cik != cik:
|
|
214
|
+
raise SECError('sec_company_mismatch')
|
|
215
|
+
result = parse_annual_statements(payload)
|
|
216
|
+
result.update(source_url=url, retrieved_at=retrieved, cache_hit=cached, symbol=symbol)
|
|
217
|
+
return result
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
from .akshare_source import AkshareSource
|
|
2
|
+
from .yfinance_source import YFinanceSource
|
|
3
|
+
from .tushare_source import TushareSource
|
|
4
|
+
from .fred_source import FREDSource
|
|
5
|
+
from .edgar_source import EDGARSource
|
|
6
|
+
from .alpha_vantage_source import AlphaVantageSource
|
|
7
|
+
from .world_bank_source import WorldBankSource
|
|
8
|
+
from .web_scraper_source import WebScraperSource
|
|
9
|
+
from .finnhub_source import FinnhubSource
|
|
10
|
+
|
|
11
|
+
__all__ = [
|
|
12
|
+
"AkshareSource", "YFinanceSource", "TushareSource",
|
|
13
|
+
"FREDSource", "EDGARSource", "AlphaVantageSource",
|
|
14
|
+
"WorldBankSource", "WebScraperSource", "FinnhubSource",
|
|
15
|
+
]
|