Fast fix session error

Author: AlexSua

chatgpt/authentication.py

@@ -123,7 +123,7 @@ def get_session(self):
         except HTTPTLSError as e:
             raise ChatgptError(
                 "Error getting the session. You may have a wrong access token; try to login again or insert an access_token yourself.",
-                ChatgptErrorCodes.LOGIN_ERROR) from e
+                ChatgptErrorCodes.SESSION_ERROR) from e
 
     def login(self, username, password):
         """Get the access token for chatgpt

chatgpt/chatgpt.py

@@ -309,7 +309,7 @@ def chat(self, message: List[str], retry_on_401: bool = True, direct_response: b
             message = [message]
         try:
             if self._access_token_expire is not None:
-                if self._access_token_expire < get_utc_now_datetime():
+                if self._access_token_expire < get_utc_now_datetime() and self._chatgpt_session_expire > get_utc_now_datetime():
                     self.get_session()
 
             if self._access_token is None:
@@ -398,7 +398,14 @@ def chat(self, message: List[str], retry_on_401: bool = True, direct_response: b
         except ChatgptError as ex:
             exception_message = ex.message
             exception_code = ex.code
-            if exception_code == ChatgptErrorCodes.LOGIN_ERROR or exception_code == ChatgptErrorCodes.TIMEOUT_ERROR and retry_on_401:
+            if exception_code == ChatgptErrorCodes.SESSION_ERROR:
+                self._chatgpt_session_expire = get_utc_now_datetime()
+                self._tls_session._cookies = {}
+                self._cookies = {}
+                self._session._cookies = {}
+                self._access_token = None
+                return self.chat(message, False, direct_response=direct_response, stream=stream)
+            elif exception_code == ChatgptErrorCodes.LOGIN_ERROR or exception_code == ChatgptErrorCodes.TIMEOUT_ERROR and retry_on_401:
                 return self.chat(message, False, direct_response=direct_response, stream=stream)
 
         except TLSClientExeption as ex:

chatgpt/errors.py

@@ -10,6 +10,7 @@ class ChatgptErrorCodes(Enum):
     CONFIG_FILE_ERROR = "config_file_error"
     UNKNOWN_ERROR = "unknown_error"
     LOGIN_ERROR = "login_error"
+    SESSION_ERROR = "session_error"
     TIMEOUT_ERROR = "timeout_error"
     CONNECTION_ERROR = "connection_error"