UStateTreeAIComponentSchema

Use it on the UStateTreeAIComponent, which is placed on a AIController.