fix imports

mindctrl-addon/config.yaml

@@ -1,6 +1,6 @@
 name: "Mindctrl"
 description: "LLMOps for Homeassistant"
-version: "0.0.40"
+version: "0.0.41"
 url: "https://github.com/akshaya-a/mindctrl/tree/main/mindctrl-addon"
 slug: "mindctrl"
 init: false

mindctrl-addon/rootfs/usr/bin/multiserver/main.py

@@ -16,9 +16,9 @@
 
 import collections
 
-from mlmodels import log_system_models, poll_registry, SUMMARIZATION_PROMPT
+from mlmodels import log_system_models, SUMMARIZATION_PROMPT
 from mqtt import setup_mqtt_client, listen_to_mqtt
-from mlflow_bridge import connect_to_mlflow
+from mlflow_bridge import connect_to_mlflow, poll_registry
 from db.config import setup_db, insert_summary
 
 

mindctrl-addon/rootfs/usr/bin/multiserver/mlflow_bridge.py

@@ -16,7 +16,7 @@ def connect_to_mlflow():
     _logger.info(f"Tracking URI: {mlflow.get_tracking_uri()}")
     _logger.info(f"Model Registry URI: {mlflow.get_registry_uri()}")
 
-from .const import CHAMPION_ALIAS, CHALLENGER_ALIAS
+from const import CHAMPION_ALIAS, CHALLENGER_ALIAS
 
 def is_deployable_alias(aliases: list[str]) -> bool:
     if not aliases:

mindctrl-addon/rootfs/usr/bin/multiserver/mlmodels.py

@@ -7,7 +7,7 @@
 from mlflow.entities.model_registry import RegisteredModel
 from mlflow import MlflowClient
 
-from .const import CHALLENGER_ALIAS, CHAMPION_ALIAS
+from const import CHALLENGER_ALIAS, CHAMPION_ALIAS
 
 _logger = logging.getLogger(__name__)