AI-powered key management