Unity ML-Agents
Discovery (unity-mcp-server)
get_ai_stack_summary— packages, ONNX assets, script hits, training YAMLsget_ai_ml_package_inventory— confirmcom.unity.ml-agentsversionlist_ml_agents_training_configs— trainer YAML pathsfind_ai_related_scripts— filter label ML-Agents
Implementation checklist
- Training: YAML defines
behaviors,trainer_type, hyperparameters; keep configs in repo root orconfig/. - Inference: Prefabs/scenes need
BehaviorParameters+ trained.onnxinModelfield; verify withlist_prefabs_with_componentforBehaviorParametersif serialized as component name. - Sensors/Observations: Stack vector observations consistently between training and runtime.
- Performance: Prefer inference on worker thread; cap decision frequency for mobile.
Common prompts
- "Does this project use ML-Agents?"
- "List ML-Agents training configs and ONNX models"
- "Which scripts reference BehaviorParameters?"