fix(cli): component name regression (#2944)

bentoml · Aug 26, 2022 · 6370972 · 6370972
1 parent d3d9cdb
commit 6370972
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/bentoml_cli/worker/http_api_server.py b/bentoml_cli/worker/http_api_server.py
@@ -109,6 +109,8 @@ def main(
     from bentoml._internal.context import component_context
     from bentoml._internal.configuration.containers import BentoMLContainer
 
+    component_context.component_type = "api_server"
+    component_context.component_index = worker_id
     configure_server_logging()
 
     BentoMLContainer.development_mode.set(False)
@@ -120,9 +122,7 @@ def main(
     svc = bentoml.load(bento_identifier, working_dir=working_dir, standalone_load=True)
 
     # setup context
-    component_context.component_type = "api_server"
     component_context.component_name = svc.name
-    component_context.component_index = worker_id
     if svc.tag is None:
         component_context.bento_name = f"*{svc.__class__.__name__}"
         component_context.bento_version = "not available"

diff --git a/bentoml_cli/worker/http_dev_api_server.py b/bentoml_cli/worker/http_dev_api_server.py
@@ -83,6 +83,7 @@ def main(
     from bentoml._internal.context import component_context
     from bentoml._internal.configuration.containers import BentoMLContainer
 
+    component_context.component_type = "dev_api_server"
     configure_server_logging()
 
     if prometheus_dir is not None:
@@ -91,7 +92,6 @@ def main(
     svc = load(bento_identifier, working_dir=working_dir, standalone_load=True)
 
     # setup context
-    component_context.component_type = "dev_api_server"
     component_context.component_name = svc.name
     if svc.tag is None:
         component_context.bento_name = f"*{svc.__class__.__name__}"

diff --git a/bentoml_cli/worker/runner.py b/bentoml_cli/worker/runner.py
@@ -79,7 +79,11 @@ def main(
     from bentoml import load
     from bentoml._internal.context import component_context
 
-    component_context.component_name = f"runner:{runner_name}:{worker_id}"
+    # setup context
+    component_context.component_type = "runner"
+    component_context.component_name = runner_name
+    component_context.component_index = worker_id
+
     from bentoml._internal.log import configure_server_logging
 
     configure_server_logging()