You still get snapshots of the working tree every time an agent calls a jj command, including reading ones. This is regular for me that I'm not worried to lose any intermediate state.
would it make sense to add jj snapshotting commands in agent hooks?