From 1feddbcc01f92d92b32072a1e931546e060b3b71 Mon Sep 17 00:00:00 2001
From: Thomas Young <35565423+HexToString@users.noreply.github.com>
Date: Wed, 11 Jan 2023 20:12:37 +0800
Subject: [PATCH] [Serving] fix simple_serving log error (#1117)

* fix simple_serving log error

* 1

1
---
 tools/common_tools/common_tools.py | 39 +++++++++++++++++++++++++++++-
 tools/setup.py                     |  2 +-
 2 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/tools/common_tools/common_tools.py b/tools/common_tools/common_tools.py
index 3a0f1f6c6..ebb96bc4c 100755
--- a/tools/common_tools/common_tools.py
+++ b/tools/common_tools/common_tools.py
@@ -132,7 +132,44 @@ def main():
             print(
                 "Model convert failed! Please check if you have installed it!")
     if args.tools == "simple_serving":
-        uvicorn.run(args.app, host=args.host, port=args.port, app_dir='.')
+        custom_logging_config = {
+            "version": 1,
+            "disable_existing_loggers": False,
+            "formatters": {
+                "default": {
+                    "()": "uvicorn.logging.DefaultFormatter",
+                    "fmt": "%(asctime)s %(levelprefix)s %(message)s",
+                    'datefmt': '%Y-%m-%d %H:%M:%S',
+                    "use_colors": None,
+                },
+            },
+            "handlers": {
+                "default": {
+                    "formatter": "default",
+                    "class": "logging.StreamHandler",
+                    "stream": "ext://sys.stderr",
+                },
+                'null': {
+                    "formatter": "default",
+                    "class": 'logging.NullHandler'
+                }
+            },
+            "loggers": {
+                "": {
+                    "handlers": ["null"],
+                    "level": "DEBUG"
+                },
+                "uvicorn.error": {
+                    "handlers": ["default"],
+                    "level": "DEBUG"
+                }
+            },
+        }
+        uvicorn.run(args.app,
+                    host=args.host,
+                    port=args.port,
+                    app_dir='.',
+                    log_config=custom_logging_config)
 
 
 if __name__ == '__main__':
diff --git a/tools/setup.py b/tools/setup.py
index 20ec6489a..477168870 100644
--- a/tools/setup.py
+++ b/tools/setup.py
@@ -7,7 +7,7 @@ install_requires = ['uvicorn==0.16.0']
 
 setuptools.setup(
     name="fastdeploy-tools",  # name of package
-    version="0.0.2",  #version of package
+    version="0.0.3",  #version of package
     description="A toolkit for FastDeploy.",
     long_description=long_description,
     long_description_content_type="text/plain",