import os import openai def set_api_key(api_key): openai.api_key = api_key return api_key def stop_api_key(value): os.environ.pop('OPENAI_API_KEY', None) openai.api_key = None return 'Stopped API key'