Use more strict type guards in ai package

[scotttrinh]

Turned on type-checked linting and fixed the low-hanging fruit of issues that arose. We can't yet add these lint rules in more places, but I wanted to at least get the benefit of these refactors before abandoning them for a while until we have time to address the issues.