[Serving] fix simple_serving log error (PaddlePaddle#1117)

* fix simple_serving log error * 1 1
leyond123 · Jan 11, 2023 · 1feddbc · 1feddbc
1 parent 70bf66b
commit 1feddbc
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 2 deletions.
diff --git 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
@@ -7,7 +7,7 @@
 
 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",