AI is not really supported well in the beta, I advise waiting till the july release before any serious AI work is started. If you have to modify the AI, there are various trigger actions which I believe allow you to change the easy AI to relate to custom buildings / units. Other AI programming requires you to manually overhaul the very easy AI script and import it each time you want to test.