Update

app.py

@@ -46,14 +46,9 @@ def fetch_tokens():
     session = requests.Session()
 
     # 检查并获取 appSession
-    if (
-        not cache["app_session_time"]
-        or datetime.now() - cache["app_session_time"] >= APP_SESSION_VALIDITY
-    ):
+    if not cache["app_session_time"] or datetime.now() - cache["app_session_time"] >= APP_SESSION_VALIDITY:
         logger.info("Fetching new appSession")
-        login_page_response = session.get(
-            "https://chat.reka.ai/bff/auth/login", allow_redirects=True
-        )
+        login_page_response = session.get("https://chat.reka.ai/bff/auth/login", allow_redirects=True)
         if login_page_response.status_code != 200:
             logger.error("Failed to load login page")
             return None
@@ -79,10 +74,7 @@ def fetch_tokens():
         cache["app_session_time"] = datetime.now()  # type: ignore
 
     # 检查并获取 accessToken
-    if (
-        not cache["access_token_time"]
-        or datetime.now() - cache["access_token_time"] >= ACCESS_TOKEN_VALIDITY
-    ):
+    if not cache["access_token_time"] or datetime.now() - cache["access_token_time"] >= ACCESS_TOKEN_VALIDITY:
         logger.info("Fetching new accessToken")
         response = session.get(
             "https://chat.reka.ai/bff/auth/access_token",
@@ -94,7 +86,7 @@ def fetch_tokens():
         cache["access_token"] = response.json().get("accessToken")
         cache["access_token_time"] = datetime.now()  # type: ignore
 
-    y(cache)
+    # y(cache)
 
     return cache["access_token"]
 
@@ -123,11 +115,7 @@ def chat_completions():
             },
         )
 
-    if (
-        request.method != "POST"
-        or request.path != "/hf/v1/chat/completions"
-        or request.headers.get("Authorization") != f"Bearer {AUTHKEY}"
-    ):
+    if request.method != "POST" or request.path != "/hf/v1/chat/completions" or request.headers.get("Authorization") != f"Bearer {AUTHKEY}":
         logger.error("Unauthorized access attempt")
         return Response("Unauthorized", status=401)
 
@@ -164,9 +152,7 @@ def chat_completions():
             conversation_history.insert(
                 i + 1,
                 {
-                    "type": "model"
-                    if conversation_history[i]["type"] == "human"
-                    else "human",
+                    "type": "model" if conversation_history[i]["type"] == "human" else "human",
                     "text": "",
                 },
             )
@@ -226,11 +212,7 @@ def chat_completions():
                     if len(last_four_texts) > 4:
                         last_four_texts.pop(0)
 
-                    if len(last_four_texts) == 4 and (
-                        len(last_four_texts[3]) < len(last_four_texts[2])
-                        or last_four_texts[3].endswith("<sep")
-                        or last_four_texts[3].endswith("<")
-                    ):
+                    if len(last_four_texts) == 4 and (len(last_four_texts[3]) < len(last_four_texts[2]) or last_four_texts[3].endswith("<sep") or last_four_texts[3].endswith("<")):
                         break
 
                     full_content = data["text"]
@@ -238,8 +220,7 @@ def chat_completions():
                     prev_content = full_content
 
                     formatted_data = {
-                        "id": "chatcmpl-"
-                        + "".join([str(time.time()), str(hash(new_content))]),
+                        "id": "chatcmpl-" + "".join([str(time.time()), str(hash(new_content))]),
                         "object": "chat.completion.chunk",
                         "created": created,
                         "model": model,