load plugins from env
This commit is contained in:
parent
98fcc07524
commit
8a3174aaff
|
@ -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:
|
||||||
|
|
Loading…
Reference in New Issue