feat(api): include version number in logs
This commit is contained in:
parent
01811abf5a
commit
003a350a6c
|
@ -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()
|
||||
|
|
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue