替换控制器的日志库

Author: shell-nlp

gpt_server/serving/controller.py

@@ -26,10 +26,7 @@
     ErrorCode,
     SERVER_ERROR_MSG,
 )
-from fastchat.utils import build_logger
-
-
-logger = build_logger("controller", "controller.log")
+from loguru import logger
 
 
 class DispatchMethod(Enum):

gpt_server/serving/openai_api_server.py

@@ -68,9 +68,7 @@
     APITokenCheckResponse,
     APITokenCheckResponseItem,
 )
-from fastchat.utils import build_logger
-
-logger = build_logger("openai_api_server", "openai_api_server.log")
+from loguru import logger
 
 conv_template_map = {}