Improve docstrings for assistant_base