aria-code 0.52.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (478) hide show
  1. aria_code/__init__.py +42 -0
  2. aria_code/_version.py +20 -0
  3. aria_code/adk_apps/__init__.py +1 -0
  4. aria_code/adk_apps/aria_code_review/__init__.py +1 -0
  5. aria_code/adk_apps/aria_code_review/agent.py +32 -0
  6. aria_code/adk_apps/aria_research/__init__.py +1 -0
  7. aria_code/adk_apps/aria_research/agent.py +31 -0
  8. aria_code/agents/__init__.py +34 -0
  9. aria_code/agents/base.py +318 -0
  10. aria_code/agents/code_review.py +200 -0
  11. aria_code/agents/deep/__init__.py +37 -0
  12. aria_code/agents/deep/calibration_loop.py +144 -0
  13. aria_code/agents/deep/critic.py +125 -0
  14. aria_code/agents/deep/deepen.py +193 -0
  15. aria_code/agents/deep/models.py +149 -0
  16. aria_code/agents/deep/pipeline.py +164 -0
  17. aria_code/agents/deep/quant_fusion.py +192 -0
  18. aria_code/agents/deep/themes.py +95 -0
  19. aria_code/agents/deep/tiers.py +106 -0
  20. aria_code/agents/engineering/__init__.py +14 -0
  21. aria_code/agents/engineering/coder.py +222 -0
  22. aria_code/agents/engineering/tester.py +145 -0
  23. aria_code/agents/financial/__init__.py +21 -0
  24. aria_code/agents/financial/cashflow_burn.py +119 -0
  25. aria_code/agents/financial/catalyst.py +279 -0
  26. aria_code/agents/financial/corporate_finance.py +175 -0
  27. aria_code/agents/financial/debate.py +224 -0
  28. aria_code/agents/financial/earnings.py +303 -0
  29. aria_code/agents/financial/fundamental.py +191 -0
  30. aria_code/agents/financial/macro.py +99 -0
  31. aria_code/agents/financial/news.py +207 -0
  32. aria_code/agents/financial/risk.py +165 -0
  33. aria_code/agents/financial/sector.py +279 -0
  34. aria_code/agents/financial/strategist.py +121 -0
  35. aria_code/agents/financial/stripe_revenue.py +127 -0
  36. aria_code/agents/financial/synthesis.py +274 -0
  37. aria_code/agents/financial/technical.py +258 -0
  38. aria_code/agents/orchestrator.py +44 -0
  39. aria_code/agents/portfolio_agent.py +363 -0
  40. aria_code/agents/realty/__init__.py +62 -0
  41. aria_code/agents/realty/asset_diagnosis.py +152 -0
  42. aria_code/agents/realty/business_match.py +166 -0
  43. aria_code/agents/realty/cashflow_verify.py +209 -0
  44. aria_code/agents/realty/contract_rules.py +210 -0
  45. aria_code/agents/realty/energy_anomaly.py +189 -0
  46. aria_code/agents/realty/exit_settlement.py +208 -0
  47. aria_code/agents/realty/fulfillment_risk.py +206 -0
  48. aria_code/agents/realty/ops_optimize.py +160 -0
  49. aria_code/agents/realty/revenue_share.py +215 -0
  50. aria_code/agents/registry.py +158 -0
  51. aria_code/agents/signal_scheme.py +111 -0
  52. aria_code/agents/sports/__init__.py +0 -0
  53. aria_code/agents/sports/football_agent.py +197 -0
  54. aria_code/agents/supervisor.py +113 -0
  55. aria_code/agents/team.py +492 -0
  56. aria_code/agents/warehouse/__init__.py +25 -0
  57. aria_code/agents/warehouse/contracts.py +26 -0
  58. aria_code/agents/warehouse/dashboard.py +155 -0
  59. aria_code/agents/warehouse/fulfillment_leadtime.py +95 -0
  60. aria_code/agents/warehouse/inbound_exceptions.py +25 -0
  61. aria_code/agents/warehouse/inventory_health.py +24 -0
  62. aria_code/agents/warehouse/local_run.py +71 -0
  63. aria_code/agents/warehouse/local_snapshot.py +132 -0
  64. aria_code/agents/warehouse/logistics_cost.py +84 -0
  65. aria_code/agents/warehouse/logistics_sync.py +21 -0
  66. aria_code/agents/warehouse/workflow.py +31 -0
  67. aria_code/aliyun_data_client.py +4 -0
  68. aria_code/apps/README.md +12 -0
  69. aria_code/apps/__init__.py +2 -0
  70. aria_code/apps/channels/README.md +15 -0
  71. aria_code/apps/channels/__init__.py +17 -0
  72. aria_code/apps/channels/intake.py +92 -0
  73. aria_code/apps/channels/registry.py +95 -0
  74. aria_code/apps/channels/tradingview.py +94 -0
  75. aria_code/apps/cli/README.md +13 -0
  76. aria_code/apps/cli/__init__.py +2 -0
  77. aria_code/apps/cli/_optional.py +120 -0
  78. aria_code/apps/cli/bootstrap.py +145 -0
  79. aria_code/apps/cli/broker_render.py +206 -0
  80. aria_code/apps/cli/codegen_paths.py +29 -0
  81. aria_code/apps/cli/commands/__init__.py +16 -0
  82. aria_code/apps/cli/commands/_business_render.py +78 -0
  83. aria_code/apps/cli/commands/_ui.py +71 -0
  84. aria_code/apps/cli/commands/analysis_cmds.py +369 -0
  85. aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
  86. aria_code/apps/cli/commands/auth_cmds.py +239 -0
  87. aria_code/apps/cli/commands/backtest_cmds.py +2543 -0
  88. aria_code/apps/cli/commands/broker_cmds.py +1234 -0
  89. aria_code/apps/cli/commands/business_workflow_cmds.py +319 -0
  90. aria_code/apps/cli/commands/canvas_cmds.py +77 -0
  91. aria_code/apps/cli/commands/catalog.py +87 -0
  92. aria_code/apps/cli/commands/core_cmds.py +2520 -0
  93. aria_code/apps/cli/commands/data_cmds.py +457 -0
  94. aria_code/apps/cli/commands/diagnostic_cmds.py +230 -0
  95. aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
  96. aria_code/apps/cli/commands/file_cmds.py +312 -0
  97. aria_code/apps/cli/commands/finance_service_cmds.py +60 -0
  98. aria_code/apps/cli/commands/fx_commodity_cmds.py +123 -0
  99. aria_code/apps/cli/commands/market.py +434 -0
  100. aria_code/apps/cli/commands/market_cmds.py +1381 -0
  101. aria_code/apps/cli/commands/market_context.py +425 -0
  102. aria_code/apps/cli/commands/market_render.py +7 -0
  103. aria_code/apps/cli/commands/model_cmds.py +2153 -0
  104. aria_code/apps/cli/commands/ops_cmds.py +721 -0
  105. aria_code/apps/cli/commands/orchestrator_cmds.py +179 -0
  106. aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
  107. aria_code/apps/cli/commands/portfolio_cmds.py +1278 -0
  108. aria_code/apps/cli/commands/report.py +437 -0
  109. aria_code/apps/cli/commands/scaffold_templates.py +617 -0
  110. aria_code/apps/cli/commands/session_cmds.py +236 -0
  111. aria_code/apps/cli/commands/session_ux_cmds.py +359 -0
  112. aria_code/apps/cli/commands/team.py +669 -0
  113. aria_code/apps/cli/commands/team_render.py +8 -0
  114. aria_code/apps/cli/commands/ui_cmds.py +401 -0
  115. aria_code/apps/cli/commands/warehouse_cmds.py +79 -0
  116. aria_code/apps/cli/commands/workflow_cmds.py +480 -0
  117. aria_code/apps/cli/commands/workspace_cmds.py +1494 -0
  118. aria_code/apps/cli/config_paths.py +73 -0
  119. aria_code/apps/cli/config_store.py +78 -0
  120. aria_code/apps/cli/constants.py +88 -0
  121. aria_code/apps/cli/context.py +34 -0
  122. aria_code/apps/cli/deterministic.py +191 -0
  123. aria_code/apps/cli/direct.py +48 -0
  124. aria_code/apps/cli/finance_service_catalog.py +111 -0
  125. aria_code/apps/cli/football_reports.py +231 -0
  126. aria_code/apps/cli/github_app_auth.py +135 -0
  127. aria_code/apps/cli/google_login.py +196 -0
  128. aria_code/apps/cli/handlers/__init__.py +11 -0
  129. aria_code/apps/cli/handlers/broker_handlers.py +122 -0
  130. aria_code/apps/cli/handlers/chart_handlers.py +1309 -0
  131. aria_code/apps/cli/handlers/market_handlers.py +2513 -0
  132. aria_code/apps/cli/handlers/realty_handlers.py +114 -0
  133. aria_code/apps/cli/handlers/sports_handlers.py +22 -0
  134. aria_code/apps/cli/handlers/strategy_advice.py +82 -0
  135. aria_code/apps/cli/helpers.py +302 -0
  136. aria_code/apps/cli/hooks.py +180 -0
  137. aria_code/apps/cli/i18n.py +297 -0
  138. aria_code/apps/cli/intent.py +136 -0
  139. aria_code/apps/cli/intent_router.py +276 -0
  140. aria_code/apps/cli/intent_signals.py +344 -0
  141. aria_code/apps/cli/lifecycle_hooks.py +48 -0
  142. aria_code/apps/cli/local_tool_schemas.py +536 -0
  143. aria_code/apps/cli/main.py +29 -0
  144. aria_code/apps/cli/market_metadata.py +135 -0
  145. aria_code/apps/cli/market_universe.py +487 -0
  146. aria_code/apps/cli/message_processing.py +310 -0
  147. aria_code/apps/cli/model_catalog.py +620 -0
  148. aria_code/apps/cli/pdf_report.py +977 -0
  149. aria_code/apps/cli/plan_mode.py +139 -0
  150. aria_code/apps/cli/plotly_html.py +15 -0
  151. aria_code/apps/cli/prediction_feedback.py +202 -0
  152. aria_code/apps/cli/preflight.py +502 -0
  153. aria_code/apps/cli/project_aria.py +60 -0
  154. aria_code/apps/cli/prompt_assembly.py +59 -0
  155. aria_code/apps/cli/prompts/__init__.py +0 -0
  156. aria_code/apps/cli/prompts/coding.py +53 -0
  157. aria_code/apps/cli/prompts/select.py +142 -0
  158. aria_code/apps/cli/prompts/system_prompts.py +552 -0
  159. aria_code/apps/cli/prompts/ui.py +436 -0
  160. aria_code/apps/cli/provider_endpoints.py +237 -0
  161. aria_code/apps/cli/provider_keys.py +73 -0
  162. aria_code/apps/cli/providers/__init__.py +1 -0
  163. aria_code/apps/cli/providers/base.py +521 -0
  164. aria_code/apps/cli/providers/chat_routing.py +128 -0
  165. aria_code/apps/cli/providers/collaboration.py +98 -0
  166. aria_code/apps/cli/providers/llm/__init__.py +1 -0
  167. aria_code/apps/cli/providers/llm/offline_responses.py +68 -0
  168. aria_code/apps/cli/providers/llm/ollama_stream.py +1594 -0
  169. aria_code/apps/cli/providers/llm/response_cache.py +60 -0
  170. aria_code/apps/cli/providers/llm/sse_stream.py +257 -0
  171. aria_code/apps/cli/providers/runtime_bridge.py +353 -0
  172. aria_code/apps/cli/providers/vertexai_stream.py +316 -0
  173. aria_code/apps/cli/runtime_consumer.py +657 -0
  174. aria_code/apps/cli/sandbox.py +32 -0
  175. aria_code/apps/cli/session_export.py +120 -0
  176. aria_code/apps/cli/session_jsonl.py +208 -0
  177. aria_code/apps/cli/session_store.py +112 -0
  178. aria_code/apps/cli/skills_catalog.py +273 -0
  179. aria_code/apps/cli/todo_tracker.py +190 -0
  180. aria_code/apps/cli/tool_executor.py +823 -0
  181. aria_code/apps/cli/tool_registry.py +34 -0
  182. aria_code/apps/cli/tool_scope.py +106 -0
  183. aria_code/apps/cli/tools/__init__.py +40 -0
  184. aria_code/apps/cli/tools/context.py +46 -0
  185. aria_code/apps/cli/tools/file_tools.py +180 -0
  186. aria_code/apps/cli/tools/market_tools.py +620 -0
  187. aria_code/apps/cli/tools/notebook_tools.py +111 -0
  188. aria_code/apps/cli/tools/system_tools.py +726 -0
  189. aria_code/apps/cli/tools/write_tools.py +845 -0
  190. aria_code/apps/cli/tradingview_bridge.py +529 -0
  191. aria_code/apps/cli/turn_planning.py +137 -0
  192. aria_code/apps/cli/update_check.py +153 -0
  193. aria_code/apps/cli/utils/__init__.py +0 -0
  194. aria_code/apps/cli/utils/market_detect.py +1597 -0
  195. aria_code/apps/daemon/README.md +14 -0
  196. aria_code/apps/vscode/README.md +115 -0
  197. aria_code/apps/vscode/package.json +70 -0
  198. aria_code/aria_cli.py +6743 -0
  199. aria_code/aria_daemon.py +1398 -0
  200. aria_code/aria_feishu_bot.py +1423 -0
  201. aria_code/aria_mcp_server.py +7 -0
  202. aria_code/aria_relay_client.py +4 -0
  203. aria_code/aria_relay_server.py +490 -0
  204. aria_code/aria_telegram_bot.py +202 -0
  205. aria_code/ariarc.py +385 -0
  206. aria_code/artifacts.py +581 -0
  207. aria_code/backtest_engine.py +4 -0
  208. aria_code/backtest_report.py +4 -0
  209. aria_code/brokers/__init__.py +78 -0
  210. aria_code/brokers/automation.py +193 -0
  211. aria_code/brokers/base.py +231 -0
  212. aria_code/brokers/capabilities.py +264 -0
  213. aria_code/brokers/cn/__init__.py +10 -0
  214. aria_code/brokers/cn/easytrader_broker.py +193 -0
  215. aria_code/brokers/cn/futu_broker.py +194 -0
  216. aria_code/brokers/cn/longbridge_broker.py +190 -0
  217. aria_code/brokers/cn/tiger_broker.py +196 -0
  218. aria_code/brokers/cn/xtquant_broker.py +225 -0
  219. aria_code/brokers/config.py +414 -0
  220. aria_code/brokers/intl/__init__.py +5 -0
  221. aria_code/brokers/intl/alpaca_broker.py +183 -0
  222. aria_code/brokers/intl/ibkr_broker.py +215 -0
  223. aria_code/brokers/intl/trading212_broker.py +183 -0
  224. aria_code/brokers/intl/webull_broker.py +156 -0
  225. aria_code/brokers/paper_broker.py +266 -0
  226. aria_code/brokers/planning.py +296 -0
  227. aria_code/brokers/registry.py +213 -0
  228. aria_code/brokers/trading.py +267 -0
  229. aria_code/canva_client.py +4 -0
  230. aria_code/change_store.py +127 -0
  231. aria_code/clients/__init__.py +0 -0
  232. aria_code/clients/aliyun_data_client.py +660 -0
  233. aria_code/clients/aria_relay_client.py +182 -0
  234. aria_code/clients/canva_client.py +284 -0
  235. aria_code/clients/figma_client.py +100 -0
  236. aria_code/clients/football_data_client.py +1671 -0
  237. aria_code/clients/kling_video_client.py +182 -0
  238. aria_code/clients/market_data_client.py +1976 -0
  239. aria_code/clients/openai_image_client.py +242 -0
  240. aria_code/clients/runway_video_client.py +157 -0
  241. aria_code/clients/warehouse_erp_client.py +105 -0
  242. aria_code/cloud_memory.py +181 -0
  243. aria_code/command_safety.py +19 -0
  244. aria_code/computer_use_tools.py +4 -0
  245. aria_code/dashboard_generator.py +579 -0
  246. aria_code/data_analysis_tools.py +4 -0
  247. aria_code/data_cleaner.py +483 -0
  248. aria_code/data_service.py +513 -0
  249. aria_code/datasources/__init__.py +23 -0
  250. aria_code/datasources/base.py +166 -0
  251. aria_code/datasources/router.py +253 -0
  252. aria_code/datasources/sources/__init__.py +15 -0
  253. aria_code/datasources/sources/akshare_source.py +269 -0
  254. aria_code/datasources/sources/alpha_vantage_source.py +203 -0
  255. aria_code/datasources/sources/edgar_source.py +219 -0
  256. aria_code/datasources/sources/finnhub_source.py +198 -0
  257. aria_code/datasources/sources/fred_source.py +223 -0
  258. aria_code/datasources/sources/tushare_source.py +143 -0
  259. aria_code/datasources/sources/web_scraper_source.py +278 -0
  260. aria_code/datasources/sources/world_bank_source.py +206 -0
  261. aria_code/datasources/sources/yfinance_source.py +152 -0
  262. aria_code/demo_player.py +207 -0
  263. aria_code/doctor.py +813 -0
  264. aria_code/domain/__init__.py +0 -0
  265. aria_code/domain/backtest_engine.py +354 -0
  266. aria_code/domain/backtest_report.py +472 -0
  267. aria_code/domain/finance_formulas.py +389 -0
  268. aria_code/domain/portfolio_ledger.py +307 -0
  269. aria_code/domain/strategy_vault.py +421 -0
  270. aria_code/evals/__init__.py +37 -0
  271. aria_code/evals/harness.py +598 -0
  272. aria_code/evals/runner.py +238 -0
  273. aria_code/external_agent_runner.py +115 -0
  274. aria_code/figma_client.py +4 -0
  275. aria_code/file_analysis_tools.py +4 -0
  276. aria_code/finance_formulas.py +4 -0
  277. aria_code/football_data_client.py +4 -0
  278. aria_code/image_gen_tools.py +4 -0
  279. aria_code/intent_classifier.py +386 -0
  280. aria_code/kling_video_client.py +4 -0
  281. aria_code/licensing.py +151 -0
  282. aria_code/local_finance_tools.py +4 -0
  283. aria_code/local_image_provider.py +4 -0
  284. aria_code/local_llm_provider.py +4 -0
  285. aria_code/macro_tools.py +4 -0
  286. aria_code/markdown_pdf.py +289 -0
  287. aria_code/market_data_client.py +4 -0
  288. aria_code/mcp_client.py +639 -0
  289. aria_code/memory_manager.py +246 -0
  290. aria_code/model_capability.py +486 -0
  291. aria_code/notification_tools.py +4 -0
  292. aria_code/openai_image_client.py +4 -0
  293. aria_code/packages/__init__.py +23 -0
  294. aria_code/packages/adk_bridge/__init__.py +11 -0
  295. aria_code/packages/adk_bridge/code_review_tools.py +49 -0
  296. aria_code/packages/adk_bridge/market_tools.py +125 -0
  297. aria_code/packages/aria_agents/__init__.py +5 -0
  298. aria_code/packages/aria_agents/manifest.py +78 -0
  299. aria_code/packages/aria_core/__init__.py +34 -0
  300. aria_code/packages/aria_core/architecture.py +269 -0
  301. aria_code/packages/aria_core/export.py +147 -0
  302. aria_code/packages/aria_core/manifest.py +65 -0
  303. aria_code/packages/aria_core/paths.py +57 -0
  304. aria_code/packages/aria_core/secure_file.py +68 -0
  305. aria_code/packages/aria_infra/__init__.py +15 -0
  306. aria_code/packages/aria_infra/arthera.py +52 -0
  307. aria_code/packages/aria_infra/doctor.py +246 -0
  308. aria_code/packages/aria_infra/product.py +40 -0
  309. aria_code/packages/aria_mcp/__init__.py +25 -0
  310. aria_code/packages/aria_mcp/bridge.py +201 -0
  311. aria_code/packages/aria_mcp/circuit.py +82 -0
  312. aria_code/packages/aria_mcp/config.py +111 -0
  313. aria_code/packages/aria_mcp/installer.py +95 -0
  314. aria_code/packages/aria_mcp/server.py +1550 -0
  315. aria_code/packages/aria_mcp/tools.py +81 -0
  316. aria_code/packages/aria_sdk/__init__.py +19 -0
  317. aria_code/packages/aria_sdk/client.py +396 -0
  318. aria_code/packages/aria_sdk/providers.py +70 -0
  319. aria_code/packages/aria_sdk/streaming.py +79 -0
  320. aria_code/packages/aria_sdk/types.py +86 -0
  321. aria_code/packages/aria_services/__init__.py +92 -0
  322. aria_code/packages/aria_services/cache.py +60 -0
  323. aria_code/packages/aria_services/context.py +356 -0
  324. aria_code/packages/aria_services/context_checkpoints.py +157 -0
  325. aria_code/packages/aria_services/data.py +11 -0
  326. aria_code/packages/aria_services/provider_health.py +268 -0
  327. aria_code/packages/aria_services/references.py +247 -0
  328. aria_code/packages/aria_services/registry.py +249 -0
  329. aria_code/packages/aria_services/research_protocol.py +57 -0
  330. aria_code/packages/aria_services/research_quality.py +186 -0
  331. aria_code/packages/aria_services/settings.py +157 -0
  332. aria_code/packages/aria_services/usage.py +151 -0
  333. aria_code/packages/aria_skills/__init__.py +46 -0
  334. aria_code/packages/aria_skills/catalog.py +149 -0
  335. aria_code/packages/aria_skills/loader.py +455 -0
  336. aria_code/packages/aria_skills/registry.py +84 -0
  337. aria_code/packages/aria_tools/__init__.py +5 -0
  338. aria_code/packages/aria_tools/financial/__init__.py +8 -0
  339. aria_code/packages/aria_tools/financial/_paths.py +130 -0
  340. aria_code/packages/aria_tools/financial/compliance.py +31 -0
  341. aria_code/packages/aria_tools/financial/factors.py +33 -0
  342. aria_code/packages/aria_tools/financial/risk_tools.py +33 -0
  343. aria_code/packages/aria_tools/financial/strategy.py +31 -0
  344. aria_code/packages/aria_tools/registry.py +128 -0
  345. aria_code/packages/quant_engine/__init__.py +34 -0
  346. aria_code/packages/quant_engine/_namespace.py +62 -0
  347. aria_code/packages/quant_engine/backtest/__init__.py +14 -0
  348. aria_code/packages/quant_engine/backtest/core.py +53 -0
  349. aria_code/packages/quant_engine/backtest/engine.py +96 -0
  350. aria_code/packages/quant_engine/portfolio/__init__.py +13 -0
  351. aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
  352. aria_code/packages/quant_engine/sports/__init__.py +80 -0
  353. aria_code/packages/quant_engine/sports/calibrator.py +354 -0
  354. aria_code/packages/quant_engine/sports/dixon_coles.py +234 -0
  355. aria_code/packages/quant_engine/sports/elo.py +300 -0
  356. aria_code/packages/quant_engine/sports/form.py +188 -0
  357. aria_code/packages/quant_engine/sports/h2h.py +195 -0
  358. aria_code/packages/quant_engine/sports/ml_model.py +377 -0
  359. aria_code/packages/quant_engine/sports/predictor.py +312 -0
  360. aria_code/packages/quant_engine/sports/tracker.py +665 -0
  361. aria_code/packages/quant_engine/stochastic/__init__.py +35 -0
  362. aria_code/packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
  363. aria_code/packages/quant_engine/stochastic/ito_calculus.py +477 -0
  364. aria_code/packages/quant_engine/stochastic/kelly_criterion.py +181 -0
  365. aria_code/packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
  366. aria_code/packages/quant_engine/stochastic/options_pricing.py +576 -0
  367. aria_code/packages/quant_engine/stochastic/stochastic_processes.py +90 -0
  368. aria_code/packs/__init__.py +85 -0
  369. aria_code/packs/base.py +168 -0
  370. aria_code/packs/finance/__init__.py +250 -0
  371. aria_code/packs/logistics/__init__.py +169 -0
  372. aria_code/packs/payments/__init__.py +139 -0
  373. aria_code/packs/realty/__init__.py +190 -0
  374. aria_code/packs/registry.py +202 -0
  375. aria_code/packs/rules.py +90 -0
  376. aria_code/plan_utils.py +194 -0
  377. aria_code/plugin_loader.py +328 -0
  378. aria_code/portfolio_ledger.py +4 -0
  379. aria_code/preview_server.py +449 -0
  380. aria_code/privacy/__init__.py +5 -0
  381. aria_code/privacy/feedback.py +123 -0
  382. aria_code/project_review.py +278 -0
  383. aria_code/project_review_api.py +235 -0
  384. aria_code/project_review_server.py +18 -0
  385. aria_code/project_tools.py +4 -0
  386. aria_code/providers/__init__.py +30 -0
  387. aria_code/providers/llm/__init__.py +19 -0
  388. aria_code/providers/llm/anthropic.py +184 -0
  389. aria_code/providers/llm/autoconfig.py +213 -0
  390. aria_code/providers/llm/base.py +139 -0
  391. aria_code/providers/llm/custom.py +158 -0
  392. aria_code/providers/llm/ollama.py +129 -0
  393. aria_code/providers/llm/openai_compat.py +313 -0
  394. aria_code/providers/llm/registry.py +444 -0
  395. aria_code/providers/local_image_provider.py +202 -0
  396. aria_code/providers/local_llm_provider.py +589 -0
  397. aria_code/realty_data_tools.py +4 -0
  398. aria_code/report_exporters.py +150 -0
  399. aria_code/report_generator.py +1764 -0
  400. aria_code/runtime/__init__.py +169 -0
  401. aria_code/runtime/acceptance.py +481 -0
  402. aria_code/runtime/agent_loop.py +1520 -0
  403. aria_code/runtime/approval.py +57 -0
  404. aria_code/runtime/budget.py +232 -0
  405. aria_code/runtime/checkpoints.py +432 -0
  406. aria_code/runtime/crash_report.py +174 -0
  407. aria_code/runtime/events.py +113 -0
  408. aria_code/runtime/gateway.py +150 -0
  409. aria_code/runtime/lsp.py +484 -0
  410. aria_code/runtime/orchestrator.py +55 -0
  411. aria_code/runtime/repo_map.py +700 -0
  412. aria_code/runtime/run_state.py +127 -0
  413. aria_code/runtime/run_store.py +450 -0
  414. aria_code/runtime/self_healing.py +422 -0
  415. aria_code/runtime/subagent.py +547 -0
  416. aria_code/runtime/task_ledger.py +86 -0
  417. aria_code/runtime/task_orchestrator.py +90 -0
  418. aria_code/runtime/tool_executor.py +217 -0
  419. aria_code/runtime/tool_policy.py +107 -0
  420. aria_code/runtime/worktrees.py +215 -0
  421. aria_code/runway_video_client.py +4 -0
  422. aria_code/safety/__init__.py +23 -0
  423. aria_code/safety/permissions.py +306 -0
  424. aria_code/safety/service.py +102 -0
  425. aria_code/setup_wizard.py +673 -0
  426. aria_code/spreadsheet_tools.py +4 -0
  427. aria_code/strategy_vault.py +4 -0
  428. aria_code/tools/__init__.py +0 -0
  429. aria_code/tools/artifact_tools.py +252 -0
  430. aria_code/tools/broker_tools.py +92 -0
  431. aria_code/tools/build_quant_engine.py +97 -0
  432. aria_code/tools/code_audit_tools.py +192 -0
  433. aria_code/tools/computer_use_tools.py +504 -0
  434. aria_code/tools/data_analysis_tools.py +809 -0
  435. aria_code/tools/enterprise_finance_tools.py +156 -0
  436. aria_code/tools/extended_tools.py +240 -0
  437. aria_code/tools/file_analysis_tools.py +885 -0
  438. aria_code/tools/image_gen_tools.py +114 -0
  439. aria_code/tools/local_finance_tools.py +3417 -0
  440. aria_code/tools/logistics_tools.py +135 -0
  441. aria_code/tools/macro_tools.py +369 -0
  442. aria_code/tools/notification_tools.py +249 -0
  443. aria_code/tools/project_tools.py +525 -0
  444. aria_code/tools/realty_data_tools.py +659 -0
  445. aria_code/tools/spreadsheet_tools.py +388 -0
  446. aria_code/tools/stripe_tools.py +115 -0
  447. aria_code/ui/__init__.py +105 -0
  448. aria_code/ui/banner.py +508 -0
  449. aria_code/ui/completer.py +501 -0
  450. aria_code/ui/console.py +281 -0
  451. aria_code/ui/image_render.py +243 -0
  452. aria_code/ui/input_box.py +488 -0
  453. aria_code/ui/picker.py +201 -0
  454. aria_code/ui/render/__init__.py +11 -0
  455. aria_code/ui/render/finance.py +1705 -0
  456. aria_code/ui/render/market.py +225 -0
  457. aria_code/ui/render/output.py +775 -0
  458. aria_code/ui/render/responsive.py +75 -0
  459. aria_code/ui/render/team.py +354 -0
  460. aria_code/ui/robot.py +235 -0
  461. aria_code/ui/startup_dashboard.py +109 -0
  462. aria_code/utils/vscode_uri.py +37 -0
  463. aria_code/video_analysis.py +147 -0
  464. aria_code/video_editor.py +271 -0
  465. aria_code/workspace/__init__.py +6 -0
  466. aria_code/workspace/files.py +193 -0
  467. aria_code/workspace/verify.py +113 -0
  468. aria_code/workspace_changes.py +199 -0
  469. aria_code/workspace_context.py +233 -0
  470. aria_code/workspace_service.py +196 -0
  471. aria_code/workspace_tools.py +150 -0
  472. aria_code-0.52.0.dist-info/METADATA +237 -0
  473. aria_code-0.52.0.dist-info/RECORD +478 -0
  474. aria_code-0.52.0.dist-info/WHEEL +5 -0
  475. aria_code-0.52.0.dist-info/entry_points.txt +3 -0
  476. aria_code-0.52.0.dist-info/licenses/LICENSE +201 -0
  477. aria_code-0.52.0.dist-info/licenses/NOTICE +49 -0
  478. aria_code-0.52.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,659 @@
1
+ """
2
+ realty_data_tools.py — 不动产市场数据层
3
+ ==========================================
4
+ 数据来源:
5
+ - AKShare — 中国房价指数/房地产投资/REITs
6
+ - FRED — 美国住房数据 (Case-Shiller/新屋开工/NAHB)
7
+ - 本地计算 — 租金收益率/物业估值/资产评级
8
+
9
+ 全部函数返回 {"success": bool, ...} 统一格式。
10
+
11
+ 安装依赖(可选):
12
+ pip install akshare # 中国数据
13
+ pip install openpyxl # Excel 导出
14
+ """
15
+
16
+ from __future__ import annotations
17
+
18
+ import logging
19
+ import os
20
+ from datetime import datetime
21
+ from pathlib import Path
22
+ from typing import Any, Dict, List, Optional
23
+
24
+ logger = logging.getLogger(__name__)
25
+
26
+ try:
27
+ import akshare as ak
28
+ _HAS_AK = True
29
+ except ImportError:
30
+ _HAS_AK = False
31
+
32
+ try:
33
+ import pandas as pd
34
+ _HAS_PD = True
35
+ except ImportError:
36
+ _HAS_PD = False
37
+
38
+ try:
39
+ import numpy as np
40
+ _HAS_NP = True
41
+ except ImportError:
42
+ _HAS_NP = False
43
+
44
+ try:
45
+ import requests as _req
46
+ _HAS_REQ = True
47
+ except ImportError:
48
+ _HAS_REQ = False
49
+
50
+ # ── 70城房价指数 ─────────────────────────────────────────────────────────────
51
+
52
+ # akshare 支持的城市(70城新房价格指数覆盖范围)
53
+ CN_CITIES_TIER1 = ["北京", "上海", "广州", "深圳"]
54
+ CN_CITIES_TIER2 = [
55
+ "成都", "杭州", "武汉", "重庆", "南京", "西安", "天津", "苏州",
56
+ "长沙", "郑州", "青岛", "沈阳", "宁波", "合肥", "厦门", "济南",
57
+ "东莞", "佛山", "福州", "南宁", "昆明", "贵阳", "大连", "哈尔滨",
58
+ "长春", "石家庄", "太原", "南昌", "兰州", "银川", "西宁", "乌鲁木齐",
59
+ "海口", "三亚", "珠海", "温州", "泉州", "烟台", "洛阳", "唐山",
60
+ "南通", "常州", "徐州", "扬州", "嘉兴", "金华", "绍兴",
61
+ "湖州", "台州", "芜湖", "湘潭", "株洲", "常德", "桂林", "柳州",
62
+ "汕头", "惠州", "江门", "湛江", "中山", "保定", "廊坊",
63
+ ]
64
+ CN_CITIES_ALL = CN_CITIES_TIER1 + CN_CITIES_TIER2
65
+
66
+
67
+ def get_house_price_index(city: str = "北京", city2: str = "上海") -> dict:
68
+ """
69
+ 获取中国 70 城新房 / 二手房价格指数。
70
+
71
+ 返回同比、环比数据(最近 12 个月)。
72
+ city: 主要城市(默认北京)
73
+ city2: 对比城市(默认上海)
74
+ """
75
+ if not _HAS_AK:
76
+ return {"success": False, "error": "akshare 未安装"}
77
+ try:
78
+ df = ak.macro_china_new_house_price(city_first=city, city_second=city2)
79
+ if df is None or df.empty:
80
+ return {"success": False, "error": "无数据"}
81
+
82
+ # Keep last 24 months
83
+ df = df.tail(24).copy()
84
+ df.columns = [c.strip() for c in df.columns]
85
+ records = df.to_dict("records")
86
+
87
+ # Split by city
88
+ city1_records = [r for r in records if str(r.get("城市","")) == city][-12:]
89
+ city2_records = [r for r in records if str(r.get("城市","")) == city2][-12:]
90
+
91
+ def _latest(recs):
92
+ if not recs: return {}
93
+ r = recs[-1]
94
+ return {
95
+ "date": str(r.get("日期",""))[:7],
96
+ "new_yoy": r.get("新建商品住宅价格指数-同比"),
97
+ "new_mom": r.get("新建商品住宅价格指数-环比"),
98
+ "second_yoy": r.get("二手住宅价格指数-同比"),
99
+ "second_mom": r.get("二手住宅价格指数-环比"),
100
+ }
101
+
102
+ return {
103
+ "success": True,
104
+ "city1": city,
105
+ "city2": city2,
106
+ "latest_city1": _latest(city1_records),
107
+ "latest_city2": _latest(city2_records),
108
+ "series_city1": city1_records,
109
+ "series_city2": city2_records,
110
+ "provider": "akshare_NBS",
111
+ }
112
+ except Exception as e:
113
+ return {"success": False, "error": str(e)}
114
+
115
+
116
+ def get_re_investment() -> dict:
117
+ """
118
+ 中国房地产开发投资额(月度累计同比)。
119
+ 反映房地产行业景气度。
120
+ """
121
+ if not _HAS_AK:
122
+ return {"success": False, "error": "akshare 未安装"}
123
+ try:
124
+ df = ak.macro_china_real_estate()
125
+ if df is None or df.empty:
126
+ return {"success": False, "error": "无数据"}
127
+ df = df.tail(12)
128
+ records = df.to_dict("records")
129
+ latest = records[-1] if records else {}
130
+ return {
131
+ "success": True,
132
+ "latest": latest,
133
+ "series": records,
134
+ "provider": "akshare_NBS",
135
+ }
136
+ except Exception as e:
137
+ return {"success": False, "error": str(e)}
138
+
139
+
140
+ def get_multi_city_comparison(cities: Optional[List[str]] = None) -> dict:
141
+ """
142
+ 多城市房价指数对比(同比涨跌幅热力图数据)。
143
+ 默认对比 8 个核心城市(两两调用拼合)。
144
+ """
145
+ if not _HAS_AK:
146
+ return {"success": False, "error": "akshare 未安装"}
147
+
148
+ if cities is None:
149
+ cities = ["北京", "上海", "深圳", "广州", "成都", "杭州", "武汉", "南京",
150
+ "重庆", "西安", "长沙", "天津", "苏州", "郑州", "厦门", "青岛"]
151
+
152
+ results: List[Dict] = []
153
+ # akshare requires city_first + city_second; batch by pairs
154
+ city_pairs = list(zip(cities[::2], cities[1::2]))
155
+ if len(cities) % 2 == 1:
156
+ city_pairs.append((cities[-1], cities[0])) # pair last with first
157
+
158
+ seen: set = set()
159
+ for c1, c2 in city_pairs:
160
+ try:
161
+ df = ak.macro_china_new_house_price(city_first=c1, city_second=c2)
162
+ if df is None or df.empty:
163
+ continue
164
+ df = df.tail(4)
165
+ for city in (c1, c2):
166
+ if city in seen:
167
+ continue
168
+ sub = df[df["城市"] == city]
169
+ if sub.empty:
170
+ continue
171
+ row = sub.iloc[-1]
172
+ results.append({
173
+ "city": city,
174
+ "tier": "一线" if city in CN_CITIES_TIER1 else "二线",
175
+ "date": str(row.get("日期",""))[:7],
176
+ "new_yoy": _safe_float(row.get("新建商品住宅价格指数-同比")),
177
+ "new_mom": _safe_float(row.get("新建商品住宅价格指数-环比")),
178
+ "second_yoy": _safe_float(row.get("二手住宅价格指数-同比")),
179
+ "second_mom": _safe_float(row.get("二手住宅价格指数-环比")),
180
+ })
181
+ seen.add(city)
182
+ except Exception as e:
183
+ logger.debug("City pair (%s,%s) failed: %s", c1, c2, e)
184
+
185
+ if not results:
186
+ return {"success": False, "error": "无法获取城市数据"}
187
+
188
+ # Sort by new_yoy descending
189
+ results.sort(key=lambda x: -(x.get("new_yoy") or 0))
190
+
191
+ return {
192
+ "success": True,
193
+ "cities": results,
194
+ "top_riser": results[0]["city"] if results else "",
195
+ "top_faller":results[-1]["city"] if results else "",
196
+ "provider": "akshare_NBS",
197
+ }
198
+
199
+
200
+ # ── REITs 分析 ───────────────────────────────────────────────────────────────
201
+
202
+ def get_reits_list() -> dict:
203
+ """获取中国 REITs 实时行情列表(东方财富)。"""
204
+ if not _HAS_AK:
205
+ return {"success": False, "error": "akshare 未安装"}
206
+ try:
207
+ df = ak.reits_realtime_em()
208
+ if df is None or df.empty:
209
+ return {"success": False, "error": "无 REIT 数据"}
210
+ cols = [c for c in ["代码","名称","最新价","涨跌额","涨跌幅","昨收","成交量","成交额"] if c in df.columns]
211
+ records = df[cols].to_dict("records")
212
+ return {
213
+ "success": True,
214
+ "count": len(records),
215
+ "reits": records,
216
+ "provider": "akshare_em",
217
+ }
218
+ except Exception as e:
219
+ return {"success": False, "error": str(e)}
220
+
221
+
222
+ def get_reit_analysis(code: str) -> dict:
223
+ """
224
+ 单只 REIT 深度分析:历史行情 + 估值指标。
225
+
226
+ code: REIT 代码,如 "508603" (唯品商业)
227
+ """
228
+ if not _HAS_AK:
229
+ return {"success": False, "error": "akshare 未安装"}
230
+ try:
231
+ import yfinance as yf
232
+ _HAS_YF = True
233
+ except ImportError:
234
+ _HAS_YF = False
235
+
236
+ result: Dict[str, Any] = {"success": True, "code": code}
237
+
238
+ # Historical prices
239
+ try:
240
+ df_hist = ak.reits_hist_em(symbol=code, period="daily",
241
+ start_date="20230101",
242
+ end_date=datetime.now().strftime("%Y%m%d"),
243
+ adjust="qfq")
244
+ if df_hist is not None and not df_hist.empty:
245
+ df_hist = df_hist.tail(252) # 1 year
246
+ last = df_hist.iloc[-1]
247
+ cols = df_hist.columns.tolist()
248
+ close_col = next((c for c in cols if "收盘" in c or "close" in c.lower()), cols[-1])
249
+ prices = df_hist[close_col].astype(float)
250
+ cur_price = float(last[close_col])
251
+ result["price"] = round(cur_price, 3)
252
+
253
+ # Performance
254
+ if len(prices) >= 2:
255
+ result["return_1y"] = round((prices.iloc[-1]/prices.iloc[0] - 1)*100, 2)
256
+ if len(prices) >= 20:
257
+ result["return_1m"] = round((prices.iloc[-1]/prices.iloc[-20] - 1)*100, 2)
258
+
259
+ # Volatility
260
+ if _HAS_NP and len(prices) >= 20:
261
+ rets = prices.pct_change().dropna()
262
+ result["volatility_annual"] = round(float(rets.std() * np.sqrt(252) * 100), 2)
263
+
264
+ result["history_tail"] = df_hist.tail(5).to_dict("records")
265
+ except Exception as e:
266
+ logger.debug("REIT hist failed %s: %s", code, e)
267
+
268
+ # Realtime info from list
269
+ try:
270
+ df_rt = ak.reits_realtime_em()
271
+ if df_rt is not None and not df_rt.empty:
272
+ row = df_rt[df_rt["代码"] == code]
273
+ if not row.empty:
274
+ r = row.iloc[0]
275
+ result["name"] = str(r.get("名称",""))
276
+ result["price"] = result.get("price") or _safe_float(r.get("最新价"))
277
+ result["chg_pct"] = _safe_float(r.get("涨跌幅"))
278
+ result["prev_close"] = _safe_float(r.get("昨收"))
279
+ except Exception as e:
280
+ logger.debug("REIT realtime failed: %s", e)
281
+
282
+ return result
283
+
284
+
285
+ # ── 租金收益率计算器 ──────────────────────────────────────────────────────────
286
+
287
+ def calc_rental_yield(params: dict) -> dict:
288
+ """
289
+ 物业租金收益率计算(毛/净收益率)。
290
+
291
+ 参数:
292
+ purchase_price: 购入价格(万元)
293
+ monthly_rent: 月租金(元)
294
+ annual_costs: 年维护成本(元,默认0)
295
+ tax_rate: 租金税率(小数,默认0.05)
296
+ loan_ratio: 贷款成数(0-1,默认0=全款)
297
+ loan_rate: 贷款年利率(小数,默认0.04)
298
+ loan_years: 贷款年数(默认30)
299
+ """
300
+ price_wan = float(params.get("purchase_price", 0))
301
+ monthly_rent= float(params.get("monthly_rent", 0))
302
+ annual_costs= float(params.get("annual_costs", 0))
303
+ tax_rate = float(params.get("tax_rate", 0.05))
304
+ loan_ratio = float(params.get("loan_ratio", 0))
305
+ loan_rate = float(params.get("loan_rate", 0.04))
306
+ loan_years = int(params.get("loan_years", 30))
307
+
308
+ if price_wan <= 0 or monthly_rent <= 0:
309
+ return {"success": False, "error": "purchase_price 和 monthly_rent 均为必填项"}
310
+
311
+ price_yuan = price_wan * 10000
312
+ annual_rent = monthly_rent * 12
313
+ tax_deduction = annual_rent * tax_rate
314
+ net_rent = annual_rent - tax_deduction - annual_costs
315
+
316
+ # Gross yield
317
+ gross_yield = annual_rent / price_yuan * 100
318
+
319
+ # Net yield (without leverage)
320
+ net_yield = net_rent / price_yuan * 100
321
+
322
+ # Leveraged yield (with mortgage)
323
+ equity = price_yuan * (1 - loan_ratio)
324
+ monthly_payment = 0.0
325
+ annual_interest = 0.0
326
+ if loan_ratio > 0:
327
+ loan_amount = price_yuan * loan_ratio
328
+ monthly_rate = loan_rate / 12
329
+ n = loan_years * 12
330
+ if monthly_rate > 0:
331
+ monthly_payment = loan_amount * monthly_rate * (1 + monthly_rate)**n / ((1 + monthly_rate)**n - 1)
332
+ annual_interest = monthly_payment * 12 - loan_amount / loan_years # approximate
333
+
334
+ leveraged_net_rent = net_rent - (monthly_payment * 12 if loan_ratio > 0 else 0)
335
+ leveraged_yield = (leveraged_net_rent / equity * 100) if equity > 0 else 0
336
+
337
+ # Payback period
338
+ payback_years = price_yuan / net_rent if net_rent > 0 else 999
339
+
340
+ # Cap rate (NOI / price)
341
+ noi = annual_rent - annual_costs
342
+ cap_rate = noi / price_yuan * 100
343
+
344
+ return {
345
+ "success": True,
346
+ "purchase_price_wan": price_wan,
347
+ "monthly_rent": monthly_rent,
348
+ "annual_rent": annual_rent,
349
+ "gross_yield_pct": round(gross_yield, 2),
350
+ "net_yield_pct": round(net_yield, 2),
351
+ "cap_rate_pct": round(cap_rate, 2),
352
+ "leveraged_yield_pct": round(leveraged_yield, 2) if loan_ratio > 0 else None,
353
+ "monthly_payment": round(monthly_payment, 2) if loan_ratio > 0 else None,
354
+ "payback_years": round(payback_years, 1),
355
+ "equity_invested": round(equity / 10000, 2),
356
+ "assessment": (
357
+ "优质标的" if gross_yield >= 5
358
+ else "合理收益" if gross_yield >= 3
359
+ else "收益偏低"
360
+ ),
361
+ "benchmark": "一线城市租金收益率通常 1.5-3%,二线 2-4%,商业地产 4-7%",
362
+ }
363
+
364
+
365
+ # ── 物业估值模型 ──────────────────────────────────────────────────────────────
366
+
367
+ def property_valuation(params: dict) -> dict:
368
+ """
369
+ 物业估值三合一模型:收益法 + DCF + 市场比较法。
370
+
371
+ 必填:
372
+ area_sqm: 建筑面积(平米)
373
+ monthly_rent: 当前月租金(元)
374
+ location_tier: 区位层级("tier1"/"tier2"/"tier3")
375
+
376
+ 可选:
377
+ vacancy_rate: 空置率(默认0.05)
378
+ cap_rate: 市场资本化率(默认由区位自动推算)
379
+ annual_growth: 租金年增长率(默认0.02)
380
+ discount_rate: 折现率(默认0.06)
381
+ hold_years: 持有年数(默认10)
382
+ price_per_sqm: 参考市场单价(元/m²,不填则用租金反推)
383
+ """
384
+ area = float(params.get("area_sqm", 100))
385
+ monthly_rent = float(params.get("monthly_rent", 0))
386
+ tier = str(params.get("location_tier", "tier2")).lower()
387
+ vacancy = float(params.get("vacancy_rate", 0.05))
388
+ annual_growth= float(params.get("annual_growth", 0.02))
389
+ hold_years = int(params.get("hold_years", 10))
390
+
391
+ # Tier-based defaults
392
+ _TIER_DEFAULTS = {
393
+ "tier1": {"cap_rate": 0.025, "discount_rate": 0.055, "price_range": (80000, 200000)},
394
+ "tier2": {"cap_rate": 0.04, "discount_rate": 0.065, "price_range": (20000, 60000)},
395
+ "tier3": {"cap_rate": 0.06, "discount_rate": 0.08, "price_range": (5000, 20000)},
396
+ }
397
+ defaults = _TIER_DEFAULTS.get(tier, _TIER_DEFAULTS["tier2"])
398
+ cap_rate = float(params.get("cap_rate", defaults["cap_rate"]))
399
+ discount_rate = float(params.get("discount_rate", defaults["discount_rate"]))
400
+ price_range = defaults["price_range"]
401
+
402
+ annual_rent = monthly_rent * 12
403
+ noi = annual_rent * (1 - vacancy)
404
+
405
+ # ── 1. 收益法 (Income Approach) ────────────────────────────────────────
406
+ income_value = noi / cap_rate if cap_rate > 0 else 0
407
+
408
+ # ── 2. DCF 法 ──────────────────────────────────────────────────────────
409
+ dcf_value = 0.0
410
+ if _HAS_NP:
411
+ cf = noi
412
+ for yr in range(1, hold_years + 1):
413
+ dcf_value += cf / (1 + discount_rate) ** yr
414
+ cf *= (1 + annual_growth)
415
+ # Terminal value (Gordon growth)
416
+ terminal_growth = min(annual_growth, 0.015)
417
+ terminal_cf = noi * (1 + annual_growth) ** hold_years
418
+ terminal_val = terminal_cf / (discount_rate - terminal_growth)
419
+ dcf_value += terminal_val / (1 + discount_rate) ** hold_years
420
+ else:
421
+ dcf_value = income_value # fallback
422
+
423
+ # ── 3. 市场比较法 (Market Comparable) ─────────────────────────────────
424
+ ref_price_per_sqm = float(params.get("price_per_sqm", 0))
425
+ if ref_price_per_sqm <= 0:
426
+ # Derive from rent using typical rent-to-price ratio
427
+ rent_per_sqm_monthly = monthly_rent / area if area > 0 else 0
428
+ # Typical P/R ratio: 300-600 months for residential, 150-250 for commercial
429
+ ratio = 400 if tier == "tier1" else 300 if tier == "tier2" else 200
430
+ ref_price_per_sqm = rent_per_sqm_monthly * ratio
431
+
432
+ market_value = ref_price_per_sqm * area
433
+
434
+ # ── Weighted average ───────────────────────────────────────────────────
435
+ weights = {"income": 0.4, "dcf": 0.4, "market": 0.2}
436
+ blended = (income_value * weights["income"] +
437
+ dcf_value * weights["dcf"] +
438
+ market_value * weights["market"])
439
+
440
+ # Price range from market benchmarks
441
+ lo = price_range[0] * area
442
+ hi = price_range[1] * area
443
+
444
+ return {
445
+ "success": True,
446
+ "area_sqm": area,
447
+ "monthly_rent": monthly_rent,
448
+ "noi_annual": round(noi, 0),
449
+ "income_approach": round(income_value / 10000, 2), # 万元
450
+ "dcf_approach": round(dcf_value / 10000, 2),
451
+ "market_approach": round(market_value / 10000, 2),
452
+ "blended_value_wan": round(blended / 10000, 2),
453
+ "market_range_wan": [round(lo/10000, 0), round(hi/10000, 0)],
454
+ "price_per_sqm": round(blended / area, 0) if area > 0 else 0,
455
+ "gross_yield_pct": round(annual_rent / blended * 100, 2) if blended > 0 else 0,
456
+ "cap_rate_used": round(cap_rate * 100, 2),
457
+ "discount_rate_used": round(discount_rate * 100, 2),
458
+ "hold_years": hold_years,
459
+ "verdict": (
460
+ "价值被低估" if market_value < blended * 0.85
461
+ else "价值被高估" if market_value > blended * 1.15
462
+ else "定价合理"
463
+ ),
464
+ }
465
+
466
+
467
+ # ── 资产综合评分 ─────────────────────────────────────────────────────────────
468
+
469
+ def asset_location_score(params: dict) -> dict:
470
+ """
471
+ 资产区位 + 业态潜力综合评分(0-100)。
472
+ 结合区位因素、空间条件、业态适配度生成量化评分。
473
+
474
+ 参数:
475
+ city: 城市名称
476
+ district: 区域/商圈
477
+ area_sqm: 面积(平米)
478
+ floor: 楼层
479
+ foot_traffic: 客流量评估 ("high"/"medium"/"low")
480
+ competition: 周边竞争 ("high"/"medium"/"low")
481
+ renovation_allowed: 是否允许改造 (bool)
482
+ open_fire_allowed: 是否允许明火 (bool)
483
+ vacant_days: 空置天数
484
+ expected_rent: 期望租金(元/月)
485
+ """
486
+ city = str(params.get("city",""))
487
+ area = float(params.get("area_sqm", 100))
488
+ floor = int(params.get("floor", 1))
489
+ traffic = str(params.get("foot_traffic", "medium")).lower()
490
+ competition = str(params.get("competition", "medium")).lower()
491
+ reno_ok = bool(params.get("renovation_allowed", True))
492
+ fire_ok = bool(params.get("open_fire_allowed", False))
493
+ vacant_days = int(params.get("vacant_days", 0))
494
+ exp_rent = float(params.get("expected_rent", 0))
495
+
496
+ score = 50 # base
497
+ breakdown = {}
498
+
499
+ # 区位得分 (0-30)
500
+ city_bonus = 30 if city in CN_CITIES_TIER1 else 20 if city in CN_CITIES_TIER2 else 10
501
+ score += city_bonus - 20 # normalize around 50
502
+ breakdown["区位城市"] = f"{city_bonus - 20:+d}"
503
+
504
+ # 客流量 (0-15)
505
+ traffic_score = {"high": 15, "medium": 8, "low": 2}.get(traffic, 8)
506
+ score += traffic_score - 8
507
+ breakdown["客流量"] = f"{traffic_score - 8:+d}"
508
+
509
+ # 楼层因素 (0-10)
510
+ floor_score = 10 if floor == 1 else 6 if floor <= 3 else 2
511
+ score += floor_score - 5
512
+ breakdown["楼层"] = f"{floor_score - 5:+d}"
513
+
514
+ # 面积适配 (0-10) — 50-300m² 最优
515
+ area_score = 10 if 50 <= area <= 300 else 5 if area < 50 else 7
516
+ score += area_score - 5
517
+ breakdown["面积适配"] = f"{area_score - 5:+d}"
518
+
519
+ # 竞争强度(负分)
520
+ comp_penalty = {"high": -8, "medium": -3, "low": 0}.get(competition, -3)
521
+ score += comp_penalty
522
+ breakdown["竞争环境"] = f"{comp_penalty:+d}"
523
+
524
+ # 改造灵活性
525
+ if reno_ok:
526
+ score += 5; breakdown["可改造"] = "+5"
527
+ if fire_ok:
528
+ score += 3; breakdown["允许明火"] = "+3"
529
+
530
+ # 空置惩罚
531
+ if vacant_days > 180:
532
+ score -= 8; breakdown["空置惩罚"] = "-8"
533
+ elif vacant_days > 90:
534
+ score -= 3; breakdown["空置惩罚"] = "-3"
535
+
536
+ # 租金合理性(与城市基准对比)
537
+ if exp_rent > 0 and area > 0:
538
+ per_sqm = exp_rent / area
539
+ benchmarks = {"tier1": 300, "tier2": 120, "tier3": 50}
540
+ tier = "tier1" if city in CN_CITIES_TIER1 else "tier2" if city in CN_CITIES_TIER2 else "tier3"
541
+ bench = benchmarks[tier]
542
+ if per_sqm < bench * 0.7:
543
+ score += 5; breakdown["低于市场租金"] = "+5"
544
+ elif per_sqm > bench * 1.3:
545
+ score -= 5; breakdown["高于市场租金"] = "-5"
546
+
547
+ score = max(0, min(100, score))
548
+
549
+ if score >= 75:
550
+ rating = "A级 — 优质资产"
551
+ elif score >= 60:
552
+ rating = "B级 — 良好资产"
553
+ elif score >= 45:
554
+ rating = "C级 — 一般资产"
555
+ else:
556
+ rating = "D级 — 需改善"
557
+
558
+ # Suitable business types based on conditions
559
+ suitable = []
560
+ if fire_ok and area >= 80:
561
+ suitable.extend(["餐饮/火锅/烤肉"])
562
+ if traffic == "high" and area <= 150:
563
+ suitable.extend(["连锁零售/便利店/奶茶"])
564
+ if reno_ok and area >= 200:
565
+ suitable.extend(["健身房/儿童乐园/培训机构"])
566
+ if floor == 1 and area >= 100:
567
+ suitable.extend(["美容美发/洗车/宠物店"])
568
+ if not suitable:
569
+ suitable = ["轻餐饮", "服务类商铺"]
570
+
571
+ return {
572
+ "success": True,
573
+ "score": score,
574
+ "rating": rating,
575
+ "breakdown": breakdown,
576
+ "suitable_businesses": list(set(suitable))[:5],
577
+ "city": city,
578
+ "area_sqm": area,
579
+ }
580
+
581
+
582
+ # ── 美国住房数据 (FRED) ──────────────────────────────────────────────────────
583
+
584
+ def get_us_housing_data() -> dict:
585
+ """
586
+ 美国住房市场数据:
587
+ - 新屋开工 (HOUST)
588
+ - NAHB 建筑商信心指数
589
+ - S&P/Case-Shiller 20城房价指数 (SPCS20RSA)
590
+ - 30年固定按揭利率 (MORTGAGE30US)
591
+ """
592
+ from macro_tools import _fred_series # reuse existing FRED helper
593
+
594
+ indicators = {
595
+ "housing_starts": ("HOUST", "新屋开工(千套,季调年化)"),
596
+ "nahb_index": ("NAHB REALTORS", "NAHB建筑商信心指数"),
597
+ "case_shiller_20": ("SPCS20RSA", "Case-Shiller 20城房价指数"),
598
+ "mortgage_30y": ("MORTGAGE30US", "30年固定按揭利率(%)"),
599
+ "existing_home_sales":("EXHOSLUSM495S","成屋销售(季调年化)"),
600
+ }
601
+
602
+ # Fallback NAHB via yfinance-equivalent approximation
603
+ _FRED_MAP = {
604
+ "housing_starts": "HOUST",
605
+ "case_shiller_20": "SPCS20RSA",
606
+ "mortgage_30y": "MORTGAGE30US",
607
+ "existing_home_sales":"EXHOSLUSM495S",
608
+ }
609
+
610
+ results = {}
611
+ for key, series_id in _FRED_MAP.items():
612
+ data = _fred_series(series_id, limit=12)
613
+ if data:
614
+ results[key] = {
615
+ "label": indicators[key][1],
616
+ "latest": data[-1],
617
+ "series": data[-6:],
618
+ }
619
+
620
+ # NAHB via hardcoded label (no clean FRED key)
621
+ nahb_data = _fred_series("NHSUSSPT", limit=12)
622
+ if nahb_data:
623
+ results["nahb_index"] = {
624
+ "label": "NAHB建筑商信心指数 (>50=乐观)",
625
+ "latest": nahb_data[-1],
626
+ "series": nahb_data[-6:],
627
+ }
628
+
629
+ if not results:
630
+ return {"success": False, "error": "FRED 数据获取失败"}
631
+
632
+ # Market assessment
633
+ mortgage = (results.get("mortgage_30y", {}).get("latest") or {}).get("value") or 0
634
+ cs_yoy = None # Case-Shiller is level index, need to compute YoY separately
635
+
636
+ assessment = []
637
+ if mortgage > 7:
638
+ assessment.append("按揭利率偏高(>7%),购房负担较重")
639
+ elif mortgage < 4:
640
+ assessment.append("按揭利率较低(<4%),购房环境宽松")
641
+ else:
642
+ assessment.append(f"按揭利率 {mortgage:.2f}%,属正常区间")
643
+
644
+ return {
645
+ "success": True,
646
+ "country": "US",
647
+ "data": results,
648
+ "assessment": assessment,
649
+ "provider": "FRED",
650
+ }
651
+
652
+
653
+ # ── Helper ────────────────────────────────────────────────────────────────────
654
+
655
+ def _safe_float(val) -> Optional[float]:
656
+ try:
657
+ return float(val) if val is not None and str(val) not in ("nan","None","") else None
658
+ except (ValueError, TypeError):
659
+ return None