From 5a4d6c57060b7047b7d8837850b3532e6292a138 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Wed, 23 Jul 2025 12:26:36 -0300
Subject: [PATCH] SIENTIAPDE-1171  feat: add minimal_retrain function to
 orchestrator_functions.py for retraining workflows

- Introduced minimal_retrain function to streamline retraining configuration.
- The function integrates common configuration with specific parameters for retraining workflows.
---
 orchestrator/utils/orchestrator_functions.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/orchestrator/utils/orchestrator_functions.py b/orchestrator/utils/orchestrator_functions.py
index f1d6510..6fb3810 100644
--- a/orchestrator/utils/orchestrator_functions.py
+++ b/orchestrator/utils/orchestrator_functions.py
@@ -13,6 +13,17 @@ def common_config(config: dict[str, Any]):
     }
 
 
+def minimal_retrain(config: dict[str, Any]):
+    return {
+        **common_config(config),
+        "workflow_type": "retrain",
+        "schedule_name": config['schedule_name'],
+        "query": config['query'],
+        "schema": "sientia_data",
+        "table_name": "log_retrain",
+    }
+
+
 def scouter(config: dict[str, Any]):
     filters = {}
     for f in config.get('filters', []):