Lithium is like cocain for silicon based lifeforms... perhaps more noncunductive shielding would work better on the AI's mood swings. Or literally throw a wet blanket over it to keep it from nuc7ing the world.