SIENTIAPDE-1350: Refactor: Improve logging, configuration, and resource management. Includes .env updates, client initialization logging, and resource closing.

scripts/run_training_test.py

@@ -24,29 +24,35 @@ import uuid
 from datetime import datetime, timedelta
 from pathlib import Path
 
+from dotenv import load_dotenv
 import psycopg2
 from psycopg2.extras import Json
 from temporalio import client
 
+
+# Carrega variáveis de ambiente do arquivo .env na raiz do projeto
+PROJECT_ROOT = Path(__file__).resolve().parent.parent
+ENV_PATH = PROJECT_ROOT / '.env'
+if ENV_PATH.exists():
+    load_dotenv(dotenv_path=ENV_PATH)
+
+
 DOCS_PATH = Path('docs/test-model-data.csv')
-MINIO_ALIAS = os.getenv('MINIO_ALIAS', 'suse')
-MINIO_BUCKET = os.getenv('MINIO_BUCKET', 'model-training')
+MINIO_ALIAS = 'suse'
+MINIO_BUCKET = 'model-training'
 
 POSTGRES_CONFIG = {
-    'host': os.getenv('POSTGRES_HOST', 'localhost'),
-    'port': os.getenv('POSTGRES_PORT', '55432'),
-    'user': os.getenv('POSTGRES_USER', 'postgres'),
-    'password': os.getenv(
-        'POSTGRES_PASSWORD',
-        'nFqc81y6kwmr2zuAIx43DhiOosFCVPpeEfTtTWZflkNjB2j1KtEeIANkhFR9mAX3',
-    ),
-    'dbname': os.getenv('POSTGRES_DBNAME', 'sientia-core-mlops-bff'),
+    'host': os.getenv('POSTGRES_HOST'),
+    'port': os.getenv('POSTGRES_PORT'),
+    'user': os.getenv('POSTGRES_USER'),
+    'password': os.getenv('POSTGRES_PASSWORD'),
+    'dbname': os.getenv('POSTGRES_DBNAME'),
 }
 
-TEMPORAL_HOST = os.getenv('TEMPORAL_HOST', 'localhost:37463')
-TEMPORAL_NAMESPACE = os.getenv('TEMPORAL_NAMESPACE', 'model-manager')
-TEMPORAL_TASK_QUEUE = os.getenv('TEMPORAL_TASK_QUEUE', 'train_model-queue')
-TEMPORAL_WORKFLOW = os.getenv('TEMPORAL_WORKFLOW', 'train_model')
+TEMPORAL_HOST = os.getenv('TEMPORAL_HOST')
+TEMPORAL_NAMESPACE = os.getenv('TEMPORAL_NAMESPACE')
+TRAIN_TASK_QUEUE = os.getenv('TRAIN_TASK_QUEUE')
+TEMPORAL_WORKFLOW = 'train_model'
 
 BASE_REQUEST_DATA = {
     'experimentName': 'model-manager-test-01',
@@ -169,6 +175,7 @@ async def trigger_temporal_workflow(workflow_input: dict) -> str:
     temporal_client = await client.Client.connect(
         target_host=TEMPORAL_HOST,
         namespace=TEMPORAL_NAMESPACE,
+        tls=os.getenv('TEMPORAL_USE_TLS', False),
     )
 
     workflow_id = f'train-model-test-{uuid.uuid4()}'
@@ -176,7 +183,7 @@ async def trigger_temporal_workflow(workflow_input: dict) -> str:
         TEMPORAL_WORKFLOW,
         workflow_input,
         id=workflow_id,
-        task_queue=TEMPORAL_TASK_QUEUE,
+        task_queue=TRAIN_TASK_QUEUE,
         execution_timeout=timedelta(minutes=5),
         run_timeout=timedelta(minutes=5),
         task_timeout=timedelta(minutes=5),