From cdadb258bf66f09673ca36472a29fbc44dd424e9 Mon Sep 17 00:00:00 2001 From: lanqian528 <5499636+lanqian528@users.noreply.github.com> Date: Sat, 26 Oct 2024 03:11:22 +0800 Subject: [PATCH] v1.5.5 use rt and at --- chatgpt/globals.py | 1 + chatgpt/refreshToken.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/chatgpt/globals.py b/chatgpt/globals.py index 7cc3aa2..90ea6ca 100644 --- a/chatgpt/globals.py +++ b/chatgpt/globals.py @@ -59,6 +59,7 @@ else: with open(TOKENS_FILE, "w", encoding="utf-8") as f: pass + if os.path.exists(ERROR_TOKENS_FILE): with open(ERROR_TOKENS_FILE, "r", encoding="utf-8") as f: for line in f: diff --git a/chatgpt/refreshToken.py b/chatgpt/refreshToken.py index 6062513..86a7b14 100644 --- a/chatgpt/refreshToken.py +++ b/chatgpt/refreshToken.py @@ -45,14 +45,14 @@ async def chat_refresh(refresh_token): access_token = r.json()['access_token'] return access_token else: - if "Unknown or invalid refresh token." in r.text: - with open(globals.ERROR_TOKENS_FILE, "a", encoding="utf-8") as f: - f.write(refresh_token + "\n") + if "invalid_grant" in r.text or "access_denied" in r.text: if refresh_token not in globals.error_token_list: globals.error_token_list.append(refresh_token) + with open(globals.ERROR_TOKENS_FILE, "a", encoding="utf-8") as f: + f.write(refresh_token + "\n") raise Exception(r.text) else: - raise Exception(r.text[:100]) + raise Exception(r.text[:300]) except Exception as e: logger.error(f"Failed to refresh access_token `{refresh_token}`: {str(e)}") raise HTTPException(status_code=500, detail=f"Failed to refresh access_token.")