From de8846d316694e31fa7e483cbdac4cb58dcf0f35 Mon Sep 17 00:00:00 2001
From: shalin <shalin.deval@mavq.com>
Date: Sun, 1 Oct 2023 03:21:38 +0530
Subject: [PATCH 1/2] Added extendLease as task_router param

---
 .../client/http/models/task_result.py         | 27 ++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/src/conductor/client/http/models/task_result.py b/src/conductor/client/http/models/task_result.py
index aab510f7c..19a7eca6f 100644
--- a/src/conductor/client/http/models/task_result.py
+++ b/src/conductor/client/http/models/task_result.py
@@ -26,7 +26,8 @@ class TaskResult(object):
         'output_data': 'dict(str, object)',
         'logs': 'list[TaskExecLog]',
         'external_output_payload_storage_path': 'str',
-        'sub_workflow_id': 'str'
+        'sub_workflow_id': 'str',
+        'extend_lease': 'bool'
     }
 
     attribute_map = {
@@ -39,10 +40,11 @@ class TaskResult(object):
         'output_data': 'outputData',
         'logs': 'logs',
         'external_output_payload_storage_path': 'externalOutputPayloadStoragePath',
-        'sub_workflow_id': 'subWorkflowId'
+        'sub_workflow_id': 'subWorkflowId',
+        'extend_lease': 'extendLease'
     }
 
-    def __init__(self, workflow_instance_id=None, task_id=None, reason_for_incompletion=None, callback_after_seconds=None, worker_id=None, status=None, output_data=None, logs=None, external_output_payload_storage_path=None, sub_workflow_id=None):  # noqa: E501
+    def __init__(self, workflow_instance_id=None, task_id=None, reason_for_incompletion=None, callback_after_seconds=None, worker_id=None, status=None, output_data=None, logs=None, external_output_payload_storage_path=None, sub_workflow_id=None, extend_lease=None):  # noqa: E501
         """TaskResult - a model defined in Swagger"""  # noqa: E501
         self._workflow_instance_id = None
         self._task_id = None
@@ -57,6 +59,8 @@ def __init__(self, workflow_instance_id=None, task_id=None, reason_for_incomplet
         self.discriminator = None
         self.workflow_instance_id = workflow_instance_id
         self.task_id = task_id
+        self._extend_lease = None
+        self.extend_lease = extend_lease
         if reason_for_incompletion is not None:
             self.reason_for_incompletion = reason_for_incompletion
         if callback_after_seconds is not None:
@@ -73,6 +77,8 @@ def __init__(self, workflow_instance_id=None, task_id=None, reason_for_incomplet
             self.external_output_payload_storage_path = external_output_payload_storage_path
         if sub_workflow_id is not None:
             self.sub_workflow_id = sub_workflow_id
+        if extend_lease is not None:
+            self.extend_lease = extend_lease
 
     @property
     def workflow_instance_id(self):
@@ -289,6 +295,21 @@ def sub_workflow_id(self, sub_workflow_id):
         """
 
         self._sub_workflow_id = sub_workflow_id
+    
+    @property
+    def extend_lease(self):
+        return self._extend_lease
+    
+    @extend_lease.setter
+    def extend_lease(self, extend_lease):
+        """Sets the extend_lease of this TaskResult.
+
+
+        :param extend_lease: The extend_lease of this TaskResult.  # noqa: E501
+        :type: str
+        """
+
+        self._extend_lease = extend_lease
 
     def to_dict(self):
         """Returns the model properties as a dict"""

From 73f7f210074d006e82056a2f7954ee7b47e2b39e Mon Sep 17 00:00:00 2001
From: shalin <shalin.deval@mavq.com>
Date: Mon, 2 Oct 2023 02:20:34 +0530
Subject: [PATCH 2/2] Refactor

---
 src/conductor/client/http/models/task_result.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/conductor/client/http/models/task_result.py b/src/conductor/client/http/models/task_result.py
index 19a7eca6f..643dd14fc 100644
--- a/src/conductor/client/http/models/task_result.py
+++ b/src/conductor/client/http/models/task_result.py
@@ -56,11 +56,10 @@ def __init__(self, workflow_instance_id=None, task_id=None, reason_for_incomplet
         self._logs = None
         self._external_output_payload_storage_path = None
         self._sub_workflow_id = None
+        self._extend_lease = None
         self.discriminator = None
         self.workflow_instance_id = workflow_instance_id
         self.task_id = task_id
-        self._extend_lease = None
-        self.extend_lease = extend_lease
         if reason_for_incompletion is not None:
             self.reason_for_incompletion = reason_for_incompletion
         if callback_after_seconds is not None: