diff --git a/api/onnx_web/main.py b/api/onnx_web/main.py index 55342536..7a43d06f 100644 --- a/api/onnx_web/main.py +++ b/api/onnx_web/main.py @@ -80,7 +80,11 @@ def run(): logger.info("shutting down workers") p.join() - logger.info("starting API server with admin token: %s", server.admin_token) + logger.info( + "starting %s API server with admin token: %s", + server.server_version, + server.admin_token, + ) atexit.register(partial(quit, pool)) return app @@ -89,7 +93,11 @@ if __name__ == "__main__": server, app, pool = main() logger.info("starting image workers") pool.start() - logger.info("starting API server with admin token: %s", server.admin_token) + logger.info( + "starting %s API server with admin token: %s", + server.server_version, + server.admin_token, + ) app.run("0.0.0.0", 5000, debug=is_debug()) logger.info("shutting down workers") pool.join() diff --git a/api/onnx_web/server/context.py b/api/onnx_web/server/context.py index 36fb8116..8af1eb2a 100644 --- a/api/onnx_web/server/context.py +++ b/api/onnx_web/server/context.py @@ -13,6 +13,7 @@ logger = getLogger(__name__) DEFAULT_CACHE_LIMIT = 5 DEFAULT_JOB_LIMIT = 10 DEFAULT_IMAGE_FORMAT = "png" +DEFAULT_SERVER_VERSION = "v0.10.0" class ServerContext: @@ -35,6 +36,7 @@ class ServerContext: job_limit: int = DEFAULT_JOB_LIMIT, memory_limit: Optional[int] = None, admin_token: Optional[str] = None, + server_version: Optional[str] = DEFAULT_SERVER_VERSION, ) -> None: self.bundle_path = bundle_path self.model_path = model_path @@ -53,6 +55,7 @@ class ServerContext: self.job_limit = job_limit self.memory_limit = memory_limit self.admin_token = admin_token or token_urlsafe() + self.server_version = server_version self.cache = ModelCache(self.cache_limit) @@ -82,6 +85,9 @@ class ServerContext: job_limit=int(environ.get("ONNX_WEB_JOB_LIMIT", DEFAULT_JOB_LIMIT)), memory_limit=memory_limit, admin_token=environ.get("ONNX_WEB_ADMIN_TOKEN", None), + server_version=environ.get( + "ONNX_WEB_SERVER_VERSION", DEFAULT_SERVER_VERSION + ), ) def torch_dtype(self):