fix: issues caught by testing /service-info

bento-platform · Feb 12, 2024 · 618983d · 618983d
1 parent c4916c3
commit 618983d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bento_aggregation_service/app.py b/bento_aggregation_service/app.py
@@ -38,5 +38,5 @@ async def _git_stdout(*args) -> str:
 
 @application.get("/service-info")
 async def service_info(config: ConfigDependency, logger: LoggerDependency):
-    return build_service_info_from_pydantic_config(
+    return await build_service_info_from_pydantic_config(
         config, logger, {"serviceKind": BENTO_SERVICE_KIND}, SERVICE_TYPE, __version__)
diff --git a/bento_aggregation_service/config.py b/bento_aggregation_service/config.py
@@ -3,7 +3,7 @@
 from functools import lru_cache
 from typing import Annotated
 
-from .constants import SERVICE_TYPE
+from .constants import SERVICE_NAME, SERVICE_TYPE
 
 __all__ = [
     "Config",
@@ -14,6 +14,7 @@
 
 class Config(BentoBaseConfig):
     service_id: str = str(":".join(list(SERVICE_TYPE.values())[:2]))
+    service_name: str = SERVICE_NAME
 
     request_timeout: int = 180  # seconds