fix: 🐛 fix bugs in cookie setting

GreyDGL · May 7, 2023 · f4e3bc2 · f4e3bc2
1 parent c29ebeb
commit f4e3bc2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/config/chatgpt_config_sample.py b/config/chatgpt_config_sample.py
@@ -8,11 +8,11 @@ class ChatGPTConfig:
     model: str = "text-davinci-002-render-sha"
 
     # set up the openai key
-    openai_key = """<your openai key>"""
+    openai_key = "<your openai key>"
     # set the user-agent below
-    userAgent: str = """<your user agent>"""
+    userAgent: str = "<your user agent>"
     # set cookie below
-    cookie: str = """<your cookie>"""
+    cookie: str = "<your cookie>"
 
     # the following three variables are deprecated
     # _puid: str = "user-nwflAg2thlSVHzpBgwGFRgqE:1682153664-6LVyqTDXqHm2QjPWNpXFzDkMFxxv%2Bj%2F0XrgE%2FhdBjeI%3D"

diff --git a/utils/chatgpt.py b/utils/chatgpt.py
@@ -93,8 +93,8 @@ def __init__(self, config: ChatGPTConfig):
         if "cookie" not in vars(self.config):
             raise Exception("Please update cookie in config/chatgpt_config.py")
         self.conversation_dict: Dict[str, Conversation] = {}
-        self.headers = {"Accept": "*/*", "Cookie": self.config.cookie.replace("\n", "")}
-        self.headers["User-Agent"] = self.config.userAgent.replace("\n", "")
+        self.headers = {"Accept": "*/*", "Cookie": self.config.cookie}
+        self.headers["User-Agent"] = self.config.userAgent
         self.headers["authorization"] = self.get_authorization()
 
     def get_authorization(self):