From 367b00f00a0551ffa8759b00f4bc539890f3e567 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Wed, 25 Mar 2026 09:41:12 -0300
Subject: [PATCH] SIENTIAPDE-1712  Enhance prediction export functionality by
 adding 'on_conflict' and 'unique_columns' parameters to the payload
 initialization in FormatAndExportPrediction class.

---
 .../workflows/sub_workflows/format_and_export_prediction.py     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/laborious/workflows/sub_workflows/format_and_export_prediction.py b/laborious/workflows/sub_workflows/format_and_export_prediction.py
index c9127c5..a9cda2a 100644
--- a/laborious/workflows/sub_workflows/format_and_export_prediction.py
+++ b/laborious/workflows/sub_workflows/format_and_export_prediction.py
@@ -198,6 +198,8 @@ class FormatAndExportPrediction:
                 'table_name': input_data['table_name'],
                 'data': prediction,
                 'timestamp_conversion': {'column': 'timestamp', 'format': DATETIME_FORMAT_WITH_TZ},
+                'on_conflict': input_data.get('on_conflict', 'replace'),
+                'unique_columns': ['model_id', 'timestamp'],
             },
             retry_policy=retry_policy,
             start_to_close_timeout=timedelta(seconds=180),