docs: Document the Python API