refactor(agent): Simplify logging in agent pkg