aria-code 0.167.0__py3-none-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- aria_bots/__init__.py +7 -0
- aria_bots/channels/README.md +97 -0
- aria_bots/channels/__init__.py +1 -0
- aria_bots/channels/approvals.py +263 -0
- aria_bots/channels/conversation.py +280 -0
- aria_bots/channels/digest.py +186 -0
- aria_bots/channels/registry.py +95 -0
- aria_bots/engine.py +144 -0
- aria_bots/feishu_bot.py +1385 -0
- aria_bots/logistics/__init__.py +1 -0
- aria_bots/logistics/carriers.py +253 -0
- aria_bots/logistics/freight.py +154 -0
- aria_bots/logistics/inventory.py +242 -0
- aria_bots/logistics/mcp_server.py +156 -0
- aria_bots/logistics/records.py +119 -0
- aria_bots/logistics/tenancy.py +152 -0
- aria_bots/market.py +91 -0
- aria_bots/relay_client.py +326 -0
- aria_bots/telegram_bot.py +299 -0
- aria_code/__init__.py +111 -0
- aria_code/_bin/aria-native.exe +0 -0
- aria_code/_bin/aria-runtime.exe +0 -0
- aria_code/_version.py +20 -0
- aria_code/account_memory.py +296 -0
- aria_code/account_session.py +166 -0
- aria_code/adk_apps/__init__.py +1 -0
- aria_code/adk_apps/aria_code_review/__init__.py +1 -0
- aria_code/adk_apps/aria_code_review/agent.py +32 -0
- aria_code/adk_apps/aria_research/__init__.py +1 -0
- aria_code/adk_apps/aria_research/agent.py +31 -0
- aria_code/agents/__init__.py +49 -0
- aria_code/agents/base.py +329 -0
- aria_code/agents/code_review.py +200 -0
- aria_code/agents/deep/__init__.py +37 -0
- aria_code/agents/deep/calibration_loop.py +144 -0
- aria_code/agents/deep/critic.py +125 -0
- aria_code/agents/deep/deepen.py +193 -0
- aria_code/agents/deep/models.py +149 -0
- aria_code/agents/deep/pipeline.py +164 -0
- aria_code/agents/deep/quant_fusion.py +192 -0
- aria_code/agents/deep/themes.py +95 -0
- aria_code/agents/deep/tiers.py +106 -0
- aria_code/agents/engineering/__init__.py +14 -0
- aria_code/agents/engineering/coder.py +222 -0
- aria_code/agents/engineering/tester.py +145 -0
- aria_code/agents/financial/__init__.py +21 -0
- aria_code/agents/financial/cashflow_burn.py +119 -0
- aria_code/agents/financial/catalyst.py +279 -0
- aria_code/agents/financial/corporate_finance.py +175 -0
- aria_code/agents/financial/debate.py +224 -0
- aria_code/agents/financial/earnings.py +315 -0
- aria_code/agents/financial/fundamental.py +178 -0
- aria_code/agents/financial/macro.py +111 -0
- aria_code/agents/financial/news.py +231 -0
- aria_code/agents/financial/risk.py +146 -0
- aria_code/agents/financial/sector.py +279 -0
- aria_code/agents/financial/strategist.py +121 -0
- aria_code/agents/financial/stripe_revenue.py +127 -0
- aria_code/agents/financial/synthesis.py +274 -0
- aria_code/agents/financial/technical.py +277 -0
- aria_code/agents/orchestrator.py +44 -0
- aria_code/agents/portfolio_agent.py +392 -0
- aria_code/agents/realty/__init__.py +62 -0
- aria_code/agents/realty/asset_diagnosis.py +152 -0
- aria_code/agents/realty/business_match.py +166 -0
- aria_code/agents/realty/cashflow_verify.py +209 -0
- aria_code/agents/realty/contract_rules.py +210 -0
- aria_code/agents/realty/energy_anomaly.py +189 -0
- aria_code/agents/realty/exit_settlement.py +208 -0
- aria_code/agents/realty/fulfillment_risk.py +206 -0
- aria_code/agents/realty/ops_optimize.py +160 -0
- aria_code/agents/realty/revenue_share.py +215 -0
- aria_code/agents/registry.py +168 -0
- aria_code/agents/signal_scheme.py +111 -0
- aria_code/agents/sports/__init__.py +0 -0
- aria_code/agents/sports/football_agent.py +197 -0
- aria_code/agents/supervisor.py +113 -0
- aria_code/agents/team.py +500 -0
- aria_code/agents/video/__init__.py +16 -0
- aria_code/agents/video/director.py +376 -0
- aria_code/agents/video/qa_agent.py +143 -0
- aria_code/agents/video/script_doctor.py +112 -0
- aria_code/agents/warehouse/__init__.py +25 -0
- aria_code/agents/warehouse/contracts.py +26 -0
- aria_code/agents/warehouse/dashboard.py +155 -0
- aria_code/agents/warehouse/fulfillment_leadtime.py +95 -0
- aria_code/agents/warehouse/inbound_exceptions.py +25 -0
- aria_code/agents/warehouse/inventory_health.py +24 -0
- aria_code/agents/warehouse/local_run.py +71 -0
- aria_code/agents/warehouse/local_snapshot.py +132 -0
- aria_code/agents/warehouse/logistics_cost.py +84 -0
- aria_code/agents/warehouse/logistics_sync.py +21 -0
- aria_code/agents/warehouse/navigation.schema.json +110 -0
- aria_code/agents/warehouse/task_agent.py +172 -0
- aria_code/agents/warehouse/workflow.py +31 -0
- aria_code/aliyun_data_client.py +4 -0
- aria_code/apps/README.md +12 -0
- aria_code/apps/__init__.py +2 -0
- aria_code/apps/channels/README.md +9 -0
- aria_code/apps/channels/__init__.py +17 -0
- aria_code/apps/channels/approvals.py +11 -0
- aria_code/apps/channels/conversation.py +11 -0
- aria_code/apps/channels/digest.py +11 -0
- aria_code/apps/channels/intake.py +92 -0
- aria_code/apps/channels/registry.py +11 -0
- aria_code/apps/channels/tradingview.py +94 -0
- aria_code/apps/cli/README.md +13 -0
- aria_code/apps/cli/__init__.py +2 -0
- aria_code/apps/cli/_optional.py +120 -0
- aria_code/apps/cli/app_server.py +656 -0
- aria_code/apps/cli/auto_update.py +125 -0
- aria_code/apps/cli/bootstrap.py +206 -0
- aria_code/apps/cli/broker_render.py +206 -0
- aria_code/apps/cli/chat_turn.py +1327 -0
- aria_code/apps/cli/cloud_models.py +231 -0
- aria_code/apps/cli/codegen_paths.py +29 -0
- aria_code/apps/cli/commands/__init__.py +34 -0
- aria_code/apps/cli/commands/_business_render.py +78 -0
- aria_code/apps/cli/commands/_ui.py +71 -0
- aria_code/apps/cli/commands/analysis_cmds.py +378 -0
- aria_code/apps/cli/commands/ashare_prediction_cmds.py +327 -0
- aria_code/apps/cli/commands/auth_cmds.py +215 -0
- aria_code/apps/cli/commands/backtest_cmds.py +2518 -0
- aria_code/apps/cli/commands/broker_cmds.py +1236 -0
- aria_code/apps/cli/commands/business_workflow_cmds.py +379 -0
- aria_code/apps/cli/commands/canvas_cmds.py +77 -0
- aria_code/apps/cli/commands/catalog.py +189 -0
- aria_code/apps/cli/commands/core_cmds.py +2748 -0
- aria_code/apps/cli/commands/data_cmds.py +521 -0
- aria_code/apps/cli/commands/diagnostic_cmds.py +285 -0
- aria_code/apps/cli/commands/diagnostic_ops_cmds.py +813 -0
- aria_code/apps/cli/commands/engineering_cmds.py +399 -0
- aria_code/apps/cli/commands/file_cmds.py +312 -0
- aria_code/apps/cli/commands/finance_service_cmds.py +60 -0
- aria_code/apps/cli/commands/fx_commodity_cmds.py +123 -0
- aria_code/apps/cli/commands/market.py +544 -0
- aria_code/apps/cli/commands/market_cmds.py +1419 -0
- aria_code/apps/cli/commands/market_context.py +432 -0
- aria_code/apps/cli/commands/market_render.py +7 -0
- aria_code/apps/cli/commands/model_cmds.py +2374 -0
- aria_code/apps/cli/commands/ops_cmds.py +721 -0
- aria_code/apps/cli/commands/orchestrator_cmds.py +181 -0
- aria_code/apps/cli/commands/pdf_export_cmds.py +125 -0
- aria_code/apps/cli/commands/portfolio_cmds.py +1316 -0
- aria_code/apps/cli/commands/report.py +476 -0
- aria_code/apps/cli/commands/scaffold_templates.py +617 -0
- aria_code/apps/cli/commands/session_cmds.py +236 -0
- aria_code/apps/cli/commands/session_ux_cmds.py +359 -0
- aria_code/apps/cli/commands/team.py +723 -0
- aria_code/apps/cli/commands/team_render.py +8 -0
- aria_code/apps/cli/commands/ui_cmds.py +401 -0
- aria_code/apps/cli/commands/warehouse_cmds.py +297 -0
- aria_code/apps/cli/commands/workflow_cmds.py +629 -0
- aria_code/apps/cli/commands/workspace_cmds.py +1543 -0
- aria_code/apps/cli/config_paths.py +73 -0
- aria_code/apps/cli/config_store.py +158 -0
- aria_code/apps/cli/constants.py +88 -0
- aria_code/apps/cli/context.py +34 -0
- aria_code/apps/cli/conversation_policy.py +53 -0
- aria_code/apps/cli/deterministic.py +200 -0
- aria_code/apps/cli/direct.py +48 -0
- aria_code/apps/cli/earnings_view.py +172 -0
- aria_code/apps/cli/endpoint_migration.py +64 -0
- aria_code/apps/cli/exec_events.py +255 -0
- aria_code/apps/cli/file_references.py +155 -0
- aria_code/apps/cli/finance_service_catalog.py +111 -0
- aria_code/apps/cli/football_reports.py +231 -0
- aria_code/apps/cli/github_app_auth.py +135 -0
- aria_code/apps/cli/google_login.py +239 -0
- aria_code/apps/cli/handlers/__init__.py +11 -0
- aria_code/apps/cli/handlers/broker_handlers.py +122 -0
- aria_code/apps/cli/handlers/chart_handlers.py +1314 -0
- aria_code/apps/cli/handlers/market_handlers.py +2570 -0
- aria_code/apps/cli/handlers/realty_handlers.py +114 -0
- aria_code/apps/cli/handlers/sports_handlers.py +22 -0
- aria_code/apps/cli/handlers/strategy_advice.py +82 -0
- aria_code/apps/cli/headless.py +352 -0
- aria_code/apps/cli/helpers.py +326 -0
- aria_code/apps/cli/hooks.py +180 -0
- aria_code/apps/cli/i18n.py +310 -0
- aria_code/apps/cli/intent.py +136 -0
- aria_code/apps/cli/intent_router.py +305 -0
- aria_code/apps/cli/intent_shadow.py +72 -0
- aria_code/apps/cli/intent_signals.py +347 -0
- aria_code/apps/cli/lifecycle_hooks.py +48 -0
- aria_code/apps/cli/local_tool_schemas.py +550 -0
- aria_code/apps/cli/main.py +39 -0
- aria_code/apps/cli/market_hours.py +100 -0
- aria_code/apps/cli/market_metadata.py +135 -0
- aria_code/apps/cli/market_universe.py +487 -0
- aria_code/apps/cli/message_processing.py +310 -0
- aria_code/apps/cli/mixin_binding.py +32 -0
- aria_code/apps/cli/model_catalog.py +652 -0
- aria_code/apps/cli/model_probe.py +169 -0
- aria_code/apps/cli/model_retirement.py +65 -0
- aria_code/apps/cli/native_branding.py +12 -0
- aria_code/apps/cli/native_bridge.py +118 -0
- aria_code/apps/cli/native_entry.py +58 -0
- aria_code/apps/cli/pdf_report.py +977 -0
- aria_code/apps/cli/plan_mode.py +139 -0
- aria_code/apps/cli/plotly_html.py +15 -0
- aria_code/apps/cli/prediction_feedback.py +202 -0
- aria_code/apps/cli/preflight.py +502 -0
- aria_code/apps/cli/project_aria.py +60 -0
- aria_code/apps/cli/prompt_assembly.py +78 -0
- aria_code/apps/cli/prompts/__init__.py +0 -0
- aria_code/apps/cli/prompts/coding.py +71 -0
- aria_code/apps/cli/prompts/select.py +193 -0
- aria_code/apps/cli/prompts/system_prompts.py +562 -0
- aria_code/apps/cli/prompts/ui.py +436 -0
- aria_code/apps/cli/provider_endpoints.py +237 -0
- aria_code/apps/cli/provider_keys.py +73 -0
- aria_code/apps/cli/providers/__init__.py +1 -0
- aria_code/apps/cli/providers/base.py +775 -0
- aria_code/apps/cli/providers/chat_routing.py +139 -0
- aria_code/apps/cli/providers/collaboration.py +98 -0
- aria_code/apps/cli/providers/llm/__init__.py +1 -0
- aria_code/apps/cli/providers/llm/offline_responses.py +68 -0
- aria_code/apps/cli/providers/llm/ollama_stream.py +1574 -0
- aria_code/apps/cli/providers/llm/response_cache.py +60 -0
- aria_code/apps/cli/providers/llm/sse_stream.py +298 -0
- aria_code/apps/cli/providers/runtime_bridge.py +586 -0
- aria_code/apps/cli/providers/vertexai_stream.py +592 -0
- aria_code/apps/cli/release_download.py +159 -0
- aria_code/apps/cli/review_runner.py +111 -0
- aria_code/apps/cli/runtime_consumer.py +769 -0
- aria_code/apps/cli/sandbox.py +32 -0
- aria_code/apps/cli/session_export.py +120 -0
- aria_code/apps/cli/session_jsonl.py +208 -0
- aria_code/apps/cli/session_refresh.py +123 -0
- aria_code/apps/cli/session_store.py +125 -0
- aria_code/apps/cli/skills_catalog.py +273 -0
- aria_code/apps/cli/task_isolation.py +211 -0
- aria_code/apps/cli/terminal_chrome.py +412 -0
- aria_code/apps/cli/todo_tracker.py +190 -0
- aria_code/apps/cli/tool_executor.py +932 -0
- aria_code/apps/cli/tool_registry.py +34 -0
- aria_code/apps/cli/tool_scope.py +113 -0
- aria_code/apps/cli/tools/__init__.py +40 -0
- aria_code/apps/cli/tools/context.py +46 -0
- aria_code/apps/cli/tools/file_tools.py +181 -0
- aria_code/apps/cli/tools/market_tools.py +626 -0
- aria_code/apps/cli/tools/notebook_tools.py +111 -0
- aria_code/apps/cli/tools/system_tools.py +656 -0
- aria_code/apps/cli/tools/write_tools.py +961 -0
- aria_code/apps/cli/tradingview_bridge.py +529 -0
- aria_code/apps/cli/transactions.py +269 -0
- aria_code/apps/cli/turn_planning.py +137 -0
- aria_code/apps/cli/update_check.py +257 -0
- aria_code/apps/cli/updater.py +373 -0
- aria_code/apps/cli/utils/__init__.py +0 -0
- aria_code/apps/cli/utils/market_detect.py +1663 -0
- aria_code/apps/cli/web_login.py +136 -0
- aria_code/apps/cli/workflow_runner.py +127 -0
- aria_code/apps/cli/workspace_route.py +121 -0
- aria_code/apps/daemon/README.md +14 -0
- aria_code/apps/vscode/README.md +115 -0
- aria_code/apps/vscode/package.json +70 -0
- aria_code/aria_cli.py +5244 -0
- aria_code/aria_daemon.py +1372 -0
- aria_code/aria_feishu_bot.py +16 -0
- aria_code/aria_mcp_server.py +61 -0
- aria_code/aria_relay_client.py +4 -0
- aria_code/aria_telegram_bot.py +11 -0
- aria_code/ariarc.py +385 -0
- aria_code/artifacts.py +581 -0
- aria_code/backtest_engine.py +4 -0
- aria_code/backtest_report.py +4 -0
- aria_code/brokers/__init__.py +78 -0
- aria_code/brokers/automation.py +193 -0
- aria_code/brokers/base.py +231 -0
- aria_code/brokers/capabilities.py +264 -0
- aria_code/brokers/cn/__init__.py +10 -0
- aria_code/brokers/cn/easytrader_broker.py +193 -0
- aria_code/brokers/cn/futu_broker.py +194 -0
- aria_code/brokers/cn/longbridge_broker.py +190 -0
- aria_code/brokers/cn/tiger_broker.py +196 -0
- aria_code/brokers/cn/xtquant_broker.py +225 -0
- aria_code/brokers/config.py +414 -0
- aria_code/brokers/intl/__init__.py +5 -0
- aria_code/brokers/intl/alpaca_broker.py +183 -0
- aria_code/brokers/intl/ibkr_broker.py +215 -0
- aria_code/brokers/intl/trading212_broker.py +183 -0
- aria_code/brokers/intl/webull_broker.py +156 -0
- aria_code/brokers/paper_broker.py +266 -0
- aria_code/brokers/planning.py +296 -0
- aria_code/brokers/registry.py +213 -0
- aria_code/brokers/trading.py +267 -0
- aria_code/canva_client.py +4 -0
- aria_code/change_store.py +127 -0
- aria_code/clients/__init__.py +0 -0
- aria_code/clients/aliyun_data_client.py +660 -0
- aria_code/clients/aria_relay_client.py +16 -0
- aria_code/clients/canva_client.py +284 -0
- aria_code/clients/figma_client.py +100 -0
- aria_code/clients/football_data_client.py +1671 -0
- aria_code/clients/kling_video_client.py +182 -0
- aria_code/clients/market_data_client.py +1985 -0
- aria_code/clients/openai_image_client.py +242 -0
- aria_code/clients/runway_video_client.py +157 -0
- aria_code/clients/warehouse_erp_client.py +105 -0
- aria_code/cloud_memory.py +190 -0
- aria_code/command_safety.py +19 -0
- aria_code/computer_use_server.py +276 -0
- aria_code/computer_use_tools.py +4 -0
- aria_code/dashboard_generator.py +579 -0
- aria_code/data_analysis_tools.py +4 -0
- aria_code/data_cleaner.py +483 -0
- aria_code/data_service.py +513 -0
- aria_code/datasources/__init__.py +23 -0
- aria_code/datasources/base.py +166 -0
- aria_code/datasources/router.py +253 -0
- aria_code/datasources/sec_statements.py +217 -0
- aria_code/datasources/sources/__init__.py +15 -0
- aria_code/datasources/sources/akshare_source.py +269 -0
- aria_code/datasources/sources/alpha_vantage_source.py +203 -0
- aria_code/datasources/sources/edgar_source.py +219 -0
- aria_code/datasources/sources/finnhub_source.py +198 -0
- aria_code/datasources/sources/fred_source.py +223 -0
- aria_code/datasources/sources/tushare_source.py +143 -0
- aria_code/datasources/sources/web_scraper_source.py +278 -0
- aria_code/datasources/sources/world_bank_source.py +206 -0
- aria_code/datasources/sources/yfinance_source.py +152 -0
- aria_code/doctor.py +860 -0
- aria_code/domain/__init__.py +0 -0
- aria_code/domain/_finance_core/__init__.py +5 -0
- aria_code/domain/_finance_core/backtest_engine.py +325 -0
- aria_code/domain/_finance_core/performance_stats.py +101 -0
- aria_code/domain/backtest_engine.py +75 -0
- aria_code/domain/backtest_report.py +488 -0
- aria_code/domain/finance_formulas.py +389 -0
- aria_code/domain/performance_stats.py +6 -0
- aria_code/domain/portfolio_ledger.py +307 -0
- aria_code/domain/strategy_vault.py +421 -0
- aria_code/evals/__init__.py +37 -0
- aria_code/evals/behavior.py +113 -0
- aria_code/evals/gap.py +93 -0
- aria_code/evals/harness.py +949 -0
- aria_code/evals/review_probe.py +132 -0
- aria_code/evals/runner.py +425 -0
- aria_code/evals/scorecard.py +322 -0
- aria_code/evals/summary.py +125 -0
- aria_code/external_agent_runner.py +115 -0
- aria_code/figma_client.py +4 -0
- aria_code/file_analysis_tools.py +4 -0
- aria_code/finance_formulas.py +4 -0
- aria_code/football_data_client.py +4 -0
- aria_code/image_gen_tools.py +4 -0
- aria_code/intent_classifier.py +459 -0
- aria_code/kling_video_client.py +4 -0
- aria_code/licensing.py +151 -0
- aria_code/local_finance_tools.py +4 -0
- aria_code/local_image_provider.py +4 -0
- aria_code/local_llm_provider.py +4 -0
- aria_code/macro_tools.py +4 -0
- aria_code/markdown_pdf.py +289 -0
- aria_code/market_data_client.py +4 -0
- aria_code/mcp_client.py +744 -0
- aria_code/memory_manager.py +273 -0
- aria_code/model_capability.py +512 -0
- aria_code/notification_tools.py +4 -0
- aria_code/openai_image_client.py +4 -0
- aria_code/packages/__init__.py +26 -0
- aria_code/packages/adk_bridge/__init__.py +11 -0
- aria_code/packages/adk_bridge/code_review_tools.py +49 -0
- aria_code/packages/adk_bridge/market_tools.py +125 -0
- aria_code/packages/aria_agents/__init__.py +5 -0
- aria_code/packages/aria_agents/manifest.py +78 -0
- aria_code/packages/aria_core/__init__.py +34 -0
- aria_code/packages/aria_core/architecture.py +269 -0
- aria_code/packages/aria_core/export.py +147 -0
- aria_code/packages/aria_core/manifest.py +65 -0
- aria_code/packages/aria_core/paths.py +57 -0
- aria_code/packages/aria_core/secure_file.py +68 -0
- aria_code/packages/aria_infra/__init__.py +15 -0
- aria_code/packages/aria_infra/arthera.py +58 -0
- aria_code/packages/aria_infra/doctor.py +246 -0
- aria_code/packages/aria_infra/product.py +40 -0
- aria_code/packages/aria_mcp/__init__.py +25 -0
- aria_code/packages/aria_mcp/bridge.py +232 -0
- aria_code/packages/aria_mcp/circuit.py +82 -0
- aria_code/packages/aria_mcp/config.py +162 -0
- aria_code/packages/aria_mcp/installer.py +95 -0
- aria_code/packages/aria_mcp/server.py +1705 -0
- aria_code/packages/aria_mcp/tools.py +81 -0
- aria_code/packages/aria_sdk/__init__.py +19 -0
- aria_code/packages/aria_sdk/client.py +393 -0
- aria_code/packages/aria_sdk/providers.py +70 -0
- aria_code/packages/aria_sdk/streaming.py +75 -0
- aria_code/packages/aria_sdk/types.py +86 -0
- aria_code/packages/aria_services/__init__.py +92 -0
- aria_code/packages/aria_services/cache.py +60 -0
- aria_code/packages/aria_services/context.py +356 -0
- aria_code/packages/aria_services/context_checkpoints.py +157 -0
- aria_code/packages/aria_services/data.py +11 -0
- aria_code/packages/aria_services/provider_health.py +304 -0
- aria_code/packages/aria_services/references.py +247 -0
- aria_code/packages/aria_services/registry.py +249 -0
- aria_code/packages/aria_services/research_protocol.py +137 -0
- aria_code/packages/aria_services/research_quality.py +186 -0
- aria_code/packages/aria_services/settings.py +160 -0
- aria_code/packages/aria_services/usage.py +151 -0
- aria_code/packages/aria_skills/__init__.py +46 -0
- aria_code/packages/aria_skills/catalog.py +149 -0
- aria_code/packages/aria_skills/loader.py +455 -0
- aria_code/packages/aria_skills/registry.py +84 -0
- aria_code/packages/aria_tools/__init__.py +5 -0
- aria_code/packages/aria_tools/financial/__init__.py +5 -0
- aria_code/packages/aria_tools/financial/_paths.py +92 -0
- aria_code/packages/aria_tools/registry.py +129 -0
- aria_code/packages/quant_engine/__init__.py +33 -0
- aria_code/packages/quant_engine/_namespace.py +24 -0
- aria_code/packages/quant_engine/backtest/__init__.py +10 -0
- aria_code/packages/quant_engine/backtest/core.py +53 -0
- aria_code/packages/quant_engine/backtest/engine.py +102 -0
- aria_code/packages/quant_engine/portfolio/__init__.py +9 -0
- aria_code/packages/quant_engine/portfolio/optimizer.py +85 -0
- aria_code/packages/quant_engine/sports/__init__.py +76 -0
- aria_code/packages/quant_engine/sports/calibrator.py +354 -0
- aria_code/packages/quant_engine/sports/dixon_coles.py +234 -0
- aria_code/packages/quant_engine/sports/elo.py +300 -0
- aria_code/packages/quant_engine/sports/form.py +188 -0
- aria_code/packages/quant_engine/sports/h2h.py +195 -0
- aria_code/packages/quant_engine/sports/predictor.py +312 -0
- aria_code/packages/quant_engine/sports/tracker.py +665 -0
- aria_code/packages/quant_engine/stochastic/__init__.py +31 -0
- aria_code/packages/quant_engine/stochastic/gbm_enhanced.py +195 -0
- aria_code/packages/quant_engine/stochastic/ito_calculus.py +477 -0
- aria_code/packages/quant_engine/stochastic/kelly_criterion.py +181 -0
- aria_code/packages/quant_engine/stochastic/monte_carlo_advanced.py +95 -0
- aria_code/packages/quant_engine/stochastic/options_pricing.py +576 -0
- aria_code/packages/quant_engine/stochastic/stochastic_processes.py +90 -0
- aria_code/packs/__init__.py +85 -0
- aria_code/packs/base.py +168 -0
- aria_code/packs/finance/__init__.py +250 -0
- aria_code/packs/healthcare/__init__.py +40 -0
- aria_code/packs/logistics/__init__.py +171 -0
- aria_code/packs/payments/__init__.py +139 -0
- aria_code/packs/realty/__init__.py +190 -0
- aria_code/packs/registry.py +202 -0
- aria_code/packs/rules.py +90 -0
- aria_code/plan_utils.py +194 -0
- aria_code/plugin_loader.py +328 -0
- aria_code/portfolio_ledger.py +4 -0
- aria_code/preview_server.py +449 -0
- aria_code/privacy/__init__.py +5 -0
- aria_code/privacy/feedback.py +123 -0
- aria_code/project_review.py +278 -0
- aria_code/project_review_api.py +235 -0
- aria_code/project_review_server.py +18 -0
- aria_code/project_tools.py +4 -0
- aria_code/providers/__init__.py +30 -0
- aria_code/providers/google_video_provider.py +256 -0
- aria_code/providers/llm/__init__.py +19 -0
- aria_code/providers/llm/anthropic.py +184 -0
- aria_code/providers/llm/arthera_gateway.py +90 -0
- aria_code/providers/llm/autoconfig.py +213 -0
- aria_code/providers/llm/base.py +139 -0
- aria_code/providers/llm/custom.py +158 -0
- aria_code/providers/llm/ollama.py +137 -0
- aria_code/providers/llm/openai_compat.py +350 -0
- aria_code/providers/llm/registry.py +457 -0
- aria_code/providers/local_image_provider.py +202 -0
- aria_code/providers/local_llm_provider.py +549 -0
- aria_code/providers/ollama_models.py +79 -0
- aria_code/providers/openai_params.py +32 -0
- aria_code/providers/tool_messages.py +100 -0
- aria_code/realty_data_tools.py +4 -0
- aria_code/report_exporters.py +150 -0
- aria_code/report_generator.py +1764 -0
- aria_code/review_service.py +497 -0
- aria_code/runtime/__init__.py +176 -0
- aria_code/runtime/acceptance.py +632 -0
- aria_code/runtime/agent_loop.py +1851 -0
- aria_code/runtime/agent_profiles.py +75 -0
- aria_code/runtime/approval.py +71 -0
- aria_code/runtime/baseline.py +219 -0
- aria_code/runtime/budget.py +232 -0
- aria_code/runtime/checkpoints.py +481 -0
- aria_code/runtime/consult.py +65 -0
- aria_code/runtime/contract.py +282 -0
- aria_code/runtime/delivery.py +432 -0
- aria_code/runtime/events.py +113 -0
- aria_code/runtime/gateway.py +167 -0
- aria_code/runtime/lsp.py +484 -0
- aria_code/runtime/native_index.py +153 -0
- aria_code/runtime/orchestrator.py +55 -0
- aria_code/runtime/processes.py +343 -0
- aria_code/runtime/project_graph.py +614 -0
- aria_code/runtime/repo_map.py +712 -0
- aria_code/runtime/research_bundle.py +105 -0
- aria_code/runtime/review.py +269 -0
- aria_code/runtime/run_state.py +127 -0
- aria_code/runtime/run_store.py +522 -0
- aria_code/runtime/scoped_runs.py +113 -0
- aria_code/runtime/self_healing.py +422 -0
- aria_code/runtime/semantic_diff.py +171 -0
- aria_code/runtime/subagent.py +721 -0
- aria_code/runtime/symbol_edit.py +117 -0
- aria_code/runtime/task_brief.py +46 -0
- aria_code/runtime/task_context.py +145 -0
- aria_code/runtime/task_ledger.py +108 -0
- aria_code/runtime/task_orchestrator.py +90 -0
- aria_code/runtime/task_outcome.py +33 -0
- aria_code/runtime/task_verification.py +266 -0
- aria_code/runtime/task_worktree.py +448 -0
- aria_code/runtime/tool_executor.py +329 -0
- aria_code/runtime/tool_policy.py +107 -0
- aria_code/runtime/transactions.py +216 -0
- aria_code/runtime/windows_job.py +115 -0
- aria_code/runtime/workflows.py +222 -0
- aria_code/runtime/worktrees.py +215 -0
- aria_code/runway_video_client.py +4 -0
- aria_code/safety/__init__.py +27 -0
- aria_code/safety/client_isolation.py +178 -0
- aria_code/safety/permissions.py +339 -0
- aria_code/safety/risk.py +699 -0
- aria_code/safety/sandbox.py +148 -0
- aria_code/safety/service.py +102 -0
- aria_code/service_endpoints.py +14 -0
- aria_code/setup_wizard.py +684 -0
- aria_code/spreadsheet_tools.py +4 -0
- aria_code/strategy_vault.py +4 -0
- aria_code/tools/__init__.py +0 -0
- aria_code/tools/artifact_tools.py +252 -0
- aria_code/tools/broker_tools.py +92 -0
- aria_code/tools/build_quant_engine.py +97 -0
- aria_code/tools/code_audit_tools.py +192 -0
- aria_code/tools/computer_use_tools.py +509 -0
- aria_code/tools/data_analysis_tools.py +831 -0
- aria_code/tools/enterprise_finance_tools.py +202 -0
- aria_code/tools/extended_tools.py +240 -0
- aria_code/tools/file_analysis_tools.py +880 -0
- aria_code/tools/image_gen_tools.py +114 -0
- aria_code/tools/local_finance_tools.py +3488 -0
- aria_code/tools/logistics_carriers.py +11 -0
- aria_code/tools/logistics_inventory.py +11 -0
- aria_code/tools/logistics_io.py +11 -0
- aria_code/tools/logistics_tenancy.py +11 -0
- aria_code/tools/logistics_tools.py +11 -0
- aria_code/tools/macro_tools.py +412 -0
- aria_code/tools/notification_tools.py +249 -0
- aria_code/tools/project_tools.py +525 -0
- aria_code/tools/realty_data_tools.py +662 -0
- aria_code/tools/spreadsheet_tools.py +399 -0
- aria_code/tools/stripe_tools.py +115 -0
- aria_code/tools/video_tools.py +500 -0
- aria_code/training/__init__.py +1 -0
- aria_code/training/compare.py +117 -0
- aria_code/training/prelude_commands.py +111 -0
- aria_code/training/prelude_eval.py +390 -0
- aria_code/training/prelude_real.py +183 -0
- aria_code/training/prelude_train.py +248 -0
- aria_code/training/sft_dataset.py +230 -0
- aria_code/training/trajectory_dataset.py +263 -0
- aria_code/ui/__init__.py +105 -0
- aria_code/ui/banner.py +404 -0
- aria_code/ui/completer.py +571 -0
- aria_code/ui/console.py +285 -0
- aria_code/ui/input_box.py +505 -0
- aria_code/ui/mentions.py +206 -0
- aria_code/ui/picker.py +248 -0
- aria_code/ui/render/__init__.py +11 -0
- aria_code/ui/render/actions.py +297 -0
- aria_code/ui/render/finance.py +1732 -0
- aria_code/ui/render/market.py +253 -0
- aria_code/ui/render/output.py +954 -0
- aria_code/ui/render/responsive.py +75 -0
- aria_code/ui/render/team.py +367 -0
- aria_code/ui/robot.py +239 -0
- aria_code/ui/startup_dashboard.py +110 -0
- aria_code/video_analysis.py +147 -0
- aria_code/video_editor.py +518 -0
- aria_code/video_workflow/__init__.py +8 -0
- aria_code/video_workflow/__main__.py +61 -0
- aria_code/video_workflow/engines/__init__.py +18 -0
- aria_code/video_workflow/engines/base.py +69 -0
- aria_code/video_workflow/engines/custom_model.py +204 -0
- aria_code/video_workflow/engines/google_veo.py +94 -0
- aria_code/video_workflow/engines/local_engine.py +71 -0
- aria_code/video_workflow/engines/registry.py +98 -0
- aria_code/video_workflow/google_models.py +278 -0
- aria_code/video_workflow/music/__init__.py +5 -0
- aria_code/video_workflow/music/foley_service.py +132 -0
- aria_code/video_workflow/music/service.py +212 -0
- aria_code/video_workflow/pipeline.py +620 -0
- aria_code/video_workflow/plan.py +202 -0
- aria_code/video_workflow/session.py +297 -0
- aria_code/workspace/__init__.py +6 -0
- aria_code/workspace/files.py +219 -0
- aria_code/workspace/verify.py +167 -0
- aria_code/workspace_changes.py +199 -0
- aria_code/workspace_service.py +196 -0
- aria_code/workspace_tools.py +150 -0
- aria_code-0.167.0.dist-info/METADATA +330 -0
- aria_code-0.167.0.dist-info/RECORD +601 -0
- aria_code-0.167.0.dist-info/WHEEL +5 -0
- aria_code-0.167.0.dist-info/entry_points.txt +5 -0
- aria_code-0.167.0.dist-info/licenses/LICENSE +201 -0
- aria_code-0.167.0.dist-info/licenses/NOTICE +49 -0
- aria_code-0.167.0.dist-info/top_level.txt +2 -0
|
@@ -0,0 +1,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())
|