I get enough of this with a pre/post tool hook that takes a jj snapshot and teaching the agent about `jj evolog`.