From 38f00302bfbd13a0a8eb889856fdccda689ecb7f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A7=9A=E5=87=AF?= <kai.yao@chaitin.com>
Date: Thu, 3 Apr 2025 19:09:38 +0800
Subject: [PATCH] fix: log format

---
 mcp_server/config/config.py |  2 +-
 mcp_server/server.py        |  1 -
 mcp_server/utils/request.py | 13 ++++++++-----
 3 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/mcp_server/config/config.py b/mcp_server/config/config.py
index 784ecb1..273c637 100644
--- a/mcp_server/config/config.py
+++ b/mcp_server/config/config.py
@@ -53,4 +53,4 @@ def set_log_level():
             case "critical":
                 level = logging.CRITICAL
 
-    logging.basicConfig(level=level)
\ No newline at end of file
+    logging.basicConfig(level=level, format="%(asctime)s - %(levelname)s - %(message)s")
\ No newline at end of file
diff --git a/mcp_server/server.py b/mcp_server/server.py
index bb8ef5a..5ddd0a1 100644
--- a/mcp_server/server.py
+++ b/mcp_server/server.py
@@ -5,7 +5,6 @@ from starlette.applications import Starlette
 from starlette.routing import Route, Mount
 from starlette.requests import Request
 import uvicorn
-from starlette.responses import PlainTextResponse
 from tools import tools
 from config import GLOBAL_CONFIG
 from middleware import AuthenticationMiddleware
diff --git a/mcp_server/utils/request.py b/mcp_server/utils/request.py
index dcf06cd..506d04f 100644
--- a/mcp_server/utils/request.py
+++ b/mcp_server/utils/request.py
@@ -19,8 +19,11 @@ async def post_slce_api(path: str,req_body: dict) -> str:
     if not path.startswith("/"):
         path = f"/{path}"
 
-    async with AsyncClient(verify=False) as client:
-        response = await client.post(f"{GLOBAL_CONFIG.SAFELINE_ADDRESS}{path}", json=req_body, headers={
-            "X-SLCE-API-TOKEN": f"{GLOBAL_CONFIG.SAFELINE_API_TOKEN}"
-        })
-        return check_slce_response(response)
\ No newline at end of file
+    try:
+        async with AsyncClient(verify=False) as client:
+            response = await client.post(f"{GLOBAL_CONFIG.SAFELINE_ADDRESS}{path}", json=req_body, headers={
+                "X-SLCE-API-TOKEN": f"{GLOBAL_CONFIG.SAFELINE_API_TOKEN}"
+            })
+            return check_slce_response(response)
+    except Exception as e:
+        return str(e)
\ No newline at end of file