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,62 @@
1
+ """
2
+ agents/realty — 经营权共创平台 AI Agent 模块
3
+ =============================================
4
+ 9 个领域专属 Agent,继承 BaseAgent,用于:
5
+ - 资产诊断与处置建议
6
+ - 业态匹配与经营方推荐
7
+ - 合同条款结构化生成
8
+ - 分账规则配置建议
9
+ - 流水核验与异常检测
10
+ - 能耗异常分析
11
+ - 合同履约风控
12
+ - 运营优化建议
13
+ - 退出清算方案生成
14
+
15
+ 用法:
16
+ from agents.realty import AssetDiagnosisAgent, RevenueShareAgent
17
+ from agents.realty import REALTY_TEAM # 默认 team 名称列表
18
+ """
19
+
20
+ from .asset_diagnosis import AssetDiagnosisAgent
21
+ from .business_match import BusinessMatchAgent
22
+ from .contract_rules import ContractRulesAgent
23
+ from .revenue_share import RevenueShareAgent
24
+ from .cashflow_verify import CashFlowVerifyAgent
25
+ from .energy_anomaly import EnergyAnomalyAgent
26
+ from .fulfillment_risk import FulfillmentRiskAgent
27
+ from .ops_optimize import OpsOptimizeAgent
28
+ from .exit_settlement import ExitSettlementAgent
29
+
30
+ # 默认完整 team
31
+ REALTY_TEAM = [
32
+ "asset_diagnosis",
33
+ "business_match",
34
+ "contract_rules",
35
+ "revenue_share",
36
+ "cashflow_verify",
37
+ "energy_anomaly",
38
+ "fulfillment_risk",
39
+ "ops_optimize",
40
+ "exit_settlement",
41
+ ]
42
+
43
+ # 风控专项 team(快速预警)
44
+ RISK_TEAM = [
45
+ "cashflow_verify",
46
+ "energy_anomaly",
47
+ "fulfillment_risk",
48
+ ]
49
+
50
+ __all__ = [
51
+ "AssetDiagnosisAgent",
52
+ "BusinessMatchAgent",
53
+ "ContractRulesAgent",
54
+ "RevenueShareAgent",
55
+ "CashFlowVerifyAgent",
56
+ "EnergyAnomalyAgent",
57
+ "FulfillmentRiskAgent",
58
+ "OpsOptimizeAgent",
59
+ "ExitSettlementAgent",
60
+ "REALTY_TEAM",
61
+ "RISK_TEAM",
62
+ ]
@@ -0,0 +1,152 @@
1
+ """
2
+ agents/realty/asset_diagnosis.py — 资产诊断 Agent
3
+ ==================================================
4
+ 判断资产适合出租、出售还是经营权共创。
5
+
6
+ 输入数据(data dict keys):
7
+ asset_info — 资产基础信息 dict(面积、位置、产权状态、空置时间等)
8
+ market_context — 周边业态、租金行情(可选)
9
+ renovation_cost — 估算改造成本(可选)
10
+
11
+ 输出:
12
+ analysis — 详细分析文本
13
+ signal — GOOD=推荐共创 / WATCH=需观察 / CONCERN=建议出租或出售
14
+ (地产健康度尺度,见 agents/signal_scheme.py::REALTY_SCHEME,
15
+ 不是股票 BUY/SELL 那套交易动作词汇)
16
+ key_points — 3-5 条关键结论
17
+ """
18
+ from __future__ import annotations
19
+
20
+ from typing import Any, Dict
21
+ from ..base import BaseAgent, AgentResult
22
+
23
+
24
+ class AssetDiagnosisAgent(BaseAgent):
25
+ name = "asset_diagnosis"
26
+ description = "资产诊断:判断处置方式(出租/出售/共创)并给出保底与分润结构建议"
27
+
28
+ _SYSTEM = (
29
+ "你是一名专业的不动产资产运营顾问,擅长评估商业空间的经营潜力。\n"
30
+ "你的任务是根据资产条件、空置状况、市场行情,判断该资产最适合哪种处置方式:\n"
31
+ " A) 传统出租(固定租金)\n"
32
+ " B) 经营权共创(保底+流水分润)\n"
33
+ " C) 出售(一次性变现)\n\n"
34
+ "输出格式要求:\n"
35
+ "1. 推荐方式及理由(2-3句话)\n"
36
+ "2. 如推荐共创,给出:保底收益建议(元/月)、分润比例建议(%)、适合业态Top3\n"
37
+ "3. 主要风险提示(1-2条)\n"
38
+ "4. 综合评分:[适合共创] / [建议观察] / [不建议共创]"
39
+ )
40
+
41
+ async def analyze(self, symbol: str, data: Dict[str, Any]) -> AgentResult:
42
+ asset = data.get("asset_info", {})
43
+ market = data.get("market_context", {})
44
+ reno = data.get("renovation_cost", 0)
45
+
46
+ # 构建资产摘要
47
+ area = asset.get("area", 0)
48
+ location = asset.get("location", "未知位置")
49
+ vacancy_days= asset.get("vacancy_days", 0)
50
+ expected_rent = asset.get("expected_rent", 0)
51
+ floor_height = asset.get("floor_height", 0)
52
+ business_types = asset.get("allowed_business", [])
53
+ prohibited = asset.get("prohibited_business", [])
54
+ property_state= asset.get("property_state", "正常")
55
+
56
+ user_prompt = (
57
+ f"资产信息:\n"
58
+ f" 位置: {location}\n"
59
+ f" 面积: {area} m² 层高: {floor_height}m\n"
60
+ f" 产权状态: {property_state} 空置天数: {vacancy_days}天\n"
61
+ f" 评估租金: {expected_rent}元/月 改造成本估算: {reno}元\n"
62
+ f" 适合业态: {', '.join(business_types) or '未指定'}\n"
63
+ f" 禁止业态: {', '.join(prohibited) or '无'}\n"
64
+ f" 市场背景: {market.get('summary', '暂无周边数据')}\n\n"
65
+ "请根据以上信息完成资产诊断。"
66
+ )
67
+
68
+ analysis = await self._call_llm(self._SYSTEM, user_prompt, max_tokens=600)
69
+ if not analysis:
70
+ analysis = _template_diagnosis(asset, reno)
71
+
72
+ signal = _score_to_signal(asset, reno)
73
+ confidence = _calc_confidence(asset)
74
+ key_points = _extract_key_points(asset, reno, market)
75
+
76
+ return AgentResult(
77
+ agent = self.name,
78
+ symbol = symbol,
79
+ analysis = analysis,
80
+ confidence = confidence,
81
+ signal = signal,
82
+ key_points = key_points,
83
+ data_used = {"asset_info": asset, "renovation_cost": reno},
84
+ )
85
+
86
+
87
+ # ── 内部辅助函数 ──────────────────────────────────────────────────────────────
88
+
89
+ def _score_to_signal(asset: Dict, reno: float) -> str:
90
+ """根据资产条件估算共创适合度"""
91
+ area = asset.get("area", 0)
92
+ vacancy_days = asset.get("vacancy_days", 0)
93
+ rent = asset.get("expected_rent", 0)
94
+ allowed = asset.get("allowed_business", [])
95
+ state = asset.get("property_state", "正常")
96
+
97
+ score = 0
98
+ if area >= 50: score += 1
99
+ if area >= 200: score += 1
100
+ if vacancy_days >= 90: score += 1 # 空置越久越需要共创
101
+ if rent > 0: score += 1
102
+ if len(allowed) >= 2: score += 1
103
+ if state != "正常": score -= 2
104
+ if reno > rent * 24: score -= 1 # 改造成本超过2年租金则不划算
105
+
106
+ if score >= 4: return "GOOD" # 非常适合共创
107
+ if score >= 2: return "WATCH" # 需进一步评估
108
+ return "CONCERN" # 建议传统出租或出售
109
+
110
+
111
+ def _calc_confidence(asset: Dict) -> float:
112
+ """数据越完整,置信度越高"""
113
+ fields = ["area", "location", "expected_rent", "allowed_business", "property_state"]
114
+ filled = sum(1 for f in fields if asset.get(f))
115
+ return round(0.5 + 0.1 * filled, 2)
116
+
117
+
118
+ def _extract_key_points(asset: Dict, reno: float, market: Dict) -> list:
119
+ points = []
120
+ area = asset.get("area", 0)
121
+ vacancy = asset.get("vacancy_days", 0)
122
+ rent = asset.get("expected_rent", 0)
123
+ allowed = asset.get("allowed_business", [])
124
+
125
+ if area >= 200:
126
+ points.append(f"面积 {area}m² 具备多业态组合潜力")
127
+ if vacancy >= 90:
128
+ points.append(f"已空置 {vacancy} 天,引入经营方具有紧迫性")
129
+ if rent > 0:
130
+ points.append(f"评估租金 {rent:,.0f} 元/月,保底基准参考值")
131
+ if allowed:
132
+ points.append(f"适合业态:{', '.join(allowed[:3])}")
133
+ if reno > 0:
134
+ points.append(f"改造成本约 {reno:,.0f} 元,需纳入分润谈判")
135
+ return points[:5]
136
+
137
+
138
+ def _template_diagnosis(asset: Dict, reno: float) -> str:
139
+ area = asset.get("area", 0)
140
+ vacancy = asset.get("vacancy_days", 0)
141
+ rent = asset.get("expected_rent", 0)
142
+ allowed = asset.get("allowed_business", [])
143
+
144
+ suitability = "建议经营权共创" if area >= 100 and vacancy >= 30 else "建议先评估市场需求"
145
+ return (
146
+ f"资产诊断(模板):\n"
147
+ f" 面积: {area}m² 空置: {vacancy}天 评估租金: {rent:,.0f}元/月\n"
148
+ f" 适合业态: {', '.join(allowed[:3]) or '待评估'}\n"
149
+ f" 改造成本估算: {reno:,.0f}元\n"
150
+ f" 诊断结论: {suitability}\n"
151
+ f" 建议保底: {int(rent * 0.7):,}元/月 建议分润比例: 8-12%"
152
+ )
@@ -0,0 +1,166 @@
1
+ """
2
+ agents/realty/business_match.py — 业态匹配 Agent
3
+ =================================================
4
+ 匹配资产条件与经营方能力,给出业态组合建议。
5
+
6
+ 输入数据(data dict keys):
7
+ asset_info — 资产基础信息(面积、位置、层高、客流等)
8
+ operators — 候选经营方列表(可选)
9
+ market_context — 周边业态、人口、消费力数据(可选)
10
+
11
+ 输出:
12
+ analysis — 业态匹配建议
13
+ signal — GOOD=高度匹配 / WATCH=一般匹配 / CONCERN=低匹配
14
+ (地产健康度尺度,见 signal_scheme.py)
15
+ key_points — 推荐业态 Top3 + 理由
16
+ """
17
+ from __future__ import annotations
18
+
19
+ from typing import Any, Dict, List
20
+ from ..base import BaseAgent, AgentResult
21
+
22
+
23
+ class BusinessMatchAgent(BaseAgent):
24
+ name = "business_match"
25
+ description = "业态匹配:根据空间条件和市场数据推荐适合的经营业态组合"
26
+
27
+ _SYSTEM = (
28
+ "你是一名商业空间招商运营专家,擅长根据空间条件匹配最优业态。\n"
29
+ "分析维度:\n"
30
+ " 1. 空间适配度(面积、层高、水电、消防条件)\n"
31
+ " 2. 市场需求(周边客群、竞争格局、消费力)\n"
32
+ " 3. 经营方能力(品牌力、资金实力、运营经验)\n"
33
+ " 4. 收益预期(流水规模、分润可行性)\n\n"
34
+ "输出格式:\n"
35
+ "1. 推荐业态 Top3(每个包含:业态名称、预期月流水范围、分润比例建议、空间需求匹配度)\n"
36
+ "2. 经营方匹配建议(如有候选方)\n"
37
+ "3. 不建议的业态及原因\n"
38
+ "4. 综合匹配评级:[高度匹配] / [一般匹配] / [谨慎入场]"
39
+ )
40
+
41
+ async def analyze(self, symbol: str, data: Dict[str, Any]) -> AgentResult:
42
+ asset = data.get("asset_info", {})
43
+ operators = data.get("operators", [])
44
+ market = data.get("market_context", {})
45
+
46
+ area = asset.get("area", 0)
47
+ location = asset.get("location", "")
48
+ floor_ht = asset.get("floor_height", 0)
49
+ power_cap = asset.get("power_capacity", "未知")
50
+ allowed = asset.get("allowed_business", [])
51
+ prohibited = asset.get("prohibited_business", [])
52
+ foot_traffic= asset.get("foot_traffic", "未知")
53
+ can_fire = asset.get("open_fire_allowed", False)
54
+ can_renovate= asset.get("renovation_allowed", True)
55
+
56
+ op_summary = ""
57
+ if operators:
58
+ op_lines = []
59
+ for op in operators[:3]:
60
+ op_lines.append(
61
+ f" - {op.get('name','未知')}: 行业={op.get('industry','未知')}, "
62
+ f"预算={op.get('budget',0):,}元, 品牌={op.get('brand_level','未知')}"
63
+ )
64
+ op_summary = "候选经营方:\n" + "\n".join(op_lines)
65
+
66
+ user_prompt = (
67
+ f"资产条件:\n"
68
+ f" 位置: {location} 面积: {area}m² 层高: {floor_ht}m\n"
69
+ f" 电容量: {power_cap} 明火许可: {'是' if can_fire else '否'}\n"
70
+ f" 可改造: {'是' if can_renovate else '否'} 客流量: {foot_traffic}\n"
71
+ f" 允许业态: {', '.join(allowed) or '无限制'}\n"
72
+ f" 禁止业态: {', '.join(prohibited) or '无'}\n"
73
+ f" 市场背景: {market.get('summary','暂无')}\n"
74
+ f"{op_summary}\n\n"
75
+ "请给出业态匹配分析。"
76
+ )
77
+
78
+ analysis = await self._call_llm(self._SYSTEM, user_prompt, max_tokens=700)
79
+ if not analysis:
80
+ analysis = _template_match(asset, operators)
81
+
82
+ signal = _match_signal(asset, market)
83
+ confidence = _match_confidence(asset, market)
84
+ key_points = _match_key_points(asset, operators, market)
85
+
86
+ return AgentResult(
87
+ agent = self.name,
88
+ symbol = symbol,
89
+ analysis = analysis,
90
+ confidence = confidence,
91
+ signal = signal,
92
+ key_points = key_points,
93
+ data_used = {"asset_info": asset, "operators_count": len(operators)},
94
+ )
95
+
96
+
97
+ # ── 辅助函数 ──────────────────────────────────────────────────────────────────
98
+
99
+ def _match_signal(asset: Dict, market: Dict) -> str:
100
+ score = 0
101
+ area = asset.get("area", 0)
102
+ allowed = asset.get("allowed_business", [])
103
+ foot_traffic = str(asset.get("foot_traffic", "")).lower()
104
+ ft_score = asset.get("foot_traffic_score", 0) # numeric 0-10
105
+
106
+ # market_score: direct int, or derive from market_maturity string
107
+ market_score = market.get("score", 0)
108
+ if not market_score:
109
+ maturity = str(market.get("market_maturity", "")).lower()
110
+ market_score = {"high": 8, "medium": 5, "low": 2, "成熟": 8, "一般": 5, "新兴": 4}.get(maturity, 0)
111
+
112
+ if area >= 100: score += 1
113
+ if area >= 300: score += 1
114
+ if len(allowed) >= 3: score += 1
115
+ if "高" in foot_traffic or "large" in foot_traffic or ft_score >= 7: score += 1
116
+ if market_score >= 7: score += 1
117
+ elif 0 < market_score <= 3: score -= 1
118
+
119
+ if score >= 4: return "GOOD"
120
+ if score >= 2: return "WATCH"
121
+ return "CONCERN"
122
+
123
+
124
+ def _match_confidence(asset: Dict, market: Dict) -> float:
125
+ fields = ["area", "location", "foot_traffic", "allowed_business", "floor_height"]
126
+ filled = sum(1 for f in fields if asset.get(f))
127
+ market_bonus = 0.1 if market.get("summary") else 0
128
+ return round(min(0.9, 0.4 + 0.1 * filled + market_bonus), 2)
129
+
130
+
131
+ def _match_key_points(asset: Dict, operators: List, market: Dict) -> List[str]:
132
+ points = []
133
+ area = asset.get("area", 0)
134
+ allowed = asset.get("allowed_business", [])
135
+ can_fire= asset.get("open_fire_allowed", False)
136
+
137
+ if allowed:
138
+ points.append(f"允许业态:{', '.join(allowed[:3])} 等 {len(allowed)} 类")
139
+ if area:
140
+ points.append(f"面积 {area}m²,适合{'大型品牌' if area >= 300 else '中小型经营方'}")
141
+ if not can_fire:
142
+ points.append("不允许明火,餐饮业态受限(建议西餐/轻食/烘焙)")
143
+ if operators:
144
+ points.append(f"共 {len(operators)} 个候选经营方,需进一步资质评估")
145
+ if market.get("competitors"):
146
+ points.append(f"周边竞争: {market['competitors']}")
147
+ return points[:5]
148
+
149
+
150
+ def _template_match(asset: Dict, operators: List) -> str:
151
+ area = asset.get("area", 0)
152
+ allowed = asset.get("allowed_business", [])
153
+ can_fire= asset.get("open_fire_allowed", False)
154
+
155
+ top3 = allowed[:3] if allowed else ["零售", "轻餐饮", "生活服务"]
156
+ if not can_fire and "餐饮" in top3:
157
+ top3 = [b for b in top3 if "餐饮" not in b] + ["轻食/咖啡"]
158
+
159
+ return (
160
+ f"业态匹配分析(模板):\n"
161
+ f" 面积: {area}m² 适合业态: {', '.join(allowed[:5]) or '无限制'}\n"
162
+ f" 推荐 Top3:\n"
163
+ + "\n".join(f" {i+1}. {b}(预期流水 {(area*200*(i+1)):,.0f}-{area*400*(i+1):,.0f}元/月)"
164
+ for i, b in enumerate(top3[:3]))
165
+ + f"\n 候选经营方数: {len(operators)}"
166
+ )
@@ -0,0 +1,209 @@
1
+ """
2
+ agents/realty/cashflow_verify.py — 流水核验 Agent
3
+ =================================================
4
+ 判断经营方申报流水是否真实,识别私账/逃费/低报行为。
5
+
6
+ 输入数据(data dict keys):
7
+ pos_transactions — POS/扫码流水列表(可选)
8
+ bank_statements — 银行流水(可选)
9
+ delivery_revenue — 外卖/团购收入(可选)
10
+ inventory_data — 库存进货数据(可选)
11
+ energy_data — 水电用量(可选)
12
+ declared_revenue — 经营方自报流水
13
+ expected_revenue — 系统预估流水(基于历史/业态/客流)
14
+
15
+ 输出:
16
+ analysis — 核验报告
17
+ signal — GOOD=流水真实 / WATCH=轻度异常需复核 / CONCERN=疑似造假/私账
18
+ / SEVERE=多项异常+差距极大(地产健康度尺度,见 signal_scheme.py)
19
+ key_points — 异常项目清单
20
+ """
21
+ from __future__ import annotations
22
+
23
+ from typing import Any, Dict, List
24
+ from ..base import BaseAgent, AgentResult
25
+
26
+
27
+ class CashFlowVerifyAgent(BaseAgent):
28
+ name = "cashflow_verify"
29
+ description = "流水核验:多源数据交叉比对,识别私账/逃费/低报流水行为"
30
+
31
+ _SYSTEM = (
32
+ "你是一名专业的财务审计和风控专家,擅长识别经营流水中的异常。\n"
33
+ "请对流水核验结果进行分析:\n"
34
+ " 1. 各数据源的一致性分析(POS/银行/外卖/库存/能耗)\n"
35
+ " 2. 申报流水与预估流水的差异分析\n"
36
+ " 3. 识别的异常模式(如:只用现金、收款码变更、能耗与流水不匹配)\n"
37
+ " 4. 风险等级判断:[正常] / [需复核] / [疑似私账] / [建议稽查]\n"
38
+ " 5. 具体建议:要求补充哪些凭证,或启动哪种核查程序"
39
+ )
40
+
41
+ async def analyze(self, symbol: str, data: Dict[str, Any]) -> AgentResult:
42
+ # 支持两种输入格式:
43
+ # 标量简化模式: pos_revenue=55000, bank_revenue=48000, cash_ratio=0.35
44
+ # 列表精细模式: pos_transactions=[{amount,payment_type,...}], bank_statements=[{amount,type,...}]
45
+ pos = data.get("pos_transactions", [])
46
+ bank = data.get("bank_statements", [])
47
+ delivery = data.get("delivery_revenue", 0)
48
+ inventory= data.get("inventory_data", {})
49
+ energy = data.get("energy_data", {})
50
+ declared = data.get("declared_revenue", 0)
51
+ expected = data.get("expected_revenue", 0)
52
+
53
+ # 标量快捷键:如果没有列表数据但有聚合数字,转换为单条虚拟记录
54
+ if not pos and data.get("pos_revenue", 0):
55
+ pos_total_val = data["pos_revenue"]
56
+ cash_r = data.get("cash_ratio", 0.0) # 0.0~1.0
57
+ if cash_r > 0:
58
+ pos = [
59
+ {"amount": pos_total_val * (1 - cash_r), "payment_type": "digital"},
60
+ {"amount": pos_total_val * cash_r, "payment_type": "cash"},
61
+ ]
62
+ else:
63
+ pos = [{"amount": pos_total_val, "payment_type": "digital"}]
64
+ if not bank and data.get("bank_revenue", 0):
65
+ bank = [{"amount": data["bank_revenue"], "type": "IN"}]
66
+ if not delivery and data.get("wechat_revenue", 0):
67
+ delivery = data.get("wechat_revenue", 0) + data.get("alipay_revenue", 0)
68
+
69
+ # 交叉核验
70
+ check = _cross_verify(pos, bank, delivery, inventory, energy, declared, expected)
71
+
72
+ user_prompt = (
73
+ f"流水核验数据:\n"
74
+ f" 经营方申报流水: {declared:,.2f}元\n"
75
+ f" 系统预估流水: {expected:,.2f}元\n"
76
+ f" 差异率: {check['gap_pct']:.1f}%({'偏低' if check['gap_pct'] < 0 else '偏高'})\n\n"
77
+ f"数据源汇总:\n"
78
+ f" POS/扫码流水: {check['pos_total']:,.2f}元({len(pos)}笔)\n"
79
+ f" 银行流水: {check['bank_total']:,.2f}元\n"
80
+ f" 外卖/团购收入: {delivery:,.2f}元\n"
81
+ f" 能耗指数(用电): {energy.get('electricity_kwh',0)} kWh\n"
82
+ f" 库存周转估算: {check['inventory_implied_revenue']:,.2f}元\n\n"
83
+ f"发现异常:\n"
84
+ + "\n".join(f" - {a}" for a in check["anomalies"])
85
+ + "\n\n请完成流水核验报告。"
86
+ )
87
+
88
+ analysis = await self._call_llm(self._SYSTEM, user_prompt, max_tokens=700)
89
+ if not analysis:
90
+ analysis = _template_verify(check, declared, expected)
91
+
92
+ signal = _verify_signal(check)
93
+ confidence = _verify_confidence(check)
94
+ key_points = _verify_key_points(check, declared, expected)
95
+
96
+ return AgentResult(
97
+ agent = self.name,
98
+ symbol = symbol,
99
+ analysis = analysis,
100
+ confidence = confidence,
101
+ signal = signal,
102
+ key_points = key_points,
103
+ data_used = {"verify_result": check},
104
+ )
105
+
106
+
107
+ # ── 核验逻辑 ──────────────────────────────────────────────────────────────────
108
+
109
+ def _cross_verify(
110
+ pos: List[Dict], bank: List[Dict], delivery: float,
111
+ inventory: Dict, energy: Dict, declared: float, expected: float,
112
+ ) -> Dict:
113
+ pos_total = sum(t.get("amount", 0) for t in pos if not t.get("is_refund"))
114
+ bank_total = sum(t.get("amount", 0) for t in bank if t.get("type") in ("IN", "in", None))
115
+
116
+ # 库存推算流水(进货金额 * 毛利率倒推)
117
+ purchase_cost = inventory.get("purchase_cost", 0)
118
+ gross_margin = inventory.get("expected_margin_pct", 40) / 100
119
+ inv_implied = purchase_cost / (1 - gross_margin) if gross_margin < 1 else 0
120
+
121
+ # 综合核验流水(取最大可信来源)
122
+ verified_sources = [s for s in [pos_total, bank_total, delivery + pos_total] if s > 0]
123
+ verified_max = max(verified_sources) if verified_sources else declared
124
+
125
+ gap_pct = ((declared - expected) / expected * 100) if expected else 0
126
+ source_gap = ((declared - verified_max) / declared * 100) if declared else 0
127
+
128
+ anomalies = []
129
+
130
+ # 申报 vs 预期 差距
131
+ if gap_pct < -30:
132
+ anomalies.append(f"申报流水比预估低{abs(gap_pct):.1f}%,偏差较大")
133
+ if gap_pct < -50:
134
+ anomalies.append("申报流水不足预估50%,建议启动稽查")
135
+
136
+ # POS vs 银行 不一致
137
+ if pos_total > 0 and bank_total > 0:
138
+ consistency = abs(pos_total - bank_total) / max(pos_total, bank_total)
139
+ if consistency > 0.2:
140
+ anomalies.append(f"POS流水({pos_total:,.0f})与银行流水({bank_total:,.0f})差异{consistency*100:.1f}%")
141
+
142
+ # 申报 vs 核验源 差距
143
+ if declared > 0 and verified_max > 0 and source_gap > 25:
144
+ anomalies.append(f"申报流水比可核验来源高{source_gap:.1f}%,可能虚报")
145
+ elif declared < verified_max * 0.7:
146
+ anomalies.append(f"申报流水({declared:,.0f})低于可核验来源({verified_max:,.0f}),疑似漏报")
147
+
148
+ # 能耗异常(有水电无流水)
149
+ elec = energy.get("electricity_kwh", 0)
150
+ if elec > 0 and declared == 0:
151
+ anomalies.append("有水电消耗记录但申报流水为零,疑似有经营未申报")
152
+
153
+ # 库存推算差距
154
+ if inv_implied > 0 and declared < inv_implied * 0.6:
155
+ anomalies.append(f"库存进货推算流水{inv_implied:,.0f}元,申报仅{declared:,.0f}元,差距较大")
156
+
157
+ # 现金比例过高
158
+ cash_txns = [t for t in pos if t.get("payment_type") in ("cash", "现金")]
159
+ if pos and len(cash_txns) / len(pos) > 0.3:
160
+ anomalies.append(f"现金交易占比{len(cash_txns)/len(pos)*100:.1f}%偏高,难以核验")
161
+
162
+ return {
163
+ "pos_total": pos_total,
164
+ "bank_total": bank_total,
165
+ "delivery": delivery,
166
+ "inventory_implied_revenue": inv_implied,
167
+ "verified_max": verified_max,
168
+ "gap_pct": gap_pct,
169
+ "source_gap_pct": source_gap,
170
+ "anomalies": anomalies,
171
+ "anomaly_count": len(anomalies),
172
+ }
173
+
174
+
175
+ def _verify_signal(check: Dict) -> str:
176
+ n = check.get("anomaly_count", 0)
177
+ gap = abs(check.get("gap_pct", 0))
178
+ if n == 0 and gap < 15: return "GOOD"
179
+ if n <= 1 and gap < 30: return "WATCH"
180
+ if n <= 2 and gap < 50: return "CONCERN"
181
+ return "SEVERE"
182
+
183
+
184
+ def _verify_confidence(check: Dict) -> float:
185
+ sources = sum(1 for k in ["pos_total", "bank_total", "delivery"] if check.get(k, 0) > 0)
186
+ return round(min(0.9, 0.5 + 0.15 * sources), 2)
187
+
188
+
189
+ def _verify_key_points(check: Dict, declared: float, expected: float) -> List[str]:
190
+ pts = [f"申报流水: {declared:,.2f}元 预估流水: {expected:,.2f}元 "
191
+ f"差异: {check['gap_pct']:+.1f}%"]
192
+ if check["pos_total"]:
193
+ pts.append(f"POS核验流水: {check['pos_total']:,.2f}元")
194
+ for a in check["anomalies"][:3]:
195
+ pts.append(f"异常: {a}")
196
+ return pts[:5]
197
+
198
+
199
+ def _template_verify(check: Dict, declared: float, expected: float) -> str:
200
+ status = "正常" if check["anomaly_count"] == 0 else "发现异常"
201
+ return (
202
+ f"流水核验报告(模板):\n"
203
+ f" 申报流水: {declared:,.2f}元 预估流水: {expected:,.2f}元 "
204
+ f"差异: {check['gap_pct']:+.1f}%\n"
205
+ f" POS流水: {check['pos_total']:,.2f}元 银行流水: {check['bank_total']:,.2f}元\n"
206
+ f" 核验状态: {status}\n"
207
+ + (" 发现异常:\n" + "\n".join(f" • {a}" for a in check["anomalies"])
208
+ if check["anomalies"] else " 无异常项目")
209
+ )