精简日志输出

core/refresh_token.py

@@ -43,9 +43,9 @@ class TokenManager:
             if result is not None:
                 self.id_token = result
                 # 修改配置中的 TOKEN
-                print(f"Before Token is {os.getenv('TOKEN', '')}")
+                # print(f"Before Token is {os.getenv('TOKEN', '')}")
                 os.environ["TOKEN"] = self.id_token
-                print(f"Now Token is {os.getenv('TOKEN', '')}")
+                # print(f"Now Token is {os.getenv('TOKEN', '')}")
                 self.last_refresh_time = time.time()
                 print(f"Token refreshed at {datetime.now()}")
             else:

core/router.py

@@ -229,7 +229,7 @@ async def chat_completions(
     # logger.info(f"Received request: {request}")
     # logger.info(f"Received request json format: {json.dumps(request.dict(), indent=4)}")
     # logger.info(f"Received request json format: {json.dumps(request.model_dump())}")
-    logger.info(f"Received request json format: {json.dumps(request.model_dump(), indent=4, ensure_ascii=False, sort_keys=True, separators=(',', ': '))}")
+    # logger.info(f"Received request json format: {json.dumps(request.model_dump(), indent=4, ensure_ascii=False, sort_keys=True, separators=(',', ': '))}")
     logger.info(f"App secret: {app_secret}")
     logger.info(f"Received chat completion request for model: {request.model}")
 
@@ -285,4 +285,7 @@ async def health_check(request: Request):
 
 @router.post("/env")
 async def environment(app_secret: str = Depends(verify_app_secret)):
-    return Response(content=json.dumps({"token": os.getenv("TOKEN", ""), "refresh_token": os.getenv("REFRESH_TOKEN", ""), "key": os.getenv("FIREBASE_API_KEY", "")}), media_type="application/json")
+    length = 0
+    if os.getenv('TOKEN', '').split(',') is not None:
+        length = len(os.getenv('TOKEN', '').split(','))
+    return Response(content=json.dumps({"token": os.getenv("TOKEN", ""), "length": length, "refresh_token": os.getenv("REFRESH_TOKEN", ""), "key": os.getenv("FIREBASE_API_KEY", "")}), media_type="application/json")

core/utils.py

@@ -234,6 +234,7 @@ async def process_streaming_response(request: ChatRequest, app_secret: str, curr
                         timeout=100,
                 ) as response:
                     response.raise_for_status()
+                    print(f"Response status code: {response.status_code}")
                     timestamp = int(datetime.now().timestamp())
                     async for line in response.aiter_lines():
                         # print(f"{type(line)}: {line}")