From 388d4c95e47199db857c117700191bc916a30bae Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Mon, 13 Apr 2026 14:57:06 -0300 Subject: [PATCH] chore: update configuration and imports for improved functionality - Updated image tag in values.yaml from "1.0.0" to "1.0.2" for the latest version. - Modified STORE_BASE_URL to include port 3000 for proper service access. - Changed import from MinioRepository to MinioRepositorySync for synchronization support. - Updated import from Postgres to PostgresSync to enhance experiment tracking capabilities. - Renamed TRAIN_TASK_QUEUE from "train_model-single-queue" to "train_model-basic-queue" for clarity. --- model_manager/activities/activities.py | 2 +- model_manager/activities/experiment_tracking.py | 6 +++--- scripts/run_training_test.py | 4 ++-- values.yaml | 4 ++-- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/model_manager/activities/activities.py b/model_manager/activities/activities.py index 5cdb61f..f81212a 100644 --- a/model_manager/activities/activities.py +++ b/model_manager/activities/activities.py @@ -7,7 +7,7 @@ with workflow.unsafe.imports_passed_through(): from sientia_do.observability.logger import Logger from sientia_do.observability.metrics_controller import MetricsController from sientia_do.observability.sientia_monitoring import SientiaMonitoring - from sientia_do.repository.minio_repository import MinioRepository + from sientia_do.repository.minio_repository_sync import MinioRepository from sientia_model.model_repository.mlflow_repository import SientiaMLflowRepository from sientia_model.model_repository.plugin_store import PluginStore diff --git a/model_manager/activities/experiment_tracking.py b/model_manager/activities/experiment_tracking.py index 1df155c..baa2fa3 100644 --- a/model_manager/activities/experiment_tracking.py +++ b/model_manager/activities/experiment_tracking.py @@ -2,7 +2,7 @@ Experiment tracking activities for managing ML experiment lifecycle. This module provides activities for tracking and updating experiment run status -in the PostgreSQL database, extending the base Postgres activity with specialized +in the PostgreSQL database, extending the synchronous Postgres client with specialized methods for experiment management. """ @@ -20,7 +20,7 @@ with workflow.unsafe.imports_passed_through(): from sientia_do.notifications.models import NotificationLevel from sientia_do.observability.logger import Logger from sientia_do.observability.metrics_controller import MetricsController - from sientia_do.temporal.activities.postgres import Postgres + from sientia_do.temporal.activities.postgres_sync import Postgres from sqlalchemy import text @@ -41,7 +41,7 @@ class ExperimentTracking(Postgres): model registration. It maintains the experiment lifecycle from initialization through training, model saving, and cleanup. - The activity uses the existing Postgres connection pool and adds experiment-specific + The activity uses a SQLAlchemy engine / connection pool and adds experiment-specific operations with proper error handling and notifications. """ diff --git a/scripts/run_training_test.py b/scripts/run_training_test.py index c1c13d6..911e0b7 100644 --- a/scripts/run_training_test.py +++ b/scripts/run_training_test.py @@ -57,7 +57,7 @@ PG = { TEMPORAL_HOST = os.getenv('TEMPORAL_HOST') TEMPORAL_NAMESPACE = os.getenv('TEMPORAL_NAMESPACE') -TRAIN_TASK_QUEUE = "train_model-single-queue" +TRAIN_TASK_QUEUE = "train_model-basic-queue" TEMPORAL_TLS = os.getenv('TEMPORAL_USE_TLS', 'false').lower() in ('1', 'true', 'yes') @@ -171,4 +171,4 @@ result = await c.execute_workflow( # type: ignore[call-overload] print(wid) print(result) -# %% +# %% \ No newline at end of file diff --git a/values.yaml b/values.yaml index ed01742..bb88584 100644 --- a/values.yaml +++ b/values.yaml @@ -21,7 +21,7 @@ global: image: repository: aignosi.azurecr.io/sientia-module pullPolicy: Always - tag: "1.0.0" + tag: "1.0.2" # Common labels applied to pods (can be extended per project). commonLabels: {} @@ -123,7 +123,7 @@ global: value: "false" - name: STORE_BASE_URL - value: "http://gitea-http.gitea.svc.cluster.local" + value: "http://gitea-http.gitea.svc.cluster.local:3000" - name: STORE_OWNER value: "aignosi" - name: STORE_REPO