add random api_key to avoid openai restrict

utils/chatgpt.py

@@ -1,16 +1,25 @@
+import random
+
 import openai
 
 
 class ChatGPTAPI:
     def __init__(self, api_key: str = '', max_input_length: int = 1024):
+        openai.api_key = self.load_api_key(api_key)
+        self.max_input_length = max_input_length
+
+    @staticmethod
+    def load_api_key(api_key: str):
         if not api_key:
             try:
                 api_key = open('data/api_key.txt', 'r').read()
             except Exception as e:
                 raise Exception(f'ChatGPT Error: No API key provided {e}')
 
-        openai.api_key = api_key
-        self.max_input_length = max_input_length
+        if '\n' in api_key:
+            api_key_list = api_key.strip().split('\n')
+            api_key = random.choice(api_key_list)
+        return api_key
 
     def __call__(self, content: str):
         assert isinstance(content, str), 'ChatGPT Error: content must be a string'
@@ -30,7 +39,5 @@ class ChatGPTAPI:
 
 if __name__ == '__main__':
     chatgpt = ChatGPTAPI()
-    r = chatgpt.truncate_string('how are you ' * 10000)
-    r_list = r.split(' ')
-    # response = chatgpt('Hello, how are you?')
-    # print(response)
+    response = chatgpt('Hello, how are you?')
+    print(response)