fix: use route instead of api name

bentoml · Jan 17, 2025 · 5852691 · 5852691
1 parent cbc3483
commit 5852691
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/_bentoml_impl/worker/service.py b/src/_bentoml_impl/worker/service.py
@@ -186,7 +186,7 @@ def main(
     BentoMLContainer.development_mode.set(development_mode)
 
     server_context.service_name = service.name
-    server_context.service_apis = [api for api in service.apis.keys()]
+    server_context.service_routes = [api.route for api in service.apis.values()]
     if service.bento is None:
         server_context.bento_name = service.name
         server_context.bento_version = "not available"

diff --git a/src/bentoml/_internal/server/http/instruments.py b/src/bentoml/_internal/server/http/instruments.py
@@ -209,7 +209,7 @@ def _setup(
         )
         # place holder metrics to initialize endpoints with 0 to facilitate rate calculation
         # otherwise, rate() can be 0 if the endpoint is hit for the first time
-        for endpoint in server_context.service_apis:
+        for endpoint in server_context.service_routes:
             self.metrics_request_total.labels(
                 endpoint=endpoint,
                 service_name=server_context.bento_name,