diff --git a/bec_widgets/widgets/services/bec_status_box/bec_status_box.py b/bec_widgets/widgets/services/bec_status_box/bec_status_box.py
index cd21e9b6..f3268c71 100644
--- a/bec_widgets/widgets/services/bec_status_box/bec_status_box.py
+++ b/bec_widgets/widgets/services/bec_status_box/bec_status_box.py
@@ -9,6 +9,7 @@ from dataclasses import dataclass
from typing import TYPE_CHECKING
from bec_lib.utils.import_utils import lazy_import_from
+from pydantic import BaseModel
from qtpy.QtCore import QObject, QTimer, Signal, Slot
from qtpy.QtWidgets import QHBoxLayout, QTreeWidget, QTreeWidgetItem
@@ -18,9 +19,10 @@ from bec_widgets.widgets.services.bec_status_box.status_item import StatusItem
if TYPE_CHECKING: # pragma: no cover
from bec_lib.client import BECClient
-
-# TODO : Put normal imports back when Pydantic gets faster
-BECStatus = lazy_import_from("bec_lib.messages", ("BECStatus",))
+ from bec_lib.messages import BECStatus, ServiceMetricMessage, StatusMessage
+else:
+ # TODO : Put normal imports back when Pydantic gets faster
+ BECStatus = lazy_import_from("bec_lib.messages", ("BECStatus",))
@dataclass
@@ -200,7 +202,11 @@ class BECStatusBox(BECWidget, CompactPopupWidget):
self.status_container[service_name].update({"info": service_info_item})
@Slot(dict, dict)
- def update_service_status(self, services_info: dict, services_metric: dict) -> None:
+ def update_service_status(
+ self,
+ services_info: dict[str, StatusMessage],
+ services_metric: dict[str, ServiceMetricMessage],
+ ) -> None:
"""Callback function services_metric from BECServiceStatusMixin.
It updates the status of all services.
@@ -209,6 +215,9 @@ class BECStatusBox(BECWidget, CompactPopupWidget):
services_metric (dict): A dictionary containing the service metrics for all running BEC services.
"""
checked = [self.box_name]
+ # FIXME: We simply replace the pydantic message with dict for now until we refactor the widget
+ for val in services_info.values():
+ val.info = val.info.model_dump() if isinstance(val.info, BaseModel) else val.info
services_info = self.update_core_services(services_info, services_metric)
checked.extend(self.CORE_SERVICES)
diff --git a/bec_widgets/widgets/services/bec_status_box/status_item.py b/bec_widgets/widgets/services/bec_status_box/status_item.py
index 2858683a..b6192535 100644
--- a/bec_widgets/widgets/services/bec_status_box/status_item.py
+++ b/bec_widgets/widgets/services/bec_status_box/status_item.py
@@ -141,7 +141,11 @@ class StatusItem(QWidget):
metrics_text = (
f"SERVICE: {self.config.service_name}
STATUS: {self.config.status}
"
)
- metrics_text += f"BEC_LIB VERSION: {self.config.info['version']}
"
+ if "version" in self.config.info:
+ metrics_text += f"BEC_LIB VERSION: {self.config.info['version']}
"
+ if "versions" in self.config.info:
+ for component, version in self.config.info["versions"].items():
+ metrics_text += f"{component.upper()} VERSION: {version}
"
if self.config.metrics:
for key, value in self.config.metrics.items():
if key == "create_time":
diff --git a/tests/unit_tests/test_bec_status_box.py b/tests/unit_tests/test_bec_status_box.py
index 363602df..974fab85 100644
--- a/tests/unit_tests/test_bec_status_box.py
+++ b/tests/unit_tests/test_bec_status_box.py
@@ -2,7 +2,7 @@
from unittest import mock
import pytest
-from bec_lib.messages import BECStatus, ServiceMetricMessage, StatusMessage
+from bec_lib.messages import BECStatus, ServiceInfo, ServiceMetricMessage, StatusMessage
from bec_widgets.widgets.services.bec_status_box.bec_status_box import (
BECServiceInfoContainer,
@@ -75,13 +75,17 @@ def test_update_service_status(status_box):
"""Also checks check redundant tree items"""
name = "test_service"
status = BECStatus.IDLE
- info = {"test": "test"}
+ info = StatusMessage(name=name, status=status, info=ServiceInfo(user="test", hostname="host"))
metrics = {"metric": "test_metric"}
status_box.add_tree_item(name, status, info, {})
not_connected_name = "invalid_service"
status_box.add_tree_item(not_connected_name, status, info, metrics)
- services_status = {name: StatusMessage(name=name, status=status, info=info)}
+ services_status = {
+ name: StatusMessage(
+ name=name, status=status, info=ServiceInfo(user="test", hostname="host")
+ )
+ }
services_metrics = {name: ServiceMetricMessage(name=name, metrics=metrics)}
with mock.patch.object(status_box, "update_core_services", return_value=services_status):
@@ -95,7 +99,7 @@ def test_update_core_services(status_box):
status_box.CORE_SERVICES = ["test_service"]
name = "test_service"
status = BECStatus.RUNNING
- info = {"test": "test"}
+ info = ServiceInfo(user="test", hostname="host")
metrics = {"metric": "test_metric"}
services_status = {name: StatusMessage(name=name, status=status, info=info)}
services_metrics = {name: ServiceMetricMessage(name=name, metrics=metrics)}
@@ -115,7 +119,7 @@ def test_update_core_services(status_box):
def test_double_click_item(status_box):
name = "test_service"
status = BECStatus.IDLE
- info = {"test": "test"}
+ info = ServiceInfo(user="test", hostname="host")
metrics = {"MyData": "This should be shown nicely"}
status_box.add_tree_item(name, status, info, metrics)
container = status_box.status_container[name]