Imported from https://github.com/czxttkl/X-AI