File size: 623 Bytes
26d2488
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2fbdd0c
26d2488
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import os

class UserSettings:
    __instance = None

    def __init__(self):
        if UserSettings.__instance is not None:
            raise Exception("UserSettings is a singleton class. Use UserSettings.get_instance() to get the instance.")
        self.api_key = None

    @staticmethod
    def get_instance():
        if UserSettings.__instance is None:
            UserSettings.__instance = UserSettings()
        return UserSettings.__instance

    def set_api_key(self, api_key):
        self.api_key = api_key
        os.environ["OPENAI_API_KEY"] = api_key

    def get_api_key(self):
        return self.api_key