Further improve ai log output