aria-code 0.167.0__py3-none-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (601) hide show
  1. aria_bots/__init__.py +7 -0
  2. aria_bots/channels/README.md +97 -0
  3. aria_bots/channels/__init__.py +1 -0
  4. aria_bots/channels/approvals.py +263 -0
  5. aria_bots/channels/conversation.py +280 -0
  6. aria_bots/channels/digest.py +186 -0
  7. aria_bots/channels/registry.py +95 -0
  8. aria_bots/engine.py +144 -0
  9. aria_bots/feishu_bot.py +1385 -0
  10. aria_bots/logistics/__init__.py +1 -0
  11. aria_bots/logistics/carriers.py +253 -0
  12. aria_bots/logistics/freight.py +154 -0
  13. aria_bots/logistics/inventory.py +242 -0
  14. aria_bots/logistics/mcp_server.py +156 -0
  15. aria_bots/logistics/records.py +119 -0
  16. aria_bots/logistics/tenancy.py +152 -0
  17. aria_bots/market.py +91 -0
  18. aria_bots/relay_client.py +326 -0
  19. aria_bots/telegram_bot.py +299 -0
  20. aria_code/__init__.py +111 -0
  21. aria_code/_bin/aria-native.exe +0 -0
  22. aria_code/_bin/aria-runtime.exe +0 -0
  23. aria_code/_version.py +20 -0
  24. aria_code/account_memory.py +296 -0
  25. aria_code/account_session.py +166 -0
  26. aria_code/adk_apps/__init__.py +1 -0
  27. aria_code/adk_apps/aria_code_review/__init__.py +1 -0
  28. aria_code/adk_apps/aria_code_review/agent.py +32 -0
  29. aria_code/adk_apps/aria_research/__init__.py +1 -0
  30. aria_code/adk_apps/aria_research/agent.py +31 -0
  31. aria_code/agents/__init__.py +49 -0
  32. aria_code/agents/base.py +329 -0
  33. aria_code/agents/code_review.py +200 -0
  34. aria_code/agents/deep/__init__.py +37 -0
  35. aria_code/agents/deep/calibration_loop.py +144 -0
  36. aria_code/agents/deep/critic.py +125 -0
  37. aria_code/agents/deep/deepen.py +193 -0
  38. aria_code/agents/deep/models.py +149 -0
  39. aria_code/agents/deep/pipeline.py +164 -0
  40. aria_code/agents/deep/quant_fusion.py +192 -0
  41. aria_code/agents/deep/themes.py +95 -0
  42. aria_code/agents/deep/tiers.py +106 -0
  43. aria_code/agents/engineering/__init__.py +14 -0
  44. aria_code/agents/engineering/coder.py +222 -0
  45. aria_code/agents/engineering/tester.py +145 -0
  46. aria_code/agents/financial/__init__.py +21 -0
  47. aria_code/agents/financial/cashflow_burn.py +119 -0
  48. aria_code/agents/financial/catalyst.py +279 -0
  49. aria_code/agents/financial/corporate_finance.py +175 -0
  50. aria_code/agents/financial/debate.py +224 -0
  51. aria_code/agents/financial/earnings.py +315 -0
  52. aria_code/agents/financial/fundamental.py +178 -0
  53. aria_code/agents/financial/macro.py +111 -0
  54. aria_code/agents/financial/news.py +231 -0
  55. aria_code/agents/financial/risk.py +146 -0
  56. aria_code/agents/financial/sector.py +279 -0
  57. aria_code/agents/financial/strategist.py +121 -0
  58. aria_code/agents/financial/stripe_revenue.py +127 -0
  59. aria_code/agents/financial/synthesis.py +274 -0
  60. aria_code/agents/financial/technical.py +277 -0
  61. aria_code/agents/orchestrator.py +44 -0
  62. aria_code/agents/portfolio_agent.py +392 -0
  63. aria_code/agents/realty/__init__.py +62 -0
  64. aria_code/agents/realty/asset_diagnosis.py +152 -0
  65. aria_code/agents/realty/business_match.py +166 -0
  66. aria_code/agents/realty/cashflow_verify.py +209 -0
  67. aria_code/agents/realty/contract_rules.py +210 -0
  68. aria_code/agents/realty/energy_anomaly.py +189 -0
  69. aria_code/agents/realty/exit_settlement.py +208 -0
  70. aria_code/agents/realty/fulfillment_risk.py +206 -0
  71. aria_code/agents/realty/ops_optimize.py +160 -0
  72. aria_code/agents/realty/revenue_share.py +215 -0
  73. aria_code/agents/registry.py +168 -0
  74. aria_code/agents/signal_scheme.py +111 -0
  75. aria_code/agents/sports/__init__.py +0 -0
  76. aria_code/agents/sports/football_agent.py +197 -0
  77. aria_code/agents/supervisor.py +113 -0
  78. aria_code/agents/team.py +500 -0
  79. aria_code/agents/video/__init__.py +16 -0
  80. aria_code/agents/video/director.py +376 -0
  81. aria_code/agents/video/qa_agent.py +143 -0
  82. aria_code/agents/video/script_doctor.py +112 -0
  83. aria_code/agents/warehouse/__init__.py +25 -0
  84. aria_code/agents/warehouse/contracts.py +26 -0
  85. aria_code/agents/warehouse/dashboard.py +155 -0
  86. aria_code/agents/warehouse/fulfillment_leadtime.py +95 -0
  87. aria_code/agents/warehouse/inbound_exceptions.py +25 -0
  88. aria_code/agents/warehouse/inventory_health.py +24 -0
  89. aria_code/agents/warehouse/local_run.py +71 -0
  90. aria_code/agents/warehouse/local_snapshot.py +132 -0
  91. aria_code/agents/warehouse/logistics_cost.py +84 -0
  92. aria_code/agents/warehouse/logistics_sync.py +21 -0
  93. aria_code/agents/warehouse/navigation.schema.json +110 -0
  94. aria_code/agents/warehouse/task_agent.py +172 -0
  95. aria_code/agents/warehouse/workflow.py +31 -0
  96. aria_code/aliyun_data_client.py +4 -0
  97. aria_code/apps/README.md +12 -0
  98. aria_code/apps/__init__.py +2 -0
  99. aria_code/apps/channels/README.md +9 -0
  100. aria_code/apps/channels/__init__.py +17 -0
  101. aria_code/apps/channels/approvals.py +11 -0
  102. aria_code/apps/channels/conversation.py +11 -0
  103. aria_code/apps/channels/digest.py +11 -0
  104. aria_code/apps/channels/intake.py +92 -0
  105. aria_code/apps/channels/registry.py +11 -0
  106. aria_code/apps/channels/tradingview.py +94 -0
  107. aria_code/apps/cli/README.md +13 -0
  108. aria_code/apps/cli/__init__.py +2 -0
  109. aria_code/apps/cli/_optional.py +120 -0
  110. aria_code/apps/cli/app_server.py +656 -0
  111. aria_code/apps/cli/auto_update.py +125 -0
  112. aria_code/apps/cli/bootstrap.py +206 -0
  113. aria_code/apps/cli/broker_render.py +206 -0
  114. aria_code/apps/cli/chat_turn.py +1327 -0
  115. aria_code/apps/cli/cloud_models.py +231 -0
  116. aria_code/apps/cli/codegen_paths.py +29 -0
  117. aria_code/apps/cli/commands/__init__.py +34 -0
  118. aria_code/apps/cli/commands/_business_render.py +78 -0
  119. aria_code/apps/cli/commands/_ui.py +71 -0
  120. aria_code/apps/cli/commands/analysis_cmds.py +378 -0
  121. aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
  122. aria_code/apps/cli/commands/auth_cmds.py +215 -0
  123. aria_code/apps/cli/commands/backtest_cmds.py +2518 -0
  124. aria_code/apps/cli/commands/broker_cmds.py +1236 -0
  125. aria_code/apps/cli/commands/business_workflow_cmds.py +379 -0
  126. aria_code/apps/cli/commands/canvas_cmds.py +77 -0
  127. aria_code/apps/cli/commands/catalog.py +189 -0
  128. aria_code/apps/cli/commands/core_cmds.py +2748 -0
  129. aria_code/apps/cli/commands/data_cmds.py +521 -0
  130. aria_code/apps/cli/commands/diagnostic_cmds.py +285 -0
  131. aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
  132. aria_code/apps/cli/commands/engineering_cmds.py +399 -0
  133. aria_code/apps/cli/commands/file_cmds.py +312 -0
  134. aria_code/apps/cli/commands/finance_service_cmds.py +60 -0
  135. aria_code/apps/cli/commands/fx_commodity_cmds.py +123 -0
  136. aria_code/apps/cli/commands/market.py +544 -0
  137. aria_code/apps/cli/commands/market_cmds.py +1419 -0
  138. aria_code/apps/cli/commands/market_context.py +432 -0
  139. aria_code/apps/cli/commands/market_render.py +7 -0
  140. aria_code/apps/cli/commands/model_cmds.py +2374 -0
  141. aria_code/apps/cli/commands/ops_cmds.py +721 -0
  142. aria_code/apps/cli/commands/orchestrator_cmds.py +181 -0
  143. aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
  144. aria_code/apps/cli/commands/portfolio_cmds.py +1316 -0
  145. aria_code/apps/cli/commands/report.py +476 -0
  146. aria_code/apps/cli/commands/scaffold_templates.py +617 -0
  147. aria_code/apps/cli/commands/session_cmds.py +236 -0
  148. aria_code/apps/cli/commands/session_ux_cmds.py +359 -0
  149. aria_code/apps/cli/commands/team.py +723 -0
  150. aria_code/apps/cli/commands/team_render.py +8 -0
  151. aria_code/apps/cli/commands/ui_cmds.py +401 -0
  152. aria_code/apps/cli/commands/warehouse_cmds.py +297 -0
  153. aria_code/apps/cli/commands/workflow_cmds.py +629 -0
  154. aria_code/apps/cli/commands/workspace_cmds.py +1543 -0
  155. aria_code/apps/cli/config_paths.py +73 -0
  156. aria_code/apps/cli/config_store.py +158 -0
  157. aria_code/apps/cli/constants.py +88 -0
  158. aria_code/apps/cli/context.py +34 -0
  159. aria_code/apps/cli/conversation_policy.py +53 -0
  160. aria_code/apps/cli/deterministic.py +200 -0
  161. aria_code/apps/cli/direct.py +48 -0
  162. aria_code/apps/cli/earnings_view.py +172 -0
  163. aria_code/apps/cli/endpoint_migration.py +64 -0
  164. aria_code/apps/cli/exec_events.py +255 -0
  165. aria_code/apps/cli/file_references.py +155 -0
  166. aria_code/apps/cli/finance_service_catalog.py +111 -0
  167. aria_code/apps/cli/football_reports.py +231 -0
  168. aria_code/apps/cli/github_app_auth.py +135 -0
  169. aria_code/apps/cli/google_login.py +239 -0
  170. aria_code/apps/cli/handlers/__init__.py +11 -0
  171. aria_code/apps/cli/handlers/broker_handlers.py +122 -0
  172. aria_code/apps/cli/handlers/chart_handlers.py +1314 -0
  173. aria_code/apps/cli/handlers/market_handlers.py +2570 -0
  174. aria_code/apps/cli/handlers/realty_handlers.py +114 -0
  175. aria_code/apps/cli/handlers/sports_handlers.py +22 -0
  176. aria_code/apps/cli/handlers/strategy_advice.py +82 -0
  177. aria_code/apps/cli/headless.py +352 -0
  178. aria_code/apps/cli/helpers.py +326 -0
  179. aria_code/apps/cli/hooks.py +180 -0
  180. aria_code/apps/cli/i18n.py +310 -0
  181. aria_code/apps/cli/intent.py +136 -0
  182. aria_code/apps/cli/intent_router.py +305 -0
  183. aria_code/apps/cli/intent_shadow.py +72 -0
  184. aria_code/apps/cli/intent_signals.py +347 -0
  185. aria_code/apps/cli/lifecycle_hooks.py +48 -0
  186. aria_code/apps/cli/local_tool_schemas.py +550 -0
  187. aria_code/apps/cli/main.py +39 -0
  188. aria_code/apps/cli/market_hours.py +100 -0
  189. aria_code/apps/cli/market_metadata.py +135 -0
  190. aria_code/apps/cli/market_universe.py +487 -0
  191. aria_code/apps/cli/message_processing.py +310 -0
  192. aria_code/apps/cli/mixin_binding.py +32 -0
  193. aria_code/apps/cli/model_catalog.py +652 -0
  194. aria_code/apps/cli/model_probe.py +169 -0
  195. aria_code/apps/cli/model_retirement.py +65 -0
  196. aria_code/apps/cli/native_branding.py +12 -0
  197. aria_code/apps/cli/native_bridge.py +118 -0
  198. aria_code/apps/cli/native_entry.py +58 -0
  199. aria_code/apps/cli/pdf_report.py +977 -0
  200. aria_code/apps/cli/plan_mode.py +139 -0
  201. aria_code/apps/cli/plotly_html.py +15 -0
  202. aria_code/apps/cli/prediction_feedback.py +202 -0
  203. aria_code/apps/cli/preflight.py +502 -0
  204. aria_code/apps/cli/project_aria.py +60 -0
  205. aria_code/apps/cli/prompt_assembly.py +78 -0
  206. aria_code/apps/cli/prompts/__init__.py +0 -0
  207. aria_code/apps/cli/prompts/coding.py +71 -0
  208. aria_code/apps/cli/prompts/select.py +193 -0
  209. aria_code/apps/cli/prompts/system_prompts.py +562 -0
  210. aria_code/apps/cli/prompts/ui.py +436 -0
  211. aria_code/apps/cli/provider_endpoints.py +237 -0
  212. aria_code/apps/cli/provider_keys.py +73 -0
  213. aria_code/apps/cli/providers/__init__.py +1 -0
  214. aria_code/apps/cli/providers/base.py +775 -0
  215. aria_code/apps/cli/providers/chat_routing.py +139 -0
  216. aria_code/apps/cli/providers/collaboration.py +98 -0
  217. aria_code/apps/cli/providers/llm/__init__.py +1 -0
  218. aria_code/apps/cli/providers/llm/offline_responses.py +68 -0
  219. aria_code/apps/cli/providers/llm/ollama_stream.py +1574 -0
  220. aria_code/apps/cli/providers/llm/response_cache.py +60 -0
  221. aria_code/apps/cli/providers/llm/sse_stream.py +298 -0
  222. aria_code/apps/cli/providers/runtime_bridge.py +586 -0
  223. aria_code/apps/cli/providers/vertexai_stream.py +592 -0
  224. aria_code/apps/cli/release_download.py +159 -0
  225. aria_code/apps/cli/review_runner.py +111 -0
  226. aria_code/apps/cli/runtime_consumer.py +769 -0
  227. aria_code/apps/cli/sandbox.py +32 -0
  228. aria_code/apps/cli/session_export.py +120 -0
  229. aria_code/apps/cli/session_jsonl.py +208 -0
  230. aria_code/apps/cli/session_refresh.py +123 -0
  231. aria_code/apps/cli/session_store.py +125 -0
  232. aria_code/apps/cli/skills_catalog.py +273 -0
  233. aria_code/apps/cli/task_isolation.py +211 -0
  234. aria_code/apps/cli/terminal_chrome.py +412 -0
  235. aria_code/apps/cli/todo_tracker.py +190 -0
  236. aria_code/apps/cli/tool_executor.py +932 -0
  237. aria_code/apps/cli/tool_registry.py +34 -0
  238. aria_code/apps/cli/tool_scope.py +113 -0
  239. aria_code/apps/cli/tools/__init__.py +40 -0
  240. aria_code/apps/cli/tools/context.py +46 -0
  241. aria_code/apps/cli/tools/file_tools.py +181 -0
  242. aria_code/apps/cli/tools/market_tools.py +626 -0
  243. aria_code/apps/cli/tools/notebook_tools.py +111 -0
  244. aria_code/apps/cli/tools/system_tools.py +656 -0
  245. aria_code/apps/cli/tools/write_tools.py +961 -0
  246. aria_code/apps/cli/tradingview_bridge.py +529 -0
  247. aria_code/apps/cli/transactions.py +269 -0
  248. aria_code/apps/cli/turn_planning.py +137 -0
  249. aria_code/apps/cli/update_check.py +257 -0
  250. aria_code/apps/cli/updater.py +373 -0
  251. aria_code/apps/cli/utils/__init__.py +0 -0
  252. aria_code/apps/cli/utils/market_detect.py +1663 -0
  253. aria_code/apps/cli/web_login.py +136 -0
  254. aria_code/apps/cli/workflow_runner.py +127 -0
  255. aria_code/apps/cli/workspace_route.py +121 -0
  256. aria_code/apps/daemon/README.md +14 -0
  257. aria_code/apps/vscode/README.md +115 -0
  258. aria_code/apps/vscode/package.json +70 -0
  259. aria_code/aria_cli.py +5244 -0
  260. aria_code/aria_daemon.py +1372 -0
  261. aria_code/aria_feishu_bot.py +16 -0
  262. aria_code/aria_mcp_server.py +61 -0
  263. aria_code/aria_relay_client.py +4 -0
  264. aria_code/aria_telegram_bot.py +11 -0
  265. aria_code/ariarc.py +385 -0
  266. aria_code/artifacts.py +581 -0
  267. aria_code/backtest_engine.py +4 -0
  268. aria_code/backtest_report.py +4 -0
  269. aria_code/brokers/__init__.py +78 -0
  270. aria_code/brokers/automation.py +193 -0
  271. aria_code/brokers/base.py +231 -0
  272. aria_code/brokers/capabilities.py +264 -0
  273. aria_code/brokers/cn/__init__.py +10 -0
  274. aria_code/brokers/cn/easytrader_broker.py +193 -0
  275. aria_code/brokers/cn/futu_broker.py +194 -0
  276. aria_code/brokers/cn/longbridge_broker.py +190 -0
  277. aria_code/brokers/cn/tiger_broker.py +196 -0
  278. aria_code/brokers/cn/xtquant_broker.py +225 -0
  279. aria_code/brokers/config.py +414 -0
  280. aria_code/brokers/intl/__init__.py +5 -0
  281. aria_code/brokers/intl/alpaca_broker.py +183 -0
  282. aria_code/brokers/intl/ibkr_broker.py +215 -0
  283. aria_code/brokers/intl/trading212_broker.py +183 -0
  284. aria_code/brokers/intl/webull_broker.py +156 -0
  285. aria_code/brokers/paper_broker.py +266 -0
  286. aria_code/brokers/planning.py +296 -0
  287. aria_code/brokers/registry.py +213 -0
  288. aria_code/brokers/trading.py +267 -0
  289. aria_code/canva_client.py +4 -0
  290. aria_code/change_store.py +127 -0
  291. aria_code/clients/__init__.py +0 -0
  292. aria_code/clients/aliyun_data_client.py +660 -0
  293. aria_code/clients/aria_relay_client.py +16 -0
  294. aria_code/clients/canva_client.py +284 -0
  295. aria_code/clients/figma_client.py +100 -0
  296. aria_code/clients/football_data_client.py +1671 -0
  297. aria_code/clients/kling_video_client.py +182 -0
  298. aria_code/clients/market_data_client.py +1985 -0
  299. aria_code/clients/openai_image_client.py +242 -0
  300. aria_code/clients/runway_video_client.py +157 -0
  301. aria_code/clients/warehouse_erp_client.py +105 -0
  302. aria_code/cloud_memory.py +190 -0
  303. aria_code/command_safety.py +19 -0
  304. aria_code/computer_use_server.py +276 -0
  305. aria_code/computer_use_tools.py +4 -0
  306. aria_code/dashboard_generator.py +579 -0
  307. aria_code/data_analysis_tools.py +4 -0
  308. aria_code/data_cleaner.py +483 -0
  309. aria_code/data_service.py +513 -0
  310. aria_code/datasources/__init__.py +23 -0
  311. aria_code/datasources/base.py +166 -0
  312. aria_code/datasources/router.py +253 -0
  313. aria_code/datasources/sec_statements.py +217 -0
  314. aria_code/datasources/sources/__init__.py +15 -0
  315. aria_code/datasources/sources/akshare_source.py +269 -0
  316. aria_code/datasources/sources/alpha_vantage_source.py +203 -0
  317. aria_code/datasources/sources/edgar_source.py +219 -0
  318. aria_code/datasources/sources/finnhub_source.py +198 -0
  319. aria_code/datasources/sources/fred_source.py +223 -0
  320. aria_code/datasources/sources/tushare_source.py +143 -0
  321. aria_code/datasources/sources/web_scraper_source.py +278 -0
  322. aria_code/datasources/sources/world_bank_source.py +206 -0
  323. aria_code/datasources/sources/yfinance_source.py +152 -0
  324. aria_code/doctor.py +860 -0
  325. aria_code/domain/__init__.py +0 -0
  326. aria_code/domain/_finance_core/__init__.py +5 -0
  327. aria_code/domain/_finance_core/backtest_engine.py +325 -0
  328. aria_code/domain/_finance_core/performance_stats.py +101 -0
  329. aria_code/domain/backtest_engine.py +75 -0
  330. aria_code/domain/backtest_report.py +488 -0
  331. aria_code/domain/finance_formulas.py +389 -0
  332. aria_code/domain/performance_stats.py +6 -0
  333. aria_code/domain/portfolio_ledger.py +307 -0
  334. aria_code/domain/strategy_vault.py +421 -0
  335. aria_code/evals/__init__.py +37 -0
  336. aria_code/evals/behavior.py +113 -0
  337. aria_code/evals/gap.py +93 -0
  338. aria_code/evals/harness.py +949 -0
  339. aria_code/evals/review_probe.py +132 -0
  340. aria_code/evals/runner.py +425 -0
  341. aria_code/evals/scorecard.py +322 -0
  342. aria_code/evals/summary.py +125 -0
  343. aria_code/external_agent_runner.py +115 -0
  344. aria_code/figma_client.py +4 -0
  345. aria_code/file_analysis_tools.py +4 -0
  346. aria_code/finance_formulas.py +4 -0
  347. aria_code/football_data_client.py +4 -0
  348. aria_code/image_gen_tools.py +4 -0
  349. aria_code/intent_classifier.py +459 -0
  350. aria_code/kling_video_client.py +4 -0
  351. aria_code/licensing.py +151 -0
  352. aria_code/local_finance_tools.py +4 -0
  353. aria_code/local_image_provider.py +4 -0
  354. aria_code/local_llm_provider.py +4 -0
  355. aria_code/macro_tools.py +4 -0
  356. aria_code/markdown_pdf.py +289 -0
  357. aria_code/market_data_client.py +4 -0
  358. aria_code/mcp_client.py +744 -0
  359. aria_code/memory_manager.py +273 -0
  360. aria_code/model_capability.py +512 -0
  361. aria_code/notification_tools.py +4 -0
  362. aria_code/openai_image_client.py +4 -0
  363. aria_code/packages/__init__.py +26 -0
  364. aria_code/packages/adk_bridge/__init__.py +11 -0
  365. aria_code/packages/adk_bridge/code_review_tools.py +49 -0
  366. aria_code/packages/adk_bridge/market_tools.py +125 -0
  367. aria_code/packages/aria_agents/__init__.py +5 -0
  368. aria_code/packages/aria_agents/manifest.py +78 -0
  369. aria_code/packages/aria_core/__init__.py +34 -0
  370. aria_code/packages/aria_core/architecture.py +269 -0
  371. aria_code/packages/aria_core/export.py +147 -0
  372. aria_code/packages/aria_core/manifest.py +65 -0
  373. aria_code/packages/aria_core/paths.py +57 -0
  374. aria_code/packages/aria_core/secure_file.py +68 -0
  375. aria_code/packages/aria_infra/__init__.py +15 -0
  376. aria_code/packages/aria_infra/arthera.py +58 -0
  377. aria_code/packages/aria_infra/doctor.py +246 -0
  378. aria_code/packages/aria_infra/product.py +40 -0
  379. aria_code/packages/aria_mcp/__init__.py +25 -0
  380. aria_code/packages/aria_mcp/bridge.py +232 -0
  381. aria_code/packages/aria_mcp/circuit.py +82 -0
  382. aria_code/packages/aria_mcp/config.py +162 -0
  383. aria_code/packages/aria_mcp/installer.py +95 -0
  384. aria_code/packages/aria_mcp/server.py +1705 -0
  385. aria_code/packages/aria_mcp/tools.py +81 -0
  386. aria_code/packages/aria_sdk/__init__.py +19 -0
  387. aria_code/packages/aria_sdk/client.py +393 -0
  388. aria_code/packages/aria_sdk/providers.py +70 -0
  389. aria_code/packages/aria_sdk/streaming.py +75 -0
  390. aria_code/packages/aria_sdk/types.py +86 -0
  391. aria_code/packages/aria_services/__init__.py +92 -0
  392. aria_code/packages/aria_services/cache.py +60 -0
  393. aria_code/packages/aria_services/context.py +356 -0
  394. aria_code/packages/aria_services/context_checkpoints.py +157 -0
  395. aria_code/packages/aria_services/data.py +11 -0
  396. aria_code/packages/aria_services/provider_health.py +304 -0
  397. aria_code/packages/aria_services/references.py +247 -0
  398. aria_code/packages/aria_services/registry.py +249 -0
  399. aria_code/packages/aria_services/research_protocol.py +137 -0
  400. aria_code/packages/aria_services/research_quality.py +186 -0
  401. aria_code/packages/aria_services/settings.py +160 -0
  402. aria_code/packages/aria_services/usage.py +151 -0
  403. aria_code/packages/aria_skills/__init__.py +46 -0
  404. aria_code/packages/aria_skills/catalog.py +149 -0
  405. aria_code/packages/aria_skills/loader.py +455 -0
  406. aria_code/packages/aria_skills/registry.py +84 -0
  407. aria_code/packages/aria_tools/__init__.py +5 -0
  408. aria_code/packages/aria_tools/financial/__init__.py +5 -0
  409. aria_code/packages/aria_tools/financial/_paths.py +92 -0
  410. aria_code/packages/aria_tools/registry.py +129 -0
  411. aria_code/packages/quant_engine/__init__.py +33 -0
  412. aria_code/packages/quant_engine/_namespace.py +24 -0
  413. aria_code/packages/quant_engine/backtest/__init__.py +10 -0
  414. aria_code/packages/quant_engine/backtest/core.py +53 -0
  415. aria_code/packages/quant_engine/backtest/engine.py +102 -0
  416. aria_code/packages/quant_engine/portfolio/__init__.py +9 -0
  417. aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
  418. aria_code/packages/quant_engine/sports/__init__.py +76 -0
  419. aria_code/packages/quant_engine/sports/calibrator.py +354 -0
  420. aria_code/packages/quant_engine/sports/dixon_coles.py +234 -0
  421. aria_code/packages/quant_engine/sports/elo.py +300 -0
  422. aria_code/packages/quant_engine/sports/form.py +188 -0
  423. aria_code/packages/quant_engine/sports/h2h.py +195 -0
  424. aria_code/packages/quant_engine/sports/predictor.py +312 -0
  425. aria_code/packages/quant_engine/sports/tracker.py +665 -0
  426. aria_code/packages/quant_engine/stochastic/__init__.py +31 -0
  427. aria_code/packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
  428. aria_code/packages/quant_engine/stochastic/ito_calculus.py +477 -0
  429. aria_code/packages/quant_engine/stochastic/kelly_criterion.py +181 -0
  430. aria_code/packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
  431. aria_code/packages/quant_engine/stochastic/options_pricing.py +576 -0
  432. aria_code/packages/quant_engine/stochastic/stochastic_processes.py +90 -0
  433. aria_code/packs/__init__.py +85 -0
  434. aria_code/packs/base.py +168 -0
  435. aria_code/packs/finance/__init__.py +250 -0
  436. aria_code/packs/healthcare/__init__.py +40 -0
  437. aria_code/packs/logistics/__init__.py +171 -0
  438. aria_code/packs/payments/__init__.py +139 -0
  439. aria_code/packs/realty/__init__.py +190 -0
  440. aria_code/packs/registry.py +202 -0
  441. aria_code/packs/rules.py +90 -0
  442. aria_code/plan_utils.py +194 -0
  443. aria_code/plugin_loader.py +328 -0
  444. aria_code/portfolio_ledger.py +4 -0
  445. aria_code/preview_server.py +449 -0
  446. aria_code/privacy/__init__.py +5 -0
  447. aria_code/privacy/feedback.py +123 -0
  448. aria_code/project_review.py +278 -0
  449. aria_code/project_review_api.py +235 -0
  450. aria_code/project_review_server.py +18 -0
  451. aria_code/project_tools.py +4 -0
  452. aria_code/providers/__init__.py +30 -0
  453. aria_code/providers/google_video_provider.py +256 -0
  454. aria_code/providers/llm/__init__.py +19 -0
  455. aria_code/providers/llm/anthropic.py +184 -0
  456. aria_code/providers/llm/arthera_gateway.py +90 -0
  457. aria_code/providers/llm/autoconfig.py +213 -0
  458. aria_code/providers/llm/base.py +139 -0
  459. aria_code/providers/llm/custom.py +158 -0
  460. aria_code/providers/llm/ollama.py +137 -0
  461. aria_code/providers/llm/openai_compat.py +350 -0
  462. aria_code/providers/llm/registry.py +457 -0
  463. aria_code/providers/local_image_provider.py +202 -0
  464. aria_code/providers/local_llm_provider.py +549 -0
  465. aria_code/providers/ollama_models.py +79 -0
  466. aria_code/providers/openai_params.py +32 -0
  467. aria_code/providers/tool_messages.py +100 -0
  468. aria_code/realty_data_tools.py +4 -0
  469. aria_code/report_exporters.py +150 -0
  470. aria_code/report_generator.py +1764 -0
  471. aria_code/review_service.py +497 -0
  472. aria_code/runtime/__init__.py +176 -0
  473. aria_code/runtime/acceptance.py +632 -0
  474. aria_code/runtime/agent_loop.py +1851 -0
  475. aria_code/runtime/agent_profiles.py +75 -0
  476. aria_code/runtime/approval.py +71 -0
  477. aria_code/runtime/baseline.py +219 -0
  478. aria_code/runtime/budget.py +232 -0
  479. aria_code/runtime/checkpoints.py +481 -0
  480. aria_code/runtime/consult.py +65 -0
  481. aria_code/runtime/contract.py +282 -0
  482. aria_code/runtime/delivery.py +432 -0
  483. aria_code/runtime/events.py +113 -0
  484. aria_code/runtime/gateway.py +167 -0
  485. aria_code/runtime/lsp.py +484 -0
  486. aria_code/runtime/native_index.py +153 -0
  487. aria_code/runtime/orchestrator.py +55 -0
  488. aria_code/runtime/processes.py +343 -0
  489. aria_code/runtime/project_graph.py +614 -0
  490. aria_code/runtime/repo_map.py +712 -0
  491. aria_code/runtime/research_bundle.py +105 -0
  492. aria_code/runtime/review.py +269 -0
  493. aria_code/runtime/run_state.py +127 -0
  494. aria_code/runtime/run_store.py +522 -0
  495. aria_code/runtime/scoped_runs.py +113 -0
  496. aria_code/runtime/self_healing.py +422 -0
  497. aria_code/runtime/semantic_diff.py +171 -0
  498. aria_code/runtime/subagent.py +721 -0
  499. aria_code/runtime/symbol_edit.py +117 -0
  500. aria_code/runtime/task_brief.py +46 -0
  501. aria_code/runtime/task_context.py +145 -0
  502. aria_code/runtime/task_ledger.py +108 -0
  503. aria_code/runtime/task_orchestrator.py +90 -0
  504. aria_code/runtime/task_outcome.py +33 -0
  505. aria_code/runtime/task_verification.py +266 -0
  506. aria_code/runtime/task_worktree.py +448 -0
  507. aria_code/runtime/tool_executor.py +329 -0
  508. aria_code/runtime/tool_policy.py +107 -0
  509. aria_code/runtime/transactions.py +216 -0
  510. aria_code/runtime/windows_job.py +115 -0
  511. aria_code/runtime/workflows.py +222 -0
  512. aria_code/runtime/worktrees.py +215 -0
  513. aria_code/runway_video_client.py +4 -0
  514. aria_code/safety/__init__.py +27 -0
  515. aria_code/safety/client_isolation.py +178 -0
  516. aria_code/safety/permissions.py +339 -0
  517. aria_code/safety/risk.py +699 -0
  518. aria_code/safety/sandbox.py +148 -0
  519. aria_code/safety/service.py +102 -0
  520. aria_code/service_endpoints.py +14 -0
  521. aria_code/setup_wizard.py +684 -0
  522. aria_code/spreadsheet_tools.py +4 -0
  523. aria_code/strategy_vault.py +4 -0
  524. aria_code/tools/__init__.py +0 -0
  525. aria_code/tools/artifact_tools.py +252 -0
  526. aria_code/tools/broker_tools.py +92 -0
  527. aria_code/tools/build_quant_engine.py +97 -0
  528. aria_code/tools/code_audit_tools.py +192 -0
  529. aria_code/tools/computer_use_tools.py +509 -0
  530. aria_code/tools/data_analysis_tools.py +831 -0
  531. aria_code/tools/enterprise_finance_tools.py +202 -0
  532. aria_code/tools/extended_tools.py +240 -0
  533. aria_code/tools/file_analysis_tools.py +880 -0
  534. aria_code/tools/image_gen_tools.py +114 -0
  535. aria_code/tools/local_finance_tools.py +3488 -0
  536. aria_code/tools/logistics_carriers.py +11 -0
  537. aria_code/tools/logistics_inventory.py +11 -0
  538. aria_code/tools/logistics_io.py +11 -0
  539. aria_code/tools/logistics_tenancy.py +11 -0
  540. aria_code/tools/logistics_tools.py +11 -0
  541. aria_code/tools/macro_tools.py +412 -0
  542. aria_code/tools/notification_tools.py +249 -0
  543. aria_code/tools/project_tools.py +525 -0
  544. aria_code/tools/realty_data_tools.py +662 -0
  545. aria_code/tools/spreadsheet_tools.py +399 -0
  546. aria_code/tools/stripe_tools.py +115 -0
  547. aria_code/tools/video_tools.py +500 -0
  548. aria_code/training/__init__.py +1 -0
  549. aria_code/training/compare.py +117 -0
  550. aria_code/training/prelude_commands.py +111 -0
  551. aria_code/training/prelude_eval.py +390 -0
  552. aria_code/training/prelude_real.py +183 -0
  553. aria_code/training/prelude_train.py +248 -0
  554. aria_code/training/sft_dataset.py +230 -0
  555. aria_code/training/trajectory_dataset.py +263 -0
  556. aria_code/ui/__init__.py +105 -0
  557. aria_code/ui/banner.py +404 -0
  558. aria_code/ui/completer.py +571 -0
  559. aria_code/ui/console.py +285 -0
  560. aria_code/ui/input_box.py +505 -0
  561. aria_code/ui/mentions.py +206 -0
  562. aria_code/ui/picker.py +248 -0
  563. aria_code/ui/render/__init__.py +11 -0
  564. aria_code/ui/render/actions.py +297 -0
  565. aria_code/ui/render/finance.py +1732 -0
  566. aria_code/ui/render/market.py +253 -0
  567. aria_code/ui/render/output.py +954 -0
  568. aria_code/ui/render/responsive.py +75 -0
  569. aria_code/ui/render/team.py +367 -0
  570. aria_code/ui/robot.py +239 -0
  571. aria_code/ui/startup_dashboard.py +110 -0
  572. aria_code/video_analysis.py +147 -0
  573. aria_code/video_editor.py +518 -0
  574. aria_code/video_workflow/__init__.py +8 -0
  575. aria_code/video_workflow/__main__.py +61 -0
  576. aria_code/video_workflow/engines/__init__.py +18 -0
  577. aria_code/video_workflow/engines/base.py +69 -0
  578. aria_code/video_workflow/engines/custom_model.py +204 -0
  579. aria_code/video_workflow/engines/google_veo.py +94 -0
  580. aria_code/video_workflow/engines/local_engine.py +71 -0
  581. aria_code/video_workflow/engines/registry.py +98 -0
  582. aria_code/video_workflow/google_models.py +278 -0
  583. aria_code/video_workflow/music/__init__.py +5 -0
  584. aria_code/video_workflow/music/foley_service.py +132 -0
  585. aria_code/video_workflow/music/service.py +212 -0
  586. aria_code/video_workflow/pipeline.py +620 -0
  587. aria_code/video_workflow/plan.py +202 -0
  588. aria_code/video_workflow/session.py +297 -0
  589. aria_code/workspace/__init__.py +6 -0
  590. aria_code/workspace/files.py +219 -0
  591. aria_code/workspace/verify.py +167 -0
  592. aria_code/workspace_changes.py +199 -0
  593. aria_code/workspace_service.py +196 -0
  594. aria_code/workspace_tools.py +150 -0
  595. aria_code-0.167.0.dist-info/METADATA +330 -0
  596. aria_code-0.167.0.dist-info/RECORD +601 -0
  597. aria_code-0.167.0.dist-info/WHEEL +5 -0
  598. aria_code-0.167.0.dist-info/entry_points.txt +5 -0
  599. aria_code-0.167.0.dist-info/licenses/LICENSE +201 -0
  600. aria_code-0.167.0.dist-info/licenses/NOTICE +49 -0
  601. aria_code-0.167.0.dist-info/top_level.txt +2 -0
@@ -0,0 +1,500 @@
1
+ """video_tools.py — 视频分析、镜头生成与自动化剪辑制作工具集
2
+
3
+ 向 aria-code 终端交互式会话及桌面端客户端注册以下能力:
4
+ 1. analyze_video_script: 多模态分析剧本与输入图片,生成包含角色特征锚点与 DiT 物理动效的结构化分镜头剧本 (Storyboard)
5
+ 2. generate_video_clip: 单镜头生成(通过 VideoEngineRegistry 调度 Google Cloud Veo、自研私有模型或本地引擎)
6
+ 3. produce_video_from_script: 全自动工作流,一键将图片素材和脚本制作成最终 MP4 视频,支持多引擎与音频自动压音
7
+ 4. configure_video_engine: 查看或配置当前使用的视频生成引擎(如设置自研模型端点或切换为 Google Cloud)
8
+ """
9
+
10
+ from __future__ import annotations
11
+
12
+ import logging
13
+ import os
14
+ from pathlib import Path
15
+ from typing import Any, Dict, List, Optional
16
+
17
+ from aria_code.video_workflow.engines import VideoEngineRegistry, VideoShotRequest, CustomModelEngine
18
+
19
+ logger = logging.getLogger(__name__)
20
+
21
+
22
+ def tool_analyze_video_script(params: Dict[str, Any]) -> Dict[str, Any]:
23
+ """LOCAL_TOOLS: 分析文案脚本和图片,生成分镜剧本。"""
24
+ script = str(params.get("script") or "").strip()
25
+ if not script:
26
+ return {"success": False, "error": "缺少脚本文案 (script)"}
27
+
28
+ images = params.get("images") or []
29
+ if isinstance(images, str):
30
+ images = [images]
31
+
32
+ aspect_ratio = str(params.get("aspect_ratio") or "16:9")
33
+ target_duration = float(params.get("target_duration") or 15.0)
34
+ style = str(params.get("style") or "cinematic")
35
+
36
+ from aria_code.agents.video.director import VideoDirectorAgent
37
+
38
+ agent = VideoDirectorAgent()
39
+ storyboard = agent.decompose_script_and_images(
40
+ script,
41
+ images=images,
42
+ target_duration=target_duration,
43
+ aspect_ratio=aspect_ratio,
44
+ style=style,
45
+ )
46
+ return {"success": True, "storyboard": storyboard}
47
+
48
+
49
+ def tool_generate_video_clip(params: Dict[str, Any]) -> Dict[str, Any]:
50
+ """LOCAL_TOOLS: 生成单个镜头视频片段,调度自研引擎适配层。"""
51
+ prompt = str(params.get("prompt") or "").strip()
52
+ image_path = str(params.get("image_path") or "").strip() or None
53
+
54
+ if not prompt and not image_path:
55
+ return {"success": False, "error": "prompt 或 image_path 至少需要提供一项"}
56
+
57
+ aspect_ratio = str(params.get("aspect_ratio") or "16:9")
58
+ duration = int(round(float(params.get("duration") or 4.0)))
59
+ provider = str(params.get("provider") or "auto").lower()
60
+ camera_motion = params.get("camera_motion")
61
+ custom_parameters = params.get("custom_parameters") or {}
62
+
63
+ registry = VideoEngineRegistry.get_instance()
64
+ engine = registry.get_engine(None if provider == "auto" else provider)
65
+
66
+ req = VideoShotRequest(
67
+ prompt=prompt or "cinematic shot",
68
+ image_path=image_path,
69
+ duration_seconds=duration,
70
+ aspect_ratio=aspect_ratio,
71
+ camera_motion=camera_motion,
72
+ custom_parameters=custom_parameters,
73
+ )
74
+
75
+ result = engine.generate_shot(req)
76
+ if result.success and result.path:
77
+ return {
78
+ "success": True,
79
+ "path": result.path,
80
+ "video_url": result.video_url,
81
+ "engine": result.engine_name,
82
+ "model": result.model_version,
83
+ "duration": result.duration,
84
+ "aspect_ratio": result.aspect_ratio,
85
+ }
86
+
87
+ # 如果首选引擎失败,尝试用本地引擎兜底
88
+ if image_path and Path(image_path).exists() and engine.name != "local":
89
+ logger.warning(f"[video_tools] Primary engine failed ({result.error}), falling back to local engine.")
90
+ local_engine = registry.get_engine("local")
91
+ fallback_res = local_engine.generate_shot(req)
92
+ if fallback_res.success and fallback_res.path:
93
+ return {
94
+ "success": True,
95
+ "path": fallback_res.path,
96
+ "engine": "local_fallback",
97
+ "model": "ken_burns_v2",
98
+ "duration": fallback_res.duration,
99
+ "aspect_ratio": fallback_res.aspect_ratio,
100
+ "note": f"Primary engine failed ({result.error}), generated via local fallback.",
101
+ }
102
+
103
+ return {
104
+ "success": False,
105
+ "error": f"Failed to generate video clip: {result.error}",
106
+ }
107
+
108
+
109
+ def tool_produce_video_from_script(params: Dict[str, Any]) -> Dict[str, Any]:
110
+ """LOCAL_TOOLS: 视频导演全流程制作,从脚本文案与图片一键输出成片。"""
111
+ script = str(params.get("script") or "").strip()
112
+ if not script:
113
+ return {"success": False, "error": "缺少脚本文案 (script)"}
114
+
115
+ images = params.get("images") or []
116
+ if isinstance(images, str):
117
+ images = [images]
118
+
119
+ aspect_ratio = str(params.get("aspect_ratio") or "16:9")
120
+ target_duration = float(params.get("target_duration") or 15.0)
121
+ provider = str(params.get("provider") or "auto").lower()
122
+ audio_path = str(params.get("audio_path") or "").strip() or None
123
+ bgm_path = str(params.get("bgm_path") or "").strip() or None
124
+ engine_parameters = params.get("engine_parameters")
125
+
126
+ from aria_code.agents.video.director import VideoDirectorAgent
127
+
128
+ agent = VideoDirectorAgent()
129
+ return agent.produce_video_workflow(
130
+ script,
131
+ images=images,
132
+ target_duration=target_duration,
133
+ aspect_ratio=aspect_ratio,
134
+ provider=provider,
135
+ audio_path=audio_path,
136
+ bgm_path=bgm_path,
137
+ engine_parameters=engine_parameters,
138
+ )
139
+
140
+
141
+ def tool_configure_video_engine(params: Dict[str, Any]) -> Dict[str, Any]:
142
+ """LOCAL_TOOLS: 查看或配置视频生成引擎。"""
143
+ registry = VideoEngineRegistry.get_instance()
144
+ action = str(params.get("action") or "list").lower()
145
+
146
+ if action == "list" or action == "status":
147
+ return {
148
+ "success": True,
149
+ "active_engine": registry.get_active_engine_name(),
150
+ "engines": registry.list_engines_status(),
151
+ }
152
+
153
+ if action == "set_active":
154
+ engine_name = str(params.get("engine") or "").strip().lower()
155
+ if not engine_name:
156
+ return {"success": False, "error": "Missing 'engine' name (e.g. google, custom, local)"}
157
+ try:
158
+ registry.set_active_engine(engine_name)
159
+ return {
160
+ "success": True,
161
+ "message": f"Active video engine switched to '{engine_name}'.",
162
+ "active_engine": engine_name,
163
+ }
164
+ except ValueError as exc:
165
+ return {"success": False, "error": str(exc)}
166
+
167
+ if action == "configure_custom":
168
+ endpoint_url = params.get("endpoint_url")
169
+ auth_token = params.get("auth_token")
170
+ model_name = params.get("model_name")
171
+ custom_engine = registry.get_engine("custom")
172
+ if isinstance(custom_engine, CustomModelEngine):
173
+ if endpoint_url:
174
+ custom_engine.endpoint_url = str(endpoint_url)
175
+ if auth_token is not None:
176
+ custom_engine.auth_token = str(auth_token)
177
+ if model_name:
178
+ custom_engine.model_name = str(model_name)
179
+ return {
180
+ "success": True,
181
+ "message": "Custom video model engine configured successfully.",
182
+ "config": custom_engine.health_check(),
183
+ }
184
+ return {"success": False, "error": "Custom engine not registered."}
185
+
186
+ return {"success": False, "error": f"Unknown action '{action}'. Use 'list', 'set_active', or 'configure_custom'."}
187
+
188
+
189
+ def tool_create_video_project(params: Dict[str, Any]) -> Dict[str, Any]:
190
+ """LOCAL_TOOLS: 创建桌面端视频项目,规划卡片化分镜。"""
191
+ from aria_code.video_workflow.session import VideoProjectSession
192
+
193
+ title = str(params.get("title") or "").strip()
194
+ script = str(params.get("script") or "").strip()
195
+ if not script:
196
+ return {"success": False, "error": "Missing 'script' text."}
197
+
198
+ images = params.get("images") or []
199
+ if isinstance(images, str):
200
+ images = [images]
201
+
202
+ aspect_ratio = str(params.get("aspect_ratio") or "16:9")
203
+ target_duration = float(params.get("target_duration") or 15.0)
204
+ provider = str(params.get("provider") or "auto")
205
+
206
+ session = VideoProjectSession.get_instance()
207
+ return session.create_project(
208
+ title=title,
209
+ script=script,
210
+ images=images,
211
+ aspect_ratio=aspect_ratio,
212
+ target_duration=target_duration,
213
+ provider=provider,
214
+ )
215
+
216
+
217
+ def tool_get_video_project(params: Dict[str, Any]) -> Dict[str, Any]:
218
+ """LOCAL_TOOLS: 获取指定项目的分镜卡片与母带状态。"""
219
+ from aria_code.video_workflow.session import VideoProjectSession
220
+
221
+ project_id = str(params.get("project_id") or "").strip()
222
+ if not project_id:
223
+ return {"success": False, "error": "Missing 'project_id'."}
224
+
225
+ session = VideoProjectSession.get_instance()
226
+ return session.get_project(project_id)
227
+
228
+
229
+ def tool_regenerate_video_shot(params: Dict[str, Any]) -> Dict[str, Any]:
230
+ """LOCAL_TOOLS: 桌面端核心能力 — 单镜头独立重新生成,并秒级重新拼接母带。"""
231
+ from aria_code.video_workflow.session import VideoProjectSession
232
+
233
+ project_id = str(params.get("project_id") or "").strip()
234
+ shot_index = int(params.get("shot_index", 0))
235
+ if not project_id or shot_index <= 0:
236
+ return {"success": False, "error": "'project_id' and valid 'shot_index' (>0) are required."}
237
+
238
+ new_prompt = params.get("new_prompt")
239
+ engine_name = params.get("engine")
240
+
241
+ session = VideoProjectSession.get_instance()
242
+ return session.regenerate_single_shot(
243
+ project_id=project_id,
244
+ shot_index=shot_index,
245
+ new_prompt=new_prompt,
246
+ engine_name=engine_name,
247
+ )
248
+
249
+
250
+ def tool_qa_review_video_clip(params: Dict[str, Any]) -> Dict[str, Any]:
251
+ """LOCAL_TOOLS: 视觉质检 Agent 对视频镜头进行多模态抽样质检与自愈建议。"""
252
+ from aria_code.agents.video.qa_agent import VisualQAAgent
253
+
254
+ video_path = str(params.get("video_path") or "").strip()
255
+ if not video_path:
256
+ return {"success": False, "error": "Missing 'video_path'."}
257
+
258
+ prompt = str(params.get("prompt") or "")
259
+ shot_index = int(params.get("shot_index", 1))
260
+
261
+ agent = VisualQAAgent()
262
+ qa_res = agent.review_shot(shot_index=shot_index, video_path=video_path, original_prompt=prompt)
263
+ import dataclasses
264
+ return {"success": True, "qa_result": dataclasses.asdict(qa_res)}
265
+
266
+
267
+ VIDEO_TOOL_SCHEMAS: List[Dict[str, Any]] = [
268
+ {
269
+ "type": "function",
270
+ "function": {
271
+ "name": "analyze_video_script",
272
+ "description": (
273
+ "智能视频导演分镜分析:解析剧本与图片,锁定角色特征锚点,生成具备 DiT 物理动力学与 3D 运镜参数的结构化分镜剧本 (Storyboard)。"
274
+ ),
275
+ "parameters": {
276
+ "type": "object",
277
+ "properties": {
278
+ "script": {"type": "string", "description": "视频文案、解说词或故事概要"},
279
+ "images": {
280
+ "type": "array",
281
+ "items": {"type": "string"},
282
+ "description": "本地图片素材路径列表(用于参考图、角色锚定与图生视频)",
283
+ },
284
+ "aspect_ratio": {
285
+ "type": "string",
286
+ "enum": ["16:9", "9:16", "1:1"],
287
+ "description": "目标视频画面画幅,默认 16:9",
288
+ },
289
+ "target_duration": {"type": "number", "description": "目标视频总时长(秒),默认 15"},
290
+ "style": {"type": "string", "description": "视觉质感与风格,例如 cinematic, apple_keynote, modern"},
291
+ },
292
+ "required": ["script"],
293
+ },
294
+ },
295
+ },
296
+ {
297
+ "type": "function",
298
+ "function": {
299
+ "name": "generate_video_clip",
300
+ "description": (
301
+ "生成单个视频镜头片段。支持调度 Google Cloud Veo 2.0、自研私有模型端点或本地离线引擎。"
302
+ ),
303
+ "parameters": {
304
+ "type": "object",
305
+ "properties": {
306
+ "prompt": {"type": "string", "description": "镜头物理动效与画面提示词"},
307
+ "image_path": {"type": "string", "description": "参考图片路径(用于图生视频或本地运镜)"},
308
+ "duration": {"type": "number", "description": "镜头时长(秒),默认 4"},
309
+ "aspect_ratio": {"type": "string", "enum": ["16:9", "9:16", "1:1"], "description": "画幅比例"},
310
+ "provider": {
311
+ "type": "string",
312
+ "enum": ["auto", "google", "custom", "local"],
313
+ "description": "生成引擎:auto 自动匹配,google 使用 Google Cloud Veo,custom 使用自研模型,local 纯本地离线",
314
+ },
315
+ "camera_motion": {
316
+ "type": "object",
317
+ "description": "3D 相机运镜参数,例如 {\"zoom\": 0.2, \"pan\": 0.1, \"tilt\": 0.0}",
318
+ },
319
+ "custom_parameters": {
320
+ "type": "object",
321
+ "description": "透传给自研模型的私有参数(如 lora, steps, cfg_scale)",
322
+ },
323
+ },
324
+ },
325
+ },
326
+ },
327
+ {
328
+ "type": "function",
329
+ "function": {
330
+ "name": "produce_video_from_script",
331
+ "description": (
332
+ "全自动视频制作工作流:从剧本和图片一键输出成片。包含分镜规划、模型调度、多轨混音(带语音自动压音 Sidechain Ducking)与字幕压制。"
333
+ ),
334
+ "parameters": {
335
+ "type": "object",
336
+ "properties": {
337
+ "script": {"type": "string", "description": "短视频脚本或解说文案"},
338
+ "images": {
339
+ "type": "array",
340
+ "items": {"type": "string"},
341
+ "description": "供视频使用的图片素材路径列表",
342
+ },
343
+ "aspect_ratio": {
344
+ "type": "string",
345
+ "enum": ["16:9", "9:16", "1:1"],
346
+ "description": "画幅比例:横屏 16:9,竖屏短视频 9:16,方形 1:1",
347
+ },
348
+ "target_duration": {"type": "number", "description": "期望总时长(秒),默认 15"},
349
+ "audio_path": {"type": "string", "description": "配音音频文件路径(可选)"},
350
+ "bgm_path": {"type": "string", "description": "背景音乐文件路径(可选)"},
351
+ "provider": {
352
+ "type": "string",
353
+ "enum": ["auto", "google", "custom", "local"],
354
+ "description": "视频生成引擎,默认 auto",
355
+ },
356
+ "engine_parameters": {
357
+ "type": "object",
358
+ "description": "透传给视频引擎的专属参数",
359
+ },
360
+ },
361
+ "required": ["script"],
362
+ },
363
+ },
364
+ },
365
+ {
366
+ "type": "function",
367
+ "function": {
368
+ "name": "configure_video_engine",
369
+ "description": (
370
+ "查看或配置视频生成模型引擎。支持查看引擎健康状态、切换默认引擎 (Google Cloud / 自研模型 / 本地),或设置自研模型端点 URL 及密钥。"
371
+ ),
372
+ "parameters": {
373
+ "type": "object",
374
+ "properties": {
375
+ "action": {
376
+ "type": "string",
377
+ "enum": ["list", "set_active", "configure_custom"],
378
+ "description": "操作类型:list 查看所有引擎状态,set_active 切换当前活动引擎,configure_custom 配置自研模型端点",
379
+ },
380
+ "engine": {
381
+ "type": "string",
382
+ "enum": ["google", "custom", "local"],
383
+ "description": "当 action 为 set_active 时,指定的引擎名称",
384
+ },
385
+ "endpoint_url": {"type": "string", "description": "自研私有模型的 HTTP 推理服务端点 URL"},
386
+ "auth_token": {"type": "string", "description": "自研模型服务的认证 Token"},
387
+ "model_name": {"type": "string", "description": "自研模型名称,例如 aria-custom-dit-v1"},
388
+ },
389
+ "required": ["action"],
390
+ },
391
+ },
392
+ },
393
+ {
394
+ "type": "function",
395
+ "function": {
396
+ "name": "create_video_project",
397
+ "description": "创建桌面端视频项目,将剧本与素材拆解为可视化的分镜工程并持久化存储。",
398
+ "parameters": {
399
+ "type": "object",
400
+ "properties": {
401
+ "title": {"type": "string", "description": "视频标题"},
402
+ "script": {"type": "string", "description": "视频文案或剧本"},
403
+ "images": {"type": "array", "items": {"type": "string"}, "description": "素材图片路径列表"},
404
+ "aspect_ratio": {"type": "string", "enum": ["16:9", "9:16", "1:1"], "description": "画幅比例"},
405
+ "target_duration": {"type": "number", "description": "目标时长(秒)"},
406
+ "provider": {"type": "string", "enum": ["auto", "google", "custom", "local"], "description": "默认生成引擎"},
407
+ },
408
+ "required": ["script"],
409
+ },
410
+ },
411
+ },
412
+ {
413
+ "type": "function",
414
+ "function": {
415
+ "name": "get_video_project",
416
+ "description": "获取指定视频项目详情(分镜卡片、当前状态、生成进度与成片路径)。",
417
+ "parameters": {
418
+ "type": "object",
419
+ "properties": {
420
+ "project_id": {"type": "string", "description": "项目 ID"},
421
+ },
422
+ "required": ["project_id"],
423
+ },
424
+ },
425
+ },
426
+ {
427
+ "type": "function",
428
+ "function": {
429
+ "name": "regenerate_video_shot",
430
+ "description": "桌面端单镜头独立重生成:仅重新渲染指定镜头,自动保留其余已有镜头缓存并秒级完成母带重缝合。",
431
+ "parameters": {
432
+ "type": "object",
433
+ "properties": {
434
+ "project_id": {"type": "string", "description": "项目 ID"},
435
+ "shot_index": {"type": "integer", "description": "需要重生成的镜头序号(从 1 开始)"},
436
+ "new_prompt": {"type": "string", "description": "可选的修改后提示词"},
437
+ "engine": {"type": "string", "enum": ["google", "custom", "local"], "description": "可选的重生成引擎"},
438
+ },
439
+ "required": ["project_id", "shot_index"],
440
+ },
441
+ },
442
+ },
443
+ {
444
+ "type": "function",
445
+ "function": {
446
+ "name": "qa_review_video_clip",
447
+ "description": "视觉质检 Agent 对生成的视频镜头进行多模态抽帧质检、打分与缺陷诊断。",
448
+ "parameters": {
449
+ "type": "object",
450
+ "properties": {
451
+ "video_path": {"type": "string", "description": "被质检的视频 MP4 文件路径"},
452
+ "prompt": {"type": "string", "description": "原始提示词"},
453
+ "shot_index": {"type": "integer", "description": "镜头序号"},
454
+ },
455
+ "required": ["video_path"],
456
+ },
457
+ },
458
+ },
459
+ ]
460
+
461
+
462
+ def register_video_tools(tool_registry: Dict, schema_registry: List) -> int:
463
+ """注册视频制作相关工具到 Aria Code 终端交互循环。"""
464
+ added = 0
465
+ tools_map = {
466
+ "analyze_video_script": tool_analyze_video_script,
467
+ "generate_video_clip": tool_generate_video_clip,
468
+ "produce_video_from_script": tool_produce_video_from_script,
469
+ "configure_video_engine": tool_configure_video_engine,
470
+ "create_video_project": tool_create_video_project,
471
+ "get_video_project": tool_get_video_project,
472
+ "regenerate_video_shot": tool_regenerate_video_shot,
473
+ "qa_review_video_clip": tool_qa_review_video_clip,
474
+ }
475
+
476
+ # LOCAL_TOOLS holds (handler, description) pairs, because ToolExecutor
477
+ # indexes local_tools[name][0]. Registering the bare function instead left
478
+ # every one of these tools raising
479
+ # TypeError: 'function' object is not subscriptable
480
+ # the first time the agent tried to call it — registered, listed, and
481
+ # unusable. The description is taken from the tool's own schema so the two
482
+ # cannot drift apart.
483
+ descriptions = {
484
+ schema.get("function", {}).get("name"): schema.get("function", {}).get("description", "")
485
+ for schema in VIDEO_TOOL_SCHEMAS
486
+ if isinstance(schema, dict)
487
+ }
488
+ for name, fn in tools_map.items():
489
+ if name not in tool_registry:
490
+ tool_registry[name] = (fn, descriptions.get(name, "").strip() or name)
491
+ added += 1
492
+
493
+ existing_names = {s.get("function", {}).get("name") for s in schema_registry if isinstance(s, dict)}
494
+ for schema in VIDEO_TOOL_SCHEMAS:
495
+ s_name = schema.get("function", {}).get("name")
496
+ if s_name and s_name not in existing_names:
497
+ schema_registry.append(schema)
498
+
499
+ return added
500
+
@@ -0,0 +1 @@
1
+ """Training data and evaluation for Aria's own model (docs/training-protocol.md)."""
@@ -0,0 +1,117 @@
1
+ """Is a candidate model really better than the baseline? Paired, with uncertainty.
2
+
3
+ A single eval run is one draw: Text2WetLab measured single-attempt scores
4
+ moving by up to 0.70 on a task between two identical runs. So, as a strategy
5
+ is judged on its excess return with a confidence interval rather than one
6
+ good month:
7
+
8
+ - **Paired by task.** Only tasks both runs scored are compared; each task's
9
+ difference is candidate pass rate minus baseline pass rate over its repeats.
10
+ - **Bootstrap interval.** The mean difference is resampled over tasks
11
+ (10,000 draws, fixed seed) for a 95% interval. An interval that includes 0
12
+ is not evidence of improvement.
13
+ - **Sign test.** How many tasks got better versus worse, with an exact
14
+ two-sided binomial p-value.
15
+ - **Errors are not failures.** ERROR and INVALID attempts (quota, timeouts, a
16
+ broken fixture) are left out of the rates, and counted.
17
+ - **Cost.** Mean seconds per attempt for each side: a score bought with
18
+ twice the latency is reported as such.
19
+
20
+ python -m aria_code.training.compare baseline/*.json --candidate tuned/*.json
21
+ """
22
+
23
+ from __future__ import annotations
24
+
25
+ import argparse
26
+ import json
27
+ import math
28
+ import random
29
+ from collections import defaultdict
30
+ from pathlib import Path
31
+ from typing import Iterable
32
+
33
+ SCORED = ("pass", "fail")
34
+
35
+
36
+ def rates(report_paths: Iterable[Path]) -> tuple[dict[str, float], dict[str, int], float]:
37
+ """Per-task pass rate over scored attempts, unscored counts, mean seconds."""
38
+ passed: dict[str, int] = defaultdict(int)
39
+ scored: dict[str, int] = defaultdict(int)
40
+ unscored: dict[str, int] = defaultdict(int)
41
+ seconds: list[float] = []
42
+ for path in report_paths:
43
+ data = json.loads(Path(path).read_text(encoding="utf-8"))
44
+ for r in data.get("results") or []:
45
+ tid, outcome = str(r["task_id"]), str(r["outcome"])
46
+ if outcome in SCORED:
47
+ scored[tid] += 1
48
+ passed[tid] += outcome == "pass"
49
+ seconds.append(float(r.get("seconds") or 0))
50
+ else:
51
+ unscored[outcome] += 1
52
+ rate = {t: passed[t] / scored[t] for t in scored}
53
+ return rate, dict(unscored), (sum(seconds) / len(seconds) if seconds else 0.0)
54
+
55
+
56
+ def bootstrap_ci(diffs: list[float], draws: int = 10_000, seed: int = 7, level: float = 0.95) -> tuple[float, float]:
57
+ if not diffs:
58
+ return (0.0, 0.0)
59
+ rng = random.Random(seed)
60
+ n = len(diffs)
61
+ means = sorted(sum(rng.choice(diffs) for _ in range(n)) / n for _ in range(draws))
62
+ lo = means[int((1 - level) / 2 * draws)]
63
+ hi = means[min(draws - 1, int((1 + level) / 2 * draws))]
64
+ return (lo, hi)
65
+
66
+
67
+ def sign_test(better: int, worse: int) -> float:
68
+ """Exact two-sided binomial p-value under 'no difference' (ties dropped)."""
69
+ n = better + worse
70
+ if n == 0:
71
+ return 1.0
72
+ k = min(better, worse)
73
+ tail = sum(math.comb(n, i) for i in range(k + 1)) / 2 ** n
74
+ return min(1.0, 2 * tail)
75
+
76
+
77
+ def compare(baseline: Iterable[Path], candidate: Iterable[Path]) -> dict:
78
+ base, base_unscored, base_s = rates(baseline)
79
+ cand, cand_unscored, cand_s = rates(candidate)
80
+ common = sorted(set(base) & set(cand))
81
+ diffs = [cand[t] - base[t] for t in common]
82
+ better = sum(d > 0 for d in diffs)
83
+ worse = sum(d < 0 for d in diffs)
84
+ lo, hi = bootstrap_ci(diffs)
85
+ mean = sum(diffs) / len(diffs) if diffs else 0.0
86
+ return {
87
+ "tasks_compared": len(common),
88
+ "baseline_pass_rate": round(sum(base[t] for t in common) / len(common), 4) if common else None,
89
+ "candidate_pass_rate": round(sum(cand[t] for t in common) / len(common), 4) if common else None,
90
+ "mean_difference": round(mean, 4),
91
+ "ci95": [round(lo, 4), round(hi, 4)],
92
+ "tasks_better": better,
93
+ "tasks_worse": worse,
94
+ "tasks_same": len(common) - better - worse,
95
+ "sign_test_p": round(sign_test(better, worse), 4),
96
+ "improved": lo > 0,
97
+ "only_in_baseline": sorted(set(base) - set(cand)),
98
+ "only_in_candidate": sorted(set(cand) - set(base)),
99
+ "unscored": {"baseline": base_unscored, "candidate": cand_unscored},
100
+ "mean_seconds": {"baseline": round(base_s, 1), "candidate": round(cand_s, 1)},
101
+ "per_task": {t: {"baseline": round(base[t], 3), "candidate": round(cand[t], 3)} for t in common},
102
+ }
103
+
104
+
105
+ def main(argv: list[str] | None = None) -> int:
106
+ ap = argparse.ArgumentParser(description="Paired comparison of two sets of eval reports.")
107
+ ap.add_argument("baseline", nargs="+", type=Path)
108
+ ap.add_argument("--candidate", nargs="+", type=Path, required=True)
109
+ args = ap.parse_args(argv)
110
+ result = compare(args.baseline, args.candidate)
111
+ summary = {k: v for k, v in result.items() if k != "per_task"}
112
+ print(json.dumps(summary, ensure_ascii=False, indent=2))
113
+ return 0
114
+
115
+
116
+ if __name__ == "__main__":
117
+ raise SystemExit(main())