aria-code 0.167.0__py3-none-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (601) hide show
  1. aria_bots/__init__.py +7 -0
  2. aria_bots/channels/README.md +97 -0
  3. aria_bots/channels/__init__.py +1 -0
  4. aria_bots/channels/approvals.py +263 -0
  5. aria_bots/channels/conversation.py +280 -0
  6. aria_bots/channels/digest.py +186 -0
  7. aria_bots/channels/registry.py +95 -0
  8. aria_bots/engine.py +144 -0
  9. aria_bots/feishu_bot.py +1385 -0
  10. aria_bots/logistics/__init__.py +1 -0
  11. aria_bots/logistics/carriers.py +253 -0
  12. aria_bots/logistics/freight.py +154 -0
  13. aria_bots/logistics/inventory.py +242 -0
  14. aria_bots/logistics/mcp_server.py +156 -0
  15. aria_bots/logistics/records.py +119 -0
  16. aria_bots/logistics/tenancy.py +152 -0
  17. aria_bots/market.py +91 -0
  18. aria_bots/relay_client.py +326 -0
  19. aria_bots/telegram_bot.py +299 -0
  20. aria_code/__init__.py +111 -0
  21. aria_code/_bin/aria-native.exe +0 -0
  22. aria_code/_bin/aria-runtime.exe +0 -0
  23. aria_code/_version.py +20 -0
  24. aria_code/account_memory.py +296 -0
  25. aria_code/account_session.py +166 -0
  26. aria_code/adk_apps/__init__.py +1 -0
  27. aria_code/adk_apps/aria_code_review/__init__.py +1 -0
  28. aria_code/adk_apps/aria_code_review/agent.py +32 -0
  29. aria_code/adk_apps/aria_research/__init__.py +1 -0
  30. aria_code/adk_apps/aria_research/agent.py +31 -0
  31. aria_code/agents/__init__.py +49 -0
  32. aria_code/agents/base.py +329 -0
  33. aria_code/agents/code_review.py +200 -0
  34. aria_code/agents/deep/__init__.py +37 -0
  35. aria_code/agents/deep/calibration_loop.py +144 -0
  36. aria_code/agents/deep/critic.py +125 -0
  37. aria_code/agents/deep/deepen.py +193 -0
  38. aria_code/agents/deep/models.py +149 -0
  39. aria_code/agents/deep/pipeline.py +164 -0
  40. aria_code/agents/deep/quant_fusion.py +192 -0
  41. aria_code/agents/deep/themes.py +95 -0
  42. aria_code/agents/deep/tiers.py +106 -0
  43. aria_code/agents/engineering/__init__.py +14 -0
  44. aria_code/agents/engineering/coder.py +222 -0
  45. aria_code/agents/engineering/tester.py +145 -0
  46. aria_code/agents/financial/__init__.py +21 -0
  47. aria_code/agents/financial/cashflow_burn.py +119 -0
  48. aria_code/agents/financial/catalyst.py +279 -0
  49. aria_code/agents/financial/corporate_finance.py +175 -0
  50. aria_code/agents/financial/debate.py +224 -0
  51. aria_code/agents/financial/earnings.py +315 -0
  52. aria_code/agents/financial/fundamental.py +178 -0
  53. aria_code/agents/financial/macro.py +111 -0
  54. aria_code/agents/financial/news.py +231 -0
  55. aria_code/agents/financial/risk.py +146 -0
  56. aria_code/agents/financial/sector.py +279 -0
  57. aria_code/agents/financial/strategist.py +121 -0
  58. aria_code/agents/financial/stripe_revenue.py +127 -0
  59. aria_code/agents/financial/synthesis.py +274 -0
  60. aria_code/agents/financial/technical.py +277 -0
  61. aria_code/agents/orchestrator.py +44 -0
  62. aria_code/agents/portfolio_agent.py +392 -0
  63. aria_code/agents/realty/__init__.py +62 -0
  64. aria_code/agents/realty/asset_diagnosis.py +152 -0
  65. aria_code/agents/realty/business_match.py +166 -0
  66. aria_code/agents/realty/cashflow_verify.py +209 -0
  67. aria_code/agents/realty/contract_rules.py +210 -0
  68. aria_code/agents/realty/energy_anomaly.py +189 -0
  69. aria_code/agents/realty/exit_settlement.py +208 -0
  70. aria_code/agents/realty/fulfillment_risk.py +206 -0
  71. aria_code/agents/realty/ops_optimize.py +160 -0
  72. aria_code/agents/realty/revenue_share.py +215 -0
  73. aria_code/agents/registry.py +168 -0
  74. aria_code/agents/signal_scheme.py +111 -0
  75. aria_code/agents/sports/__init__.py +0 -0
  76. aria_code/agents/sports/football_agent.py +197 -0
  77. aria_code/agents/supervisor.py +113 -0
  78. aria_code/agents/team.py +500 -0
  79. aria_code/agents/video/__init__.py +16 -0
  80. aria_code/agents/video/director.py +376 -0
  81. aria_code/agents/video/qa_agent.py +143 -0
  82. aria_code/agents/video/script_doctor.py +112 -0
  83. aria_code/agents/warehouse/__init__.py +25 -0
  84. aria_code/agents/warehouse/contracts.py +26 -0
  85. aria_code/agents/warehouse/dashboard.py +155 -0
  86. aria_code/agents/warehouse/fulfillment_leadtime.py +95 -0
  87. aria_code/agents/warehouse/inbound_exceptions.py +25 -0
  88. aria_code/agents/warehouse/inventory_health.py +24 -0
  89. aria_code/agents/warehouse/local_run.py +71 -0
  90. aria_code/agents/warehouse/local_snapshot.py +132 -0
  91. aria_code/agents/warehouse/logistics_cost.py +84 -0
  92. aria_code/agents/warehouse/logistics_sync.py +21 -0
  93. aria_code/agents/warehouse/navigation.schema.json +110 -0
  94. aria_code/agents/warehouse/task_agent.py +172 -0
  95. aria_code/agents/warehouse/workflow.py +31 -0
  96. aria_code/aliyun_data_client.py +4 -0
  97. aria_code/apps/README.md +12 -0
  98. aria_code/apps/__init__.py +2 -0
  99. aria_code/apps/channels/README.md +9 -0
  100. aria_code/apps/channels/__init__.py +17 -0
  101. aria_code/apps/channels/approvals.py +11 -0
  102. aria_code/apps/channels/conversation.py +11 -0
  103. aria_code/apps/channels/digest.py +11 -0
  104. aria_code/apps/channels/intake.py +92 -0
  105. aria_code/apps/channels/registry.py +11 -0
  106. aria_code/apps/channels/tradingview.py +94 -0
  107. aria_code/apps/cli/README.md +13 -0
  108. aria_code/apps/cli/__init__.py +2 -0
  109. aria_code/apps/cli/_optional.py +120 -0
  110. aria_code/apps/cli/app_server.py +656 -0
  111. aria_code/apps/cli/auto_update.py +125 -0
  112. aria_code/apps/cli/bootstrap.py +206 -0
  113. aria_code/apps/cli/broker_render.py +206 -0
  114. aria_code/apps/cli/chat_turn.py +1327 -0
  115. aria_code/apps/cli/cloud_models.py +231 -0
  116. aria_code/apps/cli/codegen_paths.py +29 -0
  117. aria_code/apps/cli/commands/__init__.py +34 -0
  118. aria_code/apps/cli/commands/_business_render.py +78 -0
  119. aria_code/apps/cli/commands/_ui.py +71 -0
  120. aria_code/apps/cli/commands/analysis_cmds.py +378 -0
  121. aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
  122. aria_code/apps/cli/commands/auth_cmds.py +215 -0
  123. aria_code/apps/cli/commands/backtest_cmds.py +2518 -0
  124. aria_code/apps/cli/commands/broker_cmds.py +1236 -0
  125. aria_code/apps/cli/commands/business_workflow_cmds.py +379 -0
  126. aria_code/apps/cli/commands/canvas_cmds.py +77 -0
  127. aria_code/apps/cli/commands/catalog.py +189 -0
  128. aria_code/apps/cli/commands/core_cmds.py +2748 -0
  129. aria_code/apps/cli/commands/data_cmds.py +521 -0
  130. aria_code/apps/cli/commands/diagnostic_cmds.py +285 -0
  131. aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
  132. aria_code/apps/cli/commands/engineering_cmds.py +399 -0
  133. aria_code/apps/cli/commands/file_cmds.py +312 -0
  134. aria_code/apps/cli/commands/finance_service_cmds.py +60 -0
  135. aria_code/apps/cli/commands/fx_commodity_cmds.py +123 -0
  136. aria_code/apps/cli/commands/market.py +544 -0
  137. aria_code/apps/cli/commands/market_cmds.py +1419 -0
  138. aria_code/apps/cli/commands/market_context.py +432 -0
  139. aria_code/apps/cli/commands/market_render.py +7 -0
  140. aria_code/apps/cli/commands/model_cmds.py +2374 -0
  141. aria_code/apps/cli/commands/ops_cmds.py +721 -0
  142. aria_code/apps/cli/commands/orchestrator_cmds.py +181 -0
  143. aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
  144. aria_code/apps/cli/commands/portfolio_cmds.py +1316 -0
  145. aria_code/apps/cli/commands/report.py +476 -0
  146. aria_code/apps/cli/commands/scaffold_templates.py +617 -0
  147. aria_code/apps/cli/commands/session_cmds.py +236 -0
  148. aria_code/apps/cli/commands/session_ux_cmds.py +359 -0
  149. aria_code/apps/cli/commands/team.py +723 -0
  150. aria_code/apps/cli/commands/team_render.py +8 -0
  151. aria_code/apps/cli/commands/ui_cmds.py +401 -0
  152. aria_code/apps/cli/commands/warehouse_cmds.py +297 -0
  153. aria_code/apps/cli/commands/workflow_cmds.py +629 -0
  154. aria_code/apps/cli/commands/workspace_cmds.py +1543 -0
  155. aria_code/apps/cli/config_paths.py +73 -0
  156. aria_code/apps/cli/config_store.py +158 -0
  157. aria_code/apps/cli/constants.py +88 -0
  158. aria_code/apps/cli/context.py +34 -0
  159. aria_code/apps/cli/conversation_policy.py +53 -0
  160. aria_code/apps/cli/deterministic.py +200 -0
  161. aria_code/apps/cli/direct.py +48 -0
  162. aria_code/apps/cli/earnings_view.py +172 -0
  163. aria_code/apps/cli/endpoint_migration.py +64 -0
  164. aria_code/apps/cli/exec_events.py +255 -0
  165. aria_code/apps/cli/file_references.py +155 -0
  166. aria_code/apps/cli/finance_service_catalog.py +111 -0
  167. aria_code/apps/cli/football_reports.py +231 -0
  168. aria_code/apps/cli/github_app_auth.py +135 -0
  169. aria_code/apps/cli/google_login.py +239 -0
  170. aria_code/apps/cli/handlers/__init__.py +11 -0
  171. aria_code/apps/cli/handlers/broker_handlers.py +122 -0
  172. aria_code/apps/cli/handlers/chart_handlers.py +1314 -0
  173. aria_code/apps/cli/handlers/market_handlers.py +2570 -0
  174. aria_code/apps/cli/handlers/realty_handlers.py +114 -0
  175. aria_code/apps/cli/handlers/sports_handlers.py +22 -0
  176. aria_code/apps/cli/handlers/strategy_advice.py +82 -0
  177. aria_code/apps/cli/headless.py +352 -0
  178. aria_code/apps/cli/helpers.py +326 -0
  179. aria_code/apps/cli/hooks.py +180 -0
  180. aria_code/apps/cli/i18n.py +310 -0
  181. aria_code/apps/cli/intent.py +136 -0
  182. aria_code/apps/cli/intent_router.py +305 -0
  183. aria_code/apps/cli/intent_shadow.py +72 -0
  184. aria_code/apps/cli/intent_signals.py +347 -0
  185. aria_code/apps/cli/lifecycle_hooks.py +48 -0
  186. aria_code/apps/cli/local_tool_schemas.py +550 -0
  187. aria_code/apps/cli/main.py +39 -0
  188. aria_code/apps/cli/market_hours.py +100 -0
  189. aria_code/apps/cli/market_metadata.py +135 -0
  190. aria_code/apps/cli/market_universe.py +487 -0
  191. aria_code/apps/cli/message_processing.py +310 -0
  192. aria_code/apps/cli/mixin_binding.py +32 -0
  193. aria_code/apps/cli/model_catalog.py +652 -0
  194. aria_code/apps/cli/model_probe.py +169 -0
  195. aria_code/apps/cli/model_retirement.py +65 -0
  196. aria_code/apps/cli/native_branding.py +12 -0
  197. aria_code/apps/cli/native_bridge.py +118 -0
  198. aria_code/apps/cli/native_entry.py +58 -0
  199. aria_code/apps/cli/pdf_report.py +977 -0
  200. aria_code/apps/cli/plan_mode.py +139 -0
  201. aria_code/apps/cli/plotly_html.py +15 -0
  202. aria_code/apps/cli/prediction_feedback.py +202 -0
  203. aria_code/apps/cli/preflight.py +502 -0
  204. aria_code/apps/cli/project_aria.py +60 -0
  205. aria_code/apps/cli/prompt_assembly.py +78 -0
  206. aria_code/apps/cli/prompts/__init__.py +0 -0
  207. aria_code/apps/cli/prompts/coding.py +71 -0
  208. aria_code/apps/cli/prompts/select.py +193 -0
  209. aria_code/apps/cli/prompts/system_prompts.py +562 -0
  210. aria_code/apps/cli/prompts/ui.py +436 -0
  211. aria_code/apps/cli/provider_endpoints.py +237 -0
  212. aria_code/apps/cli/provider_keys.py +73 -0
  213. aria_code/apps/cli/providers/__init__.py +1 -0
  214. aria_code/apps/cli/providers/base.py +775 -0
  215. aria_code/apps/cli/providers/chat_routing.py +139 -0
  216. aria_code/apps/cli/providers/collaboration.py +98 -0
  217. aria_code/apps/cli/providers/llm/__init__.py +1 -0
  218. aria_code/apps/cli/providers/llm/offline_responses.py +68 -0
  219. aria_code/apps/cli/providers/llm/ollama_stream.py +1574 -0
  220. aria_code/apps/cli/providers/llm/response_cache.py +60 -0
  221. aria_code/apps/cli/providers/llm/sse_stream.py +298 -0
  222. aria_code/apps/cli/providers/runtime_bridge.py +586 -0
  223. aria_code/apps/cli/providers/vertexai_stream.py +592 -0
  224. aria_code/apps/cli/release_download.py +159 -0
  225. aria_code/apps/cli/review_runner.py +111 -0
  226. aria_code/apps/cli/runtime_consumer.py +769 -0
  227. aria_code/apps/cli/sandbox.py +32 -0
  228. aria_code/apps/cli/session_export.py +120 -0
  229. aria_code/apps/cli/session_jsonl.py +208 -0
  230. aria_code/apps/cli/session_refresh.py +123 -0
  231. aria_code/apps/cli/session_store.py +125 -0
  232. aria_code/apps/cli/skills_catalog.py +273 -0
  233. aria_code/apps/cli/task_isolation.py +211 -0
  234. aria_code/apps/cli/terminal_chrome.py +412 -0
  235. aria_code/apps/cli/todo_tracker.py +190 -0
  236. aria_code/apps/cli/tool_executor.py +932 -0
  237. aria_code/apps/cli/tool_registry.py +34 -0
  238. aria_code/apps/cli/tool_scope.py +113 -0
  239. aria_code/apps/cli/tools/__init__.py +40 -0
  240. aria_code/apps/cli/tools/context.py +46 -0
  241. aria_code/apps/cli/tools/file_tools.py +181 -0
  242. aria_code/apps/cli/tools/market_tools.py +626 -0
  243. aria_code/apps/cli/tools/notebook_tools.py +111 -0
  244. aria_code/apps/cli/tools/system_tools.py +656 -0
  245. aria_code/apps/cli/tools/write_tools.py +961 -0
  246. aria_code/apps/cli/tradingview_bridge.py +529 -0
  247. aria_code/apps/cli/transactions.py +269 -0
  248. aria_code/apps/cli/turn_planning.py +137 -0
  249. aria_code/apps/cli/update_check.py +257 -0
  250. aria_code/apps/cli/updater.py +373 -0
  251. aria_code/apps/cli/utils/__init__.py +0 -0
  252. aria_code/apps/cli/utils/market_detect.py +1663 -0
  253. aria_code/apps/cli/web_login.py +136 -0
  254. aria_code/apps/cli/workflow_runner.py +127 -0
  255. aria_code/apps/cli/workspace_route.py +121 -0
  256. aria_code/apps/daemon/README.md +14 -0
  257. aria_code/apps/vscode/README.md +115 -0
  258. aria_code/apps/vscode/package.json +70 -0
  259. aria_code/aria_cli.py +5244 -0
  260. aria_code/aria_daemon.py +1372 -0
  261. aria_code/aria_feishu_bot.py +16 -0
  262. aria_code/aria_mcp_server.py +61 -0
  263. aria_code/aria_relay_client.py +4 -0
  264. aria_code/aria_telegram_bot.py +11 -0
  265. aria_code/ariarc.py +385 -0
  266. aria_code/artifacts.py +581 -0
  267. aria_code/backtest_engine.py +4 -0
  268. aria_code/backtest_report.py +4 -0
  269. aria_code/brokers/__init__.py +78 -0
  270. aria_code/brokers/automation.py +193 -0
  271. aria_code/brokers/base.py +231 -0
  272. aria_code/brokers/capabilities.py +264 -0
  273. aria_code/brokers/cn/__init__.py +10 -0
  274. aria_code/brokers/cn/easytrader_broker.py +193 -0
  275. aria_code/brokers/cn/futu_broker.py +194 -0
  276. aria_code/brokers/cn/longbridge_broker.py +190 -0
  277. aria_code/brokers/cn/tiger_broker.py +196 -0
  278. aria_code/brokers/cn/xtquant_broker.py +225 -0
  279. aria_code/brokers/config.py +414 -0
  280. aria_code/brokers/intl/__init__.py +5 -0
  281. aria_code/brokers/intl/alpaca_broker.py +183 -0
  282. aria_code/brokers/intl/ibkr_broker.py +215 -0
  283. aria_code/brokers/intl/trading212_broker.py +183 -0
  284. aria_code/brokers/intl/webull_broker.py +156 -0
  285. aria_code/brokers/paper_broker.py +266 -0
  286. aria_code/brokers/planning.py +296 -0
  287. aria_code/brokers/registry.py +213 -0
  288. aria_code/brokers/trading.py +267 -0
  289. aria_code/canva_client.py +4 -0
  290. aria_code/change_store.py +127 -0
  291. aria_code/clients/__init__.py +0 -0
  292. aria_code/clients/aliyun_data_client.py +660 -0
  293. aria_code/clients/aria_relay_client.py +16 -0
  294. aria_code/clients/canva_client.py +284 -0
  295. aria_code/clients/figma_client.py +100 -0
  296. aria_code/clients/football_data_client.py +1671 -0
  297. aria_code/clients/kling_video_client.py +182 -0
  298. aria_code/clients/market_data_client.py +1985 -0
  299. aria_code/clients/openai_image_client.py +242 -0
  300. aria_code/clients/runway_video_client.py +157 -0
  301. aria_code/clients/warehouse_erp_client.py +105 -0
  302. aria_code/cloud_memory.py +190 -0
  303. aria_code/command_safety.py +19 -0
  304. aria_code/computer_use_server.py +276 -0
  305. aria_code/computer_use_tools.py +4 -0
  306. aria_code/dashboard_generator.py +579 -0
  307. aria_code/data_analysis_tools.py +4 -0
  308. aria_code/data_cleaner.py +483 -0
  309. aria_code/data_service.py +513 -0
  310. aria_code/datasources/__init__.py +23 -0
  311. aria_code/datasources/base.py +166 -0
  312. aria_code/datasources/router.py +253 -0
  313. aria_code/datasources/sec_statements.py +217 -0
  314. aria_code/datasources/sources/__init__.py +15 -0
  315. aria_code/datasources/sources/akshare_source.py +269 -0
  316. aria_code/datasources/sources/alpha_vantage_source.py +203 -0
  317. aria_code/datasources/sources/edgar_source.py +219 -0
  318. aria_code/datasources/sources/finnhub_source.py +198 -0
  319. aria_code/datasources/sources/fred_source.py +223 -0
  320. aria_code/datasources/sources/tushare_source.py +143 -0
  321. aria_code/datasources/sources/web_scraper_source.py +278 -0
  322. aria_code/datasources/sources/world_bank_source.py +206 -0
  323. aria_code/datasources/sources/yfinance_source.py +152 -0
  324. aria_code/doctor.py +860 -0
  325. aria_code/domain/__init__.py +0 -0
  326. aria_code/domain/_finance_core/__init__.py +5 -0
  327. aria_code/domain/_finance_core/backtest_engine.py +325 -0
  328. aria_code/domain/_finance_core/performance_stats.py +101 -0
  329. aria_code/domain/backtest_engine.py +75 -0
  330. aria_code/domain/backtest_report.py +488 -0
  331. aria_code/domain/finance_formulas.py +389 -0
  332. aria_code/domain/performance_stats.py +6 -0
  333. aria_code/domain/portfolio_ledger.py +307 -0
  334. aria_code/domain/strategy_vault.py +421 -0
  335. aria_code/evals/__init__.py +37 -0
  336. aria_code/evals/behavior.py +113 -0
  337. aria_code/evals/gap.py +93 -0
  338. aria_code/evals/harness.py +949 -0
  339. aria_code/evals/review_probe.py +132 -0
  340. aria_code/evals/runner.py +425 -0
  341. aria_code/evals/scorecard.py +322 -0
  342. aria_code/evals/summary.py +125 -0
  343. aria_code/external_agent_runner.py +115 -0
  344. aria_code/figma_client.py +4 -0
  345. aria_code/file_analysis_tools.py +4 -0
  346. aria_code/finance_formulas.py +4 -0
  347. aria_code/football_data_client.py +4 -0
  348. aria_code/image_gen_tools.py +4 -0
  349. aria_code/intent_classifier.py +459 -0
  350. aria_code/kling_video_client.py +4 -0
  351. aria_code/licensing.py +151 -0
  352. aria_code/local_finance_tools.py +4 -0
  353. aria_code/local_image_provider.py +4 -0
  354. aria_code/local_llm_provider.py +4 -0
  355. aria_code/macro_tools.py +4 -0
  356. aria_code/markdown_pdf.py +289 -0
  357. aria_code/market_data_client.py +4 -0
  358. aria_code/mcp_client.py +744 -0
  359. aria_code/memory_manager.py +273 -0
  360. aria_code/model_capability.py +512 -0
  361. aria_code/notification_tools.py +4 -0
  362. aria_code/openai_image_client.py +4 -0
  363. aria_code/packages/__init__.py +26 -0
  364. aria_code/packages/adk_bridge/__init__.py +11 -0
  365. aria_code/packages/adk_bridge/code_review_tools.py +49 -0
  366. aria_code/packages/adk_bridge/market_tools.py +125 -0
  367. aria_code/packages/aria_agents/__init__.py +5 -0
  368. aria_code/packages/aria_agents/manifest.py +78 -0
  369. aria_code/packages/aria_core/__init__.py +34 -0
  370. aria_code/packages/aria_core/architecture.py +269 -0
  371. aria_code/packages/aria_core/export.py +147 -0
  372. aria_code/packages/aria_core/manifest.py +65 -0
  373. aria_code/packages/aria_core/paths.py +57 -0
  374. aria_code/packages/aria_core/secure_file.py +68 -0
  375. aria_code/packages/aria_infra/__init__.py +15 -0
  376. aria_code/packages/aria_infra/arthera.py +58 -0
  377. aria_code/packages/aria_infra/doctor.py +246 -0
  378. aria_code/packages/aria_infra/product.py +40 -0
  379. aria_code/packages/aria_mcp/__init__.py +25 -0
  380. aria_code/packages/aria_mcp/bridge.py +232 -0
  381. aria_code/packages/aria_mcp/circuit.py +82 -0
  382. aria_code/packages/aria_mcp/config.py +162 -0
  383. aria_code/packages/aria_mcp/installer.py +95 -0
  384. aria_code/packages/aria_mcp/server.py +1705 -0
  385. aria_code/packages/aria_mcp/tools.py +81 -0
  386. aria_code/packages/aria_sdk/__init__.py +19 -0
  387. aria_code/packages/aria_sdk/client.py +393 -0
  388. aria_code/packages/aria_sdk/providers.py +70 -0
  389. aria_code/packages/aria_sdk/streaming.py +75 -0
  390. aria_code/packages/aria_sdk/types.py +86 -0
  391. aria_code/packages/aria_services/__init__.py +92 -0
  392. aria_code/packages/aria_services/cache.py +60 -0
  393. aria_code/packages/aria_services/context.py +356 -0
  394. aria_code/packages/aria_services/context_checkpoints.py +157 -0
  395. aria_code/packages/aria_services/data.py +11 -0
  396. aria_code/packages/aria_services/provider_health.py +304 -0
  397. aria_code/packages/aria_services/references.py +247 -0
  398. aria_code/packages/aria_services/registry.py +249 -0
  399. aria_code/packages/aria_services/research_protocol.py +137 -0
  400. aria_code/packages/aria_services/research_quality.py +186 -0
  401. aria_code/packages/aria_services/settings.py +160 -0
  402. aria_code/packages/aria_services/usage.py +151 -0
  403. aria_code/packages/aria_skills/__init__.py +46 -0
  404. aria_code/packages/aria_skills/catalog.py +149 -0
  405. aria_code/packages/aria_skills/loader.py +455 -0
  406. aria_code/packages/aria_skills/registry.py +84 -0
  407. aria_code/packages/aria_tools/__init__.py +5 -0
  408. aria_code/packages/aria_tools/financial/__init__.py +5 -0
  409. aria_code/packages/aria_tools/financial/_paths.py +92 -0
  410. aria_code/packages/aria_tools/registry.py +129 -0
  411. aria_code/packages/quant_engine/__init__.py +33 -0
  412. aria_code/packages/quant_engine/_namespace.py +24 -0
  413. aria_code/packages/quant_engine/backtest/__init__.py +10 -0
  414. aria_code/packages/quant_engine/backtest/core.py +53 -0
  415. aria_code/packages/quant_engine/backtest/engine.py +102 -0
  416. aria_code/packages/quant_engine/portfolio/__init__.py +9 -0
  417. aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
  418. aria_code/packages/quant_engine/sports/__init__.py +76 -0
  419. aria_code/packages/quant_engine/sports/calibrator.py +354 -0
  420. aria_code/packages/quant_engine/sports/dixon_coles.py +234 -0
  421. aria_code/packages/quant_engine/sports/elo.py +300 -0
  422. aria_code/packages/quant_engine/sports/form.py +188 -0
  423. aria_code/packages/quant_engine/sports/h2h.py +195 -0
  424. aria_code/packages/quant_engine/sports/predictor.py +312 -0
  425. aria_code/packages/quant_engine/sports/tracker.py +665 -0
  426. aria_code/packages/quant_engine/stochastic/__init__.py +31 -0
  427. aria_code/packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
  428. aria_code/packages/quant_engine/stochastic/ito_calculus.py +477 -0
  429. aria_code/packages/quant_engine/stochastic/kelly_criterion.py +181 -0
  430. aria_code/packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
  431. aria_code/packages/quant_engine/stochastic/options_pricing.py +576 -0
  432. aria_code/packages/quant_engine/stochastic/stochastic_processes.py +90 -0
  433. aria_code/packs/__init__.py +85 -0
  434. aria_code/packs/base.py +168 -0
  435. aria_code/packs/finance/__init__.py +250 -0
  436. aria_code/packs/healthcare/__init__.py +40 -0
  437. aria_code/packs/logistics/__init__.py +171 -0
  438. aria_code/packs/payments/__init__.py +139 -0
  439. aria_code/packs/realty/__init__.py +190 -0
  440. aria_code/packs/registry.py +202 -0
  441. aria_code/packs/rules.py +90 -0
  442. aria_code/plan_utils.py +194 -0
  443. aria_code/plugin_loader.py +328 -0
  444. aria_code/portfolio_ledger.py +4 -0
  445. aria_code/preview_server.py +449 -0
  446. aria_code/privacy/__init__.py +5 -0
  447. aria_code/privacy/feedback.py +123 -0
  448. aria_code/project_review.py +278 -0
  449. aria_code/project_review_api.py +235 -0
  450. aria_code/project_review_server.py +18 -0
  451. aria_code/project_tools.py +4 -0
  452. aria_code/providers/__init__.py +30 -0
  453. aria_code/providers/google_video_provider.py +256 -0
  454. aria_code/providers/llm/__init__.py +19 -0
  455. aria_code/providers/llm/anthropic.py +184 -0
  456. aria_code/providers/llm/arthera_gateway.py +90 -0
  457. aria_code/providers/llm/autoconfig.py +213 -0
  458. aria_code/providers/llm/base.py +139 -0
  459. aria_code/providers/llm/custom.py +158 -0
  460. aria_code/providers/llm/ollama.py +137 -0
  461. aria_code/providers/llm/openai_compat.py +350 -0
  462. aria_code/providers/llm/registry.py +457 -0
  463. aria_code/providers/local_image_provider.py +202 -0
  464. aria_code/providers/local_llm_provider.py +549 -0
  465. aria_code/providers/ollama_models.py +79 -0
  466. aria_code/providers/openai_params.py +32 -0
  467. aria_code/providers/tool_messages.py +100 -0
  468. aria_code/realty_data_tools.py +4 -0
  469. aria_code/report_exporters.py +150 -0
  470. aria_code/report_generator.py +1764 -0
  471. aria_code/review_service.py +497 -0
  472. aria_code/runtime/__init__.py +176 -0
  473. aria_code/runtime/acceptance.py +632 -0
  474. aria_code/runtime/agent_loop.py +1851 -0
  475. aria_code/runtime/agent_profiles.py +75 -0
  476. aria_code/runtime/approval.py +71 -0
  477. aria_code/runtime/baseline.py +219 -0
  478. aria_code/runtime/budget.py +232 -0
  479. aria_code/runtime/checkpoints.py +481 -0
  480. aria_code/runtime/consult.py +65 -0
  481. aria_code/runtime/contract.py +282 -0
  482. aria_code/runtime/delivery.py +432 -0
  483. aria_code/runtime/events.py +113 -0
  484. aria_code/runtime/gateway.py +167 -0
  485. aria_code/runtime/lsp.py +484 -0
  486. aria_code/runtime/native_index.py +153 -0
  487. aria_code/runtime/orchestrator.py +55 -0
  488. aria_code/runtime/processes.py +343 -0
  489. aria_code/runtime/project_graph.py +614 -0
  490. aria_code/runtime/repo_map.py +712 -0
  491. aria_code/runtime/research_bundle.py +105 -0
  492. aria_code/runtime/review.py +269 -0
  493. aria_code/runtime/run_state.py +127 -0
  494. aria_code/runtime/run_store.py +522 -0
  495. aria_code/runtime/scoped_runs.py +113 -0
  496. aria_code/runtime/self_healing.py +422 -0
  497. aria_code/runtime/semantic_diff.py +171 -0
  498. aria_code/runtime/subagent.py +721 -0
  499. aria_code/runtime/symbol_edit.py +117 -0
  500. aria_code/runtime/task_brief.py +46 -0
  501. aria_code/runtime/task_context.py +145 -0
  502. aria_code/runtime/task_ledger.py +108 -0
  503. aria_code/runtime/task_orchestrator.py +90 -0
  504. aria_code/runtime/task_outcome.py +33 -0
  505. aria_code/runtime/task_verification.py +266 -0
  506. aria_code/runtime/task_worktree.py +448 -0
  507. aria_code/runtime/tool_executor.py +329 -0
  508. aria_code/runtime/tool_policy.py +107 -0
  509. aria_code/runtime/transactions.py +216 -0
  510. aria_code/runtime/windows_job.py +115 -0
  511. aria_code/runtime/workflows.py +222 -0
  512. aria_code/runtime/worktrees.py +215 -0
  513. aria_code/runway_video_client.py +4 -0
  514. aria_code/safety/__init__.py +27 -0
  515. aria_code/safety/client_isolation.py +178 -0
  516. aria_code/safety/permissions.py +339 -0
  517. aria_code/safety/risk.py +699 -0
  518. aria_code/safety/sandbox.py +148 -0
  519. aria_code/safety/service.py +102 -0
  520. aria_code/service_endpoints.py +14 -0
  521. aria_code/setup_wizard.py +684 -0
  522. aria_code/spreadsheet_tools.py +4 -0
  523. aria_code/strategy_vault.py +4 -0
  524. aria_code/tools/__init__.py +0 -0
  525. aria_code/tools/artifact_tools.py +252 -0
  526. aria_code/tools/broker_tools.py +92 -0
  527. aria_code/tools/build_quant_engine.py +97 -0
  528. aria_code/tools/code_audit_tools.py +192 -0
  529. aria_code/tools/computer_use_tools.py +509 -0
  530. aria_code/tools/data_analysis_tools.py +831 -0
  531. aria_code/tools/enterprise_finance_tools.py +202 -0
  532. aria_code/tools/extended_tools.py +240 -0
  533. aria_code/tools/file_analysis_tools.py +880 -0
  534. aria_code/tools/image_gen_tools.py +114 -0
  535. aria_code/tools/local_finance_tools.py +3488 -0
  536. aria_code/tools/logistics_carriers.py +11 -0
  537. aria_code/tools/logistics_inventory.py +11 -0
  538. aria_code/tools/logistics_io.py +11 -0
  539. aria_code/tools/logistics_tenancy.py +11 -0
  540. aria_code/tools/logistics_tools.py +11 -0
  541. aria_code/tools/macro_tools.py +412 -0
  542. aria_code/tools/notification_tools.py +249 -0
  543. aria_code/tools/project_tools.py +525 -0
  544. aria_code/tools/realty_data_tools.py +662 -0
  545. aria_code/tools/spreadsheet_tools.py +399 -0
  546. aria_code/tools/stripe_tools.py +115 -0
  547. aria_code/tools/video_tools.py +500 -0
  548. aria_code/training/__init__.py +1 -0
  549. aria_code/training/compare.py +117 -0
  550. aria_code/training/prelude_commands.py +111 -0
  551. aria_code/training/prelude_eval.py +390 -0
  552. aria_code/training/prelude_real.py +183 -0
  553. aria_code/training/prelude_train.py +248 -0
  554. aria_code/training/sft_dataset.py +230 -0
  555. aria_code/training/trajectory_dataset.py +263 -0
  556. aria_code/ui/__init__.py +105 -0
  557. aria_code/ui/banner.py +404 -0
  558. aria_code/ui/completer.py +571 -0
  559. aria_code/ui/console.py +285 -0
  560. aria_code/ui/input_box.py +505 -0
  561. aria_code/ui/mentions.py +206 -0
  562. aria_code/ui/picker.py +248 -0
  563. aria_code/ui/render/__init__.py +11 -0
  564. aria_code/ui/render/actions.py +297 -0
  565. aria_code/ui/render/finance.py +1732 -0
  566. aria_code/ui/render/market.py +253 -0
  567. aria_code/ui/render/output.py +954 -0
  568. aria_code/ui/render/responsive.py +75 -0
  569. aria_code/ui/render/team.py +367 -0
  570. aria_code/ui/robot.py +239 -0
  571. aria_code/ui/startup_dashboard.py +110 -0
  572. aria_code/video_analysis.py +147 -0
  573. aria_code/video_editor.py +518 -0
  574. aria_code/video_workflow/__init__.py +8 -0
  575. aria_code/video_workflow/__main__.py +61 -0
  576. aria_code/video_workflow/engines/__init__.py +18 -0
  577. aria_code/video_workflow/engines/base.py +69 -0
  578. aria_code/video_workflow/engines/custom_model.py +204 -0
  579. aria_code/video_workflow/engines/google_veo.py +94 -0
  580. aria_code/video_workflow/engines/local_engine.py +71 -0
  581. aria_code/video_workflow/engines/registry.py +98 -0
  582. aria_code/video_workflow/google_models.py +278 -0
  583. aria_code/video_workflow/music/__init__.py +5 -0
  584. aria_code/video_workflow/music/foley_service.py +132 -0
  585. aria_code/video_workflow/music/service.py +212 -0
  586. aria_code/video_workflow/pipeline.py +620 -0
  587. aria_code/video_workflow/plan.py +202 -0
  588. aria_code/video_workflow/session.py +297 -0
  589. aria_code/workspace/__init__.py +6 -0
  590. aria_code/workspace/files.py +219 -0
  591. aria_code/workspace/verify.py +167 -0
  592. aria_code/workspace_changes.py +199 -0
  593. aria_code/workspace_service.py +196 -0
  594. aria_code/workspace_tools.py +150 -0
  595. aria_code-0.167.0.dist-info/METADATA +330 -0
  596. aria_code-0.167.0.dist-info/RECORD +601 -0
  597. aria_code-0.167.0.dist-info/WHEEL +5 -0
  598. aria_code-0.167.0.dist-info/entry_points.txt +5 -0
  599. aria_code-0.167.0.dist-info/licenses/LICENSE +201 -0
  600. aria_code-0.167.0.dist-info/licenses/NOTICE +49 -0
  601. aria_code-0.167.0.dist-info/top_level.txt +2 -0
@@ -0,0 +1,31 @@
1
+ """
2
+ Arthera Stochastic Calculus Module
3
+ 随机微积分模块
4
+
5
+ Components:
6
+ - ito_calculus.py : 伊藤引理 / 伊藤积分 / 随机微分方程
7
+ - gbm_enhanced.py : 增强几何布朗运动(多资产/跳跃扩散/随机波动率)
8
+ - stochastic_processes : OU / CIR / Vasicek / Hull-White 过程
9
+ - monte_carlo_advanced : 方差缩减蒙特卡罗(Antithetic/Control/Quasi-MC)
10
+ - kelly_criterion : 凯利公式(连续时间 / 多资产 / Robust版本)
11
+ """
12
+
13
+ from .ito_calculus import ItoCalculus, ItoProcess, apply_ito_lemma
14
+ from .gbm_enhanced import EnhancedGBM
15
+ from .stochastic_processes import (
16
+ OrnsteinUhlenbeck, CIRProcess, VasicekModel, HullWhiteModel
17
+ )
18
+ from .monte_carlo_advanced import MonteCarloEngine, VarianceReduction
19
+ from .kelly_criterion import KellyCriterion
20
+
21
+ __all__ = [
22
+ "ItoCalculus", "ItoProcess", "apply_ito_lemma",
23
+ "EnhancedGBM",
24
+ "OrnsteinUhlenbeck", "CIRProcess", "VasicekModel", "HullWhiteModel",
25
+ "MonteCarloEngine", "VarianceReduction",
26
+ "KellyCriterion",
27
+ ]
28
+
29
+ # Optional explicit backend modules never override this distribution.
30
+ from .._namespace import extend as _extend_namespace
31
+ _extend_namespace(__path__, "stochastic")
@@ -0,0 +1,195 @@
1
+ """
2
+ 增强型几何布朗运动 (Enhanced Geometric Brownian Motion)
3
+
4
+ 包含:
5
+ 1. 多资产相关 GBM (Multi-asset GBM with Cholesky Decomposition)
6
+ 2. 默顿跳跃扩散模型 (Merton Jump-Diffusion Model)
7
+ 3. Heston 随机波动率模型 (Heston Stochastic Volatility)
8
+ 4. 分数布朗运动 (Fractional Brownian Motion) - 选配
9
+ """
10
+
11
+ import numpy as np
12
+ import pandas as pd
13
+ from typing import Optional, Tuple, List, Union
14
+
15
+
16
+ class EnhancedGBM:
17
+ """
18
+ 增强型 GBM 仿真器
19
+ """
20
+
21
+ @staticmethod
22
+ def simulate_multi_asset(
23
+ s0: np.ndarray,
24
+ mu: np.ndarray,
25
+ sigma: np.ndarray,
26
+ corr_matrix: np.ndarray,
27
+ T: float = 1.0,
28
+ n_steps: int = 252,
29
+ n_paths: int = 1000,
30
+ seed: Optional[int] = None
31
+ ) -> np.ndarray:
32
+ """
33
+ 多资产相关 GBM 模拟
34
+
35
+ 使用 Cholesky 分解生成相关随机增量:
36
+ dS_i = μ_i S_i dt + σ_i S_i dW_i
37
+ E[dW_i dW_j] = ρ_{ij} dt
38
+
39
+ Args:
40
+ s0 : 初始价格向量, shape (n_assets,)
41
+ mu : 年化漂移率向量
42
+ sigma : 年化波动率向量
43
+ corr_matrix : 相关系数矩阵, shape (n_assets, n_assets)
44
+ T : 终止时间(年)
45
+ n_steps : 时间步数
46
+ n_paths : 路径数
47
+ seed : 随机种子
48
+
49
+ Returns:
50
+ paths: shape (n_paths, n_assets, n_steps + 1)
51
+ """
52
+ if seed is not None:
53
+ np.random.seed(seed)
54
+
55
+ n_assets = len(s0)
56
+ dt = T / n_steps
57
+
58
+ # Cholesky 分解: L @ L.T = Correlation
59
+ L = np.linalg.cholesky(corr_matrix)
60
+
61
+ paths = np.zeros((n_paths, n_assets, n_steps + 1))
62
+ paths[:, :, 0] = s0
63
+
64
+ for i in range(n_steps):
65
+ # 生成独立标准正态分布 Z ~ N(0, 1)
66
+ Z = np.random.standard_normal((n_paths, n_assets))
67
+
68
+ # 转化为相关增量 ε = Z @ L.T
69
+ epsilon = Z @ L.T
70
+
71
+ # 伊藤漂移修正
72
+ drift = (mu - 0.5 * sigma**2) * dt
73
+ diffusion = sigma * np.sqrt(dt) * epsilon
74
+
75
+ # 更新价格: S_{t+dt} = S_t * exp(drift + diffusion)
76
+ paths[:, :, i + 1] = paths[:, :, i] * np.exp(drift + diffusion)
77
+
78
+ return paths
79
+
80
+ @staticmethod
81
+ def simulate_merton_jump_diffusion(
82
+ s0: float,
83
+ mu: float,
84
+ sigma: float,
85
+ lambda_j: float,
86
+ mu_j: float,
87
+ sigma_j: float,
88
+ T: float = 1.0,
89
+ n_steps: int = 252,
90
+ n_paths: int = 1000,
91
+ seed: Optional[int] = None
92
+ ) -> np.ndarray:
93
+ """
94
+ 默顿跳跃扩散模型 (Merton Jump-Diffusion)
95
+
96
+ dS/S = (μ - λk)dt + σdW + (J - 1)dN
97
+ 其中 N 是强度为 λ 的泊松过程,ln(J) ~ N(μ_j, σ_j²)
98
+ k = E[J - 1] = exp(μ_j + 0.5σ_j²) - 1
99
+
100
+ Args:
101
+ lambda_j: 跳跃强度(每年平均跳跃次数)
102
+ mu_j : 跳跃幅度的对数均值
103
+ sigma_j : 跳跃幅度的对数标准差
104
+ """
105
+ if seed is not None:
106
+ np.random.seed(seed)
107
+
108
+ dt = T / n_steps
109
+ k = np.exp(mu_j + 0.5 * sigma_j**2) - 1
110
+
111
+ paths = np.zeros((n_paths, n_steps + 1))
112
+ paths[:, 0] = s0
113
+
114
+ for i in range(n_steps):
115
+ # 扩散部分 (Diffusion)
116
+ Z = np.random.standard_normal(n_paths)
117
+ drift = (mu - 0.5 * sigma**2 - lambda_j * k) * dt
118
+ diffusion = sigma * np.sqrt(dt) * Z
119
+
120
+ # 跳跃部分 (Jump)
121
+ # 1. 产生泊松分布的跳跃次数 (通常 dt 很小,N 主要是 0 或 1)
122
+ N = np.random.poisson(lambda_j * dt, n_paths)
123
+
124
+ # 2. 对每个跳跃计算幅度
125
+ jump_factor = np.ones(n_paths)
126
+ for path_idx in range(n_paths):
127
+ if N[path_idx] > 0:
128
+ # 总跳跃幅度 = Σ ln(J_i)
129
+ total_jump_log = np.random.normal(mu_j, sigma_j, N[path_idx]).sum()
130
+ jump_factor[path_idx] = np.exp(total_jump_log)
131
+
132
+ # 更新价格
133
+ paths[:, i + 1] = paths[:, i] * np.exp(drift + diffusion) * jump_factor
134
+
135
+ return paths
136
+
137
+ @staticmethod
138
+ def simulate_heston(
139
+ s0: float,
140
+ v0: float,
141
+ mu: float,
142
+ kappa: float,
143
+ theta: float,
144
+ sigma_v: float,
145
+ rho: float,
146
+ T: float = 1.0,
147
+ n_steps: int = 252,
148
+ n_paths: int = 1000,
149
+ seed: Optional[int] = None
150
+ ) -> Tuple[np.ndarray, np.ndarray]:
151
+ """
152
+ Heston 随机波动率模型
153
+
154
+ dS = μS dt + √v S dW1
155
+ dv = κ(θ - v)dt + σ_v √v dW2
156
+ E[dW1 dW2] = ρ dt
157
+
158
+ Args:
159
+ v0 : 初始方差 (Initial Variance)
160
+ kappa : 均值回归速度 (Mean Reversion Speed)
161
+ theta : 长期平均方差 (Long-term Variance)
162
+ sigma_v: 波动率的波动率 (Vol of Vol)
163
+ rho : 价格与波动率的相关性
164
+ """
165
+ if seed is not None:
166
+ np.random.seed(seed)
167
+
168
+ dt = T / n_steps
169
+
170
+ s_paths = np.zeros((n_paths, n_steps + 1))
171
+ v_paths = np.zeros((n_paths, n_steps + 1))
172
+ s_paths[:, 0] = s0
173
+ v_paths[:, 0] = v0
174
+
175
+ for i in range(n_steps):
176
+ # 生成相关随机变量
177
+ Z1 = np.random.standard_normal(n_paths)
178
+ Z2 = np.random.standard_normal(n_paths)
179
+ W1 = Z1
180
+ W2 = rho * Z1 + np.sqrt(1 - rho**2) * Z2
181
+
182
+ v_curr = v_paths[:, i]
183
+ # 保证方差非负 (使用 Full Truncation 或 Reflection)
184
+ v_plus = np.maximum(v_curr, 0)
185
+
186
+ # 更新方差 (Euler-Maruyama)
187
+ v_paths[:, i + 1] = v_curr + kappa * (theta - v_plus) * dt + \
188
+ sigma_v * np.sqrt(v_plus * dt) * W2
189
+
190
+ # 更新价格
191
+ s_paths[:, i + 1] = s_paths[:, i] * np.exp(
192
+ (mu - 0.5 * v_plus) * dt + np.sqrt(v_plus * dt) * W1
193
+ )
194
+
195
+ return s_paths, v_paths
@@ -0,0 +1,477 @@
1
+ """
2
+ 伊藤积分与伊藤引理实现
3
+ Itô's Lemma and Itô Stochastic Integral
4
+
5
+ 理论基础:
6
+ 设 X(t) 为伊藤过程:dX = μ(X,t)dt + σ(X,t)dW
7
+ 对光滑函数 f(X,t),伊藤引理给出:
8
+ df = (∂f/∂t + μ·∂f/∂X + ½σ²·∂²f/∂X²)dt + σ·∂f/∂X·dW
9
+
10
+ 关键应用:
11
+ 1. GBM → S(t) = S₀·exp((μ - σ²/2)t + σW(t))
12
+ 2. Black-Scholes PDE推导
13
+ 3. Feynman-Kac定理(PDE ↔ 期望)
14
+ 4. 测度变换(Girsanov定理)
15
+ """
16
+
17
+ from __future__ import annotations
18
+
19
+ import numpy as np
20
+ import pandas as pd
21
+ from dataclasses import dataclass, field
22
+ from typing import Callable, Optional, Tuple, List
23
+ import warnings
24
+
25
+
26
+ @dataclass
27
+ class ItoProcess:
28
+ """
29
+ 伊藤过程:dX = μ(X,t)dt + σ(X,t)dW
30
+
31
+ 参数:
32
+ mu_func : 漂移系数函数 μ(x, t) -> float
33
+ sigma_func: 扩散系数函数 σ(x, t) -> float
34
+ x0 : 初始值
35
+ name : 过程名称
36
+ """
37
+ mu_func: Callable[[float, float], float]
38
+ sigma_func: Callable[[float, float], float]
39
+ x0: float = 1.0
40
+ name: str = "ItoProcess"
41
+
42
+ def simulate_path(
43
+ self,
44
+ T: float = 1.0,
45
+ n_steps: int = 252,
46
+ seed: Optional[int] = None,
47
+ method: str = "euler_maruyama"
48
+ ) -> Tuple[np.ndarray, np.ndarray]:
49
+ """
50
+ 模拟伊藤过程路径
51
+
52
+ Args:
53
+ T : 终止时间(年)
54
+ n_steps: 时间步数
55
+ seed : 随机种子
56
+ method : "euler_maruyama"(一阶)或 "milstein"(带二阶修正)
57
+
58
+ Returns:
59
+ (t_grid, X_path): 时间网格和过程路径
60
+ """
61
+ if seed is not None:
62
+ np.random.seed(seed)
63
+
64
+ dt = T / n_steps
65
+ sqrt_dt = np.sqrt(dt)
66
+ t_grid = np.linspace(0, T, n_steps + 1)
67
+ X = np.zeros(n_steps + 1)
68
+ X[0] = self.x0
69
+
70
+ dW = np.random.normal(0, sqrt_dt, n_steps)
71
+
72
+ for i in range(n_steps):
73
+ x_i = X[i]
74
+ t_i = t_grid[i]
75
+ mu_i = self.mu_func(x_i, t_i)
76
+ sigma_i = self.sigma_func(x_i, t_i)
77
+
78
+ if method == "euler_maruyama":
79
+ # Euler-Maruyama:X_{n+1} = X_n + μ·Δt + σ·ΔW
80
+ X[i + 1] = x_i + mu_i * dt + sigma_i * dW[i]
81
+
82
+ elif method == "milstein":
83
+ # Milstein 方法:加入二阶修正项 ½σ·σ'·((ΔW)² - Δt)
84
+ # σ' ≈ (σ(x+δ,t) - σ(x-δ,t)) / 2δ(数值微分)
85
+ delta = x_i * 1e-5 if x_i != 0 else 1e-5
86
+ sigma_prime = (self.sigma_func(x_i + delta, t_i) -
87
+ self.sigma_func(x_i - delta, t_i)) / (2 * delta)
88
+ milstein_correction = 0.5 * sigma_i * sigma_prime * (dW[i] ** 2 - dt)
89
+ X[i + 1] = x_i + mu_i * dt + sigma_i * dW[i] + milstein_correction
90
+
91
+ else:
92
+ raise ValueError(f"Unknown method: {method}")
93
+
94
+ return t_grid, X
95
+
96
+ def simulate_ensemble(
97
+ self,
98
+ T: float = 1.0,
99
+ n_steps: int = 252,
100
+ n_paths: int = 1000,
101
+ method: str = "euler_maruyama",
102
+ antithetic: bool = False,
103
+ ) -> np.ndarray:
104
+ """
105
+ 模拟多条路径(支持对偶变量方差缩减)
106
+
107
+ Returns:
108
+ paths: shape (n_paths, n_steps+1)
109
+ """
110
+ dt = T / n_steps
111
+ sqrt_dt = np.sqrt(dt)
112
+ paths = np.zeros((n_paths, n_steps + 1))
113
+ paths[:, 0] = self.x0
114
+
115
+ if antithetic:
116
+ half = n_paths // 2
117
+ dW_base = np.random.normal(0, sqrt_dt, (half, n_steps))
118
+ dW_all = np.vstack([dW_base, -dW_base]) # 对偶变量
119
+ else:
120
+ dW_all = np.random.normal(0, sqrt_dt, (n_paths, n_steps))
121
+
122
+ t_grid = np.linspace(0, T, n_steps + 1)
123
+
124
+ for i in range(n_steps):
125
+ x_i = paths[:, i]
126
+ t_i = t_grid[i]
127
+ mu_i = np.vectorize(self.mu_func)(x_i, t_i)
128
+ sigma_i = np.vectorize(self.sigma_func)(x_i, t_i)
129
+
130
+ if method == "euler_maruyama":
131
+ paths[:, i + 1] = x_i + mu_i * dt + sigma_i * dW_all[:, i]
132
+ elif method == "milstein":
133
+ delta = np.where(x_i != 0, np.abs(x_i) * 1e-5, 1e-5)
134
+ sig_p = np.vectorize(self.sigma_func)(x_i + delta, t_i)
135
+ sig_m = np.vectorize(self.sigma_func)(x_i - delta, t_i)
136
+ sigma_prime = (sig_p - sig_m) / (2 * delta)
137
+ corr = 0.5 * sigma_i * sigma_prime * (dW_all[:, i] ** 2 - dt)
138
+ paths[:, i + 1] = x_i + mu_i * dt + sigma_i * dW_all[:, i] + corr
139
+
140
+ return paths
141
+
142
+
143
+ class ItoCalculus:
144
+ """
145
+ 伊藤微积分工具集
146
+
147
+ 提供:
148
+ 1. 伊藤积分(离散近似)
149
+ 2. 伊藤引理(数值验证)
150
+ 3. 测度变换(Girsanov)
151
+ 4. Feynman-Kac 期望定理
152
+ """
153
+
154
+ @staticmethod
155
+ def ito_integral(
156
+ integrand: np.ndarray,
157
+ brownian_increments: np.ndarray
158
+ ) -> float:
159
+ """
160
+ 伊藤积分:∫₀ᵀ f(t)dW(t) 的离散近似
161
+
162
+ 使用左端点 Riemann-Stieltjes 求和(伊藤型):
163
+ I = Σ f(t_i) · ΔW_i
164
+
165
+ 注意:与 Stratonovich 积分的区别在于使用左端点(非中点)
166
+
167
+ Args:
168
+ integrand : f(t_i) 在各时间节点的值,shape (n,)
169
+ brownian_increments: ΔW_i = W(t_{i+1}) - W(t_i),shape (n,)
170
+
171
+ Returns:
172
+ 伊藤积分近似值
173
+ """
174
+ if len(integrand) != len(brownian_increments):
175
+ raise ValueError("integrand and brownian_increments must have same length")
176
+ return float(np.sum(integrand * brownian_increments))
177
+
178
+ @staticmethod
179
+ def ito_isometry_check(
180
+ integrand: np.ndarray,
181
+ dt: float,
182
+ n_trials: int = 10000
183
+ ) -> dict:
184
+ """
185
+ 验证伊藤等距性:E[(∫f dW)²] = E[∫f² dt] = ∫E[f²]dt
186
+
187
+ Args:
188
+ integrand: 确定性被积函数(测试用)
189
+ dt : 时间步长
190
+ n_trials : 蒙特卡罗验证次数
191
+
192
+ Returns:
193
+ {'theoretical': float, 'empirical': float, 'error_pct': float}
194
+ """
195
+ n = len(integrand)
196
+ theoretical = float(np.sum(integrand ** 2) * dt)
197
+
198
+ squared_integrals = []
199
+ for _ in range(n_trials):
200
+ dW = np.random.normal(0, np.sqrt(dt), n)
201
+ I = np.sum(integrand * dW)
202
+ squared_integrals.append(I ** 2)
203
+
204
+ empirical = float(np.mean(squared_integrals))
205
+ error_pct = abs(empirical - theoretical) / max(abs(theoretical), 1e-10) * 100
206
+
207
+ return {
208
+ "theoretical": theoretical,
209
+ "empirical": empirical,
210
+ "error_pct": error_pct,
211
+ "isometry_verified": error_pct < 5.0
212
+ }
213
+
214
+ @staticmethod
215
+ def stratonovich_to_ito(
216
+ integrand: np.ndarray,
217
+ sigma_deriv: np.ndarray,
218
+ dt: float
219
+ ) -> np.ndarray:
220
+ """
221
+ Stratonovich 积分转化为伊藤积分
222
+
223
+ 关系:∫f ∘ dW = ∫f dW + ½∫f'·f dt
224
+ 修正项 = ½·σ·(∂σ/∂X)·dt(二次变差修正)
225
+
226
+ Args:
227
+ integrand : Stratonovich 被积函数 f(t_i)
228
+ sigma_deriv: ∂σ/∂X 在各时间节点的值
229
+ dt : 时间步长
230
+
231
+ Returns:
232
+ 等价的伊藤积分被积函数
233
+ """
234
+ ito_correction = 0.5 * integrand * sigma_deriv * dt
235
+ return integrand - ito_correction
236
+
237
+
238
+ def apply_ito_lemma(
239
+ f: Callable[[float, float], float],
240
+ df_dt: Callable[[float, float], float],
241
+ df_dx: Callable[[float, float], float],
242
+ d2f_dx2: Callable[[float, float], float],
243
+ mu: Callable[[float, float], float],
244
+ sigma: Callable[[float, float], float],
245
+ x_path: np.ndarray,
246
+ t_grid: np.ndarray,
247
+ ) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
248
+ """
249
+ 伊藤引理数值实现
250
+
251
+ 对伊藤过程 dX = μ(X,t)dt + σ(X,t)dW,
252
+ 函数 f(X,t) 满足:
253
+ df = [∂f/∂t + μ·∂f/∂X + ½σ²·∂²f/∂X²]dt + σ·∂f/∂X·dW
254
+ |-----------漂移项(ITO漂移)------------| |扩散项|
255
+
256
+ Args:
257
+ f : 目标函数 f(x, t)
258
+ df_dt : ∂f/∂t
259
+ df_dx : ∂f/∂X(一阶偏导)
260
+ d2f_dx2 : ∂²f/∂X²(二阶偏导)
261
+ mu : 漂移系数 μ(x, t)
262
+ sigma : 扩散系数 σ(x, t)
263
+ x_path : X 的路径,shape (n_steps+1,)
264
+ t_grid : 时间网格,shape (n_steps+1,)
265
+
266
+ Returns:
267
+ (f_path, drift_components, diffusion_components)
268
+ f_path: f(X(t)) 的路径
269
+ drift : 漂移项 dt 系数在各时间点的值
270
+ diffusion: 扩散项 dW 系数在各时间点的值
271
+
272
+ 示例(GBM 验证):
273
+ X = S(股价),f = ln(S)
274
+ μ_lnS = μ - σ²/2 ← 伊藤修正项 -σ²/2
275
+ σ_lnS = σ
276
+ → ln S(T) = ln S(0) + (μ - σ²/2)T + σW(T) ✓
277
+ """
278
+ n = len(x_path)
279
+ f_path = np.zeros(n)
280
+ drift_components = np.zeros(n)
281
+ diffusion_components = np.zeros(n)
282
+
283
+ for i in range(n):
284
+ x_i = x_path[i]
285
+ t_i = t_grid[i]
286
+ f_path[i] = f(x_i, t_i)
287
+
288
+ mu_i = mu(x_i, t_i)
289
+ sigma_i = sigma(x_i, t_i)
290
+
291
+ # 伊藤漂移:∂f/∂t + μ·∂f/∂X + ½σ²·∂²f/∂X²
292
+ drift_components[i] = (
293
+ df_dt(x_i, t_i)
294
+ + mu_i * df_dx(x_i, t_i)
295
+ + 0.5 * sigma_i ** 2 * d2f_dx2(x_i, t_i)
296
+ )
297
+ # 扩散项:σ·∂f/∂X
298
+ diffusion_components[i] = sigma_i * df_dx(x_i, t_i)
299
+
300
+ return f_path, drift_components, diffusion_components
301
+
302
+
303
+ class GirsanovTransform:
304
+ """
305
+ Girsanov 测度变换(风险中性测度)
306
+
307
+ 在风险中性世界中,股价过程从:
308
+ dS = μS dt + σS dW^P (真实世界)
309
+ 变换为:
310
+ dS = rS dt + σS dW^Q (风险中性世界)
311
+
312
+ Girsanov 密度(Radon-Nikodym 导数):
313
+ dQ/dP = exp(-θ·W(T) - ½θ²·T)
314
+ 其中 θ = (μ - r) / σ (市场价格风险)
315
+ """
316
+
317
+ def __init__(self, mu: float, r: float, sigma: float):
318
+ """
319
+ Args:
320
+ mu : 真实世界漂移率(年化)
321
+ r : 无风险利率(年化)
322
+ sigma: 波动率(年化)
323
+ """
324
+ self.mu = mu
325
+ self.r = r
326
+ self.sigma = sigma
327
+ self.theta = (mu - r) / sigma # 市场价格风险
328
+
329
+ def radon_nikodym(self, W_T: float, T: float) -> float:
330
+ """
331
+ 计算 Radon-Nikodym 导数(测度变换权重)
332
+
333
+ dQ/dP|_T = exp(-θ·W(T) - ½θ²·T)
334
+ """
335
+ return np.exp(-self.theta * W_T - 0.5 * self.theta ** 2 * T)
336
+
337
+ def real_to_risk_neutral_brownian(
338
+ self,
339
+ W_path: np.ndarray,
340
+ t_grid: np.ndarray
341
+ ) -> np.ndarray:
342
+ """
343
+ 将真实世界布朗运动 W^P 转化为风险中性布朗运动 W^Q
344
+
345
+ Girsanov 定理:W^Q(t) = W^P(t) + θ·t
346
+ """
347
+ return W_path + self.theta * t_grid
348
+
349
+ def price_option_by_measure_change(
350
+ self,
351
+ S0: float,
352
+ K: float,
353
+ T: float,
354
+ n_paths: int = 50000,
355
+ n_steps: int = 252
356
+ ) -> dict:
357
+ """
358
+ 通过 Girsanov 测度变换定价欧式看涨期权
359
+
360
+ 在 Q 测度下:S(T) = S0·exp((r - σ²/2)T + σW^Q(T))
361
+ C = e^{-rT}·E^Q[max(S(T)-K, 0)]
362
+
363
+ Returns:
364
+ {'price': float, 'delta': float, 'se': float}
365
+ """
366
+ dt = T / n_steps
367
+ sqrt_dt = np.sqrt(dt)
368
+
369
+ # 在 Q 测度下模拟(使用风险中性漂移 r - σ²/2)
370
+ dW = np.random.normal(0, sqrt_dt, (n_paths, n_steps))
371
+ log_returns = (self.r - 0.5 * self.sigma ** 2) * dt + self.sigma * dW
372
+ S_T = S0 * np.exp(np.sum(log_returns, axis=1))
373
+
374
+ payoffs = np.maximum(S_T - K, 0)
375
+ discounted = np.exp(-self.r * T) * payoffs
376
+
377
+ price = float(np.mean(discounted))
378
+ se = float(np.std(discounted) / np.sqrt(n_paths))
379
+
380
+ # 数值 Delta:ΔC/ΔS ≈ (C(S+ε) - C(S-ε)) / 2ε
381
+ eps = S0 * 0.01
382
+ payoffs_up = np.maximum(S_T * (1 + eps / S0) - K, 0)
383
+ payoffs_dn = np.maximum(S_T * (1 - eps / S0) - K, 0)
384
+ delta = float(np.exp(-self.r * T) * np.mean(payoffs_up - payoffs_dn) / (2 * eps))
385
+
386
+ return {"price": price, "delta": delta, "standard_error": se}
387
+
388
+
389
+ class FeynmanKac:
390
+ """
391
+ Feynman-Kac 定理:随机过程期望 ↔ PDE
392
+
393
+ 对于 SDE: dX = μ(X,t)dt + σ(X,t)dW,
394
+ 边值问题 u_t + μ·u_x + ½σ²·u_xx - r·u = -g(x,t)
395
+ 的解为:u(x,t) = E[∫_t^T e^{-r(s-t)} g(X_s,s)ds + e^{-r(T-t)} Φ(X_T) | X_t=x]
396
+
397
+ 应用:期权定价(Black-Scholes PDE ↔ 风险中性期望)
398
+ """
399
+
400
+ def __init__(
401
+ self,
402
+ mu: Callable,
403
+ sigma: Callable,
404
+ r: float = 0.0
405
+ ):
406
+ self.mu = mu
407
+ self.sigma = sigma
408
+ self.r = r
409
+
410
+ def solve_by_mc(
411
+ self,
412
+ x0: float,
413
+ t0: float,
414
+ T: float,
415
+ terminal_condition: Callable[[float], float],
416
+ running_cost: Optional[Callable[[float, float], float]] = None,
417
+ n_paths: int = 50000,
418
+ n_steps: int = 252,
419
+ antithetic: bool = True,
420
+ ) -> dict:
421
+ """
422
+ 用蒙特卡罗方法求解 Feynman-Kac 公式
423
+
424
+ u(x,t) ≈ (1/N) Σ [e^{-r(T-t)}·Φ(X_T^i) + ∫running_cost dt]
425
+
426
+ Args:
427
+ x0 : 初始状态
428
+ t0 : 初始时间
429
+ T : 终止时间
430
+ terminal_condition: Φ(X_T) 终端条件(如期权 payoff)
431
+ running_cost : g(X_t, t) 持续成本(默认为0)
432
+ n_paths : 路径数
433
+ n_steps : 时间步数
434
+ antithetic : 是否使用对偶变量
435
+
436
+ Returns:
437
+ {'value': float, 'se': float, 'ci_95': tuple}
438
+ """
439
+ process = ItoProcess(
440
+ mu_func=self.mu,
441
+ sigma_func=self.sigma,
442
+ x0=x0
443
+ )
444
+ tau = T - t0
445
+ paths = process.simulate_ensemble(
446
+ T=tau, n_steps=n_steps, n_paths=n_paths,
447
+ method="euler_maruyama", antithetic=antithetic
448
+ )
449
+
450
+ # 终端 payoff
451
+ X_T = paths[:, -1]
452
+ terminal_values = np.array([terminal_condition(x) for x in X_T])
453
+ discounted_terminal = np.exp(-self.r * tau) * terminal_values
454
+
455
+ # 运行成本(积分)
456
+ if running_cost is not None:
457
+ t_grid = np.linspace(t0, T, n_steps + 1)
458
+ dt = tau / n_steps
459
+ running_costs = np.zeros(n_paths)
460
+ for j in range(n_steps):
461
+ g_vals = np.array([running_cost(paths[k, j], t_grid[j]) for k in range(n_paths)])
462
+ running_costs += np.exp(-self.r * (t_grid[j] - t0)) * g_vals * dt
463
+ total_values = discounted_terminal + running_costs
464
+ else:
465
+ total_values = discounted_terminal
466
+
467
+ value = float(np.mean(total_values))
468
+ se = float(np.std(total_values) / np.sqrt(n_paths))
469
+ ci_lower = value - 1.96 * se
470
+ ci_upper = value + 1.96 * se
471
+
472
+ return {
473
+ "value": value,
474
+ "standard_error": se,
475
+ "ci_95": (ci_lower, ci_upper),
476
+ "n_paths": n_paths,
477
+ }