apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

load plugins from env

This commit is contained in:
Sean Sube 2023-11-19 18:41:55 -06:00
parent 98fcc07524
commit 8a3174aaff
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -64,6 +64,7 @@ class ServerContext:
server_version: Optional[str] = DEFAULT_SERVER_VERSION, server_version: Optional[str] = DEFAULT_SERVER_VERSION,
worker_retries: Optional[int] = DEFAULT_WORKER_RETRIES, worker_retries: Optional[int] = DEFAULT_WORKER_RETRIES,
feature_flags: Optional[List[str]] = None, feature_flags: Optional[List[str]] = None,
plugins: Optional[List[str]] = None,
) -> None: ) -> None:
self.bundle_path = bundle_path self.bundle_path = bundle_path
self.model_path = model_path self.model_path = model_path
@ -85,6 +86,7 @@ class ServerContext:
self.server_version = server_version self.server_version = server_version
self.worker_retries = worker_retries self.worker_retries = worker_retries
self.feature_flags = feature_flags or [] self.feature_flags = feature_flags or []
self.plugins = plugins or []
self.cache = ModelCache(self.cache_limit) self.cache = ModelCache(self.cache_limit)
@ -125,6 +127,7 @@ class ServerContext:
environ.get("ONNX_WEB_WORKER_RETRIES", DEFAULT_WORKER_RETRIES) environ.get("ONNX_WEB_WORKER_RETRIES", DEFAULT_WORKER_RETRIES)
), ),
feature_flags=environ.get("ONNX_WEB_FEATURE_FLAGS", "").split(","), feature_flags=environ.get("ONNX_WEB_FEATURE_FLAGS", "").split(","),
plugins=environ.get("ONNX_WEB_PLUGINS", "").split(","),
) )
def has_feature(self, flag: str) -> bool: def has_feature(self, flag: str) -> bool: