apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): include version number in logs

This commit is contained in:
Sean Sube 2023-06-26 17:24:03 -05:00
parent 01811abf5a
commit 003a350a6c
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/onnx_web/main.py
View File

@ -80,7 +80,11 @@ def run():
logger.info("shutting down workers") logger.info("shutting down workers")
p.join() 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)) atexit.register(partial(quit, pool))
return app return app
@ -89,7 +93,11 @@ if __name__ == "__main__":
server, app, pool = main() server, app, pool = main()
logger.info("starting image workers") logger.info("starting image workers")
pool.start() 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()) app.run("0.0.0.0", 5000, debug=is_debug())
logger.info("shutting down workers") logger.info("shutting down workers")
pool.join() pool.join()

6
api/onnx_web/server/context.py
View File

@ -13,6 +13,7 @@ logger = getLogger(__name__)
DEFAULT_CACHE_LIMIT = 5 DEFAULT_CACHE_LIMIT = 5
DEFAULT_JOB_LIMIT = 10 DEFAULT_JOB_LIMIT = 10
DEFAULT_IMAGE_FORMAT = "png" DEFAULT_IMAGE_FORMAT = "png"
DEFAULT_SERVER_VERSION = "v0.10.0"
class ServerContext: class ServerContext:
@ -35,6 +36,7 @@ class ServerContext:
job_limit: int = DEFAULT_JOB_LIMIT, job_limit: int = DEFAULT_JOB_LIMIT,
memory_limit: Optional[int] = None, memory_limit: Optional[int] = None,
admin_token: Optional[str] = None, admin_token: Optional[str] = None,
server_version: Optional[str] = DEFAULT_SERVER_VERSION,
) -> None: ) -> None:
self.bundle_path = bundle_path self.bundle_path = bundle_path
self.model_path = model_path self.model_path = model_path
@ -53,6 +55,7 @@ class ServerContext:
self.job_limit = job_limit self.job_limit = job_limit
self.memory_limit = memory_limit self.memory_limit = memory_limit
self.admin_token = admin_token or token_urlsafe() self.admin_token = admin_token or token_urlsafe()
self.server_version = server_version
self.cache = ModelCache(self.cache_limit) self.cache = ModelCache(self.cache_limit)
@ -82,6 +85,9 @@ class ServerContext:
job_limit=int(environ.get("ONNX_WEB_JOB_LIMIT", DEFAULT_JOB_LIMIT)), job_limit=int(environ.get("ONNX_WEB_JOB_LIMIT", DEFAULT_JOB_LIMIT)),
memory_limit=memory_limit, memory_limit=memory_limit,
admin_token=environ.get("ONNX_WEB_ADMIN_TOKEN", None), admin_token=environ.get("ONNX_WEB_ADMIN_TOKEN", None),
server_version=environ.get(
"ONNX_WEB_SERVER_VERSION", DEFAULT_SERVER_VERSION
),
) )
def torch_dtype(self): def torch_dtype(self):