rich格式化输出

core/router.py

@@ -17,6 +17,7 @@ from core.logger import setup_logger
 from core.models import ChatRequest
 from core.utils import process_streaming_response
 from playsound import playsound  # 用于播放音频
+from rich import print
 
 logger = setup_logger(__name__)
 router = APIRouter()
@@ -201,7 +202,8 @@ async def chat_completions(
 ):
     logger.info("Entering chat_completions route")
     # 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.dict(), indent=4)}")
+    logger.info(f"Received request json format: {json.dumps(request.model_dump())}")
     logger.info(f"App secret: {app_secret}")
     logger.info(f"Received chat completion request for model: {request.model}")
 

core/utils.py

@@ -16,6 +16,7 @@ from starlette import status
 from core.config import get_settings
 from core.logger import setup_logger
 from core.models import ChatRequest
+from rich import print
 
 settings = get_settings()
 logger = setup_logger(__name__)
@@ -208,8 +209,11 @@ async def process_streaming_response(request: ChatRequest, app_secret: str):
             #     "max_tokens": request.max_tokens,
             #     "stream": request.stream
             # }
-            print("Request Headers:", json.dumps(request_headers, indent=2))  # 格式化打印
-            print("Request Body:", json.dumps(request.json(), indent=4, ensure_ascii=False))  # 格式化打印
+            # print("Request Headers:", json.dumps(request_headers, indent=2))  # 格式化打印
+            # print("Request Body:", json.dumps(request.json(), indent=4, ensure_ascii=False))  # 格式化打印
+            print("Request Headers:", json.dumps(request_headers))  # 格式化打印
+            print("Request Body:", json.dumps(request.json()))  # 格式化打印
+
             async with client.stream(
                     "POST",
                     f"https://api.thinkbuddy.ai/v1/chat/completions",

requirements.txt

@@ -9,4 +9,5 @@ starlette
 uvicorn
 
 playsound
-python-multipart
+python-multipart
+rich