fix docstrings in ai.py